@wootsup/yt-builder-mcp 1.0.0 → 1.1.0

package/dist/setup-cli.js

@@ -24,7 +24,14 @@
 import { ALL_CLIENTS } from './clients/index.js';
 import { installSkill as defaultInstallSkill } from './install-skill.js';
 import { SERVER_VERSION } from './server.js';
+import { confirmTTY as defaultConfirmTTY } from './setup-cli-confirm.js';
 import { DEFAULT_WIZARD_DEPS, runWizard as runWizardImpl } from './setup-wizard.js';
+import { addSiteCommand, AddSiteError, DEFAULT_ADD_SITE_DEPS, } from './sites/cli/add-site.js';
+import { DEFAULT_LIST_SITES_DEPS, listSitesCommand, } from './sites/cli/list-sites.js';
+import { DEFAULT_REMOVE_SITE_DEPS, removeSiteCommand, RemoveSiteError, } from './sites/cli/remove-site.js';
+import { DEFAULT_SET_DEFAULT_DEPS, setDefaultCommand, SetDefaultError, } from './sites/cli/set-default.js';
+import { DEFAULT_TEST_SITE_DEPS, renderTestSiteLines, testSiteCommand, TestSiteError, } from './sites/cli/test-site.js';
+import { defaultSitesFilePath } from './sites/paths.js';
 // Re-export the wizard surface so consumers that imported from
 // `./setup-cli` before the I5 split keep working without changes.
 export { DEFAULT_WIZARD_DEPS, majorMinor, runWizard, } from './setup-wizard.js';
@@ -36,12 +43,26 @@ Commands:
                    ~/.claude/skills/ and register it in ~/AGENTS.md.
                    Note: ~/.claude/skills/ is the universal-marker path
                    recognised by Claude Desktop and other AI clients.
+
+Multi-site commands (W9):
+  add-site         Add a site to ~/.config/yt-builder-mcp/sites.json.
+                   Flags: --url <url>, --token <bearer> OR --token-ref
+                   <op://...>, --platform auto|wordpress|joomla,
+                   --label "...", --site-id <slug>, --default, --yes.
+  list-sites       Show the configured sites (no network calls).
+  remove-site <id> Remove a site; auto-promotes the next as default.
+                   Flag: --yes to skip the confirmation prompt.
+  set-default <id> Flip is_default to the given site_id.
+  test-site <id>   Probe /health (no auth) + /etag (auth) for one site.
+
   --version, -v    Print the package version.
   --help, -h, help Show this message.
 
 Setup flags (for non-interactive / CI usage):
   --non-interactive       Skip all prompts; fail-fast on missing input.
-  --url <wp-url>          WordPress base URL (required with --non-interactive).
+  --url <site-url>        Site base URL (required with --non-interactive).
+                          WordPress or Joomla — the wizard auto-detects the
+                          platform from the URL shape.
   --token <bearer>        Bearer key (required with --non-interactive).
   --client <id>           AI client id to configure; repeatable. Valid ids:
                           claude-desktop, claude-code, cursor, zed, continue,
@@ -54,9 +75,12 @@ Example (non-interactive):
     --url https://example.com \\
     --token "$YTB_TOKEN"
 
-Pickup mode (one-click setup via wp-admin):
-  --pickup <url>          Pickup URL revealed by the wp-admin "Setup wizard"
-                          flow; e.g. https://example.com/wp-json/yt-builder-mcp/v1/setup/pickup
+Pickup mode (one-click setup from the admin "Setup wizard" flow):
+  --pickup <url>          Pickup URL revealed by the host CMS admin (WordPress:
+                          wp-admin → Tools → "YT Builder MCP"; Joomla:
+                          Administrator → Components → "YT Builder MCP").
+                          WordPress:  https://example.com/wp-json/yt-builder-mcp/v1/setup/pickup
+                          Joomla:      https://example.com/api/index.php/v1/yt-builder-mcp/setup/pickup
   --nonce <code>          One-shot nonce revealed alongside the pickup URL.
                           Pairs 1:1 with --pickup (both or neither).
 
