from pathlib import Path
# Run in the runner checkout. Select a Claude model via PROOF_MODEL_SELECTION JSON.
p = Path('scripts/scenarios/start-app.ts')
s = p.read_text().replace('defaultModelSelection: REPLAY,', 'defaultModelSelection: JSON.parse(process.env.PROOF_MODEL_SELECTION!),')
s = s.replace('enabled: false,', 'enabled: driver === "claudeAgent",\n            ...(driver === "claudeAgent" ? {config: {homePath: process.env.PROOF_CLAUDE_CONFIG_DIR}} : {}),')
p.write_text(s)
p = Path('scripts/e2e.config.ts')
s = p.read_text()
start = s.index('    {\n      name: "ios",')
end = s.index('\n  ],', start)
p.write_text(s[:start] + s[end:])