/**
 * Actualiza imports de contextos que ya no existen
 */

import { readFileSync, writeFileSync, readdirSync } from 'node:fs';
import { join } from 'node:path';
import { fileURLToPath } from 'node:url';
import { dirname } from 'node:path';

const __filename = fileURLToPath(import.meta.url);
const __dirname = dirname(__filename);
const terraDir = join(__dirname, '../src');

// Mapeo de nombres de contextos antiguos a nuevos patrones
const contextMap: Record<string, { class: string; newPattern: string }> = {
	AccordionRootContext: { class: 'AccordionRoot', newPattern: 'AccordionRoot.ctx' },
	AccordionItemContext: { class: 'AccordionItemState', newPattern: 'AccordionItemState.ctx' },
	AvatarRootContext: { class: 'AvatarRootState', newPattern: 'AvatarRootState.ctx' },
	CalendarCellContext: { class: 'CalendarCellState', newPattern: 'CalendarCellState.ctx' },
	CalendarRootContext: { class: 'CalendarRootState', newPattern: 'CalendarRootState.ctx' },
	CheckboxGroupContext: { class: 'CheckboxGroupState', newPattern: 'CheckboxGroupState.ctx' },
	CheckboxRootContext: { class: 'CheckboxRootState', newPattern: 'CheckboxRootState.ctx' },
	CollapsibleRootContext: { class: 'CollapsibleRootState', newPattern: 'CollapsibleRootState.ctx' },
	CommandGroupContainerContext: {
		class: 'CommandGroupContainerState',
		newPattern: 'CommandGroupContainerState.ctx'
	},
	CommandListContext: { class: 'CommandListState', newPattern: 'CommandListState.ctx' },
	CommandRootContext: { class: 'CommandRootState', newPattern: 'CommandRootState.ctx' },
	DateFieldRootContext: { class: 'DateFieldRootState', newPattern: 'DateFieldRootState.ctx' },
	DatePickerRootContext: { class: 'DatePickerRootState', newPattern: 'DatePickerRootState.ctx' },
	DateRangeFieldRootContext: {
		class: 'DateRangeFieldRootState',
		newPattern: 'DateRangeFieldRootState.ctx'
	},
	DateRangePickerRootContext: {
		class: 'DateRangePickerRootState',
		newPattern: 'DateRangePickerRootState.ctx'
	},
	DialogRootContext: { class: 'DialogRootState', newPattern: 'DialogRootState.ctx' },
	EditableRootContext: { class: 'EditableRootState', newPattern: 'EditableRootState.ctx' },
	FieldRootContext: { class: 'FieldRootState', newPattern: 'FieldRootState.ctx' },
	FileUploadItemContext: { class: 'FileUploadItemState', newPattern: 'FileUploadItemState.ctx' },
	FileUploadRootContext: { class: 'FileUploadRootState', newPattern: 'FileUploadRootState.ctx' },
	FormFieldContext: { class: 'FormFieldState', newPattern: 'FormFieldState.ctx' },
	FormRootContext: { class: 'FormRootState', newPattern: 'FormRootState.ctx' },
	LinkPreviewRootContext: { class: 'LinkPreviewRootState', newPattern: 'LinkPreviewRootState.ctx' },
	MenuCheckboxGroupContext: {
		class: 'MenuCheckboxGroupState',
		newPattern: 'MenuCheckboxGroupState.ctx'
	},
	MenuContentContext: { class: 'MenuContentState', newPattern: 'MenuContentState.ctx' },
	MenuMenuContext: { class: 'MenuMenuState', newPattern: 'MenuMenuState.ctx' },
	MenuRadioGroupContext: { class: 'MenuRadioGroupState', newPattern: 'MenuRadioGroupState.ctx' },
	MenuRootContext: { class: 'MenuRootState', newPattern: 'MenuRootState.ctx' },
	MenubarMenuContext: { class: 'MenubarMenuState', newPattern: 'MenubarMenuState.ctx' },
	MenubarRootContext: { class: 'MenubarRootState', newPattern: 'MenubarRootState.ctx' },
	NumberFieldRootContext: { class: 'NumberFieldRootState', newPattern: 'NumberFieldRootState.ctx' },
	PaginationRootContext: { class: 'PaginationRootState', newPattern: 'PaginationRootState.ctx' },
	PopoverRootContext: { class: 'PopoverRootState', newPattern: 'PopoverRootState.ctx' },
	RadioGroupRootContext: { class: 'RadioGroupRootState', newPattern: 'RadioGroupRootState.ctx' },
	RangeCalendarCellContext: {
		class: 'RangeCalendarCellState',
		newPattern: 'RangeCalendarCellState.ctx'
	},
	RatingGroupRootContext: { class: 'RatingGroupRootState', newPattern: 'RatingGroupRootState.ctx' },
	ScrollAreaScrollbarAxisContext: { class: 'ScrollbarAxis', newPattern: 'ScrollbarAxis.ctx' },
	ScrollAreaScrollbarSharedContext: {
		class: 'ScrollAreaScrollbarSharedState',
		newPattern: 'ScrollAreaScrollbarSharedState.ctx'
	},
	ScrollAreaScrollbarVisibleContext: {
		class: 'ScrollAreaScrollbarVisibleState',
		newPattern: 'ScrollAreaScrollbarVisibleState.ctx'
	},
	ScrollAreaScrollbarContext: {
		class: 'ScrollAreaScrollbarState',
		newPattern: 'ScrollAreaScrollbarState.ctx'
	},
	ScrollAreaRootContext: { class: 'ScrollAreaRootState', newPattern: 'ScrollAreaRootState.ctx' },
	SelectContentContext: { class: 'SelectContentState', newPattern: 'SelectContentState.ctx' },
	SelectGroupContext: { class: 'SelectGroupState', newPattern: 'SelectGroupState.ctx' },
	SelectRootContext: { class: 'SelectRoot', newPattern: 'SelectRoot.ctx' },
	SliderRootContext: { class: 'SliderRoot', newPattern: 'SliderRoot.ctx' },
	SpinnerRootContext: { class: 'SpinnerRootState', newPattern: 'SpinnerRootState.ctx' },
	SplitterRootContext: { class: 'SplitterRootState', newPattern: 'SplitterRootState.ctx' },
	StepperItemContext: { class: 'StepperItemState', newPattern: 'StepperItemState.ctx' },
	StepperRootContext: { class: 'StepperRootState', newPattern: 'StepperRootState.ctx' },
	SwitchRootContext: { class: 'SwitchRootState', newPattern: 'SwitchRootState.ctx' },
	TabsRootContext: { class: 'TabsRootState', newPattern: 'TabsRootState.ctx' },
	TagsInputRootContext: { class: 'TagsInputRootState', newPattern: 'TagsInputRootState.ctx' },
	TagsInputTagContext: { class: 'TagsInputTagState', newPattern: 'TagsInputTagState.ctx' },
	TimeFieldRootContext: { class: 'TimeFieldRootState', newPattern: 'TimeFieldRootState.ctx' },
	TimeRangeFieldRootContext: {
		class: 'TimeRangeFieldRootState',
		newPattern: 'TimeRangeFieldRootState.ctx'
	},
	ToastRootContext: { class: 'ToastRootState', newPattern: 'ToastRootState.ctx' },
	ToggleGroupRootContext: { class: 'ToggleGroupRootState', newPattern: 'ToggleGroupRootState.ctx' },
	ToolbarGroupContext: { class: 'ToolbarGroupState', newPattern: 'ToolbarGroupState.ctx' },
	ToolbarRootContext: { class: 'ToolbarRootState', newPattern: 'ToolbarRootState.ctx' },
	TooltipProviderContext: { class: 'TooltipProviderState', newPattern: 'TooltipProviderState.ctx' },
	TooltipRootContext: { class: 'TooltipRootState', newPattern: 'TooltipRootState.ctx' },
	FloatingTooltipRootContext: { class: 'FloatingRootState', newPattern: 'FloatingRootState.ctx' },
	FloatingContentContext: { class: 'FloatingContentState', newPattern: 'FloatingContentState.ctx' }
};

