oh-my-codex 0.16.3 → 0.16.4

package/dist/config/generator.d.ts

@@ -10,10 +10,12 @@
  *   3. [table] sections (shell_environment_policy.set, mcp_servers, tui)
  */
 import type { UnifiedMcpRegistryServer } from "./mcp-registry.js";
+import { type CodexHookFeatureFlag } from "./codex-feature-flags.js";
 import type { HudPreset } from "../hud/types.js";
 interface MergeOptions {
     includeTui?: boolean;
     codexHooksFile?: string;
+    codexHookFeatureFlag?: CodexHookFeatureFlag;
     modelOverride?: string;
     sharedMcpServers?: UnifiedMcpRegistryServer[];
     sharedMcpRegistrySource?: string;
@@ -21,6 +23,7 @@ interface MergeOptions {
     statusLinePreset?: HudPreset;
     forceStatusLinePreset?: boolean;
     notifyCommand?: string[] | false;
+    includeFirstPartyMcp?: boolean;
 }
 export interface ModelContextRecommendation {
     model: string;
@@ -43,6 +46,7 @@ export declare function getRootModelName(config: string): string | undefined;
 export declare function formatTomlStringArray(values: readonly string[]): string;
 export declare function getRootTomlArray(config: string, key: string): string[] | null;
 export declare function isOmxManagedNotifyCommand(command: readonly string[] | null | undefined, pkgRoot?: string): boolean;
+export declare function sanitizePreviousNotifyCommand(command: readonly string[] | null | undefined, pkgRoot?: string): string[] | null;
 export declare function stripOmxSeededBehavioralDefaults(config: string): string;
 /**
  * Remove any existing OMX-owned top-level keys so we can re-insert them
@@ -51,7 +55,7 @@ export declare function stripOmxSeededBehavioralDefaults(config: string): string
 export declare function stripOmxTopLevelKeys(config: string): string;
 export declare function stripManagedCodexHookTrustState(config: string): string;
 export declare function upsertManagedCodexHookTrustState(config: string, pkgRoot: string, codexHooksFile: string | undefined): string;
-export declare function upsertPluginModeRuntimeFeatureFlags(config: string): string;
+export declare function upsertPluginModeRuntimeFeatureFlags(config: string, codexHookFeatureFlag?: CodexHookFeatureFlag): string;
 /**
  * Remove OMX-owned feature flags from the [features] section.
  * If the section becomes empty after removal, remove the section header too.
@@ -61,7 +65,7 @@ export declare function stripOmxFeatureFlags(config: string): string;
  * Preserve native Codex hook enablement without re-adding other OMX feature
  * flags. Used by uninstall when user-owned hooks remain in hooks.json.
  */
-export declare function upsertCodexHooksFeatureFlag(config: string): string;
+export declare function upsertCodexHooksFeatureFlag(config: string, codexHookFeatureFlag?: CodexHookFeatureFlag): string;
 export declare function stripOmxEnvSettings(config: string): string;
 export declare function stripExistingOmxBlocks(config: string): {
     cleaned: string;
@@ -71,6 +75,11 @@ export declare function stripExistingSharedMcpRegistryBlock(config: string): {
     cleaned: string;
     removed: number;
 };
+export declare function extractSharedMcpRegistryServersFromConfig(config: string): {
+    servers: UnifiedMcpRegistryServer[];
+    sourcePath?: string;
+};
+export declare function mergeSharedMcpRegistryBlock(config: string, servers: UnifiedMcpRegistryServer[], sourcePath?: string): string;
 /**
  * Merge OMX config into existing config.toml
  * Preserves existing user settings, appends OMX block if not present.

package/dist/config/generator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"generator.d.ts","sourceRoot":"","sources":["../../src/config/generator.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAQH,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,mBAAmB,CAAC;AAGlE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAEjD,UAAU,YAAY;IACpB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,gBAAgB,CAAC,EAAE,wBAAwB,EAAE,CAAC;IAC9C,uBAAuB,CAAC,EAAE,MAAM,CAAC;IACjC,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,gBAAgB,CAAC,EAAE,SAAS,CAAC;IAC7B,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAChC,aAAa,CAAC,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC;CAClC;AAiBD,MAAM,WAAW,0BAA0B;IACzC,KAAK,EAAE,MAAM,CAAC;IACd,kBAAkB,EAAE,MAAM,CAAC;IAC3B,0BAA0B,EAAE,MAAM,CAAC;CACpC;AAED,eAAO,MAAM,mBAAmB,YAAyB,CAAC;AAC1D,eAAO,MAAM,kCAAkC,SAAS,CAAC;AACzD,eAAO,MAAM,4CAA4C,SAAS,CAAC;AAEnE,wBAAgB,6BAA6B,CAC3C,KAAK,EAAE,MAAM,GACZ,0BAA0B,GAAG,IAAI,CAQnC;AAMD,eAAO,MAAM,0BAA0B,wmBACgkB,CAAC;AACxmB,eAAO,MAAM,iCAAiC,4oBAC6lB,CAAC;AAqB5oB,eAAO,MAAM,mBAAmB,EAAE,MAAM,CAAC,SAAS,EAAE,SAAS,MAAM,EAAE,CAIpE,CAAC;AAEF,eAAO,MAAM,0BAA0B,EAAE,SAAqB,CAAC;AAE/D,wBAAgB,mBAAmB,CACjC,MAAM,GAAE,SAAsC,GAC7C,MAAM,CAKR;AAmCD,wBAAgB,8BAA8B,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAoBrE;AAED,wBAAsB,sCAAsC,CAC1D,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,OAAO,CAAC,CASlB;AAED,wBAAgB,wBAAwB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAEhE;AAMD,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAEnE;AAyED,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,SAAS,MAAM,EAAE,GAAG,MAAM,CAEvE;AAED,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,IAAI,CAgB7E;AAED,wBAAgB,yBAAyB,CACvC,OAAO,EAAE,SAAS,MAAM,EAAE,GAAG,IAAI,GAAG,SAAS,EAC7C,OAAO,CAAC,EAAE,MAAM,GACf,OAAO,CAeT;AAoED,wBAAgB,gCAAgC,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CA2CvE;AA4DD;;;GAGG;AACH,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAE3D;AA+GD,wBAAgB,+BAA+B,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CA6BtE;AAED,wBAAgB,gCAAgC,CAC9C,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,MAAM,EACf,cAAc,EAAE,MAAM,GAAG,SAAS,GACjC,MAAM,CAaR;AAED,wBAAgB,mCAAmC,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CA8E1E;AAoOD;;;GAGG;AACH,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAuD3D;AAED;;;GAGG;AACH,wBAAgB,2BAA2B,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAgDlE;AAED,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAS1D;AAmQD,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,MAAM,GAAG;IACtD,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;CACjB,CAqCA;AAED,wBAAgB,mCAAmC,CAAC,MAAM,EAAE,MAAM,GAAG;IACnE,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;CACjB,CAqCA;AA+OD;;;;;;;;;;GAUG;AACH,wBAAgB,iBAAiB,CAC/B,cAAc,EAAE,MAAM,EACtB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,YAAiB,GACzB,MAAM,CAyER;AAED;;;;;;;;;;GAUG;AACH,wBAAsB,oBAAoB,CACxC,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,YAAiB,GACzB,OAAO,CAAC,OAAO,CAAC,CAgBlB;AAED,wBAAsB,WAAW,CAC/B,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,YAAiB,GACzB,OAAO,CAAC,IAAI,CAAC,CAoBf"}
+{"version":3,"file":"generator.d.ts","sourceRoot":"","sources":["../../src/config/generator.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAQH,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,mBAAmB,CAAC;AAClE,OAAO,EAKL,KAAK,oBAAoB,EAC1B,MAAM,0BAA0B,CAAC;AAMlC,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAEjD,UAAU,YAAY;IACpB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,oBAAoB,CAAC,EAAE,oBAAoB,CAAC;IAC5C,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,gBAAgB,CAAC,EAAE,wBAAwB,EAAE,CAAC;IAC9C,uBAAuB,CAAC,EAAE,MAAM,CAAC;IACjC,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,gBAAgB,CAAC,EAAE,SAAS,CAAC;IAC7B,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAChC,aAAa,CAAC,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC;IACjC,oBAAoB,CAAC,EAAE,OAAO,CAAC;CAChC;AAqBD,MAAM,WAAW,0BAA0B;IACzC,KAAK,EAAE,MAAM,CAAC;IACd,kBAAkB,EAAE,MAAM,CAAC;IAC3B,0BAA0B,EAAE,MAAM,CAAC;CACpC;AAED,eAAO,MAAM,mBAAmB,YAAyB,CAAC;AAC1D,eAAO,MAAM,kCAAkC,SAAS,CAAC;AACzD,eAAO,MAAM,4CAA4C,SAAS,CAAC;AAEnE,wBAAgB,6BAA6B,CAC3C,KAAK,EAAE,MAAM,GACZ,0BAA0B,GAAG,IAAI,CAQnC;AAMD,eAAO,MAAM,0BAA0B,wmBACgkB,CAAC;AACxmB,eAAO,MAAM,iCAAiC,4oBAC6lB,CAAC;AAqB5oB,eAAO,MAAM,mBAAmB,EAAE,MAAM,CAAC,SAAS,EAAE,SAAS,MAAM,EAAE,CAIpE,CAAC;AAEF,eAAO,MAAM,0BAA0B,EAAE,SAAqB,CAAC;AAE/D,wBAAgB,mBAAmB,CACjC,MAAM,GAAE,SAAsC,GAC7C,MAAM,CAKR;AAmCD,wBAAgB,8BAA8B,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAoBrE;AAED,wBAAsB,sCAAsC,CAC1D,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,OAAO,CAAC,CASlB;AAED,wBAAgB,wBAAwB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAEhE;AAMD,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAEnE;AAyED,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,SAAS,MAAM,EAAE,GAAG,MAAM,CAEvE;AAED,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,IAAI,CAgB7E;AASD,wBAAgB,yBAAyB,CACvC,OAAO,EAAE,SAAS,MAAM,EAAE,GAAG,IAAI,GAAG,SAAS,EAC7C,OAAO,CAAC,EAAE,MAAM,GACf,OAAO,CAeT;AAED,wBAAgB,6BAA6B,CAC3C,OAAO,EAAE,SAAS,MAAM,EAAE,GAAG,IAAI,GAAG,SAAS,EAC7C,OAAO,CAAC,EAAE,MAAM,GACf,MAAM,EAAE,GAAG,IAAI,CAIjB;AAoED,wBAAgB,gCAAgC,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CA2CvE;AA4DD;;;GAGG;AACH,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAE3D;AAqJD,wBAAgB,+BAA+B,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CA6BtE;AAyCD,wBAAgB,gCAAgC,CAC9C,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,MAAM,EACf,cAAc,EAAE,MAAM,GAAG,SAAS,GACjC,MAAM,CAgBR;AAED,wBAAgB,mCAAmC,CACjD,MAAM,EAAE,MAAM,EACd,oBAAoB,GAAE,oBAAsD,GAC3E,MAAM,CA4DR;AAoOD;;;GAGG;AACH,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAuD3D;AAED;;;GAGG;AACH,wBAAgB,2BAA2B,CACzC,MAAM,EAAE,MAAM,EACd,oBAAoB,GAAE,oBAAsD,GAC3E,MAAM,CA+BR;AAED,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAS1D;AA+QD,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,MAAM,GAAG;IACtD,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;CACjB,CAqCA;AAED,wBAAgB,mCAAmC,CAAC,MAAM,EAAE,MAAM,GAAG;IACnE,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;CACjB,CAqCA;AAoCD,wBAAgB,yCAAyC,CAAC,MAAM,EAAE,MAAM,GAAG;IACzE,OAAO,EAAE,wBAAwB,EAAE,CAAC;IACpC,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB,CA2CA;AA8KD,wBAAgB,2BAA2B,CACzC,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,wBAAwB,EAAE,EACnC,UAAU,CAAC,EAAE,MAAM,GAClB,MAAM,CAiBR;AAsED;;;;;;;;;;GAUG;AACH,wBAAgB,iBAAiB,CAC/B,cAAc,EAAE,MAAM,EACtB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,YAAiB,GACzB,MAAM,CA8ER;AAED;;;;;;;;;;GAUG;AACH,wBAAsB,oBAAoB,CACxC,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,YAAiB,GACzB,OAAO,CAAC,OAAO,CAAC,CAgBlB;AAED,wBAAsB,WAAW,CAC/B,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,YAAiB,GACzB,OAAO,CAAC,IAAI,CAAC,CAoBf"}

package/dist/config/generator.js

@@ -11,15 +11,19 @@
  */
 import { readFile, writeFile } from "fs/promises";
 import { existsSync } from "fs";
-import { isAbsolute, join, resolve } from "path";
+import { join, resolve } from "path";
 import TOML from "@iarna/toml";
 import { AGENT_DEFINITIONS } from "../agents/definitions.js";
 import { DEFAULT_FRONTIER_MODEL } from "./models.js";
-import { getOmxFirstPartySetupMcpServers } from "./omx-first-party-mcp.js";
+import { DEFAULT_CODEX_HOOK_FEATURE_FLAG, CODEX_HOOK_FEATURE_FLAGS, formatCodexHookFeatureFlagLine, normalizeCodexHookFeatureFlag, } from "./codex-feature-flags.js";
+import { OMX_FIRST_PARTY_MCP_SERVER_NAMES, getOmxFirstPartySetupMcpServers, } from "./omx-first-party-mcp.js";
 import { buildManagedCodexHookTrustToml } from "./codex-hooks.js";
 function escapeTomlString(value) {
     return value.replace(/\\/g, "\\\\").replace(/"/g, '\\"');
 }
+function isRecord(value) {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+}
 // ---------------------------------------------------------------------------
 // Top-level OMX keys (must live before any [table] header)
 // ---------------------------------------------------------------------------
@@ -206,23 +210,37 @@ export function getRootTomlArray(config, key) {
         return null;
     }
 }
+function resolveNotifyEntrypoint(command) {
+    if (!/(?:^|[\\/])node(?:\.exe)?$/i.test(command[0] ?? "")) {
+        return command[0];
+    }
+    return command.slice(1).find((arg) => !arg.startsWith("-"));
+}
 export function isOmxManagedNotifyCommand(command, pkgRoot) {
     if (!command)
         return false;
+    const entrypoint = resolveNotifyEntrypoint(command);
+    if (!entrypoint)
+        return false;
+    if (!/(?:^|[\\/])notify-(?:hook|dispatcher)\.js$/.test(entrypoint)) {
+        return false;
+    }
     const managedScripts = pkgRoot
         ? new Set([
             resolve(pkgRoot, "dist", "scripts", "notify-hook.js"),
             resolve(pkgRoot, "dist", "scripts", "notify-dispatcher.js"),
         ])
         : new Set();
-    return command.some((part) => {
-        if (!/(?:^|[\\/])notify-(?:hook|dispatcher)\.js$/.test(part))
-            return false;
-        if (pkgRoot) {
-            return isAbsolute(part) && managedScripts.has(resolve(part));
-        }
-        return /(?:^|[\\/])oh-my-codex(?:[\\/]|$)/.test(part);
-    });
+    if (pkgRoot && managedScripts.has(resolve(entrypoint)))
+        return true;
+    return /(?:^|[\\/])oh-my-codex(?:[\\/]|$)/.test(entrypoint);
+}
+export function sanitizePreviousNotifyCommand(command, pkgRoot) {
+    if (!command || command.length === 0)
+        return null;
+    if (isOmxManagedNotifyCommand(command, pkgRoot))
+        return null;
+    return [...command];
 }
 function getOmxTopLevelLines(pkgRoot, existingConfig = "", modelOverride, notifyCommand = getDefaultNotifyCommand(pkgRoot)) {
     const rootValues = parseRootKeyValues(existingConfig);
@@ -346,16 +364,56 @@ export function stripOmxTopLevelKeys(config) {
 // ---------------------------------------------------------------------------
 // [features] upsert
 // ---------------------------------------------------------------------------
-function upsertFeatureFlags(config) {
+function isFeatureFlagLine(line, featureFlag) {
+    return new RegExp(`^\\s*${featureFlag}\\s*=`).test(line);
+}
+function isAnyCodexHookFeatureFlagLine(line) {
+    return CODEX_HOOK_FEATURE_FLAGS.some((flag) => isFeatureFlagLine(line, flag));
+}
+function upsertCodexHookFeatureFlagInSection(lines, featuresStart, sectionEnd, codexHookFeatureFlag) {
+    const featureFlag = normalizeCodexHookFeatureFlag(codexHookFeatureFlag);
+    let featureFlagIdx = -1;
+    let fallbackAliasIdx = -1;
+    for (let i = featuresStart + 1; i < sectionEnd; i++) {
+        if (isFeatureFlagLine(lines[i], featureFlag)) {
+            featureFlagIdx = i;
+        }
+        else if (isAnyCodexHookFeatureFlagLine(lines[i]) && fallbackAliasIdx < 0) {
+            fallbackAliasIdx = i;
+        }
+    }
+    if (featureFlagIdx < 0 && fallbackAliasIdx >= 0) {
+        featureFlagIdx = fallbackAliasIdx;
+    }
+    if (featureFlagIdx >= 0) {
+        lines[featureFlagIdx] = formatCodexHookFeatureFlagLine(featureFlag);
+    }
+    else {
+        lines.splice(sectionEnd, 0, formatCodexHookFeatureFlagLine(featureFlag));
+        featureFlagIdx = sectionEnd;
+        sectionEnd += 1;
+    }
+    for (let i = sectionEnd - 1; i > featuresStart; i--) {
+        if (i !== featureFlagIdx && isAnyCodexHookFeatureFlagLine(lines[i])) {
+            lines.splice(i, 1);
+            sectionEnd -= 1;
+            if (featureFlagIdx > i)
+                featureFlagIdx -= 1;
+        }
+    }
+    return { sectionEnd, featureFlagIndex: featureFlagIdx };
+}
+function upsertFeatureFlags(config, codexHookFeatureFlag = DEFAULT_CODEX_HOOK_FEATURE_FLAG) {
     const lines = config.split(/\r?\n/);
     const featuresStart = lines.findIndex((line) => /^\s*\[features\]\s*$/.test(line));
+    const hookFeatureFlagLine = formatCodexHookFeatureFlagLine(codexHookFeatureFlag);
     if (featuresStart < 0) {
         const base = config.trimEnd();
         const featureBlock = [
             "[features]",
             "multi_agent = true",
             "child_agents_md = true",
-            "codex_hooks = true",
+            hookFeatureFlagLine,
             "goals = true",
             "",
         ].join("\n");
@@ -381,9 +439,6 @@ function upsertFeatureFlags(config) {
     }
     let multiAgentIdx = -1;
     let childAgentsIdx = -1;
-    let codexHooksIdx = -1;
-    let unsupportedHooksIdx = -1;
-    let goalsIdx = -1;
     for (let i = featuresStart + 1; i < sectionEnd; i++) {
         if (/^\s*multi_agent\s*=/.test(lines[i])) {
             multiAgentIdx = i;
@@ -391,15 +446,6 @@ function upsertFeatureFlags(config) {
         else if (/^\s*child_agents_md\s*=/.test(lines[i])) {
             childAgentsIdx = i;
         }
-        else if (/^\s*hooks\s*=/.test(lines[i])) {
-            unsupportedHooksIdx = i;
-        }
-        else if (/^\s*codex_hooks\s*=/.test(lines[i])) {
-            codexHooksIdx = i;
-        }
-        else if (/^\s*goals\s*=/.test(lines[i])) {
-            goalsIdx = i;
-        }
     }
     if (multiAgentIdx >= 0) {
         lines[multiAgentIdx] = "multi_agent = true";
@@ -415,27 +461,11 @@ function upsertFeatureFlags(config) {
         lines.splice(sectionEnd, 0, "child_agents_md = true");
         sectionEnd += 1;
     }
-    if (codexHooksIdx >= 0) {
-        lines[codexHooksIdx] = "codex_hooks = true";
-    }
-    else if (unsupportedHooksIdx >= 0) {
-        lines[unsupportedHooksIdx] = "codex_hooks = true";
-        codexHooksIdx = unsupportedHooksIdx;
-        unsupportedHooksIdx = -1;
-    }
-    else {
-        lines.splice(sectionEnd, 0, "codex_hooks = true");
-        codexHooksIdx = sectionEnd;
-        sectionEnd += 1;
-    }
-    for (let i = sectionEnd - 1; i > featuresStart; i--) {
-        if (i !== codexHooksIdx && /^\s*hooks\s*=/.test(lines[i])) {
-            lines.splice(i, 1);
-            sectionEnd -= 1;
-            if (codexHooksIdx > i)
-                codexHooksIdx -= 1;
-            if (goalsIdx > i)
-                goalsIdx -= 1;
+    ({ sectionEnd } = upsertCodexHookFeatureFlagInSection(lines, featuresStart, sectionEnd, codexHookFeatureFlag));
+    let goalsIdx = -1;
+    for (let i = featuresStart + 1; i < sectionEnd; i++) {
+        if (/^\s*goals\s*=/.test(lines[i])) {
+            goalsIdx = i;
         }
     }
     if (goalsIdx >= 0) {
@@ -469,9 +499,38 @@ export function stripManagedCodexHookTrustState(config) {
     }
     return kept.join("\n").replace(/\n{3,}/g, "\n\n").trimEnd();
 }
+function managedCodexHookTrustStateHeaders(hookTrustToml) {
+    const headers = new Set();
+    for (const line of hookTrustToml.split(/\r?\n/)) {
+        const trimmed = line.trim();
+        if (/^\[hooks\.state\./.test(trimmed)) {
+            headers.add(trimmed);
+        }
+    }
+    return headers;
+}
+function stripUnfencedManagedCodexHookTrustState(config, hookTrustToml) {
+    const managedHeaders = managedCodexHookTrustStateHeaders(hookTrustToml);
+    if (managedHeaders.size === 0)
+        return config;
+    const lines = config.split(/\r?\n/);
+    const kept = [];
+    for (let i = 0; i < lines.length;) {
+        if (!managedHeaders.has(lines[i].trim())) {
+            kept.push(lines[i]);
+            i += 1;
+            continue;
+        }
+        i += 1;
+        while (i < lines.length && !TOML_TABLE_HEADER_PATTERN.test(lines[i])) {
+            i += 1;
+        }
+    }
+    return kept.join("\n").replace(/\n{3,}/g, "\n\n").trimEnd();
+}
 export function upsertManagedCodexHookTrustState(config, pkgRoot, codexHooksFile) {
-    const stripped = stripManagedCodexHookTrustState(config);
     const hookTrustToml = buildManagedCodexHookTrustToml(codexHooksFile, pkgRoot);
+    const stripped = stripUnfencedManagedCodexHookTrustState(stripManagedCodexHookTrustState(config), hookTrustToml);
     if (!hookTrustToml)
         return `${stripped}\n`;
     return [
@@ -484,14 +543,15 @@ export function upsertManagedCodexHookTrustState(config, pkgRoot, codexHooksFile
         "",
     ].filter((line, index) => index !== 0 || line.length > 0).join("\n");
 }
-export function upsertPluginModeRuntimeFeatureFlags(config) {
+export function upsertPluginModeRuntimeFeatureFlags(config, codexHookFeatureFlag = DEFAULT_CODEX_HOOK_FEATURE_FLAG) {
     const lines = config.split(/\r?\n/);
     const featuresStart = lines.findIndex((line) => /^\s*\[features\]\s*$/.test(line));
+    const hookFeatureFlagLine = formatCodexHookFeatureFlagLine(codexHookFeatureFlag);
     if (featuresStart < 0) {
         const base = config.trimEnd();
         const featureBlock = [
             "[features]",
-            "codex_hooks = true",
+            hookFeatureFlagLine,
             "goals = true",
             "",
         ].join("\n");
@@ -515,43 +575,13 @@ export function upsertPluginModeRuntimeFeatureFlags(config) {
             sectionEnd -= 1;
         }
     }
-    let codexHooksIdx = -1;
-    let unsupportedHooksIdx = -1;
+    ({ sectionEnd } = upsertCodexHookFeatureFlagInSection(lines, featuresStart, sectionEnd, codexHookFeatureFlag));
     let goalsIdx = -1;
     for (let i = featuresStart + 1; i < sectionEnd; i++) {
-        if (/^\s*hooks\s*=/.test(lines[i])) {
-            unsupportedHooksIdx = i;
-        }
-        else if (/^\s*codex_hooks\s*=/.test(lines[i])) {
-            codexHooksIdx = i;
-        }
-        else if (/^\s*goals\s*=/.test(lines[i])) {
+        if (/^\s*goals\s*=/.test(lines[i])) {
             goalsIdx = i;
         }
     }
-    if (codexHooksIdx >= 0) {
-        lines[codexHooksIdx] = "codex_hooks = true";
-    }
-    else if (unsupportedHooksIdx >= 0) {
-        lines[unsupportedHooksIdx] = "codex_hooks = true";
-        codexHooksIdx = unsupportedHooksIdx;
-        unsupportedHooksIdx = -1;
-    }
-    else {
-        lines.splice(sectionEnd, 0, "codex_hooks = true");
-        codexHooksIdx = sectionEnd;
-        sectionEnd++;
-    }
-    for (let i = sectionEnd - 1; i > featuresStart; i--) {
-        if (i !== codexHooksIdx && /^\s*hooks\s*=/.test(lines[i])) {
-            lines.splice(i, 1);
-            sectionEnd--;
-            if (codexHooksIdx > i)
-                codexHooksIdx--;
-            if (goalsIdx > i)
-                goalsIdx--;
-        }
-    }
     if (goalsIdx >= 0) {
         lines[goalsIdx] = "goals = true";
     }
@@ -767,12 +797,13 @@ export function stripOmxFeatureFlags(config) {
  * Preserve native Codex hook enablement without re-adding other OMX feature
  * flags. Used by uninstall when user-owned hooks remain in hooks.json.
  */
-export function upsertCodexHooksFeatureFlag(config) {
+export function upsertCodexHooksFeatureFlag(config, codexHookFeatureFlag = DEFAULT_CODEX_HOOK_FEATURE_FLAG) {
     const lines = config.split(/\r?\n/);
     const featuresStart = lines.findIndex((line) => /^\s*\[features\]\s*$/.test(line));
+    const hookFeatureFlagLine = formatCodexHookFeatureFlagLine(codexHookFeatureFlag);
     if (featuresStart < 0) {
         const base = config.trimEnd();
-        const featureBlock = ["[features]", "codex_hooks = true", ""].join("\n");
+        const featureBlock = ["[features]", hookFeatureFlagLine, ""].join("\n");
         return base.length === 0 ? featureBlock : `${base}\n${featureBlock}`;
     }
     let sectionEnd = lines.length;
@@ -782,31 +813,7 @@ export function upsertCodexHooksFeatureFlag(config) {
             break;
         }
     }
-    let codexHooksIdx = -1;
-    for (let i = featuresStart + 1; i < sectionEnd; i++) {
-        if (/^\s*codex_hooks\s*=/.test(lines[i])) {
-            codexHooksIdx = i;
-            lines[i] = "codex_hooks = true";
-            break;
-        }
-        if (codexHooksIdx < 0 && /^\s*hooks\s*=/.test(lines[i])) {
-            codexHooksIdx = i;
-            lines[i] = "codex_hooks = true";
-        }
-    }
-    if (codexHooksIdx < 0) {
-        lines.splice(sectionEnd, 0, "codex_hooks = true");
-        codexHooksIdx = sectionEnd;
-        sectionEnd += 1;
-    }
-    for (let i = sectionEnd - 1; i > featuresStart; i--) {
-        if (i !== codexHooksIdx && /^\s*hooks\s*=/.test(lines[i])) {
-            lines.splice(i, 1);
-            sectionEnd -= 1;
-            if (codexHooksIdx > i)
-                codexHooksIdx -= 1;
-        }
-    }
+    upsertCodexHookFeatureFlagInSection(lines, featuresStart, sectionEnd, codexHookFeatureFlag);
     return lines.join("\n");
 }
 export function stripOmxEnvSettings(config) {
@@ -822,6 +829,13 @@ export function stripOmxEnvSettings(config) {
  * Check whether a TOML table name belongs to a legacy OMX-managed agent entry.
  * Handles both `agents.name` and `agents."name"` forms.
  */
+function isOmxFirstPartyMcpSection(tableName) {
+    const match = tableName.match(/^mcp_servers\.(?:"([^"]+)"|([A-Za-z0-9_-]+))$/);
+    const name = match?.[1] ?? match?.[2];
+    return Boolean(name &&
+        (OMX_FIRST_PARTY_MCP_SERVER_NAMES.includes(name) ||
+            name === "omx_team_run"));
+}
 function isLegacyOmxAgentSection(tableName) {
     const m = tableName.match(/^agents\.(?:"([^"]+)"|(\w[\w-]*))$/);
     if (!m)
@@ -834,7 +848,8 @@ function isLegacyOmxAgentSection(tableName) {
  * This covers legacy configs that were written before markers were added,
  * or configs where the marker was accidentally removed.
  *
- * Targets: [mcp_servers.omx_*], legacy [agents.<name>] entries, [tui]
+ * Targets: exact first-party [mcp_servers.<name>] entries, retired
+ * [mcp_servers.omx_team_run], and legacy [agents.<name>] entries.
  */
 function stripOrphanedOmxSections(config) {
     const lines = config.split(/\r?\n/);
@@ -848,7 +863,7 @@ function stripOrphanedOmxSections(config) {
             // Note: [tui] is NOT stripped here because it could be user-owned.
             // The marker-based stripExistingOmxBlocks already handles [tui]
             // when it lives inside the OMX marker block.
-            const isOmxSection = /^mcp_servers\.omx_/.test(tableName) ||
+            const isOmxSection = isOmxFirstPartyMcpSection(tableName) ||
                 isLegacyOmxAgentSection(tableName);
             if (isOmxSection) {
                 // Remove preceding OMX comment lines and blank lines
@@ -1088,6 +1103,72 @@ export function stripExistingSharedMcpRegistryBlock(config) {
     }
     return { cleaned, removed };
 }
+function getExistingSharedMcpRegistryBlocks(config) {
+    const blocks = [];
+    let cursor = 0;
+    while (cursor < config.length) {
+        const markerIdx = config.indexOf(SHARED_MCP_REGISTRY_MARKER, cursor);
+        if (markerIdx < 0)
+            break;
+        let blockStart = config.lastIndexOf("\n", markerIdx);
+        blockStart = blockStart >= 0 ? blockStart + 1 : 0;
+        const previousLineEnd = blockStart - 1;
+        if (previousLineEnd >= 0) {
+            const previousLineStart = config.lastIndexOf("\n", previousLineEnd - 1);
+            const previousLine = config.slice(previousLineStart + 1, previousLineEnd);
+            if (/^# =+$/.test(previousLine.trim())) {
+                blockStart = previousLineStart >= 0 ? previousLineStart + 1 : 0;
+            }
+        }
+        let blockEnd = config.length;
+        const endIdx = config.indexOf(SHARED_MCP_REGISTRY_END_MARKER, markerIdx);
+        if (endIdx >= 0) {
+            const endLineBreak = config.indexOf("\n", endIdx);
+            blockEnd = endLineBreak >= 0 ? endLineBreak + 1 : config.length;
+        }
+        blocks.push(config.slice(blockStart, blockEnd));
+        cursor = blockEnd;
+    }
+    return blocks;
+}
+export function extractSharedMcpRegistryServersFromConfig(config) {
+    const servers = [];
+    let sourcePath;
+    for (const block of getExistingSharedMcpRegistryBlocks(config)) {
+        sourcePath ??= block.match(/^# Source:\s*(.+?)\s*$/m)?.[1];
+        let parsed;
+        try {
+            parsed = TOML.parse(block);
+        }
+        catch {
+            continue;
+        }
+        if (!isRecord(parsed) || !isRecord(parsed.mcp_servers))
+            continue;
+        for (const [name, value] of Object.entries(parsed.mcp_servers)) {
+            if (!isRecord(value) || typeof value.command !== "string")
+                continue;
+            const args = Array.isArray(value.args)
+                ? value.args.filter((arg) => typeof arg === "string")
+                : [];
+            const enabled = typeof value.enabled === "boolean" ? value.enabled : true;
+            const timeoutCandidate = typeof value.startup_timeout_sec === "number"
+                ? value.startup_timeout_sec
+                : value.startupTimeoutSec;
+            const startupTimeoutSec = typeof timeoutCandidate === "number" && Number.isFinite(timeoutCandidate)
+                ? timeoutCandidate
+                : undefined;
+            servers.push({
+                name,
+                command: value.command,
+                args,
+                enabled,
+                ...(startupTimeoutSec !== undefined ? { startupTimeoutSec } : {}),
+            });
+        }
+    }
+    return { servers, sourcePath };
+}
 function toMcpServerTableKey(name) {
     if (/^[A-Za-z0-9_-]+$/.test(name)) {
         return `mcp_servers.${name}`;
@@ -1207,11 +1288,25 @@ function getSharedMcpRegistryBlock(servers, sourcePath, existingConfig) {
     lines.push(SHARED_MCP_REGISTRY_END_MARKER);
     return lines.join("\n");
 }
+export function mergeSharedMcpRegistryBlock(config, servers, sourcePath) {
+    const stripped = stripExistingSharedMcpRegistryBlock(config);
+    const existing = stripped.cleaned;
+    const sharedRegistryBlock = getSharedMcpRegistryBlock(servers, sourcePath, existing);
+    const body = existing.trimEnd();
+    const merged = sharedRegistryBlock
+        ? body
+            ? `${body}\n\n${sharedRegistryBlock}\n`
+            : `${sharedRegistryBlock}\n`
+        : body
+            ? `${body}\n`
+            : "";
+    return addDefaultLauncherMcpStartupTimeouts(merged);
+}
 /**
  * OMX table-section block (MCP servers, TUI).
  * Contains ONLY [table] sections — no bare keys.
  */
-function getOmxTablesBlock(pkgRoot, includeTui = true, statusLinePreset = DEFAULT_STATUS_LINE_PRESET, codexHooksFile) {
+function getOmxTablesBlock(pkgRoot, includeTui = true, statusLinePreset = DEFAULT_STATUS_LINE_PRESET, codexHooksFile, includeFirstPartyMcp = false) {
     const lines = [
         "",
         "# ============================================================",
@@ -1219,17 +1314,19 @@ function getOmxTablesBlock(pkgRoot, includeTui = true, statusLinePreset = DEFAUL
         "# Managed by omx setup - manual edits preserved on next setup",
         "# ============================================================",
     ];
-    for (const server of getOmxFirstPartySetupMcpServers(pkgRoot)) {
-        lines.push("");
-        lines.push(server.title);
-        lines.push(`[mcp_servers.${server.name}]`);
-        lines.push(`command = "${escapeTomlString(server.command)}"`);
-        lines.push(`args = [${server.args
-            .map((arg) => `"${escapeTomlString(arg)}"`)
-            .join(", ")}]`);
-        lines.push(`enabled = ${server.enabled ? "true" : "false"}`);
-        if (typeof server.startupTimeoutSec === "number") {
-            lines.push(`startup_timeout_sec = ${server.startupTimeoutSec}`);
+    if (includeFirstPartyMcp) {
+        for (const server of getOmxFirstPartySetupMcpServers(pkgRoot)) {
+            lines.push("");
+            lines.push(server.title);
+            lines.push(`[mcp_servers.${server.name}]`);
+            lines.push(`command = "${escapeTomlString(server.command)}"`);
+            lines.push(`args = [${server.args
+                .map((arg) => `"${escapeTomlString(arg)}"`)
+                .join(", ")}]`);
+            lines.push(`enabled = ${server.enabled ? "true" : "false"}`);
+            if (typeof server.startupTimeoutSec === "number") {
+                lines.push(`startup_timeout_sec = ${server.startupTimeoutSec}`);
+            }
         }
     }
     const hookTrustToml = buildManagedCodexHookTrustToml(codexHooksFile, pkgRoot);
@@ -1295,11 +1392,12 @@ export function buildMergedConfig(existingConfig, pkgRoot, options = {}) {
     }
     existing = stripOrphanedManagedNotify(existing, pkgRoot);
     existing = stripManagedCodexHookTrustState(existing);
+    existing = stripUnfencedManagedCodexHookTrustState(existing, buildManagedCodexHookTrustToml(options.codexHooksFile, pkgRoot));
     if (options.modelOverride) {
         existing = stripRootLevelKeys(existing, ["model"]);
     }
     existing = stripOrphanedOmxSections(existing);
-    existing = upsertFeatureFlags(existing);
+    existing = upsertFeatureFlags(existing, options.codexHookFeatureFlag);
     existing = upsertEnvSettings(existing);
     existing = upsertAgentsSettings(existing);
     const tuiUpsert = includeTui
@@ -1311,7 +1409,7 @@ export function buildMergedConfig(existingConfig, pkgRoot, options = {}) {
     const topLines = getOmxTopLevelLines(pkgRoot, existing, options.modelOverride, options.notifyCommand === undefined
         ? getDefaultNotifyCommand(pkgRoot)
         : options.notifyCommand);
-    const tablesBlock = getOmxTablesBlock(pkgRoot, includeTui && !tuiUpsert.hadExistingTui, statusLinePreset, options.codexHooksFile);
+    const tablesBlock = getOmxTablesBlock(pkgRoot, includeTui && !tuiUpsert.hadExistingTui, statusLinePreset, options.codexHooksFile, options.includeFirstPartyMcp === true);
     const sharedRegistryBlock = getSharedMcpRegistryBlock(options.sharedMcpServers ?? [], options.sharedMcpRegistrySource, existing);
     let body = existing.trimEnd();
     if (sharedRegistryBlock) {