ai-chatbot/services/ai-cad/index.ts

import type { Model3DTaskStatus } from "@/services/canvas";

type ApiResponse<T = any> = {
  success: boolean;
  data?: T;
  error?: string;
};

export type AiCadProviderStatus = {
  configured: boolean;
  source: "provider" | "env" | "none";
  providerName?: string | null;
  baseUrl?: string | null;
};

export type AiCadWorkspaceSummary = {
  id: string;
  name: string;
  generationProvider: "kittycad-cad" | "tripo-3d";
  versionCount: number;
  completedCount: number;
  hasPendingTasks: boolean;
  latestPrompt?: string | null;
  createdAt?: string | null;
  updatedAt?: string | null;
};

export type AiCadWorkspaceDocument = {
  id: string;
  name: string;
  prompt: string;
  generationProvider: "kittycad-cad" | "tripo-3d";
  modelVersion: string;
  viewerSettings: Record<string, unknown>;
  messages: Record<string, unknown>[];
  versions: Record<string, unknown>[];
  activeVersionId?: string | null;
  createdAt?: string | null;
  updatedAt?: string | null;
};

export const aiCadGenerationService = {
  async listWorkspaces(): Promise<ApiResponse<AiCadWorkspaceSummary[]>> {
    try {
      const response = await fetch("/api/ai-cad/workspaces");
      const data = await response.json();
      return data;
    } catch (_error) {
      return { success: false, error: "读取 AI-CAD 工作台列表失败" };
    }
  },

  async createWorkspace(params?: {
    name?: string;
  }): Promise<ApiResponse<AiCadWorkspaceSummary>> {
    try {
      const response = await fetch("/api/ai-cad/workspaces", {
        method: "POST",
        headers: { "Content-Type": "application/json" },
        body: JSON.stringify(params || {}),
      });
      const data = await response.json();
      return data;
    } catch (_error) {
      return { success: false, error: "创建 AI-CAD 工作台失败" };
    }
  },

  async getWorkspace(
    workspaceId: string
  ): Promise<ApiResponse<AiCadWorkspaceDocument | null>> {
    try {
      const response = await fetch(`/api/ai-cad/workspaces/${workspaceId}`);
      const data = await response.json();
      return data;
    } catch (_error) {
      return { success: false, error: "读取 AI-CAD 工作台失败" };
    }
  },

  async saveWorkspace(
    workspaceId: string,
    params: {
      name?: string;
      prompt: string;
      generationProvider: "kittycad-cad" | "tripo-3d";
      modelVersion: string;
      viewerSettings: Record<string, unknown>;
      messages: Record<string, unknown>[];
      versions: Record<string, unknown>[];
      activeVersionId?: string | null;
    }
  ): Promise<
    ApiResponse<{
      id: string;
      name: string;
      updatedAt?: string | null;
      summary?: AiCadWorkspaceSummary;
    }>
  > {
    try {
      const response = await fetch(`/api/ai-cad/workspaces/${workspaceId}`, {
        method: "PUT",
        headers: { "Content-Type": "application/json" },
        body: JSON.stringify(params),
      });
      const data = await response.json();
      return data;
    } catch (_error) {
      return { success: false, error: "保存 AI-CAD 工作台失败" };
    }
  },

  async deleteWorkspace(
    workspaceId: string
  ): Promise<ApiResponse<{ id: string }>> {
    try {
      const response = await fetch(`/api/ai-cad/workspaces/${workspaceId}`, {
        method: "DELETE",
      });
      const data = await response.json();
      return data;
    } catch (_error) {
      return { success: false, error: "删除 AI-CAD 工作台失败" };
    }
  },

  async getProviderStatus(): Promise<ApiResponse<AiCadProviderStatus>> {
    try {
      const response = await fetch("/api/ai-cad/provider");
      const data = await response.json();
      return data;
    } catch (_error) {
      return { success: false, error: "读取 AI-CAD 提供商状态失败" };
    }
  },

  async createTask(params: {
    prompt: string;
    outputFormat?: string;
    projectName?: string;
    iterationSourceTaskId?: string;
  }): Promise<
    ApiResponse<{
      taskId: string;
      status: Model3DTaskStatus;
      conversationId?: string | null;
      modelVersion?: string | null;
      outputFormat?: string | null;
      generationMode?: "create" | "iterate";
      sourceTaskId?: string | null;
      raw?: any;
    }>
  > {
    try {
      const response = await fetch("/api/ai-cad/tasks", {
        method: "POST",
        headers: { "Content-Type": "application/json" },
        body: JSON.stringify(params),
      });
      const data = await response.json();
      return data;
    } catch (_error) {
      return { success: false, error: "创建 AI-CAD 任务失败" };
    }
  },

  async getTask(taskId: string): Promise<
    ApiResponse<{
      taskId: string;
      type: string;
      status: Model3DTaskStatus;
      progress?: number | null;
      output?: Record<string, any>;
      modelUrl?: string | null;
      previewImageUrl?: string | null;
      errorMessage?: string | null;
      conversationId?: string | null;
      modelVersion?: string | null;
      outputFormat?: string | null;
      codeUrl?: string | null;
      downloadUrls?: Record<string, string>;
    }>
  > {
    try {
      const response = await fetch(`/api/ai-cad/tasks/${taskId}`);
      const data = await response.json();
      return data;
    } catch (_error) {
      return { success: false, error: "查询 AI-CAD 任务失败" };
    }
  },
};