@wuchuheng/electron/template/src/main/database/data-source.ts

Electron project template

import { DataSource } from 'typeorm';
import packageJson from '../../../package.json';
import { logger } from '../utils/logger';
import { Welcome } from './entities/welcom';
import { seedDatabase } from './seed';
const databaseName = `${packageJson.name}-${packageJson.version}.sqlite`;

let db: DataSource;

let isInitialized = false;
export const getIsInitialized = () => isInitialized;

export const initDB = async (): Promise<void> => {
  const isDev = process.env.NODE_ENV === 'development';

  try {
    logger.info('Initializing database connection');

    db = new DataSource({
      type: 'better-sqlite3',
      database: isDev ? 'dev.sqlite' : databaseName,
      entities: [Welcome],
      subscribers: [],
      synchronize: true,
      logging: isDev,
    });

    await db.initialize();

    isInitialized = true;
    logger.info(`Database connection established to ${isDev ? 'dev.sqlite' : databaseName}`);

    // Seed the database after initialization
    await seedDatabase();
  } catch (error) {
    logger.error(`Database initialization error: ${error instanceof Error ? error.message : String(error)}`);
    throw error;
  }
};

export const getDataSource = () => {
  if (!db || !isInitialized) {
    throw new Error('Database not initialized. Call initialize() first.');
  }
  return db;
};