yunzai-micro-plugin/src/server/index.ts

Yunzai开发管理面板

import chalk from 'chalk'
import server from './app/index.js'
import MicroWs from './app/ws.js'

import { IncomingMessage } from 'http'
import { WebSocketServer, WebSocket } from 'ws';
import { getAllWebAddress } from '#utils'
import { Cfg } from '#cfg'

let wss: WebSocketServer
let microWs = new MicroWs()


/**
 * 启动接口
 * @param port 
 */
const startServer = async (port: number): Promise<'ok' | void> => {

  wss = new WebSocketServer({ server });
  wss.on('connection', (ws: WebSocket, req: IncomingMessage) => {
    microWs.onOpen(ws, req)
  });

  await new Promise((resolve, reject) => {
    server.listen(port, async (err?: Error) => {
      if (err) {
        reject(err);
      } else {
        const { local, remote } = await getAllWebAddress()
        logger.info(chalk.cyanBright('----------Micro---------'))
        logger.info(chalk.cyanBright(`微代码开发服务器启动成功！您可尝试选择访问以下地址进行开发管理：`))
        logger.info(chalk.cyanBright(`公网地址：${remote[0]}`))
        logger.info(chalk.cyanBright(`内网地址：${local[0]}`))
        logger.info(chalk.cyanBright(`回环地址：http://127.0.0.1:${port}`))
        logger.info(chalk.cyanBright('------------------------'))
        if (!Cfg.masterQQ || Cfg.masterQQ.length == 0) {
          logger.mark('[Micro]未找到主人QQ，请确定是否已配置')
        }
        try {
          Bot && await Bot?.pickFriend(Number(Cfg.masterQQ[0]))?.sendMsg(
            `开发服务器启动成功，您可打开浏览器进入以下地址开发管理：\n` +
            `公网地址：${remote[0]}\n` +
            `内网地址：${local[0]}` +
            `你也可以下载Micro Panel的安卓APP来对机器人进行管理` +
            `Micro Panel下载地址：https://tianstudio.lanzoub.com/b004iib74j（密码：micro）`
          )
        } catch (error) {
          logger.mark('[Micro]' + error.message)
        }
        resolve('ok');
      }
    });
  });
}

/**
 * 关闭接口
 * @returns 
 */
const stopServer = (): Promise<any> => {
  return new Promise((resolve, reject) => {
    if (wss) {
      wss.close((err) => {
        if (err) {
          reject(err);
        }
        if (server) {
          server.close((err) => {
            if (err) {
              reject(err);
            }
            resolve('ok');
          });
        } else {
          resolve('ok');
        }
      });
    } else {
      resolve('ok');
    }
  });
}

/**
 * 重启接口
 * @param port 
 */
const restartServer = async (port: number): Promise<any> => {
  await stopServer();
  await startServer(port);
}

export default microWs

export { startServer, stopServer, restartServer };