Something went wrong. Try again.
[READ-ONLY] Mirror of https://github.com/FoxxMD/multi-scrobbler. Scrobble plays from multiple sources to multiple clients docs.multi-scrobbler.app
Something went wrong. Try again.
6.7 kB · 183 lines
TypeScript
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183import { configDir } from '../index.ts';import * as path from 'path';import { childLogger, type Logger } from '@foxxmd/logging';import { loggerNoop } from '../MaybeLogger.ts';import { COMPACTABLE, type CompactableProperty, DEFAULT_RETENTION_COMPACT_AFTER, DEFAULT_RETENTION_DELETE_AFTER, type RetentionConfigValue, type RetentionOption, type RetentionValue, type RetentionValueUnparsed } from '../infrastructure/config/database.ts';import type {DurationValue} from '../infrastructure/Atomic.ts';import type {Duration} from 'dayjs/plugin/duration.js';import dayjs from 'dayjs';import { parseDurationFromDurationValue } from '../../utils/TimeUtils.ts';import assert from 'node:assert';import * as sqlite from 'node:sqlite';import { parseBoolStrict } from '../../utils.ts';import { SimpleError } from '../errors/MSErrors.ts';
export const MEMORY_DB_NAME = ':memory:';export const isMemoryDb = (name: string): boolean => name === MEMORY_DB_NAME;
export const getDbPath = (name: string = 'ms', workingDirectory?: string): string => { if (isMemoryDb(name)) { return MEMORY_DB_NAME; } return path.resolve(workingDirectory ?? configDir, `${name}.db`);}
export const getDbBackupPath = (dbPath: string, suffix?: string): string => { const pathInfo = path.parse(dbPath); const backupPath = `${path.join(pathInfo.dir, pathInfo.name)}.db${suffix !== undefined ? `.${suffix}` : ''}.bak`; return backupPath;}
export const backupDb = async (db: sqlite.DatabaseSync, dbPath: string, opts: { logger?: Logger, suffix?: string } = {}): Promise<void | Uint8Array> => {
const { logger: parentLogger = loggerNoop, suffix = dayjs().unix().toString(), } = opts;
const logger = childLogger(parentLogger, 'Migrations');
if(dbPath === MEMORY_DB_NAME) { // TODO serialize return; }
const backupPath = getDbBackupPath(dbPath, suffix); logger.info(`Backing up database to => ${backupPath}`); await sqlite.backup(db, backupPath, { progress: ({totalPages, remainingPages}) => { logger.debug(`Backup in progress => ${totalPages - remainingPages} / ${totalPages}`); } }) logger.info('Backed up!');}
const parseRetentionValue = (val: RetentionValueUnparsed): RetentionValue => { if(typeof val === 'string' || typeof val === 'boolean') { try { const boolVal = parseBoolStrict(val); assert(boolVal === false, 'retention value cannot be true'); return boolVal; } catch (e) { // swallow } } else if(dayjs.isDuration(val)) { return val; } else { return parseDurationFromDurationValue(val); } throw new SimpleError('retention value be of one: false, number, or string');}
const parseRetentionFromEnv = (type: string, defaultVal: number = DEFAULT_RETENTION_DELETE_AFTER): RetentionOption<RetentionValue> => { const deleteAfterEnv = process.env[`RETENTION_${type}_AFTER`] ?? defaultVal, deleteCompletedEnv = process.env[`RETENTION_${type}_COMPLETED_AFTER`] ?? deleteAfterEnv, deleteFailedEnv = process.env[`RETENTION_${type}_FAILED_AFTER`] ?? deleteAfterEnv, deleteDupedEnv = process.env[`RETENTION_${type}_DUPED_AFTER`] ?? deleteAfterEnv;
return { completed: parseRetentionValue(deleteCompletedEnv), failed: parseRetentionValue(deleteFailedEnv), duped: parseRetentionValue(deleteDupedEnv) }}
const isRetentionOptionDurations = (val: RetentionOption<RetentionValue>): val is RetentionOption<Duration> => { return dayjs.isDuration(val.completed) && dayjs.isDuration(val.duped) && dayjs.isDuration(val.failed);}
let retentionDeleteAfterFromEnv: RetentionOption<Duration>,retentionCompactAfterFromEnv: RetentionOption<RetentionValue>;
export const getRetentionDeleteAfterFromEnv = () => { if (retentionDeleteAfterFromEnv === undefined) { const deleteEnv = parseRetentionFromEnv('DELETE'); if(isRetentionOptionDurations(deleteEnv)) { retentionDeleteAfterFromEnv = deleteEnv; } else { throw new SimpleError('retention deleteAfter values from env must all be one of: number or string'); } } return retentionDeleteAfterFromEnv;}export const getRetentionCompactAfterFromEnv = () => { if (retentionCompactAfterFromEnv === undefined) { const compactEnv = parseRetentionFromEnv('COMPACT', DEFAULT_RETENTION_COMPACT_AFTER); retentionCompactAfterFromEnv = compactEnv; } return retentionCompactAfterFromEnv;}
export const parseRetentionOptions = (opts: RetentionConfigValue<DurationValue> = {}, defaults: RetentionOption<RetentionValue>): RetentionOption<RetentionValue> => { if (typeof opts === 'number' || typeof opts === 'string') { const dur = parseDurationFromDurationValue(opts); return { completed: dur, duped: dur, failed: dur } }
if (opts === undefined) { return defaults; }
if(dayjs.isDuration(opts)) { return { completed: opts, failed: opts, duped: opts } }
const { completed = defaults.completed, failed = defaults.failed, duped = defaults.duped } = opts;
return { completed: parseRetentionValue(completed), failed: parseRetentionValue(failed), duped: parseRetentionValue(duped), }}
export const parseRetentionOptionsDurations = (opts: RetentionConfigValue<Exclude<RetentionValueUnparsed, false>> = {}, defaults: RetentionOption<Duration>): RetentionOption<Duration> => { if (typeof opts === 'number' || typeof opts === 'string') { const dur = parseDurationFromDurationValue(opts); return { completed: dur, duped: dur, failed: dur } }
if(dayjs.isDuration(opts)) { return { completed: opts, failed: opts, duped: opts } }
if (opts === undefined) { return defaults; }
const { completed = defaults.completed, failed = defaults.failed, duped = defaults.duped } = opts;
return { completed: dayjs.isDuration(completed) ? completed : parseDurationFromDurationValue(completed), failed: dayjs.isDuration(failed) ? failed : parseDurationFromDurationValue(failed), duped: dayjs.isDuration(duped) ? duped : parseDurationFromDurationValue(duped), }}
export const isCompactableProperty = (val: string): val is CompactableProperty => val === COMPACTABLE.input || val === COMPACTABLE.transform;