@robosystems/client 0.1.12 → 0.1.14

package/extensions/hooks.js

@@ -1,18 +1,20 @@
-'use client'
-
+'use client';
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.useQuery = useQuery;
+exports.useStreamingQuery = useStreamingQuery;
+exports.useOperation = useOperation;
+exports.useMultipleOperations = useMultipleOperations;
+exports.useSDKClients = useSDKClients;
 /**
  * React hooks for SDK extensions
  * Provides easy-to-use hooks for Next.js/React applications
  */
-
-import { useCallback, useEffect, useRef, useState } from 'react'
-import { client } from '../client.gen'
-import { getSDKExtensionsConfig } from './config'
-import { OperationProgress, OperationResult } from './OperationClient'
-import { OperationClient } from './OperationClient'
-import { QueryOptions, QueryResult } from './QueryClient'
-import { QueryClient } from './QueryClient'
-
+const react_1 = require("react");
+const client_gen_1 = require("../client.gen");
+const config_1 = require("./config");
+const OperationClient_1 = require("./OperationClient");
+const QueryClient_1 = require("./QueryClient");
 /**
  * Hook for executing Cypher queries with loading states and error handling
  *
@@ -26,89 +28,69 @@ import { QueryClient } from './QueryClient'
  * }
  * ```
  */
