125 lines
3.3 KiB
JavaScript
125 lines
3.3 KiB
JavaScript
const { writeKey, writeNotifyList, writeNotifyConfig } = require('./utils/storage')
|
|
const { KeyDB, GroupDB, NotifyDB, NotifyConfigDB } = require('./utils/db-class')
|
|
|
|
function initKeyDB() {
|
|
return new Promise((resolve, reject) => {
|
|
const keyDB = new KeyDB().getInstance()
|
|
keyDB.count({}, async (err, count) => {
|
|
if (err) {
|
|
consola.log('初始化keyDB错误:', err)
|
|
reject(err)
|
|
} else {
|
|
if (count === 0) {
|
|
consola.log('初始化keyDB✔')
|
|
const defaultData = {
|
|
user: 'admin',
|
|
pwd: 'admin',
|
|
commonKey: '',
|
|
publicKey: '',
|
|
privateKey: ''
|
|
}
|
|
await writeKey(defaultData)
|
|
}
|
|
}
|
|
resolve()
|
|
})
|
|
})
|
|
}
|
|
|
|
async function initGroupDB() {
|
|
const groupDB = new GroupDB().getInstance()
|
|
let count = await groupDB.countAsync({})
|
|
if (count === 0) {
|
|
consola.log('初始化groupDB✔')
|
|
const defaultData = [{ '_id': 'default', 'name': '默认分组', 'index': 0 }]
|
|
return groupDB.insertAsync(defaultData)
|
|
}
|
|
return Promise.resolve()
|
|
}
|
|
|
|
function initNotifyDB() {
|
|
return new Promise((resolve, reject) => {
|
|
const notifyDB = new NotifyDB().getInstance()
|
|
notifyDB.find({}, async (err, notifyList) => {
|
|
if (err) {
|
|
consola.log('初始化notifyDB错误:', err)
|
|
reject(err)
|
|
} else {
|
|
let defaultData = [{
|
|
'type': 'login',
|
|
'desc': '登录面板提醒',
|
|
'sw': false
|
|
}, {
|
|
'type': 'err_login',
|
|
'desc': '登录错误提醒(连续5次)',
|
|
'sw': false
|
|
}, {
|
|
'type': 'updatePwd',
|
|
'desc': '修改密码提醒',
|
|
'sw': false
|
|
}, {
|
|
'type': 'host_login',
|
|
'desc': '服务器登录提醒',
|
|
'sw': false
|
|
}, {
|
|
'type': 'onekey_complete',
|
|
'desc': '批量指令执行完成提醒',
|
|
'sw': false
|
|
}, {
|
|
'type': 'host_expired',
|
|
'desc': '服务器到期提醒',
|
|
'sw': false
|
|
}]
|
|
if (notifyList.length === 0) {
|
|
consola.log('初始化notifyDB✔')
|
|
} else {
|
|
consola.log('同步notifyDB✔')
|
|
defaultData = defaultData.map(defaultItem => {
|
|
let item = notifyList.find(notify => notify.type === defaultItem.type)
|
|
defaultItem.sw = item ? item.sw : false
|
|
return item
|
|
})
|
|
}
|
|
await writeNotifyList(defaultData)
|
|
}
|
|
resolve()
|
|
})
|
|
})
|
|
}
|
|
|
|
function initNotifyConfigDB() {
|
|
return new Promise((resolve, reject) => {
|
|
const notifyConfigDB = new NotifyConfigDB().getInstance()
|
|
notifyConfigDB.count({}, async (err, count) => {
|
|
if (err) {
|
|
consola.log('初始化NotifyConfigDB错误:', err)
|
|
reject(err)
|
|
} else {
|
|
if (count === 0) {
|
|
consola.log('初始化NotifyConfigDB✔')
|
|
const defaultData = {
|
|
type: 'sct',
|
|
sct: {
|
|
sendKey: ''
|
|
},
|
|
email: {
|
|
service: 'QQ',
|
|
user: '',
|
|
pass: ''
|
|
}
|
|
}
|
|
await writeNotifyConfig(defaultData)
|
|
}
|
|
}
|
|
resolve()
|
|
})
|
|
})
|
|
}
|
|
|
|
module.exports = async () => {
|
|
await initKeyDB()
|
|
await initNotifyDB()
|
|
await initGroupDB()
|
|
await initNotifyConfigDB()
|
|
}
|