xianren_studio/ui/src/api.ts

import { invoke } from "@tauri-apps/api/core";
import { listen } from "@tauri-apps/api/event";

export interface AppInfo {
  version: string;
  platform: string;
  models_dir: string;
  engine_bin: string | null;
  engine_exists: boolean;
  db_path: string;
}

export interface ModelInfo {
  id: string;
  repo_id: string;
  source: string;
  kind: string;
  file_name: string;
  file_path: string;
  file_size: number;
  quant: string | null;
  family: string | null;
  status: string;
  sha256: string | null;
  base_url: string | null;
  api_key: string | null;
  api_model: string | null;
  meta: Record<string, unknown>;
  created_at: string;
}

export interface RepoSummary {
  repo_id: string;
  author: string;
  name: string;
  downloads: number | null;
  likes: number | null;
  tags: string[];
}

export interface ModelFile {
  path: string;
  size: number;
  sha256: string | null;
}

export interface EngineStatus {
  running: boolean;
  port: number | null;
  pid: number | null;
  model: string | null;
  backend: string | null;
  ctx_size: number | null;
  ngl: number | null;
  uptime_secs: number | null;
}

export interface Conversation {
  id: string;
  title: string;
  model_id: string | null;
  system_prompt: string | null;
  created_at: string;
  updated_at: string;
}

export interface Message {
  id: string;
  conversation_id: string;
  role: string;
  content: string;
  tokens_in: number | null;
  tokens_out: number | null;
  created_at: string;
}

export interface ChatParams {
  temperature: number;
  top_p: number;
  max_tokens: number;
  ctx_size: number;
  ngl: number;
}

export interface ChatSendPayload {
  conversation_id: string;
  model_id: string;
  content: string;
  params: ChatParams;
}

export interface ChatTokenEvent {
  conversation_id: string;
  text: string;
}

export interface ChatDoneEvent {
  conversation_id: string;
  message_id: string;
  content: string;
  tokens_in: number | null;
  tokens_out: number | null;
  elapsed_ms: number;
}

export interface ChatErrorEvent {
  conversation_id: string;
  message: string;
}

export interface EngineDeployEvent {
  model_id: string;
  file_name: string;
  state: string;
  message: string | null;
}

export interface DownloadProgressEvent {
  id: string;
  url: string;
  file_name: string;
  downloaded: number;
  total: number | null;
  percent: number | null;
  speed_bps: number;
  status: string;
}

export interface ServerStatus {
  running: boolean;
  port: number | null;
}

export const api = {
  appInfo: () => invoke<AppInfo>("app_info"),
  listModels: () => invoke<ModelInfo[]>("list_models"),
  importModel: (path: string) => invoke<ModelInfo>("import_model", { path }),
  removeModel: (id: string) => invoke<void>("remove_model", { id }),
  scanModels: () => invoke<ModelInfo[]>("scan_models"),
  addRemoteModel: (name: string, baseUrl: string, apiKey: string, apiModel: string) =>
    invoke<ModelInfo>("add_remote_model", { name, baseUrl, apiKey, apiModel }),
  searchModels: (query: string, source: string) =>
    invoke<RepoSummary[]>("search_models", { query, source }),
  listModelFiles: (repoId: string, source: string) =>
    invoke<ModelFile[]>("list_model_files", { repoId, source }),
  settingsGet: () => invoke<Record<string, string>>("settings_get"),
  settingsSet: (key: string, value: string) =>
    invoke<void>("settings_set", { key, value }),
  listConversations: () => invoke<Conversation[]>("list_conversations"),
  createConversation: (title: string) =>
    invoke<Conversation>("create_conversation", { title }),
  deleteConversation: (id: string) =>
    invoke<void>("delete_conversation", { id }),
  getMessages: (conversationId: string) =>
    invoke<Message[]>("get_messages", { conversationId }),
  engineStart: (modelId: string, params: ChatParams) =>
    invoke<EngineStatus>("engine_start", { modelId, params }),
  deployModel: (modelId: string, params: ChatParams) =>
    invoke<void>("deploy_model", { modelId, params }),
  engineStop: () => invoke<void>("engine_stop"),
  engineStatus: () => invoke<EngineStatus>("engine_status"),
  chatSend: (payload: ChatSendPayload) => invoke<string>("chat_send", { payload }),
  downloadEnqueue: (
    url: string,
    fileName: string,
    sha256?: string,
    repoId?: string,
    source?: string,
  ) =>
    invoke<string>("download_enqueue", {
      url,
      fileName,
      sha256: sha256 ?? null,
      repoId: repoId ?? null,
      source: source ?? null,
    }),
  serverStart: (port: number, apiKey: string) =>
    invoke<ServerStatus>("server_start", { port, apiKey }),
  serverStop: () => invoke<void>("server_stop"),
  serverStatus: () => invoke<ServerStatus>("server_status"),
  openPath: (path: string) => invoke<void>("open_path", { path }),
};

export function onEvent<T>(channel: string, handler: (payload: T) => void) {
  return listen<T>(channel, (event) => handler(event.payload));
}