function findAllFiles(dir: string): string[] {
	const files: string[] = [];
	const entries = readdirSync(dir, { withFileTypes: true });
	for (const entry of entries) {
		const fullPath = join(dir, entry.name);
		if (entry.isDirectory() && !entry.name.startsWith('.') && entry.name !== 'node_modules') {
			files.push(...findAllFiles(fullPath));
		} else if (entry.isFile() && (entry.name.endsWith('.svelte') || entry.name.endsWith('.ts'))) {
			files.push(fullPath);
		}
	}
	return files;
}

function fixFile(filePath: string): boolean {
	const content = readFileSync(filePath, 'utf-8');
	let modified = content;

	// Para cada contexto en el mapa
	for (const [oldContext, info] of Object.entries(contextMap)) {
		// 1. Eliminar el contexto de los imports si no se usa elsewhere en el archivo
		const importRegex = new RegExp(`,?\\s*${oldContext}\\s*,?\\s*`, 'g');
		const hasOldImport = new RegExp(`\\b${oldContext}\\b`).test(content);

		if (!hasOldImport) continue;

		// 2. Reemplazar referencias al contexto con el nuevo patrón
		modified = modified.replace(
			new RegExp(`\\b${oldContext}\\.set\\(`, 'g'),
			`${info.newPattern}.set(`
		);
		modified = modified.replace(
			new RegExp(`\\b${oldContext}\\.get\\(`, 'g'),
			`${info.newPattern}.get(`
		);
		modified = modified.replace(
			new RegExp(`\\b${oldContext}\\.getOr\\(`, 'g'),
			`${info.newPattern}.getOr(`
		);
	}

	if (modified !== content) {
		writeFileSync(filePath, modified, 'utf-8');
		return true;
	}
	return false;
}

const files = findAllFiles(terraDir);
let modified = 0;

console.log('🔧 Actualizando referencias a contextos...\n');

for (const file of files) {
	if (fixFile(file)) {
		modified++;
		const relPath = file.replace(/.*src\\/, '').replace(/.*src\//, '');
		console.log(`  ✅ ${relPath}`);
	}
}

console.log(`\n📊 Resumen:`);
console.log(`   Modificados: ${modified}`);
console.log(`   Total procesados: ${files.length}`);