为备份与重载插件补充资源清理逻辑

Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent)

Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>

Author: TiaraBasori

src/plugin/bf.ts

@@ -6,13 +6,26 @@ import * as fs from "fs";
 import * as path from "path";
 import * as crypto from "crypto";
 import * as os from "os";
-import { spawn } from "child_process";
+import { spawn, type ChildProcess } from "child_process";
 import { Low } from "lowdb";
 import { JSONFile } from "lowdb/node";
 import { getPrefixes } from "@utils/pluginManager";
 
 const prefixes = getPrefixes();
 const mainPrefix = prefixes[0];
+const activeChildProcesses = new Set<ChildProcess>();
+
+function trackChildProcess<T extends ChildProcess>(child: T): T {
+  const release = () => {
+    activeChildProcesses.delete(child);
+  };
+
+  activeChildProcesses.add(child);
+  child.once("close", release);
+  child.once("exit", release);
+  child.once("error", release);
+  return child;
+}
 
 // 时区设置
 const CN_TIME_ZONE = "Asia/Shanghai";
@@ -128,6 +141,10 @@ class ConfigManager {
     await this.setTargets(filtered);
     return filtered;
   }
+
+  static cleanup(): void {
+    this.db = null;
+  }
 }
 
 // 工具函数
@@ -169,13 +186,13 @@ async function createBackup(dirs: string[], outputPath: string): Promise<void> {
 
     // 创建tar.gz
     await new Promise<void>((resolve, reject) => {
-      const tar = spawn("tar", [
+      const tar = trackChildProcess(spawn("tar", [
         "-czf",
         outputPath,
         "-C",
         tempDir,
         "telebox_backup",
-      ]);
+      ]));
 
       tar.on("close", (code) => {
         if (code === 0) resolve();
@@ -216,7 +233,7 @@ async function extractBackup(archivePath: string): Promise<string> {
   fs.mkdirSync(extractDir, { recursive: true });
 
   await new Promise<void>((resolve, reject) => {
-    const tar = spawn("tar", ["-xzf", archivePath, "-C", extractDir]);
+    const tar = trackChildProcess(spawn("tar", ["-xzf", archivePath, "-C", extractDir]));
 
     tar.on("close", (code) => {
       if (code === 0) resolve();
@@ -279,6 +296,13 @@ const help_text = `<code>${mainPrefix}bf</code> 备份 plugins + assets 目录
 // 插件类
 class BfPlugin extends Plugin {
   cleanup(): void {
+    for (const child of activeChildProcesses) {
+      try {
+        child.kill("SIGTERM");
+      } catch {}
+    }
+    activeChildProcesses.clear();
+    ConfigManager.cleanup();
   }
 
   description = `\n📦 备份插件\n\n${help_text}
@@ -404,7 +428,7 @@ class BfPlugin extends Plugin {
           const dirName = path.basename(programDir);
 
           await new Promise<void>((resolve, reject) => {
-            const tar = spawn("tar", [
+            const tar = trackChildProcess(spawn("tar", [
               "-cf",
               "-",
               "-C",
@@ -415,9 +439,9 @@ class BfPlugin extends Plugin {
               "--exclude=temp",
               "--exclude=logs",
               dirName,
-            ], { stdio: ["pipe", "pipe", "pipe"] });
+            ], { stdio: ["pipe", "pipe", "pipe"] }));
 
-            const gzip = spawn("gzip", ["-1"], { stdio: ["pipe", "pipe", "pipe"] });
+            const gzip = trackChildProcess(spawn("gzip", ["-1"], { stdio: ["pipe", "pipe", "pipe"] }));
 
             const output = fs.createWriteStream(backupPath);
 

src/plugin/reload.ts

@@ -27,6 +27,7 @@ const exitDir = createDirectoryInTemp("exit");
 const exitFile = path.join(exitDir, "msg.json");
 const assetsDir = createDirectoryInAssets("reload");
 const configPath = path.join(assetsDir, "config.json");
+const pendingExitTimers = new Set<ReturnType<typeof setTimeout>>();
 
 interface ReloadConfig {
   leakfixEnabled: boolean;
@@ -43,6 +44,19 @@ async function initConfig() {
   return db;
 }
 
+function scheduleTrackedTimeout(
+  callback: () => void | Promise<void>,
+  delay: number
+): ReturnType<typeof setTimeout> {
+  let timer: ReturnType<typeof setTimeout>;
+  timer = setTimeout(() => {
+    pendingExitTimers.delete(timer);
+    void Promise.resolve(callback());
+  }, delay);
+  pendingExitTimers.add(timer);
+  return timer;
+}
+
 const editExitMsg = async () => {
   try {
     const data = fs.readFileSync(exitFile, "utf-8");
@@ -134,9 +148,9 @@ async function memoryMonitorTask() {
                    `正在重启 TeleBox...`,
           parseMode: "html"
         });
-        setTimeout(() => process.exit(0), 1000);
+        scheduleTrackedTimeout(() => process.exit(0), 1000);
       } else if (client) {
-        setTimeout(() => process.exit(0), 1000);
+        scheduleTrackedTimeout(() => process.exit(0), 1000);
       }
     } else {
       console.log(`[Memory Monitor] 内存使用 ${memory.heapUsed.toFixed(2)}MB / ${threshold}MB，正常`);
@@ -166,7 +180,11 @@ const HELP_TEXT = `🔄 Reload - 插件重载与内存管理
 
 class ReloadPlugin extends Plugin {
   cleanup(): void {
-    // 当前插件不持有需要在 reload 时额外释放的长期资源。
+    for (const timer of pendingExitTimers) {
+      clearTimeout(timer);
+    }
+    pendingExitTimers.clear();
+    this.lastReloadMemory = null;
   }
 
   description = HELP_TEXT;
@@ -218,7 +236,7 @@ class ReloadPlugin extends Plugin {
 
     pmr: async (msg) => {
       await msg.delete();
-      setTimeout(async () => {
+      scheduleTrackedTimeout(async () => {
         try {
           await execAsync("pm2 restart telebox");
         } catch (error) {
@@ -348,4 +366,4 @@ class ReloadPlugin extends Plugin {
   };
 }
 
-export default new ReloadPlugin();
+export default new ReloadPlugin();