@oxiaom/adoremix 1.0.0

package/src/service/linux.js

@@ -0,0 +1,139 @@
+'use strict';
+
+const fs = require('fs');
+const fse = require('fs-extra');
+const path = require('path');
+const { execSync } = require('child_process');
+const os = require('os');
+
+const logger = require('../logger');
+const paths = require('../paths');
+
+const SYSTEM_UNIT_PATH = '/etc/systemd/system/adoremix.service';
+const USER_UNIT_DIR = path.join(os.homedir(), '.config', 'systemd', 'user');
+const USER_UNIT_PATH = path.join(USER_UNIT_DIR, 'adoremix.service');
+
+function isSystemMode() {
+  return process.getuid && process.getuid() === 0;
+}
+
+function ensureUserExists(username) {
+  try {
+    execSync(`id -u ${username}`, { stdio: 'ignore' });
+    return false;
+  } catch (e) {
+    execSync(`useradd -r -s /usr/sbin/nologin ${username}`);
+    return true;
+  }
+}
+
+function chownWorkdir(workdir, username) {
+  execSync(`chown -R ${username}:${username} ${workdir}`);
+}
+
+function fillTemplate(workdir) {
+  const tplPath = path.join(__dirname, '..', '..', 'templates', 'adoremix.service.tmpl');
+  let tpl = fs.readFileSync(tplPath, 'utf8');
+  const user = isSystemMode() ? 'adoremix' : os.userInfo().username;
+  const node = paths.nodeExecutable();
+  const cli = paths.cliBin();
+  return tpl
+    .replace(/__USER__/g, user)
+    .replace(/__GROUP__/g, user)
+    .replace(/__WORKDIR__/g, workdir)
+    .replace(/__NODE__/g, node)
+    .replace(/__CLI__/g, cli);
+}
+
+function installSystem(workdir) {
+  let created = false;
+  try {
+    created = ensureUserExists('adoremix');
+    if (created) logger.ok('创建系统用户 adoremix');
+  } catch (e) {
+    logger.warn(`无法创建 adoremix 用户：${e.message}（服务将以 root 运行）`);
+  }
+  if (created) {
+    try { chownWorkdir(workdir, 'adoremix'); } catch (e) {
+      logger.warn(`chown 失败：${e.message}`);
+    }
+  }
+  const content = fillTemplate(workdir);
+  fs.writeFileSync(SYSTEM_UNIT_PATH, content, 'utf8');
+  logger.ok(`写入 unit 文件 ${SYSTEM_UNIT_PATH}`);
+  execSync('systemctl daemon-reload');
+  execSync('systemctl enable adoremix');
+  logger.ok('systemctl enable adoremix');
+  execSync('systemctl start adoremix');
+  logger.ok('systemctl start adoremix');
+  logger.log('');
+  logger.log('查看状态：systemctl status adoremix');
+  logger.log('查看日志：journalctl -u adoremix -f');
+}
+
+function installUser(workdir) {
+  fse.ensureDirSync(USER_UNIT_DIR);
+  const content = fillTemplate(workdir);
+  fs.writeFileSync(USER_UNIT_PATH, content, 'utf8');
+  logger.ok(`写入 user unit ${USER_UNIT_PATH}`);
+  execSync('systemctl --user daemon-reload');
+  execSync('systemctl --user enable adoremix');
+  logger.ok('systemctl --user enable adoremix');
+  execSync('systemctl --user start adoremix');
+  logger.ok('systemctl --user start adoremix');
+  logger.log('');
+  logger.log('用户服务需 linger 才能开机自启：loginctl enable-linger $USER');
+  logger.log('查看状态：systemctl --user status adoremix');
+}
+
+function uninstallSystem() {
+  try { execSync('systemctl stop adoremix', { stdio: 'ignore' }); } catch (e) {}
+  try { execSync('systemctl disable adoremix', { stdio: 'ignore' }); } catch (e) {}
+  if (fs.existsSync(SYSTEM_UNIT_PATH)) fs.unlinkSync(SYSTEM_UNIT_PATH);
+  execSync('systemctl daemon-reload');
+  logger.ok('服务已卸载');
+}
+
+function uninstallUser() {
+  try { execSync('systemctl --user stop adoremix', { stdio: 'ignore' }); } catch (e) {}
+  try { execSync('systemctl --user disable adoremix', { stdio: 'ignore' }); } catch (e) {}
+  if (fs.existsSync(USER_UNIT_PATH)) fs.unlinkSync(USER_UNIT_PATH);
+  execSync('systemctl --user daemon-reload');
+  logger.ok('用户服务已卸载');
+}
+
+function statusSystem() {
+  try {
+    execSync('systemctl status adoremix --no-pager', { stdio: 'inherit' });
+  } catch (e) {
+    return 1;
+  }
+  return 0;
+}
+
+function statusUser() {
+  try {
+    execSync('systemctl --user status adoremix --no-pager', { stdio: 'inherit' });
+  } catch (e) {
+    return 1;
+  }
+  return 0;
+}
+
+module.exports = {
+  install(opts) {
+    const workdir = opts.workdir;
+    if (isSystemMode()) installSystem(workdir);
+    else installUser(workdir);
+    return 0;
+  },
+  uninstall() {
+    if (isSystemMode()) uninstallSystem();
+    else uninstallUser();
+    return 0;
+  },
+  status() {
+    return isSystemMode() ? statusSystem() : statusUser();
+  },
+  _internal: { fillTemplate, SYSTEM_UNIT_PATH, USER_UNIT_PATH }
+};

