124 lines
4.6 KiB
JavaScript
124 lines
4.6 KiB
JavaScript
const { readHostList, writeHostList, RSADecryptSync, AESEncryptSync, AESDecryptSync } = require('../utils')
|
|
|
|
async function getHostList({ res }) {
|
|
// console.log('get-host-list')
|
|
let data = await readHostList()
|
|
data?.sort((a, b) => Number(b.index || 0) - Number(a.index || 0))
|
|
for (const item of data) {
|
|
let { username, port, authType, _id: id, credential } = item
|
|
if (credential) credential = await AESDecryptSync(credential)
|
|
// console.log(credential)
|
|
const isConfig = Boolean(username && port && (item[authType]))
|
|
Object.assign(item, { id, isConfig, password: '', privateKey: '', credential })
|
|
}
|
|
res.success({ data })
|
|
}
|
|
|
|
async function addHost({
|
|
res, request
|
|
}) {
|
|
let {
|
|
body: {
|
|
name, host: newHost, index, expired, expiredNotify, group, consoleUrl, remark,
|
|
port, username, authType, password, privateKey, credential, command, tempKey
|
|
}
|
|
} = request
|
|
// console.log(request)
|
|
if (!newHost || !name) return res.fail({ msg: 'missing params: name or host' })
|
|
let hostList = await readHostList()
|
|
if (hostList?.some(({ host }) => host === newHost)) return res.fail({ msg: `主机${ newHost }已存在` })
|
|
let record = {
|
|
name, host: newHost, index, expired, expiredNotify, group, consoleUrl, remark,
|
|
port, username, authType, password, privateKey, credential, command
|
|
}
|
|
const clearTempKey = await RSADecryptSync(tempKey)
|
|
console.log('clearTempKey:', clearTempKey)
|
|
const clearSSHKey = await AESDecryptSync(record[authType], clearTempKey)
|
|
// console.log(`${ authType }原密文: `, clearSSHKey)
|
|
record[authType] = await AESEncryptSync(clearSSHKey)
|
|
console.log(`${ authType }__commonKey加密存储: `, record[authType])
|
|
hostList.push(record)
|
|
await writeHostList(hostList)
|
|
res.success()
|
|
}
|
|
|
|
async function updateHost({
|
|
res, request
|
|
}) {
|
|
let {
|
|
body: {
|
|
host: newHost, name: newName, index, oldHost, expired, expiredNotify, group, consoleUrl, remark,
|
|
port, username, authType, password, privateKey, credential, command, tempKey
|
|
}
|
|
} = request
|
|
if (!newHost || !newName || !oldHost) return res.fail({ msg: '参数错误' })
|
|
let hostList = await readHostList()
|
|
let record = {
|
|
name: newName, host: newHost, index, expired, expiredNotify, group, consoleUrl, remark,
|
|
port, username, authType, password, privateKey, credential, command
|
|
}
|
|
if (!hostList.some(({ host }) => host === oldHost)) return res.fail({ msg: `原实例[${ oldHost }]不存在,请尝试新增实例` })
|
|
|
|
let idx = hostList.findIndex(({ host }) => host === oldHost)
|
|
const oldRecord = hostList[idx]
|
|
// 如果存在原认证方式则保存下来
|
|
if (!record[authType] && oldRecord[authType]) {
|
|
record[authType] = oldRecord[authType]
|
|
} else {
|
|
const clearTempKey = await RSADecryptSync(tempKey)
|
|
console.log('clearTempKey:', clearTempKey)
|
|
const clearSSHKey = await AESDecryptSync(record[authType], clearTempKey)
|
|
// console.log(`${ authType }原密文: `, clearSSHKey)
|
|
record[authType] = await AESEncryptSync(clearSSHKey)
|
|
console.log(`${ authType }__commonKey加密存储: `, record[authType])
|
|
}
|
|
hostList.splice(idx, 1, record)
|
|
writeHostList(hostList)
|
|
res.success()
|
|
}
|
|
|
|
async function removeHost({
|
|
res, request
|
|
}) {
|
|
let { body: { host } } = request
|
|
let hostList = await readHostList()
|
|
let hostIdx = hostList.findIndex(item => item.host === host)
|
|
if (hostIdx === -1) return res.fail({ msg: `${ host }不存在` })
|
|
hostList.splice(hostIdx, 1)
|
|
writeHostList(hostList)
|
|
res.success({ data: `${ host }已移除` })
|
|
}
|
|
|
|
async function importHost({
|
|
res, request
|
|
}) {
|
|
let { body: { importHost } } = request
|
|
if (!Array.isArray(importHost)) return res.fail({ msg: '参数错误' })
|
|
let hostList = await readHostList()
|
|
// 过滤已存在的host
|
|
let hostListSet = new Set(hostList.map(item => item.host))
|
|
let newHostList = importHost.filter(item => !hostListSet.has(item.host))
|
|
if (newHostList.length === 0) return res.fail({ msg: '导入的实例已存在' })
|
|
|
|
let extraFiels = {
|
|
expired: null, expiredNotify: false, group: 'default', consoleUrl: '', remark: '',
|
|
authType: 'privateKey', password: '', privateKey: '', credential: '', command: ''
|
|
}
|
|
newHostList = newHostList.map((item, index) => {
|
|
item.port = Number(item.port) || 0
|
|
item.index = hostList.length + index + 1
|
|
return Object.assign(item, { ...extraFiels })
|
|
})
|
|
hostList.push(...newHostList)
|
|
writeHostList(hostList)
|
|
res.success({ data: { len: newHostList.length } })
|
|
}
|
|
|
|
module.exports = {
|
|
getHostList,
|
|
addHost,
|
|
updateHost,
|
|
removeHost,
|
|
importHost
|
|
}
|