balena-supervisor/src/db.ts

import type { Knex } from 'knex';
import { knex } from 'knex';
import path from 'path';
import _ from 'lodash';

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

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

export type Transaction = Knex.Transaction;

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

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

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

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

	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 db.transaction(cb);
}