-export function useQuery(graphId) {
-  const [loading, setLoading] = useState(false)
-  const [error, setError] = useState(null)
-  const [data, setData] = useState(null)
-  const [queuePosition, setQueuePosition] = useState(null)
-  const clientRef = useRef(null)
-
-  // Initialize client
-  useEffect(() => {
-    const sdkConfig = getSDKExtensionsConfig()
-    const clientConfig = client.getConfig()
-    clientRef.current = new QueryClient({
-      baseUrl.baseUrl || clientConfig.baseUrl || 'http://localhost:8000',
-      credentials.credentials,
-      headers.headers,
-    })
-
-    return () => {
-      clientRef.current?.close()
-    }
-  }, [])
-
-  const execute = useCallback(
-    async (
-      query,
-      parameters?,
-      options?
-    ) => {
-      if (!clientRef.current) return null
-
-      setLoading(true)
-      setError(null)
-      setData(null)
-      setQueuePosition(null)
-
-      try {
-        const result = (await clientRef.current.executeQuery(
-          graphId,
-          { query, parameters },
-          {
-            ...options,
-            onQueueUpdate: (position) => {
-              setQueuePosition(position)
-            },
-            onProgress: () => {
-              setQueuePosition(null) // Clear queue position when executing
-            },
-          }
-        ))
-
-        setData(result)
-        return result
-      } catch (err) {
-        const error = err
-        setError(error)
-        return null
-      } finally {
-        setLoading(false)
-        setQueuePosition(null)
-      }
-    },
-    [graphId]
-  )
-
-  // Simple query method that returns just the data
-  const query = useCallback(
-    async (cypher, parameters?) => {
-      const result = await execute(cypher, parameters)
-      return result?.data || []
-    },
-    [execute]
-  )
-
-  return {
-    execute,
-    query,
-    loading,
-    error,
-    data,
-    queuePosition,
-  }
+function useQuery(graphId) {
+    const [loading, setLoading] = (0, react_1.useState)(false);
+    const [error, setError] = (0, react_1.useState)(null);
+    const [data, setData] = (0, react_1.useState)(null);
+    const [queuePosition, setQueuePosition] = (0, react_1.useState)(null);
+    const clientRef = (0, react_1.useRef)(null);
+    // Initialize client
+    (0, react_1.useEffect)(() => {
+        const sdkConfig = (0, config_1.getSDKExtensionsConfig)();
+        const clientConfig = client_gen_1.client.getConfig();
+        clientRef.current = new QueryClient_1.QueryClient({
+            baseUrl: sdkConfig.baseUrl || clientConfig.baseUrl || 'http://localhost:8000',
+            credentials: sdkConfig.credentials,
+            headers: sdkConfig.headers,
+        });
+        return () => {
+            clientRef.current?.close();
+        };
+    }, []);
+    const execute = (0, react_1.useCallback)(async (query, parameters, options) => {
+        if (!clientRef.current)
+            return null;
+        setLoading(true);
+        setError(null);
+        setData(null);
+        setQueuePosition(null);
+        try {
+            const result = (await clientRef.current.executeQuery(graphId, { query, parameters }, {
+                ...options,
+                onQueueUpdate: (position) => {
+                    setQueuePosition(position);
+                },
+                onProgress: () => {
+                    setQueuePosition(null); // Clear queue position when executing
+                },
+            }));
+            setData(result);
+            return result;
+        }
+        catch (err) {
+            const error = err;
+            setError(error);
+            return null;
+        }
+        finally {
+            setLoading(false);
+            setQueuePosition(null);
+        }
+    }, [graphId]);
+    // Simple query method that returns just the data
+    const query = (0, react_1.useCallback)(async (cypher, parameters) => {
+        const result = await execute(cypher, parameters);
+        return result?.data || [];
+    }, [execute]);
+    return {
+        execute,
+        query,
+        loading,
+        error,
+        data,
+        queuePosition,
+    };
 }
-
 /**
  * Hook for streaming large query results
  *
@@ -124,90 +106,74 @@ export function useQuery(graphId) {
  * }
  * ```
  */
-export function useStreamingQuery(graphId) {
-  const [isStreaming, setIsStreaming] = useState(false)
-  const [error, setError] = useState(null)
-  const [rowsReceived, setRowsReceived] = useState(0)
-  const clientRef = useRef(null)
-
-  useEffect(() => {
-    const sdkConfig = getSDKExtensionsConfig()
-    const clientConfig = client.getConfig()
-    clientRef.current = new QueryClient({
-      baseUrl.baseUrl || clientConfig.baseUrl || 'http://localhost:8000',
-      credentials.credentials,
-      headers.headers,
-    })
-
-    return () => {
-      clientRef.current?.close()
-    }
-  }, [])
-
-  const stream = useCallback(
-    async function* (
-      query,
-      parameters?,
-      chunkSize = 100
-    ) {
-      if (!clientRef.current) return
-
-      setIsStreaming(true)
-      setError(null)
-      setRowsReceived(0)
-
-      try {
-        const iterator = clientRef.current.streamQuery(graphId, query, parameters, chunkSize)
-
-        let buffer = []
-        let count = 0
-
-        for await (const row of iterator) {
-          buffer.push(row)
-          count++
-
-          if (buffer.length >= chunkSize) {
-            setRowsReceived(count)
-            yield buffer
-            buffer = []
-          }
+function useStreamingQuery(graphId) {
+    const [isStreaming, setIsStreaming] = (0, react_1.useState)(false);
+    const [error, setError] = (0, react_1.useState)(null);
+    const [rowsReceived, setRowsReceived] = (0, react_1.useState)(0);
+    const clientRef = (0, react_1.useRef)(null);
+    (0, react_1.useEffect)(() => {
+        const sdkConfig = (0, config_1.getSDKExtensionsConfig)();
+        const clientConfig = client_gen_1.client.getConfig();
+        clientRef.current = new QueryClient_1.QueryClient({
+            baseUrl: sdkConfig.baseUrl || clientConfig.baseUrl || 'http://localhost:8000',
+            credentials: sdkConfig.credentials,
+            headers: sdkConfig.headers,
+        });
+        return () => {
+            clientRef.current?.close();
+        };
+    }, []);
+    const stream = (0, react_1.useCallback)(async function* (query, parameters, chunkSize = 100) {
+        if (!clientRef.current)
+            return;
+        setIsStreaming(true);
+        setError(null);
+        setRowsReceived(0);
+        try {
+            const iterator = clientRef.current.streamQuery(graphId, query, parameters, chunkSize);
+            let buffer = [];
+            let count = 0;
+            for await (const row of iterator) {
+                buffer.push(row);
+                count++;
+                if (buffer.length >= chunkSize) {
+                    setRowsReceived(count);
+                    yield buffer;
+                    buffer = [];
+                }
+            }
+            // Yield any remaining items
+            if (buffer.length > 0) {
+                setRowsReceived(count);
+                yield buffer;
+            }
         }
-
-        // Yield any remaining items
-        if (buffer.length > 0) {
-          setRowsReceived(count)
-          yield buffer
+        catch (err) {
+            setError(err);
+            throw err;
         }
-      } catch (err) {
-        setError(err)
-        throw err
-      } finally {
-        setIsStreaming(false)
-      }
-    },
-    [graphId]
-  )
-
-  const cancel = useCallback(() => {
-    clientRef.current?.close()
-    setIsStreaming(false)
-  }, [])
-
-  return {
-    stream,
-    isStreaming,
-    error,
-    rowsReceived,
-    cancel,
-  }
+        finally {
+            setIsStreaming(false);
+        }
+    }, [graphId]);
+    const cancel = (0, react_1.useCallback)(() => {
+        clientRef.current?.close();
+        setIsStreaming(false);
+    }, []);
+    return {
+        stream,
+        isStreaming,
+        error,
+        rowsReceived,
+        cancel,
+    };
 }
-
 /**
  * Hook for monitoring long-running operations
  *
  * @example
  * ```tsx
- * const { monitor, status, progress, error, result } = useOperation()
+ * const { monitor, status, progress, error, result } = useOperation<BackupResult>()
  *
  * const handleBackup = async () => {
  *   const { operation_id } = await createBackup({ ... })
@@ -216,98 +182,86 @@ export function useStreamingQuery(graphId) {
  * }
  * ```
  */
-export function useOperation(operationId?) {
-  const [status, setStatus] = useState('idle')
-  const [progress, setProgress] = useState(null)
-  const [error, setError] = useState(null)
-  const [result, setResult] = useState | null>(null)
-  const clientRef = useRef(null)
-
-  useEffect(() => {
-    const sdkConfig = getSDKExtensionsConfig()
-    const clientConfig = client.getConfig()
-    clientRef.current = new OperationClient({
-      baseUrl.baseUrl || clientConfig.baseUrl || 'http://localhost:8000',
-      credentials.credentials,
-      maxRetries.maxRetries,
-      retryDelay.retryDelay,
-    })
-
-    return () => {
-      clientRef.current?.closeAll()
-    }
-  }, [])
-
-  const monitor = useCallback(
-    async (id, timeout?) | null> => {
-      if (!clientRef.current) return null
-
-      setStatus('running')
-      setError(null)
-      setResult(null)
-      setProgress(null)
-
-      try {
-        const opResult = await clientRef.current.monitorOperation(id, {
-          onProgress: (p) => {
-            setProgress(p)
-          },
-          onQueueUpdate: (position, estimatedWait) => {
-            setProgress({
-              message: `Queue position: ${position}`,
-              progressPercent,
-              details,
-            })
-          },
-          timeout,
-        })
-
-        setResult(opResult)
-        setStatus(opResult.success ? 'completed' )
-
-        if (!opResult.success && opResult.error) {
-          setError(new Error(opResult.error))
+function useOperation(operationId) {
+    const [status, setStatus] = (0, react_1.useState)('idle');
+    const [progress, setProgress] = (0, react_1.useState)(null);
+    const [error, setError] = (0, react_1.useState)(null);
+    const [result, setResult] = (0, react_1.useState)(null);
+    const clientRef = (0, react_1.useRef)(null);
+    (0, react_1.useEffect)(() => {
+        const sdkConfig = (0, config_1.getSDKExtensionsConfig)();
+        const clientConfig = client_gen_1.client.getConfig();
+        clientRef.current = new OperationClient_1.OperationClient({
+            baseUrl: sdkConfig.baseUrl || clientConfig.baseUrl || 'http://localhost:8000',
+            credentials: sdkConfig.credentials,
+            maxRetries: sdkConfig.maxRetries,
+            retryDelay: sdkConfig.retryDelay,
+        });
+        return () => {
+            clientRef.current?.closeAll();
+        };
+    }, []);
+    const monitor = (0, react_1.useCallback)(async (id, timeout) => {
+        if (!clientRef.current)
+            return null;
+        setStatus('running');
+        setError(null);
+        setResult(null);
+        setProgress(null);
+        try {
+            const opResult = await clientRef.current.monitorOperation(id, {
+                onProgress: (p) => {
+                    setProgress(p);
+                },
+                onQueueUpdate: (position, estimatedWait) => {
+                    setProgress({
+                        message: `Queue position: ${position}`,
+                        progressPercent: 0,
+                        details: { position, estimatedWait },
+                    });
+                },
+                timeout,
+            });
+            setResult(opResult);
+            setStatus(opResult.success ? 'completed' : 'error');
+            if (!opResult.success && opResult.error) {
+                setError(new Error(opResult.error));
+            }
+            return opResult;
+        }
+        catch (err) {
+            const error = err;
+            setError(error);
+            setStatus('error');
+            return null;
         }
-
-        return opResult
-      } catch (err) {
-        const error = err
-        setError(error)
-        setStatus('error')
-        return null
-      }
-    },
-    []
-  )
-
-  const cancel = useCallback(async (id) => {
-    if (!clientRef.current) return
-
-    try {
-      await clientRef.current.cancelOperation(id)
-      setStatus('idle')
-    } catch (err) {
-      setError(err)
-    }
-  }, [])
-
-  // Auto-monitor if operationId is provided
-  useEffect(() => {
-    if (operationId && status === 'idle') {
-      monitor(operationId)
-    }
-  }, [operationId, monitor, status])
-
-  return {
-    monitor,
-    cancel,
-    status,
-    progress,
-    error,
-    result,
-  }
+    }, []);
+    const cancel = (0, react_1.useCallback)(async (id) => {
+        if (!clientRef.current)
+            return;
+        try {
+            await clientRef.current.cancelOperation(id);
+            setStatus('idle');
+        }
+        catch (err) {
+            setError(err);
+        }
+    }, []);
+    // Auto-monitor if operationId is provided
+    (0, react_1.useEffect)(() => {
+        if (operationId && status === 'idle') {
+            monitor(operationId);
+        }
+    }, [operationId, monitor, status]);
+    return {
+        monitor,
+        cancel,
+        status,
+        progress,
+        error,
+        result,
+    };
 }
-
 /**
  * Hook for monitoring multiple operations concurrently
  *
@@ -325,71 +279,58 @@ export function useOperation(operationId?) {
  * }
  * ```
  */
-export function useMultipleOperations() {
-  const [results, setResults] = useState>>(new Map())
-  const [loading, setLoading] = useState(false)
-  const [errors, setErrors] = useState>(new Map())
-  const clientRef = useRef(null)
-
-  useEffect(() => {
-    const sdkConfig = getSDKExtensionsConfig()
-    const clientConfig = client.getConfig()
-    clientRef.current = new OperationClient({
-      baseUrl.baseUrl || clientConfig.baseUrl || 'http://localhost:8000',
-      credentials.credentials,
-      maxRetries.maxRetries,
-      retryDelay.retryDelay,
-    })
-
-    return () => {
-      clientRef.current?.closeAll()
-    }
-  }, [])
-
-  const monitorAll = useCallback(
-    async (operationIds)>> => {
-      if (!clientRef.current) return new Map()
-
-      setLoading(true)
-      setResults(new Map())
-      setErrors(new Map())
-
-      try {
-        const allResults = await clientRef.current.monitorMultiple(operationIds)
-        setResults(allResults)
-
-        // Extract any errors
-        const newErrors = new Map()
-        allResults.forEach((result, id) => {
-          if (!result.success && result.error) {
-            newErrors.set(id, new Error(result.error))
-          }
-        })
-        setErrors(newErrors)
-
-        return allResults
-      } finally {
-        setLoading(false)
-      }
-    },
-    []
-  )
-
-  const allCompleted =
-    results.size > 0 && Array.from(results.values()).every((r) => r.success || r.error)
-
-  const hasErrors = errors.size > 0
-
-  return {
-    monitorAll,
-    results,
-    errors,
-    loading,
-    allCompleted,
-    hasErrors,
-  }
+function useMultipleOperations() {
+    const [results, setResults] = (0, react_1.useState)(new Map());
+    const [loading, setLoading] = (0, react_1.useState)(false);
+    const [errors, setErrors] = (0, react_1.useState)(new Map());
+    const clientRef = (0, react_1.useRef)(null);
+    (0, react_1.useEffect)(() => {
+        const sdkConfig = (0, config_1.getSDKExtensionsConfig)();
+        const clientConfig = client_gen_1.client.getConfig();
+        clientRef.current = new OperationClient_1.OperationClient({
+            baseUrl: sdkConfig.baseUrl || clientConfig.baseUrl || 'http://localhost:8000',
+            credentials: sdkConfig.credentials,
+            maxRetries: sdkConfig.maxRetries,
+            retryDelay: sdkConfig.retryDelay,
+        });
+        return () => {
+            clientRef.current?.closeAll();
+        };
+    }, []);
+    const monitorAll = (0, react_1.useCallback)(async (operationIds) => {
+        if (!clientRef.current)
+            return new Map();
+        setLoading(true);
+        setResults(new Map());
+        setErrors(new Map());
+        try {
+            const allResults = await clientRef.current.monitorMultiple(operationIds);
+            setResults(allResults);
+            // Extract any errors
+            const newErrors = new Map();
+            allResults.forEach((result, id) => {
+                if (!result.success && result.error) {
+                    newErrors.set(id, new Error(result.error));
+                }
+            });
+            setErrors(newErrors);
+            return allResults;
+        }
+        finally {
+            setLoading(false);
+        }
+    }, []);
+    const allCompleted = results.size > 0 && Array.from(results.values()).every((r) => r.success || r.error);
+    const hasErrors = errors.size > 0;
+    return {
+        monitorAll,
+        results,
+        errors,
+        loading,
+        allCompleted,
+        hasErrors,
+    };
 }
-
 /**
  * Hook that provides access to all SDK extension clients
  * Useful when you need direct access to the underlying clients
@@ -402,34 +343,29 @@ export function useMultipleOperations() {
  * const result = await clients.query.query('graph_123', 'MATCH (n) RETURN n')
  * ```
  */
-export function useSDKClients() {
-  const [clients, setClients] = useState({
-    query,
-    operations,
-  })
-
-  useEffect(() => {
-    const sdkConfig = getSDKExtensionsConfig()
-    const clientConfig = client.getConfig()
-    const baseConfig = {
-      baseUrl.baseUrl || clientConfig.baseUrl || 'http://localhost:8000',
-      credentials.credentials,
-      headers.headers,
-    }
-
-    const queryClient = new QueryClient(baseConfig)
-    const operationsClient = new OperationClient(baseConfig)
-
-    setClients({
-      query,
-      operations,
-    })
-
-    return () => {
-      queryClient.close()
-      operationsClient.closeAll()
-    }
-  }, [])
-
-  return clients
+function useSDKClients() {
+    const [clients, setClients] = (0, react_1.useState)({
+        query: null,
+        operations: null,
+    });
+    (0, react_1.useEffect)(() => {
+        const sdkConfig = (0, config_1.getSDKExtensionsConfig)();
+        const clientConfig = client_gen_1.client.getConfig();
+        const baseConfig = {
+            baseUrl: sdkConfig.baseUrl || clientConfig.baseUrl || 'http://localhost:8000',
+            credentials: sdkConfig.credentials,
+            headers: sdkConfig.headers,
+        };
+        const queryClient = new QueryClient_1.QueryClient(baseConfig);
+        const operationsClient = new OperationClient_1.OperationClient(baseConfig);
+        setClients({
+            query: queryClient,
+            operations: operationsClient,
+        });
+        return () => {
+            queryClient.close();
+            operationsClient.closeAll();
+        };
+    }, []);
+    return clients;
 }