import { bookingsDataset } from './datasets/bookings.dataset';
import { contractsDataset } from './datasets/contracts.dataset';
import { customersDataset } from './datasets/customers.dataset';
import { invoicesDataset } from './datasets/invoices.dataset';
import { locomotivesDataset } from './datasets/locomotives.dataset';
import { paymentsDataset } from './datasets/payments.dataset';
import { trainSchedulesDataset } from './datasets/train-schedules.dataset';
import { trainsDataset } from './datasets/trains.dataset';
import { wagonsDataset } from './datasets/wagons.dataset';
import { ExportDataset } from './export.types';

/**
 * Every exportable dataset.
 *
 * Adding one = a new file under `datasets/` + an entry here. No frontend edit,
 * no route, no permission seed — the dialog is driven entirely by the catalog
 * this registry serves, and a dataset reuses its module's existing `view` key.
 */
export const DATASETS: ExportDataset[] = [
  bookingsDataset,
  contractsDataset,
  customersDataset,
  invoicesDataset,
  paymentsDataset,
  trainSchedulesDataset,
  locomotivesDataset,
  trainsDataset,
  wagonsDataset,
];

const BY_KEY = new Map(DATASETS.map((d) => [d.key, d]));

export const getDataset = (key: string): ExportDataset | undefined => BY_KEY.get(key);