balena-supervisor/src/db.ts

import * as Knex from 'knex';
import * as path from 'path';

import * as constants from './lib/constants';

type DBTransactionCallback = (trx: Knex.Transaction) => void;

export type Transaction = Knex.Transaction;

const databasePath = constants.databasePath;
const knex = Knex({
	client: 'sqlite3',
	connection: {
		filename: databasePath,
	},
	useNullAsDefault: true,
});

export const initialized = (async () => {
	try {
		await knex('knex_migrations_lock').update({ is_locked: 0 });
	} catch {
		/* ignore */
	}
	return knex.migrate.latest({
		directory: path.join(__dirname, 'migrations'),
	});
})();

export function models(modelName: string): Knex.QueryBuilder {
	return knex(modelName);
}

export async function upsertModel(
	modelName: string,
	obj: any,
	id: Dictionary<unknown>,
	trx?: Knex.Transaction,
): Promise<any> {
	const k = trx || knex;

	const n = await k(modelName).update(obj).where(id);
	if (n === 0) {
		return k(modelName).insert(obj);
	}
}

export function transaction(
	cb: DBTransactionCallback,
): Promise<Knex.Transaction> {
	return knex.transaction(cb);
}