@actiondock/core 2.0.1 → 2.0.3

package/src/server/types.ts

@@ -1,5 +1,16 @@
 import type { ServerRuntimeRegistry } from "./runtime-registry";
 
+export interface CoreHttpServerInstance {
+  port: number;
+  stop: (closeActiveConnections?: boolean) => void | Promise<void>;
+}
+
+export type CoreHttpServerFactory = (options: {
+  port: number;
+  host: string;
+  fetch: (req: Request) => Promise<Response>;
+}) => CoreHttpServerInstance | Promise<CoreHttpServerInstance>;
+
 /**
  * 启动 ActionDock HTTP Runner 服务端的配置选项。
  */
@@ -22,6 +33,10 @@ export interface ServerOptions {
   maxBodyBytes?: number;
   /** 是否在 health 和 info 接口中透传本地 projectRoot 等调试路径 */
   exposeDebugInfo?: boolean;
+  /** 是否启用一体化 MCP 协议支持（默认开启） */
+  enableMcp?: boolean;
+  /** 自定义 MCP 请求处理器钩子（若挂载则 /mcp 路由交由其处理） */
+  mcpHandler?: (req: Request) => Promise<Response | null | undefined> | Response | null | undefined;
 }
 
 /**

package/src/storage/driver.ts

@@ -0,0 +1,126 @@
+import { createRequire } from "node:module";
+import type { SqliteDriver } from "./types";
+
+export type SqliteDriverFactory = (dbPath: string) => SqliteDriver;
+
+/** ESM 环境下可用的 CommonJS require，用于动态加载 node:sqlite / bun:sqlite */
+const cjsRequire = createRequire(import.meta.url);
+
+let customDriverFactory: SqliteDriverFactory | undefined;
+
+/**
+ * 注册全局默认 SQLite 驱动工厂。
+ */
+export function setSqliteDriverFactory(factory: SqliteDriverFactory): void {
+  customDriverFactory = factory;
+}
+
+/**
+ * 创建默认 SQLite 驱动实例。
+ * 优先使用外部注册的工厂，其次根据当前运行时环境自动适配。
+ */
+export function createDefaultSqliteDriver(dbPath: string): SqliteDriver {
+  if (customDriverFactory) {
+    return customDriverFactory(dbPath);
+  }
+
+  // 检查是否在 Bun 运行时环境
+  if (typeof (globalThis as any).Bun !== "undefined") {
+    try {
+      const { Database } = (globalThis as any).Bun.sqlite || cjsRequire("bun:sqlite");
+      const db = new Database(dbPath);
+      // bun:sqlite 的 statement 未 finalize 时会一直持有数据库文件句柄，
+      // Windows 下导致 db.close() 后文件仍被锁（EBUSY 无法删除）。
+      // 跟踪全部 prepared statement，close() 时统一 finalize 释放句柄。
+      const openStatements = new Set<any>();
+      return {
+        exec(sql: string) {
+          db.exec(sql);
+        },
+        prepare(sql: string) {
+          const stmt = db.prepare(sql);
+          openStatements.add(stmt);
+          return {
+            run(...args: any[]) {
+              const params = args.length === 1 && Array.isArray(args[0]) ? args[0] : args;
+              const res = stmt.run(...params);
+              return { changes: res.changes, lastInsertRowid: res.lastInsertRowid };
+            },
+            get<T>(...args: any[]): T | undefined {
+              const params = args.length === 1 && Array.isArray(args[0]) ? args[0] : args;
+              return stmt.get(...params) as T | undefined;
+            },
+            all<T>(...args: any[]): T[] {
+              const params = args.length === 1 && Array.isArray(args[0]) ? args[0] : args;
+              return stmt.all(...params) as T[];
+            },
+          };
+        },
+        transaction<T>(fn: () => T extends PromiseLike<unknown> ? never : T): T {
+          return db.transaction(fn)() as T;
+        },
+        close() {
+          for (const stmt of openStatements) {
+            try {
+              stmt.finalize();
+            } catch {
+              // 已 finalize 或重复释放时忽略
+            }
+          }
+          openStatements.clear();
+          db.close();
+        },
+      };
+    } catch {
+      // 若在 Bun 下获取 bun:sqlite 失败，回退到标准 Node 驱动尝试
+    }
+  }
+
+  // 在 Node.js 环境下使用 node:sqlite
+  try {
+    const { DatabaseSync } = cjsRequire("node:sqlite");
+    const db = new DatabaseSync(dbPath);
+    return {
+      exec(sql: string) {
+        db.exec(sql);
+      },
+      prepare(sql: string) {
+        const stmt = db.prepare(sql);
+        return {
+          run(...args: any[]) {
+            const params = args.length === 1 && Array.isArray(args[0]) ? args[0] : args;
+            const res = stmt.run(...params);
+            return { changes: res.changes, lastInsertRowid: res.lastInsertRowid };
+          },
+          get<T>(...args: any[]): T | undefined {
+            const params = args.length === 1 && Array.isArray(args[0]) ? args[0] : args;
+            return stmt.get(...params) as T | undefined;
+          },
+          all<T>(...args: any[]): T[] {
+            const params = args.length === 1 && Array.isArray(args[0]) ? args[0] : args;
+            return stmt.all(...params) as T[];
+          },
+        };
+      },
+      transaction<T>(fn: () => T extends PromiseLike<unknown> ? never : T): T {
+        db.exec("BEGIN");
+        try {
+          const res = fn();
+          if (res && typeof (res as any).then === "function") {
+            throw new Error("Async transactions are not allowed in SQLite");
+          }
+          db.exec("COMMIT");
+          return res;
+        } catch (e) {
+          db.exec("ROLLBACK");
+          throw e;
+        }
+      },
+      close() {
+        db.close();
+      },
+    };
+  } catch (err: any) {
+    throw new Error(`Failed to initialize SQLite driver: ${err?.message || String(err)}`);
+  }
+}

package/src/storage/index.ts

@@ -1,8 +1,9 @@
-import { homedir } from "node:os";
 import { join } from "node:path";
+import { getActionDockHome } from "../utils";
 import { SqliteRuntimeStorage } from "./sqlite";
 import type { RuntimeStorage, StorageOptions } from "./types";
 
+export * from "./driver";
 export * from "./mask";
 export * from "./sqlite";
 export * from "./types";
@@ -33,7 +34,7 @@ export function resolveDatabasePath(
     return join(options.projectRoot, ".actiondock", "runtime.db");
   }
   // 独立执行二进制默认存储路径: ~/.actiondock/data/<package-id>/runtime.db
-  return join(homedir(), ".actiondock", "data", packageId, "runtime.db");
+  return join(getActionDockHome(), ".actiondock", "data", packageId, "runtime.db");
 }
 
 /**
@@ -57,7 +58,7 @@ export function createStorage(
  * @param customHome 自定义家目录路径（可选）
  */
 export function createGlobalStorage(customHome?: string): RuntimeStorage {
-  const baseDir = customHome || process.env.ACTIONDOCK_HOME || homedir();
+  const baseDir = getActionDockHome(customHome);
   const dbPath = join(baseDir, ".actiondock", "global.db");
   return new SqliteRuntimeStorage({ dbPath, packageId: "__global__" });
 }