easynode/server/app/db.js

const NodeRSA = require('node-rsa')
const { randomStr } = require('./utils/tools')
const { AESEncryptAsync, SHA1Encrypt } = require('./utils/encrypt')
const { KeyDB, GroupDB, NotifyDB, NotifyConfigDB, ScriptGroupDB } = require('./utils/db-class')

async function initKeyDB() {
  const keyDB = new KeyDB().getInstance()
  let keyData = await keyDB.findOneAsync({})

  if (keyData?.user) {

    if (!keyData?.jwtToken) {
      logger.info('🔒初始化jwtToken')
      const jwtToken = randomStr(32)
      await keyDB.updateAsync({ _id: keyData._id }, { $set: { jwtToken } })
    }

    const { _id, ipWhiteList = [] } = keyData
    let allowedIPs = process.env.ALLOWED_IPS ? process.env.ALLOWED_IPS.split(',') : []
    if (allowedIPs.length > 0) {
      logger.info('[存在白名单IP环境变量,合并到本地数据库中]')
      allowedIPs = [...new Set([...ipWhiteList, ...allowedIPs])].filter(item => item)
      await keyDB.updateAsync({ _id }, { $set: { ipWhiteList: allowedIPs } })
    }
    try {
      let { ipWhiteList = [] } = await keyDB.findOneAsync({})
      if (ipWhiteList.length > 0) global.ALLOWED_IPS = ipWhiteList
    } catch (error) {
      logger.error('设置全局IP白名单失败:', error)
    }
    logger.info('公私钥已存在[重新生成会导致已保存的ssh密钥信息失效]')
    return
  }

  const randomUsername = randomStr(8)
  const randomPassword = randomStr(8)

  let newConfig = {
    user: randomUsername,
    pwd: SHA1Encrypt(randomPassword),
    commonKey: randomStr(32),
    jwtToken: randomStr(32),
    publicKey: '',
    privateKey: ''
  }
  await keyDB.insertAsync(newConfig)
  let key = new NodeRSA({ b: 2048 })
  key.setOptions({ encryptionScheme: 'pkcs1', environment: 'browser' })
  let privateKey = key.exportKey('pkcs1-private-pem')
  let publicKey = key.exportKey('pkcs8-public-pem')
  newConfig.privateKey = await AESEncryptAsync(privateKey, newConfig.commonKey) // 加密私钥
  newConfig.publicKey = publicKey // 公开公钥
  await keyDB.updateAsync({}, { $set: newConfig }, { upsert: true })

  // 在控制台打印随机生成的账号密码
  logger.info('========================================')
  logger.info('EasyNode 默认登录凭据 (请及时更改):')
  logger.info(`用户名: ${ randomUsername }`)
  logger.info(`密码: ${ randomPassword }`)
  logger.info('========================================')

}

async function initGroupDB() {
  const groupDB = new GroupDB().getInstance()
  let count = await groupDB.countAsync({})
  if (count === 0) {
    logger.info('初始化groupDB✔')
    const defaultData = [{ '_id': 'default', 'name': '默认分组', 'index': 0 }]
    return groupDB.insertAsync(defaultData)
  }
  return Promise.resolve()
}

async function initNotifyDB() {
  const notifyDB = new NotifyDB().getInstance()
  let count = await notifyDB.countAsync({})
  if (count !== 0) return
  logger.info('初始化notifyDB✔')
  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
  }]
  return notifyDB.insertAsync(defaultData)
}

async function initNotifyConfigDB() {
  const notifyConfigDB = new NotifyConfigDB().getInstance()
  let notifyConfig = await notifyConfigDB.findOneAsync({})
  logger.info('初始化NotifyConfigDB✔')
  const defaultData = {
    type: 'sct',
    sct: {
      sendKey: ''
    },
    email: {
      service: 'QQ',
      user: '',
      pass: '',
      useCustom: false,
      host: '',
      port: 465,
      secure: true,
      to: ''
    },
    tg: {
      token: '',
      chatId: ''
    },
    webhook: {
      url: '',
      method: 'POST',
      contentType: 'application/json',
      headers: '',
      template: '{\n  "title": "{{title}}",\n  "content": "{{content}}",\n  "description": "{{content}}",\n  "timestamp": "{{timestamp}}",\n  "datetime": "{{datetime}}"\n}'
    }
  }
  if (notifyConfig) {
    await notifyConfigDB.removeAsync({ _id: notifyConfig._id })
    delete notifyConfig._id
    return notifyConfigDB.insertAsync(Object.assign({}, defaultData, notifyConfig))
  }
  return notifyConfigDB.insertAsync(defaultData)
}

async function initScriptGroupDB() {
  const scriptGroupDB = new ScriptGroupDB().getInstance()
  let count = await scriptGroupDB.countAsync({})
  if (count === 0) {
    logger.info('初始化ScriptGroupDB✔')
    const defaultData = [
      { '_id': 'default', 'name': '默认分组', 'index': 0 },
      { '_id': 'builtin', 'name': '内置脚本', 'index': -1 }
    ]
    return scriptGroupDB.insertAsync(defaultData)
  }
  return Promise.resolve()
}

module.exports = async () => {
  await initKeyDB()
  await initNotifyDB()
  await initGroupDB()
  await initScriptGroupDB()
  await initNotifyConfigDB()
}