import type {
	ExpressionRole2d,
	ExpressionRole3d,
	RenderScene2d,
	RenderScene3d
} from './types';

type WasmScene = Omit<RenderScene2d, 'expression_roles' | 'model_revisions'> & {
	expression_roles: Record<string, ExpressionRole2d> | Map<string, ExpressionRole2d>;
	model_revisions: Record<string, number> | Map<string, number>;
};

type WasmScene3d = Omit<RenderScene3d, 'expression_roles' | 'model_revisions' | 'coverage_revisions'> & {
	expression_roles: Record<string, ExpressionRole3d> | Map<string, ExpressionRole3d>;
	model_revisions: Record<string, number> | Map<string, number>;
	coverage_revisions?: Record<string, number> | Map<string, number>;
};

/** Converts serde-wasm-bindgen map values into the plain records used by the UI. */
export function normalizeWasmScene(scene: WasmScene): RenderScene2d {
	return {
		...scene,
		expression_roles: toRecord(scene.expression_roles),
		model_revisions: toRecord(scene.model_revisions)
	};
}

/** Converts serde-wasm-bindgen map values in 3D scenes to UI records. */
export function normalizeWasmScene3d(scene: WasmScene3d): RenderScene3d {
	return {
		...scene,
		expression_roles: toRecord(scene.expression_roles),
		model_revisions: toRecord(scene.model_revisions),
		coverage_revisions: scene.coverage_revisions
			? toRecord(scene.coverage_revisions)
			: undefined
	};
}

function toRecord<T>(value: Record<string, T> | Map<string, T>): Record<string, T> {
	return value instanceof Map ? Object.fromEntries(value) : value;
}