import {
	type SerializedDataTable,
	type SerializedDataTableColumn,
	serializedDataTableSchema,
} from '@n8n/api-types';
import { Service } from '@n8n/di';

import type { DataTableColumn } from '@/modules/data-table/data-table-column.entity';
import type { DataTable } from '@/modules/data-table/data-table.entity';

import { definePackageSerializationPayload } from '../package-serialization.types';

type DataTablePackageKeyHandling = {
	id: 'copy';
	createdAt: 'exclude';
	updatedAt: 'exclude';
	name: 'copy';
	columns: 'transform';
	project: 'transform';
	projectId: 'transform';
};

type DataTableColumnPackageKeyHandling = {
	id: 'copy';
	createdAt: 'exclude';
	updatedAt: 'exclude';
	dataTableId: 'exclude';
	name: 'copy';
	type: 'copy';
	index: 'copy';
	dataTable: 'exclude';
};

const serializeDataTablePayload = definePackageSerializationPayload<
	DataTable,
	SerializedDataTable,
	DataTablePackageKeyHandling
>();

const serializeColumnPayload = definePackageSerializationPayload<
	DataTableColumn,
	SerializedDataTableColumn,
	DataTableColumnPackageKeyHandling
>();

@Service()
export class DataTableSerializer {
	serialize(dataTable: DataTable): SerializedDataTable {
		const columns = [...dataTable.columns]
			.sort((a, b) => a.index - b.index)
			.map((column) =>
				serializeColumnPayload({
					id: column.id,
					name: column.name,
					type: column.type,
					index: column.index,
				}),
			);

		return serializedDataTableSchema.parse(
			serializeDataTablePayload({
				id: dataTable.id,
				name: dataTable.name,
				columns,
			}),
		);
	}
}