feat: add error notification (#386)

* feat: add error notification

* feat: add error collection

* chore: error display

* chore: error string

* docs: update notes

* docs: update notes

* build: build error

* chore: errors 5 and link length truncation

Author: RainyNight9

docs/content.en/docs/release-notes/_index.md

@@ -23,6 +23,7 @@ Information about release notes of Coco Server is provided here.
 - feat: networked search data sources support search and keyboard-only operation #367
 - feat: add application management to the plugin #374
 - feat: add keyboard-only operation to history list #385
+- feat: add error notification #386
 
 ### Bug fix
 

src-tauri/src/server/http_client.rs

@@ -38,14 +38,10 @@ impl HttpClient {
         let request_builder =
             Self::get_request_builder(method, url, headers, query_params, body).await;
 
-
-        let response = request_builder
-            .send()
-            .await
-            .map_err(|e| {
-                dbg!("Failed to send request: {}", &e);
-                format!("Failed to send request: {}", e)
-            })?;
+        let response = request_builder.send().await.map_err(|e| {
+            dbg!("Failed to send request: {}", &e);
+            format!("Failed to send request: {}", e)
+        })?;
         Ok(response)
     }
 
@@ -58,14 +54,12 @@ impl HttpClient {
     ) -> RequestBuilder {
         let client = HTTP_CLIENT.lock().await; // Acquire the lock on HTTP_CLIENT
 
-
         // Build the request
         let mut request_builder = client.request(method.clone(), url);
 
         if let Some(h) = headers {
             let mut req_headers = reqwest::header::HeaderMap::new();
             for (key, value) in h.into_iter() {
-
                 match (
                     HeaderName::from_bytes(key.as_bytes()),
                     HeaderValue::from_str(value.trim()),
@@ -74,20 +68,22 @@ impl HttpClient {
                         req_headers.insert(name, val);
                     }
                     (Err(e), _) => {
-                        eprintln!("Invalid header name: {:?}", key);
+                        eprintln!("Invalid header name: {:?}, error: {}", key, e);
                     }
                     (_, Err(e)) => {
-                        eprintln!("Invalid header value for {}: {:?}", key, value);
+                        eprintln!(
+                            "Invalid header value for {}: {:?}, error: {}",
+                            key, value, e
+                        );
                     }
                 }
             }
             request_builder = request_builder.headers(req_headers);
         }
 
         if let Some(query) = query_params {
-            let query: HashMap<String, String> = query.into_iter()
-                .map(|(k, v)| (k, v.to_string()))
-                .collect();
+            let query: HashMap<String, String> =
+                query.into_iter().map(|(k, v)| (k, v.to_string())).collect();
             request_builder = request_builder.query(&query);
         }
 
@@ -173,7 +169,7 @@ impl HttpClient {
             query_params,
             body,
         )
-            .await
+        .await
     }
 
     // Convenience method for PUT requests
@@ -193,7 +189,7 @@ impl HttpClient {
             query_params,
             body,
         )
-            .await
+        .await
     }
 
     // Convenience method for DELETE requests
@@ -212,6 +208,6 @@ impl HttpClient {
             query_params,
             None,
         )
-            .await
+        .await
     }
 }

src-tauri/src/server/servers.rs

@@ -304,7 +304,7 @@ pub async fn refresh_coco_server_info<R: Runtime>(
         cache.get(&id).cloned()
     };
 
-    let mut server = match cached_server {
+    let server = match cached_server {
         Some(server) => server,
         None => return Err("Server not found.".into()),
     };

src/api/axiosRequest.ts

@@ -1,5 +1,7 @@
 import axios from "axios";
 
+import { useAppStore } from '@/stores/appStore';
+
 import {
   handleChangeRequestHeader,
   handleConfigureAuth,
@@ -40,6 +42,26 @@ axios.interceptors.response.use(
   }
 );
 
+export const handleApiError = (error: any) => {
+  const addError = useAppStore.getState().addError;
+  
+  let message = 'Request failed';
+  
+  if (error.response) {
+    // Server error response
+    message = error.response.data?.message || `Error (${error.response.status})`;
+  } else if (error.request) {
+    // Request failed to send
+    message = 'Network connection failed';
+  } else {
+    // Other errors
+    message = error.message;
+  }
+  
+  addError(message, 'error');
+  return error;
+};
+
 export const Get = <T>(
   url: string,
   params: IAnyObj = {},
@@ -63,6 +85,7 @@ export const Get = <T>(
         resolve([null, res as FcResponse<T>]);
       })
       .catch((err) => {
+        handleApiError(err);
         resolve([err, undefined]);
       });
   });
