67 lines
2.2 KiB
JavaScript
67 lines
2.2 KiB
JavaScript
const { readGroupList, writeGroupList } = require('../utils/storage')
|
|
const { HostListDB } = require('../utils/db-class')
|
|
|
|
const hostListDB = new HostListDB().getInstance()
|
|
|
|
async function getGroupList({ res }) {
|
|
let data = await readGroupList()
|
|
data = data.map(item => {
|
|
return { ...item, id: item._id }
|
|
})
|
|
data?.sort((a, b) => Number(b.index || 0) - Number(a.index || 0))
|
|
res.success({ data })
|
|
}
|
|
|
|
const addGroupList = async ({ res, request }) => {
|
|
let { body: { name, index } } = request
|
|
if (!name) return res.fail({ data: false, msg: '参数错误' })
|
|
let groupList = await readGroupList()
|
|
let group = { name, index }
|
|
groupList.push(group)
|
|
await writeGroupList(groupList)
|
|
res.success({ data: '添加成功' })
|
|
}
|
|
|
|
const updateGroupList = async ({ res, request }) => {
|
|
let { params: { id } } = request
|
|
let { body: { name, index } } = request
|
|
if (!id || !name) return res.fail({ data: false, msg: '参数错误' })
|
|
let groupList = await readGroupList()
|
|
let idx = groupList.findIndex(item => item._id === id)
|
|
if (idx === -1) return res.fail({ data: false, msg: `分组ID${ id }不存在` })
|
|
const { _id } = groupList[idx]
|
|
let group = { _id, name, index: Number(index) || 0 }
|
|
groupList.splice(idx, 1, group)
|
|
await writeGroupList(groupList)
|
|
res.success({ data: '修改成功' })
|
|
}
|
|
|
|
const removeGroup = async ({ res, request }) => {
|
|
let { params: { id } } = request
|
|
if (id === 'default') return res.fail({ data: false, msg: '保留分组, 禁止删除' })
|
|
let groupList = await readGroupList()
|
|
let idx = groupList.findIndex(item => item._id === id)
|
|
if (idx === -1) return res.fail({ msg: '分组不存在' })
|
|
|
|
// 移除分组将所有该分组下host分配到default中去
|
|
let hostList = await hostListDB.findAsync({})
|
|
if (Array.isArray(hostList) && hostList.length > 0) {
|
|
for (let item of hostList) {
|
|
if (item.group === groupList[idx]._id) {
|
|
item.group = 'default'
|
|
await hostListDB.updateAsync({ _id: item._id }, item)
|
|
}
|
|
}
|
|
}
|
|
groupList.splice(idx, 1)
|
|
await writeGroupList(groupList)
|
|
res.success({ data: '移除成功' })
|
|
}
|
|
|
|
module.exports = {
|
|
addGroupList,
|
|
getGroupList,
|
|
updateGroupList,
|
|
removeGroup
|
|
}
|