@opengeni/react 0.13.0 → 0.20.0

package/src/hooks/use-session-control.ts

@@ -1,13 +1,13 @@
-import type { SessionEvent } from "@opengeni/sdk";
-import { useCallback } from "react";
-import { useOpenGeni, type ClientOverride } from "../provider";
+import type { SessionControlResponse, SessionEvent } from "@opengeni/sdk";
+import { useCallback, useLayoutEffect, useRef } from "react";
+import { useEmbeddedSession, type EmbeddedSessionClientOverride } from "../session-context";
 import { useMutationRunner } from "./internal";
 
-export type UseSessionControlOptions = ClientOverride;
+export type UseSessionControlOptions = EmbeddedSessionClientOverride;
 
 export type UseSessionControlResult = {
-  pause: (reason?: string) => Promise<SessionEvent | null>;
-  resume: (reason?: string) => Promise<SessionEvent | null>;
+  pause: (reason?: string) => Promise<SessionControlResponse | null>;
+  resume: (reason?: string) => Promise<SessionControlResponse | null>;
   controlling: boolean;
   /** Approve a pending `requires_action` approval. */
   approve: (approvalId: string, message?: string) => Promise<SessionEvent | null>;
@@ -28,22 +28,33 @@ export function useSessionControl(
   sessionId: string | null | undefined,
   options: UseSessionControlOptions = {},
 ): UseSessionControlResult {
-  const { client, workspaceId } = useOpenGeni(options);
+  const { client, workspaceId } = useEmbeddedSession(options);
+  const approvalTargetKey = `${workspaceId}\u0000${sessionId ?? ""}`;
+  const pendingApproval = useRef<{
+    targetKey: string;
+    decisionKey: string;
+    clientEventId: string;
+  } | null>(null);
+  useLayoutEffect(() => {
+    if (pendingApproval.current?.targetKey !== approvalTargetKey) {
+      pendingApproval.current = null;
+    }
+  }, [approvalTargetKey]);
   const {
     run: runControl,
     mutating: controlling,
     mutationError: controlError,
     clearMutationError: clearControlError,
-  } = useMutationRunner();
+  } = useMutationRunner(approvalTargetKey);
   const {
     run: runApproval,
     mutating: responding,
     mutationError: approvalError,
     clearMutationError: clearApprovalError,
-  } = useMutationRunner();
+  } = useMutationRunner(approvalTargetKey);
 
   const pause = useCallback(
-    async (reason?: string): Promise<SessionEvent | null> => {
+    async (reason?: string): Promise<SessionControlResponse | null> => {
       if (!sessionId) {
         return null;
       }
@@ -55,12 +66,11 @@ export function useSessionControl(
   );
 
   const resume = useCallback(
-    async (reason?: string): Promise<SessionEvent | null> => {
+    async (reason?: string): Promise<SessionControlResponse | null> => {
       if (!sessionId) return null;
-      const result = await runControl(() =>
+      return await runControl(() =>
         client.resumeSession(workspaceId, sessionId, reason !== undefined ? { reason } : {}),
       );
-      return result?.event ?? null;
     },
     [client, workspaceId, sessionId, runControl],
   );
@@ -74,15 +84,33 @@ export function useSessionControl(
       if (!sessionId) {
         return null;
       }
-      return await runApproval(() =>
+      const decisionKey = JSON.stringify([approvalId, decision, message ?? null]);
+      if (pendingApproval.current?.decisionKey !== decisionKey) {
+        pendingApproval.current = {
+          targetKey: approvalTargetKey,
+          decisionKey,
+          clientEventId: crypto.randomUUID(),
+        };
+      }
+      const clientEventId = pendingApproval.current.clientEventId;
+      const accepted = await runApproval(() =>
         client.sendApprovalDecision(workspaceId, sessionId, {
           approvalId,
           decision,
           ...(message !== undefined ? { message } : {}),
+          clientEventId,
         }),
       );
+      if (
+        accepted !== null &&
+        pendingApproval.current?.targetKey === approvalTargetKey &&
+        pendingApproval.current.clientEventId === clientEventId
+      ) {
+        pendingApproval.current = null;
+      }
+      return accepted;
     },
-    [client, workspaceId, sessionId, runApproval],
+    [approvalTargetKey, client, workspaceId, sessionId, runApproval],
   );
 
   const approve = useCallback(