import type { ProviderHeaders } from "@earendil-works/pi-ai";

export const REMOTE_COMPACTION_V2_FEATURE = "remote_compaction_v2";

export function hasRemoteCompactionV2Input(input: readonly unknown[]): boolean {
	return input.some((item) => item !== null && typeof item === "object" && "type" in item
		&& (item.type === "compaction" || item.type === "compaction_summary" || item.type === "compaction_trigger"));
}

export function withRemoteCompactionV2Feature(headers: ProviderHeaders | undefined): ProviderHeaders {
	const merged: ProviderHeaders = { ...headers };
	for (const name of Object.keys(merged)) {
		if (name.toLowerCase() === "x-codex-beta-features") delete merged[name];
	}
	const configured = Object.entries(headers ?? {}).find(([name]) => name.toLowerCase() === "x-codex-beta-features")?.[1];
	const features = (configured ?? "")
		.split(",")
		.map((value) => value.trim())
		.filter(Boolean);
	if (!features.includes(REMOTE_COMPACTION_V2_FEATURE)) features.push(REMOTE_COMPACTION_V2_FEATURE);
	merged["x-codex-beta-features"] = features.join(",");
	return merged;
}