import { resolve } from "node:path";
import { CODEX_FALLBACK_SHELL, getCodexRuntimeShell, getDefaultCodexRuntimeShell, isFishShell } from "../../adapter/prompt/runtime-shell.ts";

const MIN_YIELD_TIME_MS = 250;
const MIN_NON_INTERACTIVE_EXEC_YIELD_TIME_MS = 5_000;
const MIN_EMPTY_WRITE_YIELD_TIME_MS = 30_000;
const MAX_YIELD_TIME_MS = 30_000;
export const MAX_EXEC_YIELD_TIME_MS = 1_800_000;
export const DEFAULT_EXEC_YIELD_TIME_MS = 10_000;
export const DEFAULT_WRITE_YIELD_TIME_MS = 250;
export const DEFAULT_MAX_EMPTY_WRITE_YIELD_TIME_MS = 1_800_000;

const BASH_SYNC_ENV_KEYS = [
	"PATH", "SHELL", "HOME", "XDG_CONFIG_HOME", "XDG_DATA_HOME", "XDG_CACHE_HOME", "BUN_INSTALL", "PNPM_HOME",
	"MISE_DATA_DIR", "MISE_CONFIG_DIR", "MISE_SHIMS_DIR", "CARGO_HOME", "GOPATH", "PI_WEB_RUN_STATE_PATH", "PI_CODEX_MODEL", "ANDROID_HOME", "ANDROID_NDK_HOME", "JAVA_HOME",
];

export function resolveWorkdir(baseCwd: string, workdir?: string): string {
	return workdir ? resolve(baseCwd, workdir) : baseCwd;
}

export function resolveShell(shell?: string): string {
	if (!shell || (process.platform === "win32" && /^(?:bash|bash\.exe)$/i.test(shell)))
		return getDefaultCodexRuntimeShell();
	return getCodexRuntimeShell(shell);
}

function shellEscape(value: string): string {
	if (/^[A-Za-z0-9_@%+=:,./-]+$/.test(value)) return value;
	return `'${value.replace(/'/g, `'"'"'`)}'`;
}

function shouldSyncBashEnv(requestedShell: string | undefined, effectiveShell: string): boolean {
	return effectiveShell === CODEX_FALLBACK_SHELL && isFishShell(requestedShell || process.env["SHELL"]!);
}

function buildSyncedBashCommand(command: string, env: NodeJS.ProcessEnv): string {
	const assignments: string[] = [];
	for (const key of BASH_SYNC_ENV_KEYS) {
		const value = key === "SHELL" ? CODEX_FALLBACK_SHELL : env[key]!;
		if (typeof value !== "string") continue;
		assignments.push(`export ${key}=${shellEscape(value)}`);
	}
	return assignments.length === 0 ? command : `${assignments.join("; ")}; ${command}`;
}

export function resolveExecution(requestedShell: string | undefined, command: string, extraEnv?: NodeJS.ProcessEnv, baseEnv: NodeJS.ProcessEnv = process.env): { shell: string; command: string; env: NodeJS.ProcessEnv } {
	const shell = resolveShell(requestedShell);
	const env: NodeJS.ProcessEnv = { ...baseEnv, ...extraEnv };
	if (!shouldSyncBashEnv(requestedShell, shell)) return { shell, command, env };
	env["SHELL"] = CODEX_FALLBACK_SHELL;
	return { shell, command: buildSyncedBashCommand(command, env), env };
}

function clampYieldTime(yieldTimeMs: number | undefined, fallback: number): number {
	return Math.min(MAX_YIELD_TIME_MS, Math.max(MIN_YIELD_TIME_MS, yieldTimeMs ?? fallback));
}

export function normalizeMinNonInteractiveExecYieldTime(value: number | undefined): number {
	return Math.min(MAX_EXEC_YIELD_TIME_MS, Math.max(MIN_YIELD_TIME_MS, value ?? MIN_NON_INTERACTIVE_EXEC_YIELD_TIME_MS));
}

export function normalizeMinEmptyWriteYieldTime(value: number | undefined): number {
	return Math.min(MAX_YIELD_TIME_MS, Math.max(MIN_YIELD_TIME_MS, value ?? MIN_EMPTY_WRITE_YIELD_TIME_MS));
}

export function clampExecYieldTime(yieldTimeMs: number | undefined, fallback: number, isInteractive: boolean, minNonInteractiveExecYieldTimeMs: number, maxYieldTimeMs = MAX_EXEC_YIELD_TIME_MS): number {
	const value = Math.min(maxYieldTimeMs, Math.max(MIN_YIELD_TIME_MS, yieldTimeMs ?? fallback));
	return isInteractive ? value : Math.min(maxYieldTimeMs, Math.max(minNonInteractiveExecYieldTimeMs, value));
}

export function clampWriteYieldTime(yieldTimeMs: number | undefined, fallback: number, isEmptyPoll: boolean, minEmptyWriteYieldTimeMs: number, maxEmptyWriteYieldTimeMs: number): number {
	return isEmptyPoll ? Math.min(maxEmptyWriteYieldTimeMs, Math.max(minEmptyWriteYieldTimeMs, yieldTimeMs ?? fallback)) : clampYieldTime(yieldTimeMs, fallback);
}