books/src/initialization.js

import config from '@/config';
import { ipcRenderer } from 'electron';
import { _ } from 'frappejs';
import SQLiteDatabase from 'frappejs/backends/sqlite';
import fs from 'fs';
import models from '../models';
import regionalModelUpdates from '../models/regionalModelUpdates';
import postStart from '../server/postStart';
import { DB_CONN_FAILURE, IPC_ACTIONS } from './messages';
import migrate from './migrate';

export async function createNewDatabase() {
  const options = {
    title: _('Select folder'),
    defaultPath: 'books.db',
  };

  let { canceled, filePath } = await ipcRenderer.invoke(
    IPC_ACTIONS.GET_SAVE_FILEPATH,
    options
  );

  if (canceled || filePath.length === 0) {
    return '';
  }

  if (!filePath.endsWith('.db')) {
    showMessageDialog({
      message: "Please select a filename ending with '.db'.",
    });
    return '';
  }

  if (fs.existsSync(filePath)) {
    fs.unlinkSync(filePath);
  }

  return filePath;
}

async function runRegionalModelUpdates() {
  if (!(await frappe.db.knex.schema.hasTable('SingleValue'))) {
    return;
  }

  const { country, setupComplete } = await frappe.db.getSingle(
    'AccountingSettings'
  );
  if (!parseInt(setupComplete)) return;
  await regionalModelUpdates({ country });
}

export async function connectToLocalDatabase(filePath) {
  if (!filePath) {
    return { connectionSuccess: false, reason: DB_CONN_FAILURE.INVALID_FILE };
  }

  frappe.login('Administrator');
  try {
    frappe.db = new SQLiteDatabase({
      dbPath: filePath,
    });
    await frappe.db.connect();
  } catch (error) {
    console.error(error);
    return { connectionSuccess: false, reason: DB_CONN_FAILURE.CANT_CONNECT };
  }

  try {
    await runRegionalModelUpdates();
  } catch (error) {
    console.error('regional model updates failed', error);
  }

  try {
    await migrate();
    await postStart();
  } catch (error) {
    if (!error.message.includes('SQLITE_CANTOPEN')) {
      throw error;
    }

    console.error(error);
    return { connectionSuccess: false, reason: DB_CONN_FAILURE.CANT_OPEN };
  }

  // set file info in config
  const { companyName } = frappe.AccountingSettings;
  let files = config.get('files') || [];
  if (
    !files.find(
      (file) => file.filePath === filePath && file.companyName === companyName
    )
  ) {
    files = [
      {
        companyName,
        filePath,
      },
      ...files.filter((file) => file.filePath !== filePath),
    ];
    config.set('files', files);
  }

  // set last selected file
  config.set('lastSelectedFilePath', filePath);
  return { connectionSuccess: true, reason: '' };
}

export function purgeCache(purgeAll = false) {
  const filterFunction = purgeAll
    ? () => true
    : (d) => frappe.docs[d][d] instanceof frappe.BaseMeta;

  Object.keys(frappe.docs)
    .filter(filterFunction)
    .forEach((d) => {
      frappe.removeFromCache(d, d);
      delete frappe[d];
    });

  if (purgeAll) {
    delete frappe.db;
    frappe.initializeAndRegister(models, true);
  }
}