@robosystems/client 0.1.15 → 0.1.16

package/sdk-extensions/hooks.js

@@ -0,0 +1,371 @@
+'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
+ */
+const react_1 = require("react");
+const client_gen_1 = require("../sdk/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
+ *
+ * @example
+ * ```tsx
+ * const { execute, loading, error, data } = useQuery('graph_123')
+ *
+ * const handleSearch = async () => {
+ *   const result = await execute('MATCH (n:Company) RETURN n LIMIT 10')
+ *   console.log(result.data)
+ * }
+ * ```
+ */
+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
+ *
+ * @example
+ * ```tsx
+ * const { stream, isStreaming, error, cancel } = useStreamingQuery('graph_123')
+ *
+ * const handleStream = async () => {
+ *   const iterator = stream('MATCH (n) RETURN n')
+ *   for await (const batch of iterator) {
+ *     console.log('Received batch:', batch)
+ *   }
+ * }
+ * ```
+ */
+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;
+            }
+        }
+        catch (err) {
+            setError(err);
+            throw err;
+        }
+        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<BackupResult>()
+ *
+ * const handleBackup = async () => {
+ *   const { operation_id } = await createBackup({ ... })
+ *   const result = await monitor(operation_id)
+ *   console.log('Backup completed:', result)
+ * }
+ * ```
+ */
+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;
+        }
+    }, []);
+    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
+ *
+ * @example
+ * ```tsx
+ * const { monitorAll, results, allCompleted, hasErrors } = useMultipleOperations()
+ *
+ * const handleMultiple = async () => {
+ *   const operations = await Promise.all([
+ *     createBackup(...),
+ *     createExport(...),
+ *   ])
+ *
+ *   const results = await monitorAll(operations.map(op => op.operation_id))
+ * }
+ * ```
+ */
+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
+ *
+ * @example
+ * ```tsx
+ * const clients = useSDKClients()
+ *
+ * // Direct access to clients
+ * const result = await clients.query.query('graph_123', 'MATCH (n) RETURN n')
+ * ```
+ */
+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;
+}