@malloy-publisher/sdk 0.0.41 → 0.0.42

package/src/components/QueryResult/QueryResult.tsx

@@ -1,14 +1,13 @@
 import { Suspense, lazy } from "react";
 import { Configuration, QueryresultsApi } from "../../client";
-import { QueryClient, useQuery } from "@tanstack/react-query";
 import { usePackage } from "../Package/PackageProvider";
 import { ApiErrorDisplay } from "../ApiErrorDisplay";
 import { Loading } from "../Loading";
+import { useQueryWithApiError } from "../../hooks/useQueryWithApiError";
 
 const RenderedResult = lazy(() => import("../RenderedResult/RenderedResult"));
 
 const queryResultsApi = new QueryresultsApi(new Configuration());
-const queryClient = new QueryClient();
 
 interface QueryResultProps {
    modelPath: string;
@@ -26,41 +25,36 @@ export default function QueryResult({
    const { server, projectName, packageName, versionId, accessToken } =
       usePackage();
 
-   const { data, isSuccess, isError, error } = useQuery(
-      {
-         queryKey: [
-            "queryResult",
-            server,
+   const { data, isSuccess, isError, error } = useQueryWithApiError({
+      queryKey: [
+         "queryResult",
+         server,
+         projectName,
+         packageName,
+         modelPath,
+         versionId,
+         query,
+         sourceName,
+         queryName,
+      ],
+      queryFn: () =>
+         queryResultsApi.executeQuery(
             projectName,
             packageName,
             modelPath,
-            versionId,
             query,
             sourceName,
             queryName,
-         ],
-         queryFn: () =>
-            queryResultsApi.executeQuery(
-               projectName,
-               packageName,
-               modelPath,
-               query,
-               sourceName,
-               queryName,
-               versionId,
-               {
-                  baseURL: server,
-                  withCredentials: !accessToken,
-                  headers: {
-                     Authorization: accessToken && `Bearer ${accessToken}`,
-                  },
+            versionId,
+            {
+               baseURL: server,
+               withCredentials: !accessToken,
+               headers: {
+                  Authorization: accessToken && `Bearer ${accessToken}`,
                },
-            ),
-         retry: false,
-         throwOnError: false,
-      },
-      queryClient,
-   );
+            },
+         ),
+   });
 
    return (
       <>

package/src/hooks/useQueryWithApiError.ts

@@ -0,0 +1,114 @@
+import {
+   useQuery,
+   useMutation,
+   QueryClient,
+   UseQueryOptions,
+   UseQueryResult,
+   UseMutationOptions,
+   UseMutationResult,
+} from "@tanstack/react-query";
+import { ApiError } from "../components/ApiErrorDisplay";
+
+// Global QueryClient instance
+const globalQueryClient = new QueryClient({
+   defaultOptions: {
+      queries: {
+         retry: false,
+         throwOnError: false,
+      },
+      mutations: {
+         retry: false,
+         throwOnError: false,
+      },
+   },
+});
+
+export function useQueryWithApiError<TData = unknown, TError = ApiError>(
+   options: Omit<UseQueryOptions<TData, TError>, "throwOnError" | "retry"> & {
+      queryFn: () => Promise<TData>;
+   },
+): UseQueryResult<TData, TError> {
+   const enhancedOptions: UseQueryOptions<TData, TError> = {
+      ...options,
+      queryFn: async () => {
+         try {
+            return await options.queryFn();
+         } catch (err) {
+            // Standardized error handling for axios errors
+            if (err && typeof err === "object" && "response" in err) {
+               const axiosError = err as {
+                  response?: {
+                     status: number;
+                     data: { code: number; message: string };
+                  };
+                  message: string;
+               };
+               if (axiosError.response?.data) {
+                  const apiError: ApiError = new Error(
+                     axiosError.response.data.message || axiosError.message,
+                  );
+                  apiError.status = axiosError.response.status;
+                  apiError.data = axiosError.response.data;
+                  throw apiError;
+               }
+            }
+            // For other errors, throw as is
+            throw err;
+         }
+      },
+      retry: false,
+      throwOnError: false,
+   };
+
+   return useQuery(enhancedOptions, globalQueryClient);
+}
+
+export function useMutationWithApiError<
+   TData = unknown,
+   TError = ApiError,
+   TVariables = void,
+>(
+   options: Omit<
+      UseMutationOptions<TData, TError, TVariables>,
+      "throwOnError" | "retry"
+   > & {
+      mutationFn: (variables: TVariables) => Promise<TData>;
+   },
+): UseMutationResult<TData, TError, TVariables> {
+   const enhancedOptions: UseMutationOptions<TData, TError, TVariables> = {
+      ...options,
+      mutationFn: async (variables: TVariables) => {
+         try {
+            return await options.mutationFn(variables);
+         } catch (err) {
+            // Standardized error handling for axios errors
+            if (err && typeof err === "object" && "response" in err) {
+               const axiosError = err as {
+                  response?: {
+                     status: number;
+                     data: { code: number; message: string };
+                  };
+                  message: string;
+               };
+               if (axiosError.response?.data) {
+                  const apiError: ApiError = new Error(
+                     axiosError.response.data.message || axiosError.message,
+                  );
+                  apiError.status = axiosError.response.status;
+                  apiError.data = axiosError.response.data;
+                  throw apiError;
+               }
+            }
+            // For other errors, throw as is
+            throw err;
+         }
+      },
+      retry: false,
+      throwOnError: false,
+   };
+
+   return useMutation(enhancedOptions, globalQueryClient);
+}
+
+// Export the global query client for direct access if needed
+export { globalQueryClient };

package/src/index.ts

@@ -1,2 +1,7 @@
 export * from "./components";
 export { default as ConnectionExplorer } from "./components/Project/ConnectionExplorer";
+import axios from "axios";
+
+// There's a bug in the OpenAPI generator that causes it to ignore baseURL in the
+// axios request if defaults.baseURL is not set.
+axios.defaults.baseURL = "IfYouAreSeeingThis_baseURL_IsNotSet";