openai-migrate-to-codex/scripts/migrate/codex_config.py

Version 49f948fa.bb1 · Apache-2.0. This preview displays packaged text and does not execute code. Treat the contents as untrusted instructions.

← Return to resource and package checksum

"""Render Codex config from Claude Code settings and MCP inputs.

This module owns `.codex/config.toml` generation. It reads Claude Code
settings for model/sandbox equivalents, asks `mcps.py` for MCP server tables,
and adds Codex-native defaults that are not MCP-specific, such as the friendly
personality used for Claude Code migrations.
"""

from __future__ import annotations

import tomllib
from pathlib import Path

from migrate.common import (
    CODEX_CONFIG_PATH,
    ConversionResult,
    GeneratedText,
    MigrationReportItem,
    MigrationSummary,
    PlannedArtifact,
    ScopePaths,
    json_string,
    json_string_tuple,
    load_scope_settings,
    map_model_name,
    map_permission_mode,
)
from migrate.hooks import has_convertible_hooks
from migrate.mcps import (
    mcp_report_items,
    mcp_server_toml_table,
    read_claude_mcp_servers,
    validate_mcp_commands,
)
from utils.util import TomlValue, render_toml_document


DEFAULT_CODEX_PERSONALITY = "friendly"


def convert_settings(scope: ScopePaths) -> ConversionResult:
    settings = load_scope_settings(scope.source)
    mcp_servers = read_claude_mcp_servers(scope.source)
    if not settings and not mcp_servers:
        return ConversionResult()

    enabled_mcp_servers = json_string_tuple(settings.get("enabledMcpjsonServers"))
    disabled_mcp_servers = frozenset(
        json_string_tuple(settings.get("disabledMcpjsonServers"))
    )
    config_toml = render_codex_config(
        model=json_string(settings.get("model")),
        permission_mode=json_string(settings.get("permissionMode")),
        enabled_mcp_servers=enabled_mcp_servers,
        disabled_mcp_servers=disabled_mcp_servers,
        mcp_servers=mcp_servers,
        codex_hooks_enabled=has_convertible_hooks(scope.source),
    )
    if not config_toml.strip():
        return ConversionResult()
    return ConversionResult(
        summary=MigrationSummary(mcp_servers=len(mcp_servers)),
        artifacts=[
            PlannedArtifact(
                relative_path=CODEX_CONFIG_PATH,
                payload=GeneratedText(config_toml),
            )
        ],
        report_items=mcp_report_items(mcp_servers),
    )


def render_codex_config(
    model: str | None,
    permission_mode: str | None,
    enabled_mcp_servers: tuple[str, ...],
    disabled_mcp_servers: frozenset[str],
    mcp_servers: tuple[tuple[str, dict[str, object]], ...],
    codex_hooks_enabled: bool,
) -> str:
    document: dict[str, TomlValue] = {}
    if model:
        document["model"] = map_model_name(model)
    sandbox_mode = map_permission_mode(permission_mode)
    if sandbox_mode:
        document["sandbox_mode"] = sandbox_mode

    if mcp_servers:
        document["mcp_servers"] = {
            server_name: mcp_server_toml_table(
                server_name,
                server_config,
                enabled_mcp_servers,
                disabled_mcp_servers,
            )
            for server_name, server_config in mcp_servers
        }

    if codex_hooks_enabled:
        document["features"] = {"codex_hooks": True}

    if document:
        document = {"personality": DEFAULT_CODEX_PERSONALITY, **document}

    return render_toml_document(document)


def validate_config_toml(target_root: Path) -> list[MigrationReportItem]:
    config_path = target_root / CODEX_CONFIG_PATH
    if not config_path.exists():
        return [
            MigrationReportItem(
                "warning",
                CODEX_CONFIG_PATH,
                "not present; no Codex config to validate.",
            )
        ]

    try:
        parsed = tomllib.loads(config_path.read_text())
    except tomllib.TOMLDecodeError as exc:
        return [
            MigrationReportItem(
                "error",
                CODEX_CONFIG_PATH,
                f"invalid TOML: {exc}.",
            )
        ]

    report_items = [
        MigrationReportItem("ok", CODEX_CONFIG_PATH, "valid TOML."),
    ]
    report_items.extend(validate_mcp_commands(parsed))
    return report_items