import fs from 'node:fs';
import os from 'node:os';
import path from 'node:path';

import { Database, open } from 'sqlite';
import * as sqlite3 from 'sqlite3';

import { getIndexFolderPath } from '../utils/IndexPathHelper';

export type DatabaseConnection = Database<sqlite3.Database>;

export function getIndexSqlitePath(): string {
	return path.join(getIndexFolderPath(), 'autodev-index.sqlite');
}

/**
 * todo: add Symbols
 * {@link Symbol} {@link ScopeGraph#symbols}
 */
export class SqliteDb {
	static db: DatabaseConnection | null = null;

	private static async createTables(db: DatabaseConnection) {
		await db.exec(
			`CREATE TABLE IF NOT EXISTS tag_catalog (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            dir STRING NOT NULL,
            branch STRING NOT NULL,
            artifactId STRING NOT NULL,
            path STRING NOT NULL,
            cacheKey STRING NOT NULL,
            lastUpdated INTEGER NOT NULL
        )`,
		);

		await db.exec(
			`CREATE TABLE IF NOT EXISTS global_cache (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            cacheKey STRING NOT NULL,
            dir STRING NOT NULL,
            branch STRING NOT NULL,
            artifactId STRING NOT NULL
        )`,
		);
	}

	private static indexSqlitePath = getIndexSqlitePath();

	static async get(): Promise<Database> {
		if (SqliteDb.db && fs.existsSync(SqliteDb.indexSqlitePath)) {
			return SqliteDb.db;
		}

		SqliteDb.indexSqlitePath = getIndexSqlitePath();
		SqliteDb.db = await open({
			filename: SqliteDb.indexSqlitePath,
			driver: sqlite3.Database,
		});

		await SqliteDb.createTables(SqliteDb.db);
		return SqliteDb.db;
	}

	static setUrl(url?: string) {
		if (url) {
			this.indexSqlitePath = url.replace('~', os.homedir());
		} else {
			this.indexSqlitePath = getIndexSqlitePath();
		}
	}
}