SessionsDb.ts

import * as Database from 'better-sqlite3';
import { Database as SqliteDatabase } from 'better-sqlite3';
import SessionsTable from '../models/SessionsTable';
import DetachedJsPathCallsTable from '../models/DetachedJsPathCallsTable';

interface IDbOptions {
  readonly?: boolean;
  fileMustExist?: boolean;
}

export default class SessionsDb {
  private static dbByBaseDir: { [dir: string]: SessionsDb } = {};
  public readonly sessions: SessionsTable;
  public readonly detachedJsPathCalls: DetachedJsPathCallsTable;
  public readonly readonly: boolean;
  private readonly baseDir: string;
  private db: SqliteDatabase;

  constructor(baseDir: string, dbOptions: IDbOptions = {}) {
    const { readonly = false, fileMustExist = false } = dbOptions;
    this.db = new Database(`${baseDir}/sessions.db`, { readonly, fileMustExist });
    this.baseDir = baseDir;
    this.readonly = readonly;
    this.sessions = new SessionsTable(this.db);
    this.detachedJsPathCalls = new DetachedJsPathCallsTable(this.db);
  }

  public findLatestSessionId(script: {
    sessionName: string;
    scriptInstanceId: string;
    scriptEntrypoint?: string;
  }) {
    const { sessionName, scriptEntrypoint, scriptInstanceId } = script;
    if (sessionName && scriptInstanceId) {
      // find default session if current not available
      const sessionRecord =
        this.sessions.findByName(sessionName, scriptInstanceId) ??
        this.sessions.findByName('default-session', scriptInstanceId);
      return sessionRecord?.id;
    }
    if (scriptEntrypoint) {
      const sessionRecords = this.sessions.findByScriptEntrypoint(scriptEntrypoint);
      if (!sessionRecords.length) return undefined;
      return sessionRecords[0].id;
    }
  }

  public findRelatedSessions(session: { scriptEntrypoint: string; scriptInstanceId: string }) {
    const otherSessions = this.sessions.findByScriptEntrypoint(session.scriptEntrypoint);
    const relatedScriptInstances: {
      id: string;
      startDate: number;
      defaultSessionId: string;
    }[] = [];
    const relatedSessions: { id: string; name: string }[] = [];
    const scriptDates = new Set<string>();
    for (const otherSession of otherSessions) {
      const key = `${otherSession.scriptInstanceId}_${otherSession.scriptStartDate}`;
      if (!scriptDates.has(key)) {
        relatedScriptInstances.push({
          id: otherSession.scriptInstanceId,
          startDate: new Date(otherSession.scriptStartDate).getTime(),
          defaultSessionId: otherSession.id,
        });
      }
      if (otherSession.scriptInstanceId === session.scriptInstanceId) {
        relatedSessions.unshift({ id: otherSession.id, name: otherSession.name });
      }
      scriptDates.add(key);
    }
    return {
      relatedSessions,
      relatedScriptInstances,
    };
  }

  public close() {
    if (this.db) {
      this.db.close();
    }
    this.db = null;
    delete SessionsDb.dbByBaseDir[this.baseDir];
  }

  public static shutdown() {
    for (const [key, db] of Object.entries(SessionsDb.dbByBaseDir)) {
      db.close();
      delete SessionsDb.dbByBaseDir[key];
    }
  }

  public static find(baseDir: string) {
    if (!this.dbByBaseDir[baseDir]) {
      this.dbByBaseDir[baseDir] = new SessionsDb(baseDir);
    }
    return this.dbByBaseDir[baseDir];
  }
}