@fengye_nobug/dsh-qq-bot 0.2.6 → 0.2.8

package/updates.js

@@ -10,7 +10,11 @@
 // 安装成功后，若 profile、home 或插件自己的 cordis.patch.yml 里 qq-bridge 的 name
 // 仍是旧的无作用域名，会改成磁盘上 package.json 里的包名。不改 config，不改设置，不改 NapCat。
 // 安装超过三分钟还没结束时，停掉这次拉起的进程组，并清掉已经退出的安装记录。
-// 超时后若磁盘上已是目标版本，仍改包名，但不自动重启。
+// 超时后若磁盘上已是目标版本，仍改包名，并安排重启。
+// 重启会读本进程的 cgroup：认得出具体服务就用 systemd-run 只重启那个服务
+// （挂在 user@ 下面的用 systemctl --user）。systemd-run 失败，或只看见用户会话、
+// 认不出单独服务时，不重启整个登录，改为按当前命令行再拉起一次。
+// 两条都做不到，才提示自己重启。
 
 import { spawn } from 'node:child_process'
 import { chmodSync, mkdtempSync, readFileSync, renameSync, rmSync, statSync, writeFileSync } from 'node:fs'
@@ -237,7 +241,7 @@ export function assessUpdate({ current, releases, repo = UPDATE_REPO, channel =
       kind: 'no-release',
       current: parsedCurrent.raw,
       updateAvailable: false,
-      message: 'GitHub 上还没有可用的正式版，没法对比新版本。',
+      message: channel === 'npm' ? 'npm 上还没有可用的正式版，没法对比新版本。' : 'GitHub 上还没有可用的正式版，没法对比新版本。',
     }
   }
   stable.sort((left, right) => compareSemver(right.parsed, left.parsed) ?? 0)
