claudecodeui/server/src/shared/database/repositories/workspace-original-paths.db.ts

import { randomUUID } from 'node:crypto';

import { getConnection } from '@/shared/database/connection.js';
import type { WorkspaceOriginalPathRow } from '@/shared/database/types.js';

export const workspaceOriginalPathsDb = {
    createWorkspacePath(workspacePath: string, customWorkspaceName: string | null = null): void {
        const db = getConnection();
        db.prepare(`
            INSERT INTO workspace_original_paths (workspace_id, workspace_path, custom_workspace_name)
            VALUES (?, ?, ?)
            ON CONFLICT(workspace_path) DO UPDATE SET
              custom_workspace_name = CASE
                WHEN workspace_original_paths.custom_workspace_name IS NULL OR workspace_original_paths.custom_workspace_name = ''
                THEN excluded.custom_workspace_name
                ELSE workspace_original_paths.custom_workspace_name
              END
        `).run(randomUUID(), workspacePath, customWorkspaceName);
    },

    getWorkspacePath(workspacePath: string): WorkspaceOriginalPathRow | null {
        const db = getConnection();
        const row = db.prepare(`
            SELECT workspace_id, workspace_path, custom_workspace_name, isStarred
            FROM workspace_original_paths
            WHERE workspace_path = ?
        `).get(workspacePath) as WorkspaceOriginalPathRow | undefined;

        return row ?? null;
    },

    getWorkspaceById(workspaceId: string): WorkspaceOriginalPathRow | null {
        const db = getConnection();
        const row = db.prepare(`
            SELECT workspace_id, workspace_path, custom_workspace_name, isStarred
            FROM workspace_original_paths
            WHERE workspace_id = ?
        `).get(workspaceId) as WorkspaceOriginalPathRow | undefined;

        return row ?? null;
    },

    getWorkspacePaths(): WorkspaceOriginalPathRow[] {
        const db = getConnection();
        return db.prepare(`
            SELECT workspace_id, workspace_path, custom_workspace_name, isStarred
            FROM workspace_original_paths
        `).all() as WorkspaceOriginalPathRow[];
    },

    getCustomWorkspaceName(workspacePath: string): string | null {
        const db = getConnection();
        const row = db.prepare(`
            SELECT custom_workspace_name
            FROM workspace_original_paths
            WHERE workspace_path = ?
        `).get(workspacePath) as Pick<WorkspaceOriginalPathRow, 'custom_workspace_name'> | undefined;

        return row?.custom_workspace_name ?? null;
    },

    updateCustomWorkspaceName(workspacePath: string, customWorkspaceName: string | null): void {
        const db = getConnection();
        db.prepare(`
            INSERT INTO workspace_original_paths (workspace_id, workspace_path, custom_workspace_name)
            VALUES (?, ?, ?)
            ON CONFLICT(workspace_path) DO UPDATE SET custom_workspace_name = excluded.custom_workspace_name
        `).run(randomUUID(), workspacePath, customWorkspaceName);
    },

    updateCustomWorkspaceNameById(workspaceId: string, customWorkspaceName: string | null): void {
        const db = getConnection();
        db.prepare(`
            UPDATE workspace_original_paths
            SET custom_workspace_name = ?
            WHERE workspace_id = ?
        `).run(customWorkspaceName, workspaceId);
    },

    updateWorkspaceIsStarred(workspacePath: string, isStarred: boolean): void {
        const db = getConnection();
        db.prepare(`
            UPDATE workspace_original_paths
            SET isStarred = ?
            WHERE workspace_path = ?
        `).run(isStarred ? 1 : 0, workspacePath);
    },

    updateWorkspaceIsStarredById(workspaceId: string, isStarred: boolean): void {
        const db = getConnection();
        db.prepare(`
            UPDATE workspace_original_paths
            SET isStarred = ?
            WHERE workspace_id = ?
        `).run(isStarred ? 1 : 0, workspaceId);
    },

    deleteWorkspacePath(workspacePath: string): void {
        const db = getConnection();
        db.prepare(`
            DELETE FROM workspace_original_paths
            WHERE workspace_path = ?
        `).run(workspacePath);
    },

    deleteWorkspaceById(workspaceId: string): void {
        const db = getConnection();
        db.prepare(`
            DELETE FROM workspace_original_paths
            WHERE workspace_id = ?
        `).run(workspaceId);
    },
};