lens/common/workspace-store.ts

import ElectronStore from "electron-store"
import { clusterStore } from "./cluster-store"

export interface WorkspaceData {
  id: string;
  name: string;
  description?: string;
}

export class Workspace implements WorkspaceData {
  public id: string
  public name: string
  public description?: string

  public constructor(data: WorkspaceData) {
    Object.assign(this, data)
  }
}

export class WorkspaceStore {
  public static defaultId = "default"
  private static instance: WorkspaceStore;
  public store: ElectronStore;

  private constructor() {
    this.store = new ElectronStore({
      name: "lens-workspace-store"
    })
  }

  public storeWorkspace(workspace: WorkspaceData) {
    const workspaces = this.getAllWorkspaces()
    const index = workspaces.findIndex((w) => w.id === workspace.id)
    if (index !== -1) {
      workspaces[index] = workspace
    } else {
      workspaces.push(workspace)
    }
    this.store.set("workspaces", workspaces)
  }

  public removeWorkspace(workspace: Workspace) {
    if (workspace.id === WorkspaceStore.defaultId) {
      throw new Error("Cannot remove default workspace")
    }
    const workspaces = this.getAllWorkspaces()
    const index = workspaces.findIndex((w) => w.id === workspace.id)
    if (index !== -1) {
      clusterStore.removeClustersByWorkspace(workspace.id)
      workspaces.splice(index, 1)
      this.store.set("workspaces", workspaces)
    }
  }

  public getAllWorkspaces(): Array<Workspace> {
    const workspacesData: WorkspaceData[]  = this.store.get("workspaces", [])

    return workspacesData.map((wsd) => new Workspace(wsd))
  }

  static getInstance(): WorkspaceStore {
    if (!WorkspaceStore.instance) {
      WorkspaceStore.instance = new WorkspaceStore()
    }
    return WorkspaceStore.instance
  }
}

const workspaceStore: WorkspaceStore = WorkspaceStore.getInstance()

if (!workspaceStore.getAllWorkspaces().find( ws => ws.id === WorkspaceStore.defaultId)) {
  workspaceStore.storeWorkspace({
    id: WorkspaceStore.defaultId,
    name: "default"
  })
}

export { workspaceStore }