@@ -259,21 +263,21 @@ export function assessUpdate({ current, releases, repo = UPDATE_REPO, channel =
     url,
   }
   if (compared > 0) {
-    const lines = [`有新版本 ${latest.parsed.raw}（你现在是 ${parsedCurrent.raw}）。`]
-    if (summary) lines.push(`说明：${summary}`)
-    lines.push(`升级命令：${command}`)
-    lines.push('装完请重启 dsh-web。')
-    if (useNpm) lines.push(`也可以在设置卡片里点「更新到 v${latest.parsed.raw}」，面板会从 npm 装上并重启。`)
-    else lines.push(`也可以在设置卡片里点「更新到 v${latest.parsed.raw}」，面板会自己装并重启。`)
-    if (!useNpm) lines.push(`发布页：${url}`)
-    return { ...base, kind: 'update', updateAvailable: true, message: lines.join('\n') }
+    const where = useNpm ? 'npm' : 'GitHub'
+    return {
+      ...base,
+      kind: 'update',
+      updateAvailable: true,
+      message: `有新版本 ${latest.parsed.raw}（现在是 ${parsedCurrent.raw}），在 ${where} 上。这里只检查，还没安装。`,
+    }
   }
   if (compared < 0) {
+    const where = useNpm ? 'npm' : 'GitHub'
     return {
       ...base,
       kind: 'ahead',
       updateAvailable: false,
-      message: `你现在的 ${parsedCurrent.raw} 比 GitHub 上最新正式版 ${latest.parsed.raw} 还新，不用按 Release 往回装。`,
+      message: `你现在的 ${parsedCurrent.raw} 比 ${where} 上最新正式版 ${latest.parsed.raw} 还新，不用往回装。`,
     }
   }
   return {
@@ -439,19 +443,6 @@ export async function queryNpmVersions(input = {}) {
   return { ok: false, code: 'network' }
 }
 
-function noteGithubFallback(assessed) {
-  if (!assessed || assessed.updateAvailable !== true) return assessed
-  const message = String(assessed.message ?? '')
-  if (message.includes('还没发到 npm') || !message.includes('也可以在设置卡片里点')) return assessed
-  return {
-    ...assessed,
-    message: message.replace(
-      '也可以在设置卡片里点',
-      '这一版还没发到 npm，一键更新会改从 GitHub 下载。\n也可以在设置卡片里点',
-    ),
-  }
-}
-
 function lookupFailure(npmCode, githubError, applying) {
   const raw = String(githubError ?? '')
   if (npmCode === 'network' && /连不上 GitHub/.test(raw)) {
@@ -490,7 +481,7 @@ async function findUpdate(input) {
       if (wanted !== '') return { ok: false, error: '这次检查到的正式版里没有这个版本，没有安装。' }
       return { ok: true, channel: 'npm', releases, assessed }
     }
-    const ghAssessed = noteGithubFallback(assessUpdate({ current: input.current, releases: gh.releases, repo, channel: 'github' }))
+    const ghAssessed = assessUpdate({ current: input.current, releases: gh.releases, repo, channel: 'github' })
     if (ghAssessed.ok === false) return { ok: false, error: ghAssessed.message }
     const npmLatest = assessed.latest || '0.0.0'
     const githubNewer = ghAssessed.updateAvailable === true && (compareSemver(ghAssessed.latest, npmLatest) ?? 0) > 0
@@ -506,7 +497,7 @@ async function findUpdate(input) {
   }
   const gh = await queryGithubReleases(input.fetchImpl, { repo, timeoutMs: input.timeoutMs })
   if (!gh.ok) return { ok: false, error: lookupFailure(npm.code, gh.error, input.applying === true) }
-  const assessed = noteGithubFallback(assessUpdate({ current: input.current, releases: gh.releases, repo, channel: 'github' }))
+  const assessed = assessUpdate({ current: input.current, releases: gh.releases, repo, channel: 'github' })
   if (assessed.ok === false) return { ok: false, error: assessed.message }
   return { ok: true, channel: 'github', releases: gh.releases, assessed }
 }
@@ -627,9 +618,6 @@ function installError(code, command, extra = {}) {
     const which = Number.isSafeInteger(pid) && pid > 1 ? `（进程 ${pid}）` : ''
     return `已经有一个安装进程还在跑${which}，这次没有再启动。等它结束再点「更新」。如果确认它已经卡死，先停掉那个进程，再点一次。`
   }
-  if (code === 'timeout' && extra.packageName) {
-    return `安装超时，已经停掉卡住的进程。新版本的文件已经在磁盘上，qq-bridge 的包名也改成了 ${extra.packageName}。请重启 dsh-web，设置页才会出来。`
-  }
   if (code === 'timeout' && extra.leftover) {
     return `安装超时，卡住的安装进程还没停干净。请先停掉 dsh 和 pnpm，再点一次「更新」。${hint}`
   }
@@ -639,13 +627,12 @@ function installError(code, command, extra = {}) {
   if (code === 'wrong-name') return '下到的安装包不是这个插件，没有安装。'
   if (code === 'wrong-version') return '下到的版本和发布标签对不上，没有安装。'
   if (code === 'bad-package') return '下到的安装包读不懂，没有安装。'
-  const base = `安装没有成功。${hint}`
-  if (extra.packageName) return `${base}Cordis 里 qq-bridge 的包名已经改成实际安装的 ${extra.packageName}。请重启 dsh-web，设置页才会出来。`
-  return base
+  return `安装没有成功。${hint}`
 }
 
-function restartError() {
-  return '新版本已经装上，但面板没有重启。请自己重启 dsh-web，装好的才会生效。重启后请刷新页面，登录口令可能会变。'
+function restartError(channel) {
+  const via = channel === 'npm' ? '从 npm ' : channel === 'github' ? '从 GitHub ' : ''
+  return `新版本已经${via}装上，但没能自动重启面板。请自己重启 dsh-web，然后刷新页面。登录口令可能会变。`
 }
 
 function headerValue(response, name) {
@@ -1538,46 +1525,75 @@ setTimeout(() => {
 }, delay)
 `.trim()
 
+/** 具体服务挂在 user@UID.service 下面时，重启要走用户实例，不能重启整个登录会话。 */
+function systemdUserManager(text) {
+  return /\/user@\d+\.service(?:\/|$)/.test(String(text ?? ''))
+}
+
+function systemdRunArgs(unit, user) {
+  const scope = user ? ['--user'] : []
+  return [
+    ...scope,
+    '--quiet',
+    '--collect',
+    '--on-active=3s',
+    '--timer-property=AccuracySec=1s',
+    '--',
+    'systemctl',
+    ...scope,
+    'restart',
+    unit,
+  ]
+}
+
+function cgroupText(input) {
+  if (input.cgroup != null) return String(input.cgroup)
+  const read = typeof input.readCgroup === 'function' ? input.readCgroup : readOwnCgroup
+  try {
+    return String(read() ?? '')
+  } catch {
+    return ''
+  }
+}
+
+function reexecPlan(input) {
+  return {
+    mode: 'reexec',
+    execPath: input.execPath || process.execPath,
+    argv: Array.isArray(input.argv) ? input.argv : process.argv.slice(1),
+    pid: input.pid || process.pid,
+    cwd: input.cwd || process.cwd(),
+    delayMs: RESTART_DELAY_MS,
+  }
+}
+
 /**
  * 重启计划。systemd 里只重启 cgroup 里那个具体服务，用 systemd-run 把动作放到服务外面。
- * 认不出单独服务时不重启整个登录会话。其余情况等响应发出后，按当前进程的命令行再拉起一次。
+ * 没传入 cgroup 时读本进程的 /proc/self/cgroup。用户服务加 --user。
+ * 认不出单独服务时计划是 manual：不重启整个登录会话。其余情况按当前进程的命令行再拉起一次。
  */
 export function planPanelRestart(input = {}) {
   const platform = input.platform || process.platform
   const env = input.env || process.env
   const invocation = platform !== 'win32' && Boolean(env && (env.INVOCATION_ID || env.NOTIFY_SOCKET))
-  const unit = invocation ? systemdUnitFromCgroup(input.cgroup ?? '') : null
-  if (invocation && !unit) return { mode: 'manual' }
+  const cgroup = invocation ? cgroupText(input) : ''
+  const unit = invocation ? systemdUnitFromCgroup(cgroup) : null
   if (unit) {
+    const user = systemdUserManager(cgroup)
     return {
       mode: 'systemd',
       command: 'systemd-run',
-      args: ['--quiet', '--collect', '--on-active=3s', '--timer-property=AccuracySec=1s', '--', 'systemctl', 'restart', unit],
+      args: systemdRunArgs(unit, user),
       unit,
+      scope: user ? 'user' : 'system',
     }
   }
-  return {
-    mode: 'reexec',
-    execPath: input.execPath || process.execPath,
-    argv: Array.isArray(input.argv) ? input.argv : process.argv.slice(1),
-    pid: input.pid || process.pid,
-    cwd: input.cwd || process.cwd(),
-    delayMs: RESTART_DELAY_MS,
-  }
+  if (invocation) return { mode: 'manual' }
+  return reexecPlan(input)
 }
 
-/**
- * @param {{ plan?: ReturnType<typeof planPanelRestart>, spawnImpl?: typeof spawn, env?: NodeJS.ProcessEnv, platform?: string, cgroup?: string, execPath?: string, argv?: string[], pid?: number, cwd?: string }} input
- */
-export async function schedulePanelRestart(input = {}) {
-  const plan = input.plan || planPanelRestart(input)
-  if (plan.mode === 'manual') return { ok: false, mode: 'manual', error: restartError() }
+async function spawnReexec(plan, input) {
   const spawnImpl = input.spawnImpl || spawn
-  if (plan.mode === 'systemd') {
-    const ran = await runChild(spawnImpl, plan.command, plan.args, { timeoutMs: 20000, stdio: 'ignore' })
-    if (!ran.ok) return { ok: false, mode: 'systemd', error: restartError() }
-    return { ok: true, mode: 'systemd' }
-  }
   const payload = JSON.stringify({
     pid: plan.pid,
     cwd: plan.cwd,
@@ -1607,6 +1623,29 @@ export async function schedulePanelRestart(input = {}) {
   return { ok: true, mode: 'reexec' }
 }
 
+/**
+ * 先按计划重启。systemd-run 失败，或认不出单独服务时，改为重启当前进程，不重启整个登录。
+ * 两条都做不到才返回失败。
+ * @param {{ plan?: ReturnType<typeof planPanelRestart>, spawnImpl?: typeof spawn, env?: NodeJS.ProcessEnv, platform?: string, cgroup?: string, readCgroup?: () => string, execPath?: string, argv?: string[], pid?: number, cwd?: string }} input
+ */
+export async function schedulePanelRestart(input = {}) {
+  const plan = input.plan || planPanelRestart(input)
+  const spawnImpl = input.spawnImpl || spawn
+  if (plan.mode === 'systemd') {
+    const ran = await runChild(spawnImpl, plan.command, plan.args, { timeoutMs: 20000, stdio: 'ignore' })
+    if (ran.ok) return { ok: true, mode: 'systemd', unit: plan.unit, scope: plan.scope }
+    const fallback = await spawnReexec(reexecPlan(input), { ...input, spawnImpl })
+    if (fallback.ok) return fallback
+    return { ok: false, mode: 'manual', error: restartError() }
+  }
+  if (plan.mode === 'manual') {
+    const fallback = await spawnReexec(reexecPlan(input), { ...input, spawnImpl })
+    if (fallback.ok) return fallback
+    return { ok: false, mode: 'manual', error: restartError() }
+  }
+  return spawnReexec(plan, { ...input, spawnImpl })
+}
+
 const CORDIS_ID = 'qq-bridge'
 
 function isWritablePackageName(name) {
@@ -1783,10 +1822,8 @@ function alignNote(input, expectedVersion) {
 }
 
 function successMessage(tag, channel, note = '') {
-  const via = channel === 'npm'
-    ? '这次是从 npm 装的。'
-    : '这一版在 npm 上还没有，所以从 GitHub 装上。'
-  const base = `已经装上 ${tag}，面板即将重启。请刷新页面；登录口令可能会变，用新的面板地址再打开。${via}现有设置还在，没有改动 NapCat。`
+  const via = channel === 'npm' ? 'npm' : 'GitHub'
+  const base = `已经从 ${via} 装上 ${tag}，面板即将重启。`
   return note ? `${base}${note}` : base
 }
 
@@ -1810,25 +1847,31 @@ function installRunSpec(input, extra) {
   }
 }
 
-/** 安装失败时，只有目标版本已经在磁盘上、并且进程已经停干净，才改包名。不在这里重启。 */
-function failInstall(input, decision, ran, command, install) {
+/** 磁盘上的已安装包是不是这次要装的版本。只读 package.json，不改文件。 */
+function installedMatches(input, version) {
+  const profileDir = profileDirOf(input)
+  if (profileDir === '' || !version) return false
+  const scoped = readInstalledManifest(join(profileDir, 'node_modules', '@fengye_nobug', 'dsh-qq-bot'))
+  const legacy = readInstalledManifest(join(profileDir, 'node_modules', 'dsh-qq-bot'))
+  return [scoped, legacy].some((item) => item && compareSemver(item.version, version) === 0)
+}
+
+/**
+ * 安装命令没返回成功时：进程已经停干净，并且磁盘上已经是目标版本，就改包名并安排重启。
+ * 进程还在，或版本对不上，不重启。
+ */
+async function failInstall(input, decision, ran, command, install, channel) {
   const code = ran?.code
   const leftover = ran?.leftover === true
-  let packageName = ''
-  const profileDir = profileDirOf(input)
-  if (!leftover && (code === 'timeout' || code === 'exit' || code === 'error') && profileDir !== '') {
-    try {
-      const aligned = alignCordisNames({
-        profileDir,
-        home: typeof input.home === 'string' ? input.home : '',
-        expectedVersion: decision.version,
-      })
-      if (aligned.changed.length > 0) packageName = aligned.name
-    } catch {
-      // 对不齐就只说明安装失败。
-    }
-  }
-  const message = installError(code, command, { packageName, leftover, pid: ran?.pid })
+  const via = channel === 'npm' || channel === 'github' ? channel : decision.channel
+  if (!leftover && (code === 'timeout' || code === 'exit' || code === 'error') && installedMatches(input, decision.version)) {
+    return finishApply(
+      input,
+      { ...decision, command: command || decision.command, install: install || decision.install },
+      via === 'npm' ? 'npm' : 'github',
+    )
+  }
+  const message = installError(code, command, { leftover, pid: ran?.pid })
   return {
     ok: false,
     stage: 'install',
@@ -1891,7 +1934,7 @@ export async function applyPluginUpdate(input = {}) {
       ran = { ok: false, code: 'error' }
     }
     if (ran?.ok === true) return finishApply(input, decision, 'npm')
-    if (ran?.code !== 'unpublished') return failInstall(input, decision, ran, decision.command, decision.install)
+    if (ran?.code !== 'unpublished') return failInstall(input, decision, ran, decision.command, decision.install, 'npm')
     channel = 'github'
   }
   const shown = shownInstall(channel, decision.version, repo)
@@ -1933,7 +1976,7 @@ export async function applyPluginUpdate(input = {}) {
     rmSync(dir, { recursive: true, force: true })
   }
   if (!ran || ran.ok !== true) {
-    return failInstall(input, { ...decision, install: shown.install, command: shown.command }, ran, shown.command, shown.install)
+    return failInstall(input, { ...decision, install: shown.install, command: shown.command }, ran, shown.command, shown.install, 'github')
   }
   return finishApply(input, { ...decision, install: shown.install, command: shown.command }, 'github')
 }
@@ -1943,12 +1986,22 @@ async function finishApply(input, decision, channel) {
   let restart
   try {
     const schedule = input.scheduleRestart || schedulePanelRestart
-    restart = await schedule({ spawnImpl: input.spawnImpl, platform: input.platform })
+    restart = await schedule({
+      spawnImpl: input.spawnImpl,
+      platform: input.platform,
+      env: input.env,
+      cgroup: input.cgroup,
+      readCgroup: input.readCgroup,
+      execPath: input.execPath,
+      argv: input.argv,
+      pid: input.pid,
+      cwd: input.cwd,
+    })
   } catch {
     restart = { ok: false, error: restartError() }
   }
   if (!restart || restart.ok !== true) {
-    const message = `${restart?.error || restartError()}${note}`
+    const message = `${restartError(channel)}${note}`
     return {
       ok: false,
       stage: 'restart',