aws-runtime-bridge 1.7.8 → 1.7.9

package/dist/config.d.ts

@@ -7,7 +7,7 @@
 export declare const port: number;
 /** 默认调度器基础 URL */
 export declare const DEFAULT_SCHEDULER_BASE_URL = "http://localhost:8080";
-export type SchedulerBaseUrlSource = "env" | "single-auto-register-target" | "default" | "ambiguous-auto-register-targets";
+export type SchedulerBaseUrlSource = "request" | "env" | "single-auto-register-target" | "default" | "ambiguous-auto-register-targets";
 export interface SchedulerBaseUrlResolution {
     url: string;
     source: SchedulerBaseUrlSource;
@@ -18,6 +18,11 @@ export interface SchedulerBaseUrlResolution {
  * 主流程：显式环境变量优先；未配置时，单个自动注册目标可作为宿主机部署的零额外配置回退；多目标不静默取第一个。
  */
 export declare function resolveSchedulerBaseUrl(): SchedulerBaseUrlResolution;
+/**
+ * 解析调度中心基础 URL。
+ * 主流程：服务端请求携带的 scheduler 地址优先；再读显式环境变量；最后回退到单个自动注册目标或默认值。
+ */
+export declare function resolveSchedulerBaseUrlFrom(requestSchedulerBaseUrl: unknown): SchedulerBaseUrlResolution;
 /** 调度器基础 URL（兼容旧导入；请求处理应优先调用 resolveSchedulerBaseUrl 获取最新值） */
 export declare const schedulerBaseUrl: string;
 /** Node 环境 */

package/dist/config.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAMH,WAAW;AACX,eAAO,MAAM,IAAI,EAAE,MAElB,CAAC;AAEF,kBAAkB;AAClB,eAAO,MAAM,0BAA0B,0BAA0B,CAAC;AAElE,MAAM,MAAM,sBAAsB,GAC9B,KAAK,GACL,6BAA6B,GAC7B,SAAS,GACT,iCAAiC,CAAC;AAEtC,MAAM,WAAW,0BAA0B;IACzC,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,EAAE,sBAAsB,CAAC;IAC/B,mBAAmB,EAAE,MAAM,EAAE,CAAC;CAC/B;AA+ED;;;GAGG;AACH,wBAAgB,uBAAuB,IAAI,0BAA0B,CAiCpE;AAED,+DAA+D;AAC/D,eAAO,MAAM,gBAAgB,EAAE,MAEH,CAAC;AAE7B,cAAc;AACd,eAAO,MAAM,OAAO,EAAE,MAA8C,CAAC;AAErE,oCAAoC;AACpC,eAAO,MAAM,oBAAoB,EAAE,OACiB,CAAC;AAErD,8BAA8B;AAC9B,eAAO,MAAM,kBAAkB,EAAE,MAAM,EAMC,CAAC;AAEzC,gEAAgE;AAChE,wBAAgB,uBAAuB,IAAI,IAAI,CAAG;AAElD,eAAe;AACf,wBAAgB,iBAAiB,IAAI,MAAM,CAE1C;AAED,8CAA8C;AAC9C,wBAAgB,mBAAmB,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAK/D;AAED,2BAA2B;AAC3B,wBAAgB,sBAAsB,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAKlE;AAED,0BAA0B;AAC1B,wBAAgB,kBAAkB,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAK9D;AAED,yBAAyB;AACzB,wBAAgB,qBAAqB,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAKjE;AAED,sBAAsB;AACtB,wBAAgB,kBAAkB,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAK9D;AAED,yBAAyB;AACzB,wBAAgB,iBAAiB,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAK7D;AAED,4BAA4B;AAC5B,wBAAgB,oBAAoB,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAKhE;AAED;;GAEG;AAEH,yBAAyB;AACzB,eAAO,MAAM,oBAAoB,QAEhC,CAAC;AAEF,0CAA0C;AAC1C,eAAO,MAAM,sBAAsB,EACrB,MAAM,GAAG,OAAO,GAAG,QAAQ,CAAC;AAE1C,eAAe;AACf,eAAO,MAAM,mBAAmB,SACiB,CAAC;AAElD,iCAAiC;AACjC,eAAO,MAAM,2BAA2B,QAEvC,CAAC;AAEF,0CAA0C;AAC1C,eAAO,MAAM,oBAAoB,QAEhC,CAAC;AAEF,qCAAqC;AACrC,eAAO,MAAM,uBAAuB,QAEnC,CAAC;AAEF;;GAEG;AAEH,eAAe;AACf,eAAO,MAAM,qBAAqB,SAA2C,CAAC;AAE9E,YAAY;AACZ,eAAO,MAAM,uBAAuB,QAAkC,CAAC;AAEvE,iBAAiB;AACjB,eAAO,MAAM,sBAAsB,QAAiC,CAAC;AAErE,oBAAoB;AACpB,eAAO,MAAM,2BAA2B,QAAsC,CAAC;AAE/E,WAAW;AACX,eAAO,MAAM,0BAA0B,QAAqC,CAAC;AAE7E,WAAW;AACX,eAAO,MAAM,uBAAuB,QAAkC,CAAC;AAEvE,aAAa;AACb,eAAO,MAAM,4BAA4B,QACH,CAAC;AAEvC,kCAAkC;AAClC,eAAO,MAAM,wBAAwB,QAC4B,CAAC;AAElE,aAAa;AACb,eAAO,MAAM,yBAAyB,QAErC,CAAC;AAEF,iBAAiB;AACjB,eAAO,MAAM,4BAA4B,QAExC,CAAC"}
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAMH,WAAW;AACX,eAAO,MAAM,IAAI,EAAE,MAElB,CAAC;AAEF,kBAAkB;AAClB,eAAO,MAAM,0BAA0B,0BAA0B,CAAC;AAElE,MAAM,MAAM,sBAAsB,GAC9B,SAAS,GACT,KAAK,GACL,6BAA6B,GAC7B,SAAS,GACT,iCAAiC,CAAC;AAEtC,MAAM,WAAW,0BAA0B;IACzC,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,EAAE,sBAAsB,CAAC;IAC/B,mBAAmB,EAAE,MAAM,EAAE,CAAC;CAC/B;AA+ED;;;GAGG;AACH,wBAAgB,uBAAuB,IAAI,0BAA0B,CAEpE;AAED;;;GAGG;AACH,wBAAgB,2BAA2B,CACzC,uBAAuB,EAAE,OAAO,GAC/B,0BAA0B,CA4C5B;AAED,+DAA+D;AAC/D,eAAO,MAAM,gBAAgB,EAAE,MAEH,CAAC;AAE7B,cAAc;AACd,eAAO,MAAM,OAAO,EAAE,MAA8C,CAAC;AAErE,oCAAoC;AACpC,eAAO,MAAM,oBAAoB,EAAE,OACiB,CAAC;AAErD,8BAA8B;AAC9B,eAAO,MAAM,kBAAkB,EAAE,MAAM,EAMC,CAAC;AAEzC,gEAAgE;AAChE,wBAAgB,uBAAuB,IAAI,IAAI,CAAG;AAElD,eAAe;AACf,wBAAgB,iBAAiB,IAAI,MAAM,CAE1C;AAED,8CAA8C;AAC9C,wBAAgB,mBAAmB,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAK/D;AAED,2BAA2B;AAC3B,wBAAgB,sBAAsB,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAKlE;AAED,0BAA0B;AAC1B,wBAAgB,kBAAkB,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAK9D;AAED,yBAAyB;AACzB,wBAAgB,qBAAqB,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAKjE;AAED,sBAAsB;AACtB,wBAAgB,kBAAkB,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAK9D;AAED,yBAAyB;AACzB,wBAAgB,iBAAiB,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAK7D;AAED,4BAA4B;AAC5B,wBAAgB,oBAAoB,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAKhE;AAED;;GAEG;AAEH,yBAAyB;AACzB,eAAO,MAAM,oBAAoB,QAEhC,CAAC;AAEF,0CAA0C;AAC1C,eAAO,MAAM,sBAAsB,EACrB,MAAM,GAAG,OAAO,GAAG,QAAQ,CAAC;AAE1C,eAAe;AACf,eAAO,MAAM,mBAAmB,SACiB,CAAC;AAElD,iCAAiC;AACjC,eAAO,MAAM,2BAA2B,QAEvC,CAAC;AAEF,0CAA0C;AAC1C,eAAO,MAAM,oBAAoB,QAEhC,CAAC;AAEF,qCAAqC;AACrC,eAAO,MAAM,uBAAuB,QAEnC,CAAC;AAEF;;GAEG;AAEH,eAAe;AACf,eAAO,MAAM,qBAAqB,SAA2C,CAAC;AAE9E,YAAY;AACZ,eAAO,MAAM,uBAAuB,QAAkC,CAAC;AAEvE,iBAAiB;AACjB,eAAO,MAAM,sBAAsB,QAAiC,CAAC;AAErE,oBAAoB;AACpB,eAAO,MAAM,2BAA2B,QAAsC,CAAC;AAE/E,WAAW;AACX,eAAO,MAAM,0BAA0B,QAAqC,CAAC;AAE7E,WAAW;AACX,eAAO,MAAM,uBAAuB,QAAkC,CAAC;AAEvE,aAAa;AACb,eAAO,MAAM,4BAA4B,QACH,CAAC;AAEvC,kCAAkC;AAClC,eAAO,MAAM,wBAAwB,QAC4B,CAAC;AAElE,aAAa;AACb,eAAO,MAAM,yBAAyB,QAErC,CAAC;AAEF,iBAAiB;AACjB,eAAO,MAAM,4BAA4B,QAExC,CAAC"}

package/dist/config.js

@@ -76,6 +76,21 @@ function readAutoRegisterTargetUrls() {
  * 主流程：显式环境变量优先；未配置时，单个自动注册目标可作为宿主机部署的零额外配置回退；多目标不静默取第一个。
  */
 export function resolveSchedulerBaseUrl() {
+    return resolveSchedulerBaseUrlFrom(undefined);
+}
+/**
+ * 解析调度中心基础 URL。
+ * 主流程：服务端请求携带的 scheduler 地址优先；再读显式环境变量；最后回退到单个自动注册目标或默认值。
+ */
+export function resolveSchedulerBaseUrlFrom(requestSchedulerBaseUrl) {
+    const normalizedRequestSchedulerBaseUrl = normalizeSchedulerHttpBaseUrl(requestSchedulerBaseUrl);
+    if (normalizedRequestSchedulerBaseUrl) {
+        return {
+            url: normalizedRequestSchedulerBaseUrl,
+            source: "request",
+            ambiguousTargetUrls: [],
+        };
+    }
     const envSchedulerBaseUrl = normalizeSchedulerHttpBaseUrl(process.env.AWS_RUNTIME_SCHEDULER_BASE_URL);
     if (envSchedulerBaseUrl) {
         return {

package/dist/config.test.js

@@ -42,6 +42,18 @@ afterEach(() => {
     }
 });
 describe("scheduler base URL resolution", () => {
+    it("keeps scheduler URL supplied by the server request as the highest priority", async () => {
+        const runtimeHome = createRuntimeHome();
+        useRuntimeHome(runtimeHome);
+        process.env.AWS_RUNTIME_SCHEDULER_BASE_URL = "http://env.local:7380";
+        writeBridgeConfig(runtimeHome, [{ serverUrl: "http://target.local:7380" }]);
+        const { resolveSchedulerBaseUrlFrom } = await import("./config.js");
+        expect(resolveSchedulerBaseUrlFrom("http://server.local:7380/api")).toEqual({
+            url: "http://server.local:7380",
+            source: "request",
+            ambiguousTargetUrls: [],
+        });
+    });
     it("keeps explicit AWS_RUNTIME_SCHEDULER_BASE_URL as the highest priority", async () => {
         const runtimeHome = createRuntimeHome();
         useRuntimeHome(runtimeHome);

package/dist/routes/instance.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"instance.d.ts","sourceRoot":"","sources":["../../src/routes/instance.ts"],"names":[],"mappings":"AAsBA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAYrD,wBAAgB,qBAAqB,CACnC,EAAE,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,OAAO,KAAK,OAAO,CAAC,IAAI,CAAC,GAC/D,IAAI,CAEN;AAED,eAAO,MAAM,cAAc,4CAAW,CAAC;AAyBvC,wBAAgB,4BAA4B,CAAC,gBAAgB,EAAE,MAAM,GAAG;IACtE,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE;QACJ,EAAE,EAAE,OAAO,CAAC;QACZ,aAAa,CAAC,EAAE,SAAS,CAAC;QAC1B,oBAAoB,EAAE,OAAO,CAAC;QAC9B,qBAAqB,EAAE,OAAO,CAAC;QAC/B,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,CAAC;CACH,CAqCA;AAED,MAAM,WAAW,4BAA4B;IAC3C,EAAE,EAAE,KAAK,CAAC;IACV,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,EAAE,gBAAgB,CAAC;IAC/B,aAAa,EAAE,SAAS,CAAC;IACzB,gBAAgB,EAAE,MAAM,CAAC;IACzB,IAAI,EAAE,MAAM,CAAC;CACd;AAED,UAAU,iCAAiC;IACzC,EAAE,EAAE,KAAK,CAAC;IACV,KAAK,EAAE,8BAA8B,CAAC;IACtC,YAAY,EAAE,gBAAgB,CAAC;IAC/B,aAAa,EAAE,SAAS,CAAC;IACzB,gBAAgB,EAAE,MAAM,CAAC;IACzB,sBAAsB,EAAE,MAAM,EAAE,CAAC;IACjC,IAAI,EAAE,MAAM,CAAC;CACd;AAED;;;GAGG;AACH,wBAAgB,iCAAiC,CAC/C,KAAK,EAAE,KAAK,EACZ,0BAA0B,EAAE,MAAM,GACjC,4BAA4B,CAiB9B;AAED;;;GAGG;AACH,wBAAgB,sCAAsC,CACpD,UAAU,EAAE,MAAM,EAAE,GACnB,iCAAiC,CAUnC;AAED;;;GAGG;AACH,wBAAgB,+BAA+B,CAAC,YAAY,EAAE,MAAM,EAAE,GAAG,MAAM,EAAE,CAQhF;AAED;;GAEG;AACH,wBAAgB,4BAA4B,CAC1C,cAAc,EAAE,MAAM,EAAE,EACxB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,iBAAiB,CAAC,GAC5C,MAAM,CAaR;AAED,wBAAgB,mBAAmB,CACjC,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,iBAAiB,CAAC,GAC5C,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,iBAAiB,EAAE,WAAW,GAAG,SAAS,GAAG,YAAY,GAAG,OAAO,CAAC,CAAC,CAY3F"}
+{"version":3,"file":"instance.d.ts","sourceRoot":"","sources":["../../src/routes/instance.ts"],"names":[],"mappings":"AAsBA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAYrD,wBAAgB,qBAAqB,CACnC,EAAE,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,OAAO,KAAK,OAAO,CAAC,IAAI,CAAC,GAC/D,IAAI,CAEN;AAED,eAAO,MAAM,cAAc,4CAAW,CAAC;AAyBvC,wBAAgB,4BAA4B,CAAC,gBAAgB,EAAE,MAAM,GAAG;IACtE,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE;QACJ,EAAE,EAAE,OAAO,CAAC;QACZ,aAAa,CAAC,EAAE,SAAS,CAAC;QAC1B,oBAAoB,EAAE,OAAO,CAAC;QAC9B,qBAAqB,EAAE,OAAO,CAAC;QAC/B,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,CAAC;CACH,CAqCA;AAED,MAAM,WAAW,4BAA4B;IAC3C,EAAE,EAAE,KAAK,CAAC;IACV,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,EAAE,gBAAgB,CAAC;IAC/B,aAAa,EAAE,SAAS,CAAC;IACzB,gBAAgB,EAAE,MAAM,CAAC;IACzB,IAAI,EAAE,MAAM,CAAC;CACd;AAED,UAAU,iCAAiC;IACzC,EAAE,EAAE,KAAK,CAAC;IACV,KAAK,EAAE,8BAA8B,CAAC;IACtC,YAAY,EAAE,gBAAgB,CAAC;IAC/B,aAAa,EAAE,SAAS,CAAC;IACzB,gBAAgB,EAAE,MAAM,CAAC;IACzB,sBAAsB,EAAE,MAAM,EAAE,CAAC;IACjC,IAAI,EAAE,MAAM,CAAC;CACd;AAED;;;GAGG;AACH,wBAAgB,iCAAiC,CAC/C,KAAK,EAAE,KAAK,EACZ,0BAA0B,EAAE,MAAM,GACjC,4BAA4B,CAiB9B;AAoDD;;;GAGG;AACH,wBAAgB,sCAAsC,CACpD,UAAU,EAAE,MAAM,EAAE,GACnB,iCAAiC,CAUnC;AAED;;;GAGG;AACH,wBAAgB,+BAA+B,CAAC,YAAY,EAAE,MAAM,EAAE,GAAG,MAAM,EAAE,CAQhF;AAED;;GAEG;AACH,wBAAgB,4BAA4B,CAC1C,cAAc,EAAE,MAAM,EAAE,EACxB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,iBAAiB,CAAC,GAC5C,MAAM,CAaR;AAED,wBAAgB,mBAAmB,CACjC,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,iBAAiB,CAAC,GAC5C,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,iBAAiB,EAAE,WAAW,GAAG,SAAS,GAAG,YAAY,GAAG,OAAO,CAAC,CAAC,CAY3F"}

package/dist/routes/instance.js

@@ -2,14 +2,14 @@ import { Router } from "express";
 import axios from "axios";
 import { createHash, timingSafeEqual } from "node:crypto";
 import { validateToken } from "../middleware/auth.js";
-import { resolveSchedulerBaseUrl } from "../config.js";
+import { resolveSchedulerBaseUrlFrom } from "../config.js";
 import { getRuntimeAccessToken } from "../services/runtime-binding.js";
 import { loadInstanceState, saveInstanceState, } from "../services/instance-state.js";
 import { initInstance } from "../services/instance-init-service.js";
 import { discoverCcSwitchConfiguredItems, loadCcSwitchSdk } from "../services/cc-switch-sdk.js";
 import { syncLegacyStateFromSdk } from "../services/instance-init-service.js";
 import { detectToolStatuses, ensureToolsInstalled, SUPPORTED_INSTALLABLE_TOOLS, SUPPORTED_UNINSTALLABLE_TOOLS, uninstallTools, } from "../services/tool-installer.js";
-import { getConfiguredConnectionKeys } from "../services/auto-register.js";
+import { getConfiguredConnectionKeys, requestRuntimeAccessTokenRefreshForServer } from "../services/auto-register.js";
 import { createLogger } from "../utils/logger.js";
 const log = createLogger("instance");
 const PANEL_TOOL_STATUS_KEYS = ["claude", "opencode", "codex"];
@@ -92,6 +92,40 @@ export function buildSchedulerPingFailureResponse(error, configuredSchedulerBase
             : "aws-runtime-bridge 已连通，但它回连调度中心失败。请确认 AWS_RUNTIME_SCHEDULER_BASE_URL 指向的调度中心地址可从 bridge 机器访问，并且运行时访问令牌仍有效。",
     };
 }
+function getAxiosStatusCode(error) {
+    if (typeof error !== "object" || error === null) {
+        return undefined;
+    }
+    const response = Reflect.get(error, "response");
+    if (typeof response !== "object" || response === null) {
+        return undefined;
+    }
+    const status = Reflect.get(response, "status");
+    return typeof status === "number" ? status : undefined;
+}
+async function pingSchedulerWithToken(schedulerBaseUrl, runtimeAccessToken) {
+    try {
+        const schedulerResponse = await axios.get(`${schedulerBaseUrl}/api/runtime/ping`, { headers: { "X-Runtime-Token": runtimeAccessToken } });
+        return {
+            schedulerReachable: schedulerResponse.status >= 200 && schedulerResponse.status < 300,
+            tokenRefreshed: false,
+        };
+    }
+    catch (error) {
+        if (getAxiosStatusCode(error) !== 401) {
+            throw error;
+        }
+        const refreshResult = await requestRuntimeAccessTokenRefreshForServer(schedulerBaseUrl);
+        if (!refreshResult.success || !refreshResult.runtimeAccessToken) {
+            throw error;
+        }
+        const retryResponse = await axios.get(`${schedulerBaseUrl}/api/runtime/ping`, { headers: { "X-Runtime-Token": refreshResult.runtimeAccessToken } });
+        return {
+            schedulerReachable: retryResponse.status >= 200 && retryResponse.status < 300,
+            tokenRefreshed: true,
+        };
+    }
+}
 /**
  * Build the response for ambiguous scheduler targets.
  * Main flow: never silently pick the first target; require an explicit scheduler URL or future scheduler identity routing.
@@ -154,8 +188,8 @@ instanceRouter.get("/connection-check", (req, res) => {
     const result = buildConnectionCheckResponse(connectionKeyMd5);
     res.status(result.status).json(result.body);
 });
-instanceRouter.get("/ping", validateToken, async (_req, res) => {
-    const schedulerBaseUrlResolution = resolveSchedulerBaseUrl();
+instanceRouter.get("/ping", validateToken, async (req, res) => {
+    const schedulerBaseUrlResolution = resolveSchedulerBaseUrlFrom(req.header("X-Scheduler-Base-Url"));
     if (schedulerBaseUrlResolution.source === "ambiguous-auto-register-targets") {
         res
             .status(400)
@@ -169,11 +203,12 @@ instanceRouter.get("/ping", validateToken, async (_req, res) => {
             res.status(401).json({ ok: false, error: "runtime_access_token_required" });
             return;
         }
-        const schedulerResponse = await axios.get(`${schedulerBaseUrl}/api/runtime/ping`, { headers: { "X-Runtime-Token": runtimeAccessToken } });
+        const schedulerPing = await pingSchedulerWithToken(schedulerBaseUrl, runtimeAccessToken);
         res.json({
             ok: true,
             runtimeBridge: "healthy",
-            schedulerReachable: schedulerResponse.status >= 200 && schedulerResponse.status < 300,
+            schedulerReachable: schedulerPing.schedulerReachable,
+            tokenRefreshed: schedulerPing.tokenRefreshed,
             schedulerBaseUrl,
         });
     }

package/dist/routes/instance.test.js

@@ -2,8 +2,17 @@
  * Instance 路由单元测试
  */
 import { describe, it, expect, vi, afterEach } from 'vitest';
+vi.mock('axios', () => ({
+    default: {
+        get: vi.fn(),
+    },
+}));
 vi.mock('../services/auto-register.js', () => ({
     getConfiguredConnectionKeys: vi.fn(() => []),
+    requestRuntimeAccessTokenRefreshForServer: vi.fn(),
+}));
+vi.mock('../services/runtime-binding.js', () => ({
+    getRuntimeAccessToken: vi.fn(() => 'stale-runtime-token-123456'),
 }));
 afterEach(() => {
     vi.clearAllMocks();
@@ -203,6 +212,67 @@ describe('instance route validation', () => {
         expect(response.runtimeBridge).toBe('healthy');
         expect(response.hint).toContain('运行时访问令牌仍有效');
     });
+    it('refreshes runtime token and retries scheduler ping after 401', async () => {
+        process.env.AWS_RUNTIME_SCHEDULER_BASE_URL = 'http://localhost:8080';
+        const { default: axios } = await import('axios');
+        const { requestRuntimeAccessTokenRefreshForServer } = await import('../services/auto-register.js');
+        const { instanceRouter } = await import('./instance.js');
+        vi.mocked(requestRuntimeAccessTokenRefreshForServer).mockResolvedValue({
+            success: true,
+            runtimeAccessToken: 'fresh-runtime-token-123456',
+            updated: true,
+        });
+        vi.mocked(axios.get)
+            .mockRejectedValueOnce({ response: { status: 401 }, message: 'Request failed with status code 401' })
+            .mockResolvedValueOnce({ status: 200 });
+        const handler = instanceRouter.stack.find((layer) => layer.route?.path === '/ping')?.route?.stack[1]?.handle;
+        expect(handler).toBeTypeOf('function');
+        const json = vi.fn();
+        const status = vi.fn(() => ({ json }));
+        await handler?.({ header: vi.fn(() => '') }, { json, status }, vi.fn());
+        expect(requestRuntimeAccessTokenRefreshForServer).toHaveBeenCalledWith('http://localhost:8080');
+        expect(axios.get).toHaveBeenNthCalledWith(1, 'http://localhost:8080/api/runtime/ping', {
+            headers: { 'X-Runtime-Token': 'stale-runtime-token-123456' },
+        });
+        expect(axios.get).toHaveBeenNthCalledWith(2, 'http://localhost:8080/api/runtime/ping', {
+            headers: { 'X-Runtime-Token': 'fresh-runtime-token-123456' },
+        });
+        expect(json).toHaveBeenCalledWith({
+            ok: true,
+            runtimeBridge: 'healthy',
+            schedulerReachable: true,
+            tokenRefreshed: true,
+            schedulerBaseUrl: 'http://localhost:8080',
+        });
+        expect(status).not.toHaveBeenCalled();
+    });
+    it('uses scheduler base URL supplied by server ping request before local bridge config', async () => {
+        process.env.AWS_RUNTIME_SCHEDULER_BASE_URL = 'http://localhost:8080';
+        const { default: axios } = await import('axios');
+        const { getRuntimeAccessToken } = await import('../services/runtime-binding.js');
+        const { instanceRouter } = await import('./instance.js');
+        vi.mocked(getRuntimeAccessToken).mockReturnValue('stale-runtime-token-123456');
+        vi.mocked(axios.get).mockReset();
+        vi.mocked(axios.get).mockResolvedValueOnce({ status: 200 });
+        const handler = instanceRouter.stack.find((layer) => layer.route?.path === '/ping')?.route?.stack[1]?.handle;
+        expect(handler).toBeTypeOf('function');
+        const json = vi.fn();
+        const status = vi.fn(() => ({ json }));
+        const header = vi.fn((name) => (name === 'X-Scheduler-Base-Url' ? 'http://server.local:7380/api' : ''));
+        await handler?.({ header }, { json, status }, vi.fn());
+        expect(getRuntimeAccessToken).toHaveBeenCalledWith(undefined, 'http://server.local:7380');
+        expect(axios.get).toHaveBeenCalledWith('http://server.local:7380/api/runtime/ping', {
+            headers: { 'X-Runtime-Token': 'stale-runtime-token-123456' },
+        });
+        expect(json).toHaveBeenCalledWith({
+            ok: true,
+            runtimeBridge: 'healthy',
+            schedulerReachable: true,
+            tokenRefreshed: false,
+            schedulerBaseUrl: 'http://server.local:7380',
+        });
+        expect(status).not.toHaveBeenCalled();
+    });
     it('explains ambiguous auto-register scheduler targets without picking the first one', async () => {
         const { buildAmbiguousSchedulerBaseUrlResponse } = await import('./instance.js');
         const response = buildAmbiguousSchedulerBaseUrlResponse([

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aws-runtime-bridge",
-  "version": "1.7.8",
+  "version": "1.7.9",
   "description": "AgentsWorkStudio runtime bridge service for machine-level agent runtime integration",
   "type": "module",
   "main": "dist/index.js",