aws-runtime-bridge 1.1.1 → 1.1.5

package/dist/index.js

@@ -32,6 +32,11 @@ import { logger } from "./utils/logger.js";
 import { autoRegister, unregister, bridgeRestartCleanup, } from "./services/auto-register.js";
 import { ensureAwsClientAgentMcpReleased } from "./services/aws-client-agent-mcp.js";
 import { ensureStartupConfig } from "./services/startup-config-wizard.js";
+import { handleCliCommand } from "./services/cli-commands.js";
+const cliCommandResult = await handleCliCommand();
+if (cliCommandResult.handled) {
+    process.exit(cliCommandResult.exitCode);
+}
 // 验证生产环境配置
 await ensureStartupConfig();
 validateProductionToken();

package/dist/services/auto-register.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"auto-register.d.ts","sourceRoot":"","sources":["../../src/services/auto-register.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AA8CH;;GAEG;AACH,UAAU,kBAAkB;IAC1B,eAAe;IACf,OAAO,EAAE,OAAO,CAAC;IACjB,oBAAoB;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,oCAAoC;IACpC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,mCAAmC;IACnC,OAAO,EAAE,MAAM,CAAC;IAChB,0BAA0B;IAC1B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,oBAAoB;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW;IACX,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,WAAW;IACX,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,aAAa;IACb,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,yCAAyC;IACzC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,4BAA4B;IAC5B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,mCAAmC;IACnC,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,aAAa;IACb,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,iBAAiB;IACjB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAwBD,wBAAgB,2BAA2B,IAAI,MAAM,EAAE,CAEtD;AA6DD;;GAEG;AACH,wBAAgB,6BAA6B,IAAI,MAAM,CAGtD;AAyJD;;;;;;GAMG;AACH,wBAAgB,UAAU,IAAI,kBAAkB,CA2C/C;AAED,wBAAgB,WAAW,IAAI,kBAAkB,EAAE,CA0ClD;AA8LD;;;;;;;;;;GAUG;AACH,wBAAsB,YAAY,CAChC,YAAY,CAAC,EAAE,OAAO,CAAC,kBAAkB,CAAC,GACzC,OAAO,CAAC,OAAO,CAAC,CAalB;AAyHD;;GAEG;AACH,wBAAsB,gCAAgC,IAAI,OAAO,CAAC;IAChE,OAAO,EAAE,OAAO,CAAC;IACjB,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,CAAC,CAoED;AAED,wBAAsB,UAAU,IAAI,OAAO,CAAC,OAAO,CAAC,CAuBnD;AAED;;GAEG;AACH,wBAAgB,oBAAoB;gBAnwBtB,OAAO;iBACN,MAAM;mBACJ,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;kBACvB,IAAI;YACV,MAAM;EAiwBf;AAED;;GAEG;AACH,wBAAgB,YAAY,IAAI,OAAO,CAEtC;AAED;;GAEG;AACH,wBAAgB,aAAa,IAAI,MAAM,GAAG,SAAS,CAElD;AAED;;;;;;;;;;;GAWG;AACH,wBAAsB,oBAAoB,IAAI,OAAO,CAAC;IACpD,OAAO,EAAE,OAAO,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,CAAC,CA0ED"}
+{"version":3,"file":"auto-register.d.ts","sourceRoot":"","sources":["../../src/services/auto-register.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AA+CH;;GAEG;AACH,UAAU,kBAAkB;IAC1B,eAAe;IACf,OAAO,EAAE,OAAO,CAAC;IACjB,oBAAoB;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,oCAAoC;IACpC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,mCAAmC;IACnC,OAAO,EAAE,MAAM,CAAC;IAChB,0BAA0B;IAC1B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,oBAAoB;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW;IACX,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,WAAW;IACX,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,aAAa;IACb,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,yCAAyC;IACzC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,4BAA4B;IAC5B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,mCAAmC;IACnC,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,aAAa;IACb,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,iBAAiB;IACjB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAwBD,wBAAgB,2BAA2B,IAAI,MAAM,EAAE,CAEtD;AA6DD;;GAEG;AACH,wBAAgB,6BAA6B,IAAI,MAAM,CAGtD;AAyJD;;;;;;GAMG;AACH,wBAAgB,UAAU,IAAI,kBAAkB,CA2C/C;AAED,wBAAgB,WAAW,IAAI,kBAAkB,EAAE,CA0ClD;AA4OD;;;;;;;;;;GAUG;AACH,wBAAsB,YAAY,CAChC,YAAY,CAAC,EAAE,OAAO,CAAC,kBAAkB,CAAC,GACzC,OAAO,CAAC,OAAO,CAAC,CAalB;AAyHD;;GAEG;AACH,wBAAsB,gCAAgC,IAAI,OAAO,CAAC;IAChE,OAAO,EAAE,OAAO,CAAC;IACjB,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,CAAC,CAqED;AAED,wBAAsB,UAAU,IAAI,OAAO,CAAC,OAAO,CAAC,CAuBnD;AAED;;GAEG;AACH,wBAAgB,oBAAoB;gBAlzBtB,OAAO;iBACN,MAAM;mBACJ,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;kBACvB,IAAI;YACV,MAAM;EAgzBf;AAED;;GAEG;AACH,wBAAgB,YAAY,IAAI,OAAO,CAEtC;AAED;;GAEG;AACH,wBAAgB,aAAa,IAAI,MAAM,GAAG,SAAS,CAElD;AAED;;;;;;;;;;;GAWG;AACH,wBAAsB,oBAAoB,IAAI,OAAO,CAAC;IACpD,OAAO,EAAE,OAAO,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,CAAC,CA2ED"}

package/dist/services/auto-register.js

@@ -13,7 +13,7 @@ import path from "node:path";
 import fs from "node:fs";
 import { logger } from "../utils/logger.js";
 import { schedulerBaseUrl, runtimeToken } from "../config.js";
-import { getRuntimeAccessToken, getRuntimeBindingPublicState, saveRuntimeBinding, } from "./runtime-binding.js";
+import { getRuntimeAccessToken, getRuntimeBindingPublicState, loadRuntimeBinding, saveRuntimeBinding, } from "./runtime-binding.js";
 // 默认配置
 const DEFAULT_CONFIG = {
     enabled: false,
@@ -410,6 +410,34 @@ function resolveRuntimeBridgeBaseUrl(config) {
         : getLocalIpAddress(config.serverUrl || schedulerBaseUrl);
     return `http://${config.registerIp || config.virtualIp || localIp}:${port}`;
 }
+function loadPrimaryLifecycleConfig() {
+    return loadConfigs()[0] || loadConfig();
+}
+function resolveRegisteredSchedulerBaseUrl(instanceId) {
+    if (!instanceId) {
+        return undefined;
+    }
+    const registration = Object.entries(registrationState.registrations).find(([, registeredInstanceId]) => registeredInstanceId === instanceId);
+    return registration?.[0];
+}
+function resolveLifecycleSchedulerBaseUrl(config, instanceId) {
+    const registeredServerUrl = resolveRegisteredSchedulerBaseUrl(instanceId);
+    if (registeredServerUrl) {
+        return registeredServerUrl;
+    }
+    const configuredServerUrl = normalizeOptionalString(config.serverUrl);
+    if (configuredServerUrl &&
+        (configuredServerUrl !== schedulerBaseUrl ||
+            Boolean(process.env.AWS_RUNTIME_SCHEDULER_BASE_URL))) {
+        return configuredServerUrl;
+    }
+    const binding = loadRuntimeBinding();
+    const pairedSchedulerBaseUrl = normalizeOptionalString(binding.schedulerBaseUrl);
+    if (binding.status === "paired" && pairedSchedulerBaseUrl) {
+        return pairedSchedulerBaseUrl;
+    }
+    return configuredServerUrl || schedulerBaseUrl;
+}
 /**
  * 执行注册请求
  */
@@ -445,7 +473,8 @@ async function doRegister(config) {
  * 执行注销请求
  */
 async function doUnregister(instanceId) {
-    const response = await axios.post(`${schedulerBaseUrl}/api/instances/unregister`, { instanceId }, {
+    const targetServerUrl = resolveLifecycleSchedulerBaseUrl(loadPrimaryLifecycleConfig(), instanceId);
+    const response = await axios.post(`${targetServerUrl}/api/instances/unregister`, { instanceId }, {
         headers: {
             "Content-Type": "application/json",
             "X-Runtime-Token": runtimeToken,
@@ -565,7 +594,7 @@ async function autoRegisterSingle(config) {
  * 注销实例
  */
 export async function requestRuntimeAccessTokenRefresh() {
-    const config = loadConfig();
+    const config = loadPrimaryLifecycleConfig();
     if (!config.userKey) {
         return {
             success: false,
@@ -573,11 +602,12 @@ export async function requestRuntimeAccessTokenRefresh() {
         };
     }
     const state = getRuntimeBindingPublicState();
-    const localIp = getLocalIpAddress();
+    const targetServerUrl = resolveLifecycleSchedulerBaseUrl(config);
+    const localIp = getLocalIpAddress(targetServerUrl);
     const runtimeBridgePort = process.env.AWS_RUNTIME_BRIDGE_PORT || 18081;
     const runtimeBridgeBaseUrl = `http://${config.registerIp || config.virtualIp || localIp}:${runtimeBridgePort}`;
     try {
-        const response = await axios.post(`${schedulerBaseUrl}/api/instances/runtime-tokens/refresh`, {
+        const response = await axios.post(`${targetServerUrl}/api/instances/runtime-tokens/refresh`, {
             tenantId: config.tenantId,
             userKey: config.userKey,
             instanceId: state.instanceId,
@@ -603,7 +633,7 @@ export async function requestRuntimeAccessTokenRefresh() {
             accessToken: response.data.runtimeAccessToken,
             instanceId: state.instanceId,
             userId: response.data.userId || config.userKey,
-            schedulerBaseUrl,
+            schedulerBaseUrl: targetServerUrl,
         });
         return {
             success: true,
@@ -674,8 +704,9 @@ export function getInstanceId() {
  * @returns 清理结果
  */
 export async function bridgeRestartCleanup() {
-    const config = loadConfig();
+    const config = loadPrimaryLifecycleConfig();
     const instanceId = registrationState.instanceId;
+    const targetServerUrl = resolveLifecycleSchedulerBaseUrl(config, instanceId);
     // 构建清理请求
     const requestBody = {};
     if (instanceId) {
@@ -683,7 +714,7 @@ export async function bridgeRestartCleanup() {
     }
     // 如果有配置，也传入 runtimeBridgeBaseUrl
     if (config.enabled) {
-        const localIp = getLocalIpAddress();
+        const localIp = getLocalIpAddress(targetServerUrl);
         const port = process.env.AWS_RUNTIME_BRIDGE_PORT || 18081;
         requestBody.runtimeBridgeBaseUrl = `http://${config.registerIp || config.virtualIp || localIp}:${port}`;
     }
@@ -695,7 +726,7 @@ export async function bridgeRestartCleanup() {
         logger.info(`[AutoRegister] Bridge 重启清理：正在通知调度中心...`);
         logger.info(`[AutoRegister]   instanceId: ${requestBody.instanceId || "(无)"}`);
         logger.info(`[AutoRegister]   runtimeBridgeBaseUrl: ${requestBody.runtimeBridgeBaseUrl || "(无)"}`);
-        const response = await axios.post(`${schedulerBaseUrl}/api/instances/bridge-restart-cleanup`, requestBody, {
+        const response = await axios.post(`${targetServerUrl}/api/instances/bridge-restart-cleanup`, requestBody, {
             headers: {
                 "Content-Type": "application/json",
                 "X-Runtime-Token": runtimeToken,

package/dist/services/auto-register.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=auto-register.test.d.ts.map

package/dist/services/auto-register.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"auto-register.test.d.ts","sourceRoot":"","sources":["../../src/services/auto-register.test.ts"],"names":[],"mappings":""}

package/dist/services/auto-register.test.js

@@ -0,0 +1,154 @@
+import { mkdirSync, mkdtempSync, rmSync, writeFileSync } from "node:fs";
+import os from "node:os";
+import path from "node:path";
+import { afterEach, describe, expect, it, vi } from "vitest";
+vi.mock("node:os", async () => {
+    const actual = await vi.importActual("node:os");
+    const mocked = {
+        ...actual,
+        homedir: () => process.env.AWS_TEST_HOME || actual.homedir(),
+    };
+    return {
+        ...mocked,
+        default: mocked,
+    };
+});
+vi.mock("axios", () => ({
+    default: {
+        post: vi.fn(),
+    },
+}));
+const originalEnv = { ...process.env };
+const tempRoots = [];
+function createRuntimeHome() {
+    const root = mkdtempSync(path.join(os.tmpdir(), "aws-auto-register-"));
+    tempRoots.push(root);
+    return root;
+}
+function writeAutoRegisterConfig(homeDir, targets) {
+    const configPath = path.join(homeDir, ".aws-bridge", "config.json");
+    const autoRegisterTargets = (typeof targets === "string"
+        ? [{ serverUrl: targets }]
+        : targets).map((target, index) => ({
+        serverUrl: target.serverUrl,
+        instanceName: target.instanceName || `test-bridge-${index + 1}`,
+        userKey: target.userKey || `aws_live_test_key_${index + 1}`,
+        registerIp: target.registerIp || "127.0.0.1",
+    }));
+    mkdirSync(path.dirname(configPath), { recursive: true });
+    writeFileSync(configPath, `${JSON.stringify({
+        connectionKey: "bridge-key",
+        autoRegisterTargets,
+    }, null, 2)}\n`, "utf-8");
+}
+function useRuntimeHome(runtimeHome) {
+    process.env.AWS_TEST_HOME = runtimeHome;
+    process.env.HOME = runtimeHome;
+    process.env.USERPROFILE = runtimeHome;
+    process.env.AWS_RUNTIME_HOME_DIR = runtimeHome;
+    process.env.AWS_RUNTIME_SCHEDULER_BASE_URL = "";
+}
+async function mockSuccessfulPost(response) {
+    const { default: axios } = await import("axios");
+    vi.mocked(axios.post).mockResolvedValue({ data: response });
+    return vi.mocked(axios.post);
+}
+afterEach(() => {
+    process.env = { ...originalEnv };
+    vi.resetModules();
+    vi.clearAllMocks();
+    vi.restoreAllMocks();
+    for (const root of tempRoots.splice(0)) {
+        rmSync(root, { recursive: true, force: true });
+    }
+});
+describe("auto-register scheduler URL selection", () => {
+    it("uses configured serverUrl for lifecycle callbacks when env scheduler URL is absent", async () => {
+        const runtimeHome = createRuntimeHome();
+        useRuntimeHome(runtimeHome);
+        writeAutoRegisterConfig(runtimeHome, "http://127.0.0.1:7380");
+        const post = await mockSuccessfulPost({
+            success: true,
+            instanceId: "bridge-instance-1",
+            userId: "user-1",
+            message: "ok",
+            runtimeAccessToken: "runtime-token-123456",
+        });
+        const { autoRegister, bridgeRestartCleanup, requestRuntimeAccessTokenRefresh, unregister } = await import("./auto-register.js");
+        await expect(autoRegister()).resolves.toBe(true);
+        await expect(bridgeRestartCleanup()).resolves.toMatchObject({ success: true });
+        post.mockResolvedValueOnce({
+            data: {
+                success: true,
+                runtimeAccessToken: "refreshed-token-123456",
+                userId: "user-1",
+            },
+        });
+        await expect(requestRuntimeAccessTokenRefresh()).resolves.toMatchObject({
+            success: true,
+            runtimeAccessToken: "refreshed-token-123456",
+        });
+        post.mockResolvedValueOnce({ data: { success: true, message: "unregistered" } });
+        await expect(unregister()).resolves.toBe(true);
+        expect(post.mock.calls[0]?.[0]).toBe("http://127.0.0.1:7380/api/instances/register");
+        expect(post.mock.calls[1]?.[0]).toBe("http://127.0.0.1:7380/api/instances/bridge-restart-cleanup");
+        expect(post.mock.calls[2]?.[0]).toBe("http://127.0.0.1:7380/api/instances/runtime-tokens/refresh");
+        expect(post.mock.calls[3]?.[0]).toBe("http://127.0.0.1:7380/api/instances/unregister");
+    });
+    it("uses the registered target URL instead of the first configured target", async () => {
+        const runtimeHome = createRuntimeHome();
+        useRuntimeHome(runtimeHome);
+        writeAutoRegisterConfig(runtimeHome, [
+            { serverUrl: "http://scheduler-a.local:7380", userKey: "aws_live_key_a" },
+            { serverUrl: "http://scheduler-b.local:7380", userKey: "aws_live_key_b" },
+        ]);
+        const post = await mockSuccessfulPost({ success: false, message: "unexpected" });
+        post.mockResolvedValueOnce({
+            data: {
+                success: true,
+                instanceId: "bridge-a",
+                userId: "user-a",
+                runtimeAccessToken: "runtime-token-a-123456",
+            },
+        });
+        post.mockResolvedValueOnce({
+            data: {
+                success: true,
+                instanceId: "bridge-b",
+                userId: "user-b",
+                runtimeAccessToken: "runtime-token-b-123456",
+            },
+        });
+        post.mockResolvedValueOnce({ data: { success: true, message: "cleanup ok" } });
+        const { autoRegister, bridgeRestartCleanup } = await import("./auto-register.js");
+        await expect(autoRegister()).resolves.toBe(true);
+        await expect(bridgeRestartCleanup()).resolves.toMatchObject({ success: true });
+        expect(post.mock.calls[0]?.[0]).toBe("http://scheduler-a.local:7380/api/instances/register");
+        expect(post.mock.calls[1]?.[0]).toBe("http://scheduler-b.local:7380/api/instances/register");
+        expect(post.mock.calls[2]?.[0]).toBe("http://scheduler-b.local:7380/api/instances/bridge-restart-cleanup");
+    });
+    it("falls back to paired runtime binding scheduler URL when config has no serverUrl", async () => {
+        const runtimeHome = createRuntimeHome();
+        useRuntimeHome(runtimeHome);
+        const post = await mockSuccessfulPost({
+            success: true,
+            runtimeAccessToken: "binding-refresh-token-123456",
+            userId: "user-from-binding",
+        });
+        const { saveRuntimeBinding } = await import("./runtime-binding.js");
+        const { requestRuntimeAccessTokenRefresh } = await import("./auto-register.js");
+        saveRuntimeBinding({
+            accessToken: "existing-binding-token-123456",
+            instanceId: "bridge-from-binding",
+            userId: "user-from-binding",
+            schedulerBaseUrl: "http://paired-scheduler.local:7380",
+        });
+        process.env.AWS_AUTO_REGISTER = "true";
+        process.env.AWS_USER_KEY = "aws_live_env_key";
+        await expect(requestRuntimeAccessTokenRefresh()).resolves.toMatchObject({
+            success: true,
+            runtimeAccessToken: "binding-refresh-token-123456",
+        });
+        expect(post.mock.calls[0]?.[0]).toBe("http://paired-scheduler.local:7380/api/instances/runtime-tokens/refresh");
+    });
+});

package/dist/services/aws-client-agent-mcp.d.ts

@@ -14,11 +14,16 @@ export interface ResolveAwsClientAgentMcpCommandOptions {
     exists?: (filePath: string) => boolean;
     release?: () => string | null;
 }
+export interface ReleaseBundledAwsClientAgentMcpOptions {
+    packageRoot?: string;
+    releasedRoot?: string;
+    now?: () => string;
+}
 export interface AwsMcpPreparedInfo extends AwsMcpCommandSpec {
     source: "override" | "bundled" | "global";
 }
 export declare function getReleasedMcpDistPath(): string;
-export declare function releaseBundledAwsClientAgentMcp(): string | null;
+export declare function releaseBundledAwsClientAgentMcp(options?: ReleaseBundledAwsClientAgentMcpOptions): string | null;
 export declare function resolveAwsClientAgentMcpCommand(options?: ResolveAwsClientAgentMcpCommandOptions): AwsMcpPreparedInfo;
 export declare function getAwsClientAgentMcpPreparedInfo(options?: ResolveAwsClientAgentMcpCommandOptions): AwsMcpPreparedInfo;
 export declare function buildAwsMcpServerConfig(input: AwsMcpServerConfigInput): AwsMcpServerConfig;

package/dist/services/aws-client-agent-mcp.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"aws-client-agent-mcp.d.ts","sourceRoot":"","sources":["../../src/services/aws-client-agent-mcp.ts"],"names":[],"mappings":"AAWA,eAAO,MAAM,mBAAmB,YAAY,CAAC;AAuB7C,MAAM,WAAW,uBAAuB;IACtC,OAAO,EAAE,MAAM,CAAC;IAChB,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,EAAE,CAAC;CAChB;AAED,MAAM,WAAW,kBAAmB,SAAQ,iBAAiB;IAC3D,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAC7B;AAED,MAAM,WAAW,sCAAsC;IACrD,MAAM,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC;IACvC,OAAO,CAAC,EAAE,MAAM,MAAM,GAAG,IAAI,CAAC;CAC/B;AAED,MAAM,WAAW,kBAAmB,SAAQ,iBAAiB;IAC3D,MAAM,EAAE,UAAU,GAAG,SAAS,GAAG,QAAQ,CAAC;CAC3C;AAqBD,wBAAgB,sBAAsB,IAAI,MAAM,CAE/C;AAMD,wBAAgB,+BAA+B,IAAI,MAAM,GAAG,IAAI,CAsB/D;AA2ED,wBAAgB,+BAA+B,CAC7C,OAAO,GAAE,sCAA2C,GACnD,kBAAkB,CAEpB;AAED,wBAAgB,gCAAgC,CAC9C,OAAO,GAAE,sCAA2C,GACnD,kBAAkB,CA+BpB;AAED,wBAAgB,uBAAuB,CACrC,KAAK,EAAE,uBAAuB,GAC7B,kBAAkB,CAwBpB;AAED,wBAAgB,+BAA+B,IAAI,IAAI,CAgBtD"}
+{"version":3,"file":"aws-client-agent-mcp.d.ts","sourceRoot":"","sources":["../../src/services/aws-client-agent-mcp.ts"],"names":[],"mappings":"AAqBA,eAAO,MAAM,mBAAmB,YAAY,CAAC;AAuB7C,MAAM,WAAW,uBAAuB;IACtC,OAAO,EAAE,MAAM,CAAC;IAChB,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,EAAE,CAAC;CAChB;AAED,MAAM,WAAW,kBAAmB,SAAQ,iBAAiB;IAC3D,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAC7B;AAED,MAAM,WAAW,sCAAsC;IACrD,MAAM,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC;IACvC,OAAO,CAAC,EAAE,MAAM,MAAM,GAAG,IAAI,CAAC;CAC/B;AAED,MAAM,WAAW,sCAAsC;IACrD,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,GAAG,CAAC,EAAE,MAAM,MAAM,CAAC;CACpB;AAaD,MAAM,WAAW,kBAAmB,SAAQ,iBAAiB;IAC3D,MAAM,EAAE,UAAU,GAAG,SAAS,GAAG,QAAQ,CAAC;CAC3C;AAqBD,wBAAgB,sBAAsB,IAAI,MAAM,CAE/C;AA0FD,wBAAgB,+BAA+B,CAC7C,OAAO,GAAE,sCAA2C,GACnD,MAAM,GAAG,IAAI,CAgDf;AA2ED,wBAAgB,+BAA+B,CAC7C,OAAO,GAAE,sCAA2C,GACnD,kBAAkB,CAEpB;AAED,wBAAgB,gCAAgC,CAC9C,OAAO,GAAE,sCAA2C,GACnD,kBAAkB,CAsBpB;AAED,wBAAgB,uBAAuB,CACrC,KAAK,EAAE,uBAAuB,GAC7B,kBAAkB,CAwBpB;AAED,wBAAgB,+BAA+B,IAAI,IAAI,CAgBtD"}

package/dist/services/aws-client-agent-mcp.js

@@ -1,9 +1,10 @@
-import { cpSync, existsSync, mkdirSync } from "node:fs";
+import { createHash } from "node:crypto";
+import { cpSync, existsSync, mkdirSync, readdirSync, readFileSync, rmSync, statSync, writeFileSync, } from "node:fs";
 import path from "node:path";
 import { fileURLToPath } from "node:url";
 import { getRuntimeHomeDir, port, schedulerBaseUrl } from "../config.js";
-import { getRuntimeAccessToken, loadRuntimeBinding, } from "./runtime-binding.js";
 import { logger } from "../utils/logger.js";
+import { getRuntimeAccessToken, loadRuntimeBinding, } from "./runtime-binding.js";
 export const AWS_MCP_SERVER_NAME = "aws-mcp";
 const AWS_MCP_ALLOWED_ENV_KEYS = [
     "AWS_INTERNAL_API_KEY",
@@ -30,8 +31,8 @@ function getPackageRoot() {
     const currentFile = fileURLToPath(import.meta.url);
     return path.resolve(path.dirname(currentFile), "..", "..");
 }
-function getBundledEntryPath() {
-    return path.join(getPackageRoot(), "package", "aws-client-agent-mcp", "dist", "index.js");
+function getBundledEntryPath(packageRoot = getPackageRoot()) {
+    return path.join(packageRoot, "package", "aws-client-agent-mcp", "dist", "index.js");
 }
 function getReleasedMcpRoot() {
     return path.join(getRuntimeHomeDir(), ".aws-bridge", "mcp");
@@ -39,24 +40,107 @@ function getReleasedMcpRoot() {
 export function getReleasedMcpDistPath() {
     return path.join(getReleasedMcpRoot(), "dist");
 }
-function getReleasedEntryPath() {
-    return path.join(getReleasedMcpDistPath(), "index.js");
+function getReleaseManifestPath(releasedRoot = getReleasedMcpRoot()) {
+    return path.join(releasedRoot, ".release.json");
+}
+function readPackageVersion(packageJsonPath) {
+    try {
+        const metadata = JSON.parse(readFileSync(packageJsonPath, "utf-8"));
+        return metadata.version || "unknown";
+    }
+    catch {
+        return "unknown";
+    }
+}
+function getBridgeVersion(packageRoot) {
+    return readPackageVersion(path.join(packageRoot, "package.json"));
+}
+function getBundledMcpVersion(packageRoot) {
+    return readPackageVersion(path.join(packageRoot, "package", "aws-client-agent-mcp", "package.json"));
+}
+function listDistFiles(dirPath, rootPath = dirPath) {
+    return readdirSync(dirPath, { withFileTypes: true }).flatMap((entry) => {
+        const entryPath = path.join(dirPath, entry.name);
+        if (entry.isDirectory()) {
+            return listDistFiles(entryPath, rootPath);
+        }
+        if (!entry.isFile()) {
+            return [];
+        }
+        return [path.relative(rootPath, entryPath).split(path.sep).join("/")];
+    });
+}
+function hashDistDirectory(distPath) {
+    const hash = createHash("sha256");
+    for (const relativePath of listDistFiles(distPath).sort()) {
+        const filePath = path.join(distPath, ...relativePath.split("/"));
+        const stat = statSync(filePath);
+        hash.update(relativePath);
+        hash.update("\0");
+        hash.update(String(stat.size));
+        hash.update("\0");
+        hash.update(readFileSync(filePath));
+        hash.update("\0");
+    }
+    return `sha256:${hash.digest("hex")}`;
 }
-export function releaseBundledAwsClientAgentMcp() {
-    const bundledDist = path.dirname(getBundledEntryPath());
-    const releasedDist = getReleasedMcpDistPath();
+function readReleaseManifest(releasedRoot) {
+    try {
+        const parsed = JSON.parse(readFileSync(getReleaseManifestPath(releasedRoot), "utf-8"));
+        if (typeof parsed.bridgeVersion === "string" &&
+            typeof parsed.mcpVersion === "string" &&
+            typeof parsed.distHash === "string") {
+            return {
+                bridgeVersion: parsed.bridgeVersion,
+                mcpVersion: parsed.mcpVersion,
+                distHash: parsed.distHash,
+                releasedAt: typeof parsed.releasedAt === "string" ? parsed.releasedAt : "",
+            };
+        }
+    }
+    catch {
+        return null;
+    }
+    return null;
+}
+function shouldRefreshReleasedMcp(releasedEntry, manifest, nextManifest) {
+    return (!existsSync(releasedEntry) ||
+        !manifest ||
+        manifest.bridgeVersion !== nextManifest.bridgeVersion ||
+        manifest.mcpVersion !== nextManifest.mcpVersion ||
+        manifest.distHash !== nextManifest.distHash);
+}
+export function releaseBundledAwsClientAgentMcp(options = {}) {
+    const packageRoot = options.packageRoot || getPackageRoot();
+    const releasedRoot = options.releasedRoot || getReleasedMcpRoot();
+    const bundledDist = path.dirname(getBundledEntryPath(packageRoot));
+    const releasedDist = path.join(releasedRoot, "dist");
     const releasedEntry = path.join(releasedDist, "index.js");
     if (!existsSync(path.join(bundledDist, "index.js"))) {
         return null;
     }
-    if (!existsSync(releasedEntry)) {
-        mkdirSync(releasedDist, { recursive: true });
+    const nextManifest = {
+        bridgeVersion: getBridgeVersion(packageRoot),
+        mcpVersion: getBundledMcpVersion(packageRoot),
+        distHash: hashDistDirectory(bundledDist),
+    };
+    const currentManifest = readReleaseManifest(releasedRoot);
+    if (shouldRefreshReleasedMcp(releasedEntry, currentManifest, nextManifest)) {
+        rmSync(releasedDist, { recursive: true, force: true });
+        mkdirSync(releasedRoot, { recursive: true });
         cpSync(bundledDist, releasedDist, {
             errorOnExist: false,
             force: true,
             recursive: true,
         });
-        logger.info(`[runtime-bridge] ${AWS_MCP_SERVER_NAME} MCP 已释放到: ${releasedDist}`);
+        writeFileSync(getReleaseManifestPath(releasedRoot), `${JSON.stringify({
+            ...nextManifest,
+            releasedAt: options.now ? options.now() : new Date().toISOString(),
+        }, null, 2)}\n`, "utf-8");
+        logger.info(`[runtime-bridge] ${AWS_MCP_SERVER_NAME} MCP 已更新到: ${releasedDist}`);
+    }
+    else {
+        logger.info(`[runtime-bridge] ${AWS_MCP_SERVER_NAME} MCP 已是最新: ${releasedDist}`);
     }
     return releasedEntry;
 }
@@ -125,14 +209,6 @@ export function getAwsClientAgentMcpPreparedInfo(options = {}) {
         };
     }
     const exists = options.exists || existsSync;
-    const releasedEntry = getReleasedEntryPath();
-    if (exists(releasedEntry)) {
-        return {
-            source: "bundled",
-            command: process.execPath,
-            args: [releasedEntry],
-        };
-    }
     const release = options.release || releaseBundledAwsClientAgentMcp;
     const released = release();
     if (released && exists(released)) {

package/dist/services/aws-client-agent-mcp.test.js

@@ -1,4 +1,4 @@
-import { mkdtempSync } from 'node:fs';
+import { existsSync, mkdirSync, mkdtempSync, readFileSync, rmSync, writeFileSync, } from 'node:fs';
 import os from 'node:os';
 import path from 'node:path';
 import { afterEach, describe, expect, it, vi } from 'vitest';
@@ -8,6 +8,15 @@ afterEach(() => {
     vi.resetModules();
     vi.restoreAllMocks();
 });
+function createBundledMcpPackage(version = '1.0.0') {
+    const packageRoot = mkdtempSync(path.join(os.tmpdir(), 'aws-mcp-package-'));
+    const bundledDist = path.join(packageRoot, 'package', 'aws-client-agent-mcp', 'dist');
+    mkdirSync(bundledDist, { recursive: true });
+    writeFileSync(path.join(packageRoot, 'package.json'), JSON.stringify({ name: 'aws-runtime-bridge', version: '9.9.9' }), 'utf-8');
+    writeFileSync(path.join(packageRoot, 'package', 'aws-client-agent-mcp', 'package.json'), JSON.stringify({ name: 'aws-client-agent-mcp', version }), 'utf-8');
+    writeFileSync(path.join(bundledDist, 'index.js'), 'console.log("mcp v1");\n', 'utf-8');
+    return packageRoot;
+}
 describe('aws-client-agent-mcp service', () => {
     it('prefers explicit command override from env', async () => {
         process.env.AWS_CLIENT_AGENT_MCP_COMMAND = 'custom-aws-client-agent-mcp';
@@ -19,18 +28,79 @@ describe('aws-client-agent-mcp service', () => {
             args: ['--stdio', '--trace'],
         });
     });
-    it('uses bundled dist entry when present', async () => {
+    it('uses bundled dist entry after release check', async () => {
         process.env.AWS_CLIENT_AGENT_MCP_COMMAND = '';
         const mod = await import('./aws-client-agent-mcp.js');
-        const exists = vi.fn().mockReturnValue(true);
-        const resolved = mod.resolveAwsClientAgentMcpCommand({ exists });
         const releasedEntry = path.join(os.homedir(), '.aws-bridge', 'mcp', 'dist', 'index.js');
+        const exists = vi.fn((filePath) => filePath === releasedEntry);
+        const release = vi.fn(() => releasedEntry);
+        const resolved = mod.resolveAwsClientAgentMcpCommand({ exists, release });
         expect(resolved).toEqual({
             source: 'bundled',
             command: process.execPath,
             args: [releasedEntry],
         });
         expect(exists).toHaveBeenCalled();
+        expect(release).toHaveBeenCalledOnce();
+    });
+    it('writes release manifest and skips copy when bundled mcp is unchanged', async () => {
+        process.env.AWS_CLIENT_AGENT_MCP_COMMAND = '';
+        const packageRoot = createBundledMcpPackage('1.0.0');
+        const releasedRoot = mkdtempSync(path.join(os.tmpdir(), 'aws-mcp-release-'));
+        try {
+            const mod = await import('./aws-client-agent-mcp.js');
+            const firstRelease = mod.releaseBundledAwsClientAgentMcp({
+                packageRoot,
+                releasedRoot,
+                now: () => '2026-05-04T00:00:00.000Z',
+            });
+            const releasedIndex = path.join(releasedRoot, 'dist', 'index.js');
+            const manifestPath = path.join(releasedRoot, '.release.json');
+            expect(firstRelease).toBe(releasedIndex);
+            expect(existsSync(releasedIndex)).toBe(true);
+            const firstManifest = JSON.parse(readFileSync(manifestPath, 'utf-8'));
+            expect(firstManifest).toMatchObject({
+                bridgeVersion: '9.9.9',
+                mcpVersion: '1.0.0',
+                releasedAt: '2026-05-04T00:00:00.000Z',
+            });
+            expect(firstManifest.distHash).toMatch(/^sha256:/);
+            writeFileSync(releasedIndex, 'console.log("local marker");\n', 'utf-8');
+            mod.releaseBundledAwsClientAgentMcp({
+                packageRoot,
+                releasedRoot,
+                now: () => '2026-05-05T00:00:00.000Z',
+            });
+            expect(readFileSync(releasedIndex, 'utf-8')).toBe('console.log("local marker");\n');
+            expect(JSON.parse(readFileSync(manifestPath, 'utf-8')).releasedAt).toBe('2026-05-04T00:00:00.000Z');
+        }
+        finally {
+            rmSync(packageRoot, { recursive: true, force: true });
+            rmSync(releasedRoot, { recursive: true, force: true });
+        }
+    });
+    it('refreshes released mcp when bundled dist changes', async () => {
+        process.env.AWS_CLIENT_AGENT_MCP_COMMAND = '';
+        const packageRoot = createBundledMcpPackage('1.0.0');
+        const releasedRoot = mkdtempSync(path.join(os.tmpdir(), 'aws-mcp-release-'));
+        try {
+            const mod = await import('./aws-client-agent-mcp.js');
+            const bundledIndex = path.join(packageRoot, 'package', 'aws-client-agent-mcp', 'dist', 'index.js');
+            const releasedIndex = path.join(releasedRoot, 'dist', 'index.js');
+            mod.releaseBundledAwsClientAgentMcp({ packageRoot, releasedRoot });
+            writeFileSync(bundledIndex, 'console.log("mcp v2");\n', 'utf-8');
+            mod.releaseBundledAwsClientAgentMcp({
+                packageRoot,
+                releasedRoot,
+                now: () => '2026-05-05T00:00:00.000Z',
+            });
+            expect(readFileSync(releasedIndex, 'utf-8')).toBe('console.log("mcp v2");\n');
+            expect(JSON.parse(readFileSync(path.join(releasedRoot, '.release.json'), 'utf-8')).releasedAt).toBe('2026-05-05T00:00:00.000Z');
+        }
+        finally {
+            rmSync(packageRoot, { recursive: true, force: true });
+            rmSync(releasedRoot, { recursive: true, force: true });
+        }
     });
     it('releases bundled mcp into home directory when runtime dist is missing', async () => {
         process.env.AWS_CLIENT_AGENT_MCP_COMMAND = '';

package/dist/services/cli-commands.d.ts

@@ -0,0 +1,20 @@
+import type { StartupConfigWizardResult } from "./startup-config-wizard.js";
+export interface CliCommandResult {
+    handled: boolean;
+    exitCode: number;
+}
+export interface CommandRunResult {
+    status: number | null;
+    error?: Error;
+}
+export interface CliCommandOptions {
+    argv?: string[];
+    packageRoot?: string;
+    stdout?: (message: string) => void;
+    stderr?: (message: string) => void;
+    runCommand?: (command: string, args: string[]) => CommandRunResult;
+    configure?: () => Promise<StartupConfigWizardResult>;
+}
+export declare function readPackageVersion(packageRoot?: string): string;
+export declare function handleCliCommand(options?: CliCommandOptions): Promise<CliCommandResult>;
+//# sourceMappingURL=cli-commands.d.ts.map

package/dist/services/cli-commands.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli-commands.d.ts","sourceRoot":"","sources":["../../src/services/cli-commands.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,yBAAyB,EAAE,MAAM,4BAA4B,CAAC;AAG5E,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,gBAAgB;IAC/B,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,KAAK,CAAC,EAAE,KAAK,CAAC;CACf;AAED,MAAM,WAAW,iBAAiB;IAChC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IACnC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IACnC,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,gBAAgB,CAAC;IACnE,SAAS,CAAC,EAAE,MAAM,OAAO,CAAC,yBAAyB,CAAC,CAAC;CACtD;AAYD,wBAAgB,kBAAkB,CAAC,WAAW,SAA0B,GAAG,MAAM,CAMhF;AAcD,wBAAsB,gBAAgB,CACpC,OAAO,GAAE,iBAAsB,GAC9B,OAAO,CAAC,gBAAgB,CAAC,CAqC3B"}