@@ -66,16 +90,31 @@ Pickup mode (one-click setup via wp-admin):
   expires after 5 minutes. If --url or --token are also passed they are
   ignored (a warning is printed to stderr).
 
-Example (pickup):
+Example (pickup — WordPress):
   npx -y @wootsup/yt-builder-mcp setup \\
     --pickup https://example.com/wp-json/yt-builder-mcp/v1/setup/pickup \\
     --nonce "$NONCE" \\
     --client claude-desktop
 
+Example (pickup — Joomla):
+  npx -y @wootsup/yt-builder-mcp setup \\
+    --pickup https://example.com/api/index.php/v1/yt-builder-mcp/setup/pickup \\
+    --nonce "$NONCE" \\
+    --client claude-desktop
+
 Environment (when launched by an AI client):
-  YTB_MCP_WP_URL         WordPress base URL (e.g. https://example.com).
-  YTB_MCP_BEARER_TOKEN   Bearer key from wp-admin → Tools → "YT Builder MCP".
+  YTB_MCP_SITE_URL       Host CMS base URL (e.g. https://example.com).
+                         Works for BOTH WordPress and Joomla.
+  YTB_MCP_WP_URL         DEPRECATED alias for YTB_MCP_SITE_URL — still
+                         honoured for pre-Wave-7 configurations.
+  YTB_MCP_BEARER_TOKEN   Bearer key from:
+                           - WordPress: wp-admin → Tools → "YT Builder MCP"
+                           - Joomla: Administrator → Components → "YT Builder MCP"
                          Format: ytb_(live|test)_<payload>.<signature>
+  YTB_MCP_PLATFORM       Optional explicit platform: 'wordpress' or 'joomla'.
+                         Set this to 'joomla' when YTB_MCP_SITE_URL is an
+                         origin-only Joomla URL (no /api/index.php/ in path).
+                         Defaults to URL-shape auto-detection.
   YTB_MCP_TEST_MODE=1    Skip stdio loop (used by smoke tests).
 
 Documentation: https://github.com/wootsup/yt-builder-mcp
@@ -265,12 +304,269 @@ export function buildPickupDeps(parsed, base = DEFAULT_WIZARD_DEPS) {
         confirmContinue: async () => false,
     };
 }
+/**
+ * Parse `add-site` flags from the tail of argv. Accepts both
+ * `--key value` and `--key=value` styles. Unknown flags are reported
+ * as errors so typos never silently disable validation.
+ */
+export function parseAddSiteArgs(argv) {
+    let url = '';
+    let token = '';
+    let tokenRef = '';
+    let label = '';
+    let siteId = '';
+    let platform;
+    let isDefault = false;
+    let yes = false;
+    let sitesFile;
+    const errors = [];
+    function consume(name, inline, i) {
+        if (inline !== undefined)
+            return { value: inline, nextIndex: i };
+        const next = argv[i + 1];
+        if (next === undefined || next.startsWith('--')) {
+            errors.push(`Flag ${name} requires a value.`);
+            return { value: '', nextIndex: i };
+        }
+        return { value: next, nextIndex: i + 1 };
+    }
+    for (let i = 0; i < argv.length; i++) {
+        const raw = argv[i];
+        const eq = raw.indexOf('=');
+        const flag = eq >= 0 ? raw.slice(0, eq) : raw;
+        const inline = eq >= 0 ? raw.slice(eq + 1) : undefined;
+        switch (flag) {
+            case '--url': {
+                const r = consume('--url', inline, i);
+                url = r.value.trim().replace(/\/+$/, '');
+                i = r.nextIndex;
+                break;
+            }
+            case '--token': {
+                const r = consume('--token', inline, i);
+                token = r.value.trim();
+                i = r.nextIndex;
+                break;
+            }
+            case '--token-ref': {
+                const r = consume('--token-ref', inline, i);
+                tokenRef = r.value.trim();
+                i = r.nextIndex;
+                break;
+            }
+            case '--platform': {
+                const r = consume('--platform', inline, i);
+                const v = r.value.trim();
+                if (v === 'auto' || v === 'wordpress' || v === 'joomla') {
+                    platform = v;
+                }
+                else if (v.length > 0) {
+                    errors.push(`--platform must be one of auto, wordpress, joomla (got "${v}").`);
+                }
+                i = r.nextIndex;
+                break;
+            }
+            case '--label': {
+                const r = consume('--label', inline, i);
+                label = r.value;
+                i = r.nextIndex;
+                break;
+            }
+            case '--site-id': {
+                const r = consume('--site-id', inline, i);
+                siteId = r.value.trim();
+                i = r.nextIndex;
+                break;
+            }
+            case '--default': {
+                isDefault = true;
+                if (inline !== undefined) {
+                    errors.push('Flag --default does not take a value.');
+                }
+                break;
+            }
+            case '--yes':
+            case '-y': {
+                yes = true;
+                if (inline !== undefined) {
+                    errors.push(`Flag ${flag} does not take a value.`);
+                }
+                break;
+            }
+            case '--sites-file': {
+                const r = consume('--sites-file', inline, i);
+                sitesFile = r.value.trim();
+                i = r.nextIndex;
+                break;
+            }
+            case '--help':
+            case '-h': {
+                errors.push('HELP_REQUESTED');
+                break;
+            }
+            default: {
+                errors.push(`Unknown add-site flag: ${flag}`);
+            }
+        }
+    }
+    const parsed = {
+        url,
+        token,
+        tokenRef,
+        label,
+        siteId: siteId.length > 0 ? siteId : 'default',
+        default: isDefault,
+        yes,
+        ...(platform !== undefined ? { platform } : {}),
+        ...(sitesFile !== undefined ? { sitesFile } : {}),
+        errors,
+    };
+    return parsed;
+}
+const ADD_SITE_HELP = `Usage: yt-builder-mcp add-site [flags]\n\n` +
+    `Flags:\n` +
+    `  --url <url>             (required) Site base URL (WordPress or Joomla origin).\n` +
+    `  --token <bearer>        Inline bearer (mutually exclusive with --token-ref).\n` +
+    `  --token-ref <op://...>  1Password Secret Reference for the bearer.\n` +
+    `  --platform <kind>       auto|wordpress|joomla (default: auto).\n` +
+    `  --label "<text>"        Human-readable label (≤120 chars).\n` +
+    `  --site-id <slug>        Registry key (default: "default").\n` +
+    `  --default               Mark this site as the default.\n` +
+    `  --yes, -y               Overwrite an existing site_id without confirming.\n` +
+    `  --sites-file <path>     Override sites.json location.\n\n` +
+    `Notes:\n` +
+    `  - The first site you add becomes the default automatically.\n` +
+    `  - Use --token OR --token-ref, not both.\n` +
+    `  - URL is normalised (trailing slashes stripped).\n\n` +
+    `Example:\n` +
+    `  yt-builder-mcp add-site --url https://acme.example.com \\\n` +
+    `    --token "$YTB_TOKEN" --site-id wp-acme --label "ACME WP"\n`;
+const REMOVE_SITE_HELP = `Usage: yt-builder-mcp remove-site <site_id> [--yes] [--sites-file <path>]\n\n` +
+    `Removes the site from sites.json. If the default site is removed and at\n` +
+    `least one other site remains, the next site in insertion order becomes\n` +
+    `the new default. Removing the only site clears default_site_id and\n` +
+    `leaves an empty registry.\n\n` +
+    `Confirmation is required unless --yes is passed. In non-TTY contexts\n` +
+    `(CI, piped stdin) the subcommand fails with CONFIRM_REQUIRED rather\n` +
+    `than silently deleting.\n\n` +
+    `Example:\n` +
+    `  yt-builder-mcp remove-site wp-acme --yes\n`;
+const SET_DEFAULT_HELP = `Usage: yt-builder-mcp set-default <site_id> [--yes] [--sites-file <path>]\n\n` +
+    `Changes which site every subsequent tool call routes to by default\n` +
+    `when site_id is omitted. Confirmation is required unless --yes is\n` +
+    `passed — in non-TTY contexts the subcommand fails with\n` +
+    `SET_DEFAULT_CONFIRM_REQUIRED rather than silently flipping.\n\n` +
+    `Example:\n` +
+    `  yt-builder-mcp set-default wp-beta --yes\n`;
+const TEST_SITE_HELP = `Usage: yt-builder-mcp test-site <site_id> [--sites-file <path>]\n\n` +
+    `Probes /health (unauthenticated) and /etag (authenticated) in parallel\n` +
+    `against the configured Bearer. Returns plugin reachability + Bearer\n` +
+    `validity in one call. Does NOT mutate the registry or the site.\n\n` +
+    `Example:\n` +
+    `  yt-builder-mcp test-site wp-acme\n`;
+const LIST_SITES_HELP = `Usage: yt-builder-mcp list-sites [--sites-file <path>]\n\n` +
+    `Prints the configured sites table (site_id, url, platform, label,\n` +
+    `bearer_source, is_default). No network calls — pure registry read.\n\n` +
+    `Example:\n` +
+    `  yt-builder-mcp list-sites\n`;
+function parseIdArgs(argv, allowYes) {
+    let siteId = '';
+    let yes = false;
+    let sitesFile;
+    const errors = [];
+    let helpRequested = false;
+    for (let i = 0; i < argv.length; i++) {
+        const raw = argv[i];
+        const eq = raw.indexOf('=');
+        const flag = eq >= 0 ? raw.slice(0, eq) : raw;
+        const inline = eq >= 0 ? raw.slice(eq + 1) : undefined;
+        if (flag === '--help' || flag === '-h') {
+            helpRequested = true;
+            continue;
+        }
+        if (flag === '--yes' || flag === '-y') {
+            if (!allowYes) {
+                errors.push(`Unknown flag for this subcommand: ${flag}`);
+            }
+            else {
+                yes = true;
+            }
+            if (inline !== undefined)
+                errors.push(`Flag ${flag} does not take a value.`);
+            continue;
+        }
+        if (flag === '--sites-file') {
+            const next = inline ?? argv[i + 1];
+            if (next === undefined || next.startsWith('--')) {
+                errors.push('Flag --sites-file requires a value.');
+                continue;
+            }
+            sitesFile = next.trim();
+            if (inline === undefined)
+                i += 1;
+            continue;
+        }
+        if (flag.startsWith('--') || flag.startsWith('-')) {
+            errors.push(`Unknown flag: ${flag}`);
+            continue;
+        }
+        if (siteId !== '') {
+            errors.push(`Too many positional arguments; got "${siteId}" and "${flag}".`);
+            continue;
+        }
+        siteId = flag.trim();
+    }
+    return {
+        siteId,
+        yes,
+        ...(sitesFile !== undefined ? { sitesFile } : {}),
+        errors,
+        helpRequested,
+    };
+}
 export async function runCli(argv, deps) {
     const out = deps?.log ?? ((s) => process.stdout.write(`${s}\n`));
     const err = deps?.error ?? ((s) => process.stderr.write(`${s}\n`));
     const wizard = deps?.runWizard ?? ((wizDeps) => runWizardImpl(wizDeps));
     const install = deps?.installSkill ?? defaultInstallSkill;
+    const resolveSitesPath = deps?.resolveSitesPath ?? (() => defaultSitesFilePath());
+    const addSiteDeps = deps?.addSiteDeps ?? DEFAULT_ADD_SITE_DEPS;
+    const listSitesDeps = deps?.listSitesDeps ?? DEFAULT_LIST_SITES_DEPS;
+    // W12-R2A: wire a TTY confirm hook into the destructive-subcommand
+    // deps bags. Tests that supply `removeSiteDeps`/`setDefaultDeps`
+    // directly retain full control over the confirm channel (so the
+    // existing unit-tests keep working unchanged); only the default
+    // dispatcher path gets the `defaultConfirmTTY` injection.
+    const confirmHook = deps?.confirm ?? defaultConfirmTTY;
+    const removeSiteDeps = deps?.removeSiteDeps
+        ?? { ...DEFAULT_REMOVE_SITE_DEPS, confirm: confirmHook };
+    const setDefaultDeps = deps?.setDefaultDeps
+        ?? { ...DEFAULT_SET_DEFAULT_DEPS, confirm: confirmHook };
+    const testSiteDeps = deps?.testSiteDeps ?? DEFAULT_TEST_SITE_DEPS;
+    const addSiteFromWizard = deps?.addSiteFromWizard
+        ?? ((args, path) => addSiteCommand(args, path, addSiteDeps));
     const [subcommand] = argv;
+    // Build a `persistSite` callback for the wizard that translates
+    // the collected WizardAnswers into an `addSiteCommand` call so the
+    // sites.json registry stays in sync with the env-var path. This is
+    // the W9 "wizard internally invokes add-site" wiring.
+    const buildPersistSite = () => {
+        return async (answers) => {
+            const siteId = (answers.siteId !== undefined && answers.siteId.length > 0)
+                ? answers.siteId
+                : 'default';
+            const path = resolveSitesPath();
+            const args = {
+                url: answers.wpUrl,
+                token: answers.bearer,
+                siteId,
+                default: true,
+                yes: true,
+                platform: 'auto',
+            };
+            await addSiteFromWizard(args, path);
+        };
+    };
     if (subcommand === undefined || subcommand === 'setup') {
         const setupArgs = argv[0] === 'setup' ? argv.slice(1) : argv;
         const parsed = parseSetupArgs(setupArgs);
@@ -278,6 +574,10 @@ export async function runCli(argv, deps) {
         for (const line of parsed.warnings)
             err(`yt-builder-mcp: warning: ${line}`);
         const pickupMode = (parsed.pickup ?? '') !== '' || (parsed.nonce ?? '') !== '';
+        const persistSite = buildPersistSite();
+        function withPersist(base) {
+            return { ...base, persistSite };
+        }
         if (pickupMode) {
             if (parsed.errors.length > 0) {
                 for (const line of parsed.errors)
@@ -285,7 +585,7 @@ export async function runCli(argv, deps) {
                 err(HELP_TEXT);
                 return 2;
             }
-            return wizard(buildPickupDeps(parsed));
+            return wizard(withPersist(buildPickupDeps(parsed)));
         }
         if (parsed.nonInteractive) {
             if (parsed.errors.length > 0) {
@@ -294,7 +594,7 @@ export async function runCli(argv, deps) {
                 err(HELP_TEXT);
                 return 2;
             }
-            return wizard(buildNonInteractiveDeps(parsed));
+            return wizard(withPersist(buildNonInteractiveDeps(parsed)));
         }
         if (parsed.errors.length > 0) {
             for (const line of parsed.errors)
@@ -302,7 +602,7 @@ export async function runCli(argv, deps) {
             err(HELP_TEXT);
             return 1;
         }
-        return wizard();
+        return wizard(withPersist(DEFAULT_WIZARD_DEPS));
     }
     switch (subcommand) {
         case 'install-skill':
@@ -323,6 +623,221 @@ export async function runCli(argv, deps) {
                 return 2;
             }
         }
+        case 'add-site': {
+            const tail = argv.slice(1);
+            const parsed = parseAddSiteArgs(tail);
+            if (parsed.errors.some((e) => e === 'HELP_REQUESTED')) {
+                out(ADD_SITE_HELP);
+                return 0;
+            }
+            if (parsed.errors.length > 0) {
+                for (const line of parsed.errors)
+                    err(`yt-builder-mcp: ${line}`);
+                err(ADD_SITE_HELP);
+                return 2;
+            }
+            const path = parsed.sitesFile ?? resolveSitesPath();
+            const args = {
+                url: parsed.url,
+                ...(parsed.token.length > 0 ? { token: parsed.token } : {}),
+                ...(parsed.tokenRef.length > 0 ? { tokenRef: parsed.tokenRef } : {}),
+                ...(parsed.platform !== undefined ? { platform: parsed.platform } : {}),
+                ...(parsed.label.length > 0 ? { label: parsed.label } : {}),
+                siteId: parsed.siteId,
+                default: parsed.default,
+                yes: parsed.yes,
+            };
+            try {
+                const result = await addSiteCommand(args, path, addSiteDeps);
+                const stateBits = [];
+                if (result.becameDefault)
+                    stateBits.push('default — first site');
+                else if (result.defaultRequested)
+                    stateBits.push('default');
+                if (result.overwritten)
+                    stateBits.push('overwritten');
+                const state = stateBits.length > 0 ? ` (${stateBits.join(', ')})` : '';
+                const verb = result.overwritten ? 'updated' : 'added';
+                out(`✓ Site "${result.siteId}" ${verb}${state}.`);
+                out(`  Path: ${result.path}`);
+                out('Restart your AI client to pick up the new site list.');
+                return 0;
+            }
+            catch (e) {
+                if (e instanceof AddSiteError) {
+                    err(`yt-builder-mcp: add-site failed [${e.code}]: ${e.message}`);
+                    return 2;
+                }
+                const msg = e instanceof Error ? e.message : String(e);
+                err(`yt-builder-mcp: add-site failed: ${msg}`);
+                return 2;
+            }
+        }
+        case 'list-sites': {
+            const tail = argv.slice(1);
+            const parsed = parseIdArgs(tail, false);
+            // list-sites takes no positional id; if one slipped in it's
+            // a usage error (parseIdArgs allows zero or one).
+            if (parsed.helpRequested) {
+                out(LIST_SITES_HELP);
+                return 0;
+            }
+            if (parsed.siteId !== '') {
+                err(`yt-builder-mcp: list-sites takes no positional argument (got "${parsed.siteId}").`);
+                err(LIST_SITES_HELP);
+                return 2;
+            }
+            if (parsed.errors.length > 0) {
+                for (const line of parsed.errors)
+                    err(`yt-builder-mcp: ${line}`);
+                err(LIST_SITES_HELP);
+                return 2;
+            }
+            const path = parsed.sitesFile ?? resolveSitesPath();
+            try {
+                const lines = await listSitesCommand(path, listSitesDeps);
+                for (const line of lines)
+                    out(line);
+                return 0;
+            }
+            catch (e) {
+                const msg = e instanceof Error ? e.message : String(e);
+                err(`yt-builder-mcp: list-sites failed: ${msg}`);
+                return 2;
+            }
+        }
+        case 'remove-site': {
+            const tail = argv.slice(1);
+            const parsed = parseIdArgs(tail, true);
+            if (parsed.helpRequested) {
+                out(REMOVE_SITE_HELP);
+                return 0;
+            }
+            if (parsed.siteId === '') {
+                err('yt-builder-mcp: remove-site requires a <site_id> argument.');
+                err(REMOVE_SITE_HELP);
+                return 2;
+            }
+            if (parsed.errors.length > 0) {
+                for (const line of parsed.errors)
+                    err(`yt-builder-mcp: ${line}`);
+                err(REMOVE_SITE_HELP);
+                return 2;
+            }
+            const path = parsed.sitesFile ?? resolveSitesPath();
+            try {
+                const result = await removeSiteCommand(parsed.siteId, path, { yes: parsed.yes }, removeSiteDeps);
+                if (result.cancelled) {
+                    out(`Cancelled — site "${result.siteId}" not removed.`);
+                    return 0;
+                }
+                out(`✓ Site "${result.siteId}" removed.`);
+                if (result.promoted !== undefined) {
+                    out(`  Default promoted: ${result.promoted}`);
+                }
+                if (result.nowEmpty) {
+                    out('  (registry is now empty)');
+                }
+                out(`  Path: ${result.path}`);
+                out('Restart your AI client to pick up the new site list.');
+                return 0;
+            }
+            catch (e) {
+                if (e instanceof RemoveSiteError) {
+                    err(`yt-builder-mcp: remove-site failed [${e.code}]: ${e.message}`);
+                    return 2;
+                }
+                const msg = e instanceof Error ? e.message : String(e);
+                err(`yt-builder-mcp: remove-site failed: ${msg}`);
+                return 2;
+            }
+        }
+        case 'set-default': {
+            const tail = argv.slice(1);
+            // W12-R2A: set-default is a routing-destructive op (every
+            // subsequent default-routed tool call lands on a different
+            // site). Allow --yes so CI can flip without a prompt.
+            const parsed = parseIdArgs(tail, true);
+            if (parsed.helpRequested) {
+                out(SET_DEFAULT_HELP);
+                return 0;
+            }
+            if (parsed.siteId === '') {
+                err('yt-builder-mcp: set-default requires a <site_id> argument.');
+                err(SET_DEFAULT_HELP);
+                return 2;
+            }
+            if (parsed.errors.length > 0) {
+                for (const line of parsed.errors)
+                    err(`yt-builder-mcp: ${line}`);
+                err(SET_DEFAULT_HELP);
+                return 2;
+            }
+            const path = parsed.sitesFile ?? resolveSitesPath();
+            try {
+                const result = await setDefaultCommand(parsed.siteId, path, { yes: parsed.yes }, setDefaultDeps);
+                if (result.cancelled) {
+                    out(`Cancelled — default unchanged.`);
+                    return 0;
+                }
+                const prior = result.previousDefault !== null
+                    ? `(was: ${result.previousDefault})`
+                    : '(no previous default)';
+                out(`✓ Default set to "${result.siteId}" ${prior}.`);
+                out(`  Path: ${result.path}`);
+                out('Restart your AI client to pick up the new site list.');
+                return 0;
+            }
+            catch (e) {
+                if (e instanceof SetDefaultError) {
+                    err(`yt-builder-mcp: set-default failed [${e.code}]: ${e.message}`);
+                    return 2;
+                }
+                const msg = e instanceof Error ? e.message : String(e);
+                err(`yt-builder-mcp: set-default failed: ${msg}`);
+                return 2;
+            }
+        }
+        case 'test-site': {
+            const tail = argv.slice(1);
+            const parsed = parseIdArgs(tail, false);
+            if (parsed.helpRequested) {
+                out(TEST_SITE_HELP);
+                return 0;
+            }
+            if (parsed.siteId === '') {
+                err('yt-builder-mcp: test-site requires a <site_id> argument.');
+                err(TEST_SITE_HELP);
+                return 2;
+            }
+            if (parsed.errors.length > 0) {
+                for (const line of parsed.errors)
+                    err(`yt-builder-mcp: ${line}`);
+                err(TEST_SITE_HELP);
+                return 2;
+            }
+            const path = parsed.sitesFile ?? resolveSitesPath();
+            try {
+                const result = await testSiteCommand(parsed.siteId, path, testSiteDeps);
+                for (const line of renderTestSiteLines(result))
+                    out(line);
+                const exitCode = result.unknownSite
+                    || !result.pluginReachable
+                    || !result.bearerValid
+                    ? 2
+                    : 0;
+                return exitCode;
+            }
+            catch (e) {
+                if (e instanceof TestSiteError) {
+                    err(`yt-builder-mcp: test-site failed [${e.code}]: ${e.message}`);
+                    return 2;
+                }
+                const msg = e instanceof Error ? e.message : String(e);
+                err(`yt-builder-mcp: test-site failed: ${msg}`);
+                return 2;
+            }
+        }
         case '--version':
         case '-v': {
             out(SERVER_VERSION);