orange-orm/src/client/map.js

Object Relational Mapper

const _newTable = require('../table');
const createProviders = require('./createProviders');

function mapRoot(index, fn) {

	const providers = createProviders(index);

	return map(index, context, providers, fn);

	function context(arg) {

		const tables = {};
		for (let name in context) {
			if (context[name] && context[name]._dbName)
				tables[name] = context[name];
		}

		if (arg && arg.db && typeof arg.db === 'function') {
			return index({
				...arg,
				db: providers(arg.db),
				tables, providers
			});
		}
		else
			return index({
				...arg, tables, providers
			});
	}
}


function map(index, context, providers, fn) {
	let next = fn({ table: newTable, ...context });

	for (let name in next) {
		if (next[name] && next[name]._dbName) {
			context[name] = next[name];
			context[name].map = mapTable.bind(null, context[name]);
		}
	}
	context.map = map.bind(null, index, context, providers);
	context.pg = connect.bind(null, 'pg');
	context.pglite = connect.bind(null, 'pglite');
	context.postgres = connect.bind(null, 'pg');
	context.mssql = connect.bind(null, 'mssql');
	context.mssqlNative = connect.bind(null, 'mssqlNative');
	context.mysql = connect.bind(null, 'mysql');
	context.sap = connect.bind(null, 'sap');
	context.oracle = connect.bind(null, 'oracle');
	context.sqlite = connect.bind(null, 'sqlite');
	context.d1 = connect.bind(null, 'd1');
	context.http = function(url) {
		return index({ db: url, providers});
	};

	function connect(name, ...args) {
		const provider = index[name];
		const pool = provider.apply(null, args);

		const tables = {};
		for (let name in context) {
			if (context[name] && context[name]._dbName)
				tables[name] = context[name];
		}

		return index({ db: pool, tables, providers });
	}

	return context;

	function newTable() {
		let table = _newTable.apply(null, arguments);
		table.map = mapTable.bind(null, table);
		return table;
	}
}

function mapTable(table, fn) {
	let next = fn({ column: table.column, primaryColumn: table.primaryColumn, references: table.join, hasMany, hasOne });
	for (let name in next) {
		if (next[name].as)
			next[name] = next[name].as(name);
	}

	function hasMany(to) {
		if (!to)
			throw new Error('Missing \'to\' table');
		return { by };

		function by() {
			const join = to.join(table).by.apply(null, arguments);
			return table.hasMany(join);
		}
	}

	function hasOne(to) {
		if (!to)
			throw new Error('Missing \'to\' table');
		return { by };

		function by() {
			const join = to.join(table).by.apply(null, arguments);
			return table.hasOne(join);
		}
	}

	return table;

}

module.exports = mapRoot;