package/src/service/windows.js

@@ -0,0 +1,88 @@
+'use strict';
+
+const fs = require('fs');
+const path = require('path');
+const { execSync } = require('child_process');
+
+const logger = require('../logger');
+const paths = require('../paths');
+
+const SVC_NAME = 'AdoreMixService';
+
+function fillTemplate(workdir) {
+  const tplPath = path.join(__dirname, '..', '..', 'templates', 'winservice.js.tmpl');
+  let tpl = fs.readFileSync(tplPath, 'utf8');
+  return tpl
+    .replace(/__WORKDIR__/g, workdir.replace(/\\/g, '\\\\'))
+    .replace(/__NODE__/g, paths.nodeExecutable().replace(/\\/g, '\\\\'))
+    .replace(/__CLI__/g, paths.cliBin().replace(/\\/g, '\\\\'));
+}
+
+function ensureNodeWindows(workdir) {
+  try {
+    require.resolve('node-windows');
+    return true;
+  } catch (e) {
+    logger.warn('node-windows 未安装，正在尝试本地安装...');
+    try {
+      execSync('npm install --no-save node-windows', { cwd: workdir, stdio: 'inherit' });
+      return true;
+    } catch (e2) {
+      logger.error('node-windows 安装失败，请手动在工作目录执行 npm install node-windows');
+      return false;
+    }
+  }
+}
+
+function install(opts) {
+  const workdir = opts.workdir;
+  if (!ensureNodeWindows(workdir)) return 1;
+  const wrapperPath = path.join(workdir, 'winservice.js');
+  const content = fillTemplate(workdir);
+  fs.writeFileSync(wrapperPath, content, 'utf8');
+  logger.ok(`生成 wrapper ${wrapperPath}`);
+  try {
+    execSync(`node "${wrapperPath}" install`, { cwd: workdir, stdio: 'inherit' });
+  } catch (e) {
+    logger.error(`注册失败：${e.message}`);
+    return 1;
+  }
+  logger.log('');
+  logger.log('查看服务：sc query AdoreMixService');
+  logger.log('启动/停止：sc start/stop AdoreMixService');
+  logger.log('或：net start/stop AdoreMixService');
+  return 0;
+}
+
+function uninstall(opts) {
+  const workdir = opts.workdir;
+  const wrapperPath = path.join(workdir, 'winservice.js');
+  if (!fs.existsSync(wrapperPath)) {
+    logger.warn(`未找到 ${wrapperPath}，直接 sc delete`);
+    try {
+      execSync(`sc delete ${SVC_NAME}`, { stdio: 'inherit' });
+      return 0;
+    } catch (e) {
+      return 1;
+    }
+  }
+  try {
+    execSync(`node "${wrapperPath}" uninstall`, { cwd: workdir, stdio: 'inherit' });
+  } catch (e) {
+    logger.error(`卸载失败：${e.message}`);
+    return 1;
+  }
+  return 0;
+}
+
+function status() {
+  try {
+    execSync(`sc query ${SVC_NAME}`, { stdio: 'inherit' });
+    return 0;
+  } catch (e) {
+    logger.warn(`服务未安装或查询失败`);
+    return 1;
+  }
+}
+
+module.exports = { install, uninstall, status, SVC_NAME };

package/templates/adoremix.service.tmpl

@@ -0,0 +1,19 @@
+[Unit]
+Description=AdoreMix Broadcast Server
+Documentation=https://github.com/your-org/adoremix
+After=network.target mysqld.service redis.service
+
+[Service]
+Type=simple
+User=__USER__
+Group=__GROUP__
+WorkingDirectory=__WORKDIR__
+ExecStart=__NODE__ __CLI__ start --daemon --workdir __WORKDIR__
+Restart=on-failure
+RestartSec=5
+StandardOutput=append:__WORKDIR__/logs/svc.log
+StandardError=append:__WORKDIR__/logs/svc.err
+LimitNOFILE=65536
+
+[Install]
+WantedBy=multi-user.target

package/templates/winservice.js.tmpl

@@ -0,0 +1,66 @@
+'use strict';
+// 该文件由 `adoremix service install` 在工作目录生成。
+// 它通过 node-windows 创建/卸载 Windows 服务。
+// 用法：node winservice.js install|uninstall
+
+const Service = require('node-windows').Service;
+const path = require('path');
+
+const WORKDIR = '__WORKDIR__';
+const NODE = '__NODE__';
+const CLI = '__CLI__';
+const SVC_NAME = 'AdoreMixService';
+const SVC_DESC = 'AdoreMix Broadcast Server';
+
+const svc = new Service({
+  name: SVC_NAME,
+  description: SVC_DESC,
+  script: CLI,
+  execPath: NODE,
+  args: ['start', '--daemon', '--workdir', WORKDIR],
+  cwd: WORKDIR,
+  env: [{
+    name: 'NODE_ENV',
+    value: 'production'
+  }]
+});
+
+svc.on('install', () => {
+  console.log(`[ok] 服务已安装：${SVC_NAME}`);
+  svc.start();
+});
+
+svc.on('alreadyinstalled', () => {
+  console.log(`[warn] 服务已存在：${SVC_NAME}`);
+});
+
+svc.on('start', () => {
+  console.log(`[ok] 服务已启动：${SVC_NAME}`);
+});
+
+svc.on('stop', () => {
+  console.log(`[ok] 服务已停止：${SVC_NAME}`);
+});
+
+svc.on('uninstall', () => {
+  console.log(`[ok] 服务已卸载：${SVC_NAME}`);
+});
+
+svc.on('error', (err) => {
+  console.error(`[err] node-windows 错误：${err.message}`);
+  process.exit(1);
+});
+
+const cmd = process.argv[2] || 'install';
+if (cmd === 'install') {
+  svc.install();
+} else if (cmd === 'uninstall') {
+  svc.uninstall();
+} else if (cmd === 'start') {
+  svc.start();
+} else if (cmd === 'stop') {
+  svc.stop();
+} else {
+  console.error('用法：node winservice.js install|uninstall|start|stop');
+  process.exit(1);
+}