mcp-google-multi 6.0.0-alpha.48 → 6.0.0-alpha.49

package/dist/accounts.d.ts

@@ -48,6 +48,13 @@ export declare function unknownAliasMessage(aliases: readonly string[], selector
 export declare function accountAliasSchemaFor(aliases: readonly string[]): z.ZodType<string>;
 /** Shared, load-time snapshot used by every tool's `account` field. */
 export declare const accountAliasSchema: z.ZodType<string>;
+/** Live account argument: validates at PARSE time against the accessor's
+ * CURRENT aliases, so an alias added mid-session (account_add, or a tenant
+ * alias link) is immediately valid on already-registered tools — a baked
+ * z.enum freezes the boot-time set until restart. An empty set stays
+ * permissive (same empty-safe rule as above); dispatch still resolves the
+ * alias against the live registry. */
+export declare function accountArgLive(aliases: () => readonly string[]): z.ZodType<string>;
 /**
  * BR-4: the stdio/http SERVER never boots with an empty registry — a fresh user
  * bootstraps via env / `migrate-config` / `account import` / `auth` first. The

package/dist/accounts.js

@@ -302,6 +302,20 @@ export function accountAliasSchemaFor(aliases) {
 }
 /** Shared, load-time snapshot used by every tool's `account` field. */
 export const accountAliasSchema = accountAliasSchemaFor(ACCOUNTS);
+/** Live account argument: validates at PARSE time against the accessor's
+ * CURRENT aliases, so an alias added mid-session (account_add, or a tenant
+ * alias link) is immediately valid on already-registered tools — a baked
+ * z.enum freezes the boot-time set until restart. An empty set stays
+ * permissive (same empty-safe rule as above); dispatch still resolves the
+ * alias against the live registry. */
+export function accountArgLive(aliases) {
+    return z.string().superRefine((value, ctx) => {
+        const current = aliases();
+        if (current.length > 0 && !current.includes(value)) {
+            ctx.addIssue({ code: 'custom', message: unknownAliasMessage(current) });
+        }
+    });
+}
 /**
  * BR-4: the stdio/http SERVER never boots with an empty registry — a fresh user
  * bootstraps via env / `migrate-config` / `account import` / `auth` first. The

package/dist/registry.d.ts

@@ -123,4 +123,11 @@ export declare class ToolRegistry {
     };
     installListHandler(): void;
     private toToolJson;
+    /** Advertise the CURRENT alias enum on every tools/list: account validation
+     * is (or is becoming) live, so a cached boot-time enum would go stale the
+     * moment account_add lands mid-session. The structural schema stays cached;
+     * only the account property's values are refreshed per call. Skipped for
+     * the subject-taking account tools (their `account` is the OPERAND, e.g. a
+     * brand-new alias name, not an identity to enumerate). */
+    private withLiveAccountEnum;
 }

package/dist/registry.js

@@ -451,9 +451,35 @@ export class ToolRegistry {
         return {
             name: tool.name,
             description: tool.description,
-            inputSchema,
+            inputSchema: this.withLiveAccountEnum(tool, inputSchema),
             annotations: tool.annotations,
             ...(tool.clientMeta ? { _meta: tool.clientMeta } : {}),
         };
     }
+    /** Advertise the CURRENT alias enum on every tools/list: account validation
+     * is (or is becoming) live, so a cached boot-time enum would go stale the
+     * moment account_add lands mid-session. The structural schema stays cached;
+     * only the account property's values are refreshed per call. Skipped for
+     * the subject-taking account tools (their `account` is the OPERAND, e.g. a
+     * brand-new alias name, not an identity to enumerate). */
+    withLiveAccountEnum(tool, schema) {
+        if (DEFAULT_ACCOUNT_EXCLUDE.has(tool.name))
+            return schema;
+        const s = schema;
+        const account = s?.properties?.account;
+        if (!account)
+            return schema;
+        const aliases = this.accounts().aliases;
+        if (aliases.length === 0)
+            return schema;
+        if (Array.isArray(account.anyOf)) {
+            // fan-out union: refresh the enum branch ('*' + aliases), keep the CSV branch
+            const hasEnumBranch = account.anyOf.some((b) => Array.isArray(b.enum));
+            if (!hasEnumBranch)
+                return schema;
+            const anyOf = account.anyOf.map((b) => (Array.isArray(b.enum) ? { ...b, enum: ['*', ...aliases] } : b));
+            return { ...s, properties: { ...s.properties, account: { ...account, anyOf } } };
+        }
+        return { ...s, properties: { ...s.properties, account: { ...account, enum: [...aliases] } } };
+    }
 }

package/dist/tools/account-wizard.js

@@ -297,6 +297,15 @@ export function registerAccountWizardTools(registry, server) {
             // S2: atomic write + make the alias callable without a restart (BR3).
             writeAccountRow(validated.alias, validated.email, validated.bundles, validated.admin);
             invalidateAccountSet();
+            // Live account args validate against the refreshed registry already;
+            // the list_changed nudge makes clients re-fetch tools/list, where the
+            // advertised account enums are rebuilt from the live set.
+            try {
+                server.sendToolListChanged();
+            }
+            catch {
+                // a client that cannot receive notifications loses nothing but the nudge
+            }
             // S3 + S4: consent + validate.
             const consent = await runConsent(server, validated.alias);
             if (consent.ok === 'pending')

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mcp-google-multi",
-  "version": "6.0.0-alpha.48",
+  "version": "6.0.0-alpha.49",
   "mcpName": "io.github.bakissation/mcp-google-multi",
   "description": "Local MCP server for Google Workspace (Gmail, Drive, Calendar, Sheets, Docs, Contacts, Tasks, Meet, Search Console, +Forms/Chat/Admin) across multiple accounts — OAuth-only, encrypted token storage, deny-by-default writes.",
   "type": "module",