singlestoreImports.ts — drizzle-orm Source File

Source Code

import { is } from 'drizzle-orm';
import { AnySingleStoreTable, SingleStoreTable } from 'drizzle-orm/singlestore-core';
import { safeRegister } from '../cli/commands/utils';

export const prepareFromExports = (exports: Record<string, unknown>) => {
	const tables: AnySingleStoreTable[] = [];
	/* const views: SingleStoreView[] = []; */

	const i0values = Object.values(exports);
	i0values.forEach((t) => {
		if (is(t, SingleStoreTable)) {
			tables.push(t);
		}

		/* if (is(t, SingleStoreView)) {
			views.push(t);
		} */
	});

	return { tables /* views  */ };
};

export const prepareFromSingleStoreImports = async (imports: string[]) => {
	const tables: AnySingleStoreTable[] = [];
	/* const views: SingleStoreView[] = []; */

	const { unregister } = await safeRegister();
	for (let i = 0; i < imports.length; i++) {
		const it = imports[i];
		const i0: Record<string, unknown> = require(`${it}`);
		const prepared = prepareFromExports(i0);

		tables.push(...prepared.tables);
		/* views.push(...prepared.views); */
	}
	unregister();
	return { tables: Array.from(new Set(tables)) /* , views */ };
};