tauri.ts

src/ipc/tauri.ts

1. import { Channel, invoke } from '@tauri-apps/api/core'
2. import { listen } from '@tauri-apps/api/event'
3. import { openUrl } from '@tauri-apps/plugin-opener'
5. import type { IpcClient } from './client'
6. import type { StreamEvent } from './gen/StreamEvent'
8. // Tauri turns camelCase argument names into the Rust commands' snake_case.
9. export const tauriClient: IpcClient = {
10. appInfo: () => invoke('app_info'),
11. windowReady: () => invoke('window_ready'),
12. showMain: () => invoke('show_main'),
14. listProviders: () => invoke('list_providers'),
15. saveProvider: (provider) => invoke('save_provider', { provider }),
16. deleteProvider: (id) => invoke('delete_provider', { id }),
17. setApiKey: (providerId, key) => invoke('set_api_key', { providerId, key }),
18. clearApiKey: (providerId) => invoke('clear_api_key', { providerId }),
19. hasApiKey: (providerId) => invoke('has_api_key', { providerId }),
21. listModels: (providerId) => invoke('list_models', { providerId }),
22. refreshModels: (providerId) => invoke('refresh_models', { providerId }),
24. listConversations: (includeArchived) => invoke('list_conversations', { includeArchived }),
25. getConversation: (id) => invoke('get_conversation', { id }),
26. createConversation: (providerId, modelId, presetId, origin) =>
27. invoke('create_conversation', { providerId, modelId, presetId, origin }),
28. renameConversation: (id, title) => invoke('rename_conversation', { id, title }),
29. setPinned: (id, pinned) => invoke('set_pinned', { id, pinned }),
30. setArchived: (id, archived) => invoke('set_archived', { id, archived }),
31. deleteConversation: (id) => invoke('delete_conversation', { id }),
32. setConversationModel: (id, providerId, modelId) => invoke('set_conversation_model', { id, providerId, modelId }),
33. setSystemPrompt: (id, systemPrompt, presetId) => invoke('set_system_prompt', { id, systemPrompt, presetId }),
35. sendMessage: (req, onEvent) => {
36. const channel = new Channel<StreamEvent>()
37. channel.onmessage = onEvent
38. return invoke('send_message', { req, onEvent: channel })
39. },
40. cancelStream: (streamId) => invoke('cancel_stream', { streamId }),
41. truncateFrom: (conversationId, messageId) => invoke('truncate_from', { conversationId, messageId }),
42. estimatePrompt: (a) =>
43. invoke('estimate_prompt', {
44. conversationId: a.conversationId,
45. providerId: a.providerId,
46. modelId: a.modelId,
47. text: a.text,
48. imageCount: a.imageCount,
49. extraText: a.extraText ?? null,
50. }),
52. search: (query) => invoke('search', { query }),
53. spendSummary: () => invoke('spend_summary'),
54. usageReport: (days) => invoke('usage_report', { days }),
56. listPresets: () => invoke('list_presets'),
57. savePreset: (preset) => invoke('save_preset', { preset }),
58. deletePreset: (id) => invoke('delete_preset', { id }),
60. getSettings: () => invoke('get_settings'),
61. saveSettings: (settings) => invoke('save_settings', { settings }),
63. pickAttachments: (imagesAllowed) => invoke('pick_attachments', { imagesAllowed }),
64. // Raw bytes in the body, the name in a header: no JSON number arrays.
65. attachBytes: (filename, bytes) =>
66. invoke('attach_bytes', bytes, { headers: { 'x-filename': encodeURIComponent(filename) } }),
67. attachmentBytes: (id) => invoke('attachment_bytes', { id }),
68. discardAttachment: (id) => invoke('discard_attachment', { id }),
70. exportConversation: (id) => invoke('export_conversation', { id }),
71. chooseVaultFolder: () => invoke('choose_vault_folder'),
73. quickOpenInMain: (conversationId) => invoke('quick_open_in_main', { conversationId }),
74. hideQuick: () => invoke('hide_quick'),
75. resizeQuick: (height) => invoke('resize_quick', { height }),
77. openUrl: (url) => openUrl(url),
78. on: async (event, handler) => {
79. const un = await listen(event, (e) => handler(e.payload))
80. return un
81. },
82. }