@rocksky/cli/src/drizzle.ts

Command-line interface for Rocksky – scrobble tracks, view stats, and manage your listening history

import { drizzle } from "drizzle-orm/better-sqlite3";
import { migrate } from "drizzle-orm/better-sqlite3/migrator";
import Database from "better-sqlite3";
import envpaths from "env-paths";
import fs from "node:fs";
import path from "node:path";
import { fileURLToPath } from "node:url";

const __filename = fileURLToPath(import.meta.url);
const __dirname = path.dirname(__filename);

fs.mkdirSync(envpaths("rocksky", { suffix: "" }).data, { recursive: true });
const url = `${envpaths("rocksky", { suffix: "" }).data}/rocksky.sqlite`;

const sqlite = new Database(url);
const db = drizzle(sqlite);

let initialized = false;

/**
 * Initialize the database and run migrations
 * This should be called before any database operations
 */
export async function initializeDatabase() {
  if (initialized) {
    return;
  }

  try {
    // In production (built), migrations are in ../drizzle
    // In development (src), migrations are in ../../drizzle
    let migrationsFolder = path.join(__dirname, "../drizzle");

    if (!fs.existsSync(migrationsFolder)) {
      // Try development path
      migrationsFolder = path.join(__dirname, "../../drizzle");
    }

    if (fs.existsSync(migrationsFolder)) {
      migrate(db, { migrationsFolder });
      initialized = true;
    } else {
      // No migrations folder found - this might be the first run
      // or migrations haven't been generated yet
      initialized = true;
    }
  } catch (error) {
    console.error("Failed to run migrations:", error);
    throw error;
  }
}

export default { db, initializeDatabase };