db-module.ts

import { window } from "vscode";
import { App } from "../common/app";
import { extLogger } from "../common";
import { DisposableObject } from "../pure/disposable-object";
import { DbConfigStore } from "./config/db-config-store";
import { DbManager } from "./db-manager";
import { DbPanel } from "./ui/db-panel";
import { DbSelectionDecorationProvider } from "./ui/db-selection-decoration-provider";
import { DatabasePanelCommands } from "../common/commands";

export class DbModule extends DisposableObject {
  public readonly dbManager: DbManager;
  private readonly dbConfigStore: DbConfigStore;
  private dbPanel: DbPanel | undefined;

  private constructor(app: App) {
    super();

    this.dbConfigStore = new DbConfigStore(app);
    this.dbManager = new DbManager(app, this.dbConfigStore);
  }

  public static async initialize(app: App): Promise<DbModule> {
    const dbModule = new DbModule(app);
    app.subscriptions.push(dbModule);

    await dbModule.initialize(app);
    return dbModule;
  }

  public getCommands(): DatabasePanelCommands {
    if (!this.dbPanel) {
      throw new Error("Database panel not initialized");
    }

    return {
      ...this.dbPanel.getCommands(),
    };
  }

  private async initialize(app: App): Promise<void> {
    void extLogger.log("Initializing database module");

    await this.dbConfigStore.initialize();

    this.dbPanel = new DbPanel(this.dbManager, app.credentials);

    this.push(this.dbPanel);
    this.push(this.dbConfigStore);

    const dbSelectionDecorationProvider = new DbSelectionDecorationProvider();

    window.registerFileDecorationProvider(dbSelectionDecorationProvider);
  }
}