@@ -88,6 +111,7 @@ export const Post = <T>(
         resolve([null, result.data as FcResponse<T>]);
       })
       .catch((err) => {
+        handleApiError(err);
         resolve([err, undefined]);
       });
   });

src/commands/servers.ts

@@ -13,38 +13,74 @@ import {
   DeleteAttachmentPayload,
   TranscriptionPayload,
   TranscriptionResponse,
+  MultiSourceQueryResponse,
 } from "@/types/commands";
+import { useAppStore } from '@/stores/appStore';
+
+async function invokeWithErrorHandler<T>(
+  command: string,
+  args?: Record<string, any>
+): Promise<T> {
+  const addError = useAppStore.getState().addError;
+  try {
+    const result = await invoke<T>(command, args);
+    console.log(command, result);
+
+    if (result && typeof result === 'object' && 'failed' in result) {
+      const failedResult = result as any;
+      if (failedResult.failed?.length > 0) {
+        failedResult.failed.forEach((error: any) => {
+          // addError(error.error, 'error');
+          console.log(error.error);
+        });
+      }
+    }
+
+    if (typeof result === 'string') {
+      const res = JSON.parse(result);
+      if (typeof res === 'string') {
+        throw new Error(result);
+      }
+    }
+
+    return result;
+  } catch (error: any) {
+    const errorMessage = error || 'Command execution failed';
+    addError(errorMessage, 'error');
+    throw error;
+  }
+}
 
 export function get_server_token(id: string): Promise<ServerTokenResponse> {
-  return invoke(`get_server_token`, { id });
+  return invokeWithErrorHandler(`get_server_token`, { id });
 }
 
 export function list_coco_servers(): Promise<Server[]> {
-  return invoke(`list_coco_servers`);
+  return invokeWithErrorHandler(`list_coco_servers`);
 }
 
 export function add_coco_server(endpoint: string): Promise<Server> {
-  return invoke(`add_coco_server`, { endpoint });
+  return invokeWithErrorHandler(`add_coco_server`, { endpoint });
 }
 
 export function enable_server(id: string): Promise<void> {
-  return invoke(`enable_server`, { id });
+  return invokeWithErrorHandler(`enable_server`, { id });
 }
 
 export function disable_server(id: string): Promise<void> {
-  return invoke(`disable_server`, { id });
+  return invokeWithErrorHandler(`disable_server`, { id });
 }
 
 export function remove_coco_server(id: string): Promise<void> {
-  return invoke(`remove_coco_server`, { id });
+  return invokeWithErrorHandler(`remove_coco_server`, { id });
 }
 
 export function logout_coco_server(id: string): Promise<void> {
-  return invoke(`logout_coco_server`, { id });
+  return invokeWithErrorHandler(`logout_coco_server`, { id });
 }
 
 export function refresh_coco_server_info(id: string): Promise<Server> {
-  return invoke(`refresh_coco_server_info`, { id });
+  return invokeWithErrorHandler(`refresh_coco_server_info`, { id });
 }
 
 export function handle_sso_callback({
@@ -56,27 +92,27 @@ export function handle_sso_callback({
   requestId: string;
   code: string;
 }): Promise<void> {
-  return invoke(`handle_sso_callback`, {
+  return invokeWithErrorHandler(`handle_sso_callback`, {
     serverId,
     requestId,
     code,
   });
 }
 
 export function get_connectors_by_server(id: string): Promise<Connector[]> {
-  return invoke(`get_connectors_by_server`, { id });
+  return invokeWithErrorHandler(`get_connectors_by_server`, { id });
 }
 
 export function get_datasources_by_server(id: string): Promise<DataSource[]> {
-  return invoke(`get_datasources_by_server`, { id });
+  return invokeWithErrorHandler(`get_datasources_by_server`, { id });
 }
 
 export function connect_to_server(id: string, clientId: string): Promise<void> {
-  return invoke(`connect_to_server`, { id, clientId });
+  return invokeWithErrorHandler(`connect_to_server`, { id, clientId });
 }
 
 export function disconnect(clientId: string): Promise<void> {
-  return invoke(`disconnect`, { clientId });
+  return invokeWithErrorHandler(`disconnect`, { clientId });
 }
 
 export function chat_history({
@@ -90,7 +126,7 @@ export function chat_history({
   size?: number;
   query?: string;
 }): Promise<string> {
-  return invoke(`chat_history`, {
+  return invokeWithErrorHandler(`chat_history`, {
     serverId,
     from,
     size,
@@ -109,7 +145,7 @@ export function session_chat_history({
   from?: number;
   size?: number;
 }): Promise<string> {
-  return invoke(`session_chat_history`, {
+  return invokeWithErrorHandler(`session_chat_history`, {
     serverId,
     sessionId,
     from,
@@ -124,7 +160,7 @@ export function close_session_chat({
   serverId: string;
   sessionId: string;
 }): Promise<string> {
-  return invoke(`close_session_chat`, {
+  return invokeWithErrorHandler(`close_session_chat`, {
     serverId,
     sessionId,
   });
@@ -137,7 +173,7 @@ export function open_session_chat({
   serverId: string;
   sessionId: string;
 }): Promise<string> {
-  return invoke(`open_session_chat`, {
+  return invokeWithErrorHandler(`open_session_chat`, {
     serverId,
     sessionId,
   });
@@ -150,7 +186,7 @@ export function cancel_session_chat({
   serverId: string;
   sessionId: string;
 }): Promise<string> {
-  return invoke(`cancel_session_chat`, {
+  return invokeWithErrorHandler(`cancel_session_chat`, {
     serverId,
     sessionId,
   });
@@ -167,7 +203,7 @@ export function new_chat({
   message: string;
   queryParams?: Record<string, any>;
 }): Promise<GetResponse> {
-  return invoke(`new_chat`, {
+  return invokeWithErrorHandler(`new_chat`, {
     serverId,
     websocketId,
     message,
@@ -188,7 +224,7 @@ export function send_message({
   message: string;
   queryParams?: Record<string, any>;
 }): Promise<string> {
-  return invoke(`send_message`, {
+  return invokeWithErrorHandler(`send_message`, {
     serverId,
     websocketId,
     sessionId,
@@ -198,7 +234,7 @@ export function send_message({
 }
 
 export const delete_session_chat = (serverId: string, sessionId: string) => {
-  return invoke<boolean>(`delete_session_chat`, { serverId, sessionId });
+  return invokeWithErrorHandler<boolean>(`delete_session_chat`, { serverId, sessionId });
 };
 
 export const update_session_chat = (payload: {
@@ -209,11 +245,11 @@ export const update_session_chat = (payload: {
     attachments?: string[];
   };
 }): Promise<boolean> => {
-  return invoke<boolean>("update_session_chat", payload);
+  return invokeWithErrorHandler<boolean>("update_session_chat", payload);
 };
 
 export const upload_attachment = async (payload: UploadAttachmentPayload) => {
-  const response = await invoke<UploadAttachmentResponse>("upload_attachment", {
+  const response = await invokeWithErrorHandler<UploadAttachmentResponse>("upload_attachment", {
     ...payload,
   });
 
@@ -223,13 +259,21 @@ export const upload_attachment = async (payload: UploadAttachmentPayload) => {
 };
 
 export const get_attachment = (payload: GetAttachmentPayload) => {
-  return invoke<GetAttachmentResponse>("get_attachment", { ...payload });
+  return invokeWithErrorHandler<GetAttachmentResponse>("get_attachment", { ...payload });
 };
 
 export const delete_attachment = (payload: DeleteAttachmentPayload) => {
-  return invoke<boolean>("delete_attachment", { ...payload });
+  return invokeWithErrorHandler<boolean>("delete_attachment", { ...payload });
 };
 
 export const transcription = (payload: TranscriptionPayload) => {
-  return invoke<TranscriptionResponse>("transcription", { ...payload });
+  return invokeWithErrorHandler<TranscriptionResponse>("transcription", { ...payload });
+};
+
+export const query_coco_fusion = (payload: {
+  from: number;
+  size: number;
+  query_strings: Record<string, string>;
+}) => {
+  return invokeWithErrorHandler<MultiSourceQueryResponse>("query_coco_fusion", { ...payload });
 };