@opengeni/react 0.57.0 → 1.0.2-canary.37191796234001

package/src/hooks/use-machines.ts

@@ -2,12 +2,34 @@ import { useCallback, useRef, useState } from "react";
 import type {
   RemoveEnrollmentRequest,
   RemoveEnrollmentResponse,
+  MachineOperationPolicy,
   SwapActiveSandboxResponse,
+  UpdateMachineOperationPolicyRequest,
+  UpdateMachineAgentResponse,
 } from "@opengeni/sdk";
 import { useOpenGeni, type ClientOverride } from "../provider";
-import { useMutationRunner, usePolledValue } from "./internal";
+import { useMutationRunner } from "./internal";
+import { EMPTY_MACHINES, useSharedMachinesList } from "./shared-machines-list";
 import type { MachinesResponse, MachineView, MetricSample } from "../types/machines";
 
+/** Dashboard list: cheap heartbeat read, keep the pill reasonably fresh. */
+export const MACHINES_DASHBOARD_POLL_MS = 10_000;
+/** Session surfaces share one poll for the same workspace+session. */
+export const MACHINES_SESSION_POLL_MS = 15_000;
+/** Composer polls after the first load shows a machine, or to retry a real error. */
+export const MACHINES_COMPOSER_POLL_MS = 30_000;
+
+const machinesClientIds = new WeakMap<object, number>();
+let nextMachinesClientId = 1;
+
+function machinesClientId(client: object): number {
+  const existing = machinesClientIds.get(client);
+  if (existing !== undefined) return existing;
+  const id = nextMachinesClientId++;
+  machinesClientIds.set(client, id);
+  return id;
+}
+
 /**
  * The slice of the SDK client the Machines surface needs. The method NAMES +
  * SIGNATURES match M10's `OpenGeniClient` (`listMachines`, `machineMetricsSeries`)
@@ -42,6 +64,17 @@ export type MachinesClientLike = {
     enrollmentId: string,
     request?: RemoveEnrollmentRequest,
   ) => Promise<RemoveEnrollmentResponse>;
+  /** POST .../machines/:enrollmentId/update — signed, generation-fenced update. */
+  updateMachineAgent?: (
+    workspaceId: string,
+    enrollmentId: string,
+  ) => Promise<UpdateMachineAgentResponse>;
+  /** PATCH .../machines/:enrollmentId/operation-policy — revision-fenced policy. */
+  updateMachineOperationPolicy?: (
+    workspaceId: string,
+    enrollmentId: string,
+    request: UpdateMachineOperationPolicyRequest,
+  ) => Promise<MachineOperationPolicy>;
   /**
    * POST .../sessions/:sessionId/active-sandbox — swap the session's active
    * sandbox to a machine. The default swap path; the real SDK client provides it.
@@ -87,6 +120,17 @@ export type UseMachinesResult = {
   ) => Promise<RemoveEnrollmentResponse | null>;
   canRemove: boolean;
   removingEnrollmentId: string | null;
+  /** Start/retry the signed self-update for one Connected Machine. */
+  updateAgent: (enrollmentId: string) => Promise<UpdateMachineAgentResponse | null>;
+  canUpdateAgent: boolean;
+  updatingEnrollmentId: string | null;
+  /** Update the optional per-enrollment command memory policy. */
+  updateOperationPolicy: (
+    enrollmentId: string,
+    request: UpdateMachineOperationPolicyRequest,
+  ) => Promise<MachineOperationPolicy | null>;
+  canUpdateOperationPolicy: boolean;
+  updatingOperationPolicyEnrollmentId: string | null;
   /** Whether the host wired an attach/swap path (drives the card affordance). */
   canAttach: boolean;
   /** Fetch a downsampled metric series for one enrolled machine. */
@@ -101,8 +145,6 @@ export type UseMachinesResult = {
   clearMutationError: () => void;
 };
 
-const EMPTY: MachinesResponse = { activeSandboxId: null, activeEpoch: 0, machines: [] };
-
 function swapFailureMessage(result: SwapActiveSandboxResponse): string {
   if (result.reason?.trim()) return result.reason;
   switch (result.code) {
@@ -134,7 +176,7 @@ export function useMachines(options: UseMachinesOptions = {}): UseMachinesResult
   const machinesClient = (options.machinesClient ??
     (client as unknown as MachinesClientLike)) satisfies MachinesClientLike;
   const sessionId = options.sessionId;
-  const identityKey = `${workspaceId}\u0000${sessionId ?? ""}`;
+  const identityKey = `${workspaceId}\u0000${sessionId ?? ""}\u0000${machinesClientId(machinesClient)}`;
   const identityRef = useRef(identityKey);
   identityRef.current = identityKey;
 
@@ -153,7 +195,7 @@ export function useMachines(options: UseMachinesOptions = {}): UseMachinesResult
     loading,
     error,
     refresh,
-  } = usePolledValue(load, {
+  } = useSharedMachinesList(identityKey, load, {
     pollIntervalMs: options.pollIntervalMs,
     enabled: options.enabled,
   });
@@ -170,8 +212,20 @@ export function useMachines(options: UseMachinesOptions = {}): UseMachinesResult
   }>(() => ({ identity: identityKey, enrollmentId: null }));
   const removingEnrollmentId =
     removeState.identity === identityKey ? removeState.enrollmentId : null;
+  const [updateState, setUpdateState] = useState<{
+    identity: string;
+    enrollmentId: string | null;
+  }>(() => ({ identity: identityKey, enrollmentId: null }));
+  const updatingEnrollmentId =
+    updateState.identity === identityKey ? updateState.enrollmentId : null;
+  const [operationPolicyState, setOperationPolicyState] = useState<{
+    identity: string;
+    enrollmentId: string | null;
+  }>(() => ({ identity: identityKey, enrollmentId: null }));
+  const updatingOperationPolicyEnrollmentId =
+    operationPolicyState.identity === identityKey ? operationPolicyState.enrollmentId : null;
 
-  const data = loadedData ?? EMPTY;
+  const data = loadedData ?? EMPTY_MACHINES;
   // The swap is session-scoped: a host adapter (`attachMachine`) wins; otherwise
   // the default `swapActiveSandbox` path is wired whenever a sessionId is in
   // scope. Either way attach needs a sessionId to point at.
@@ -180,6 +234,9 @@ export function useMachines(options: UseMachinesOptions = {}): UseMachinesResult
     (typeof machinesClient.attachMachine === "function" ||
       typeof machinesClient.swapActiveSandbox === "function");
   const canRemove = typeof machinesClient.removeEnrollment === "function";
+  const canUpdateAgent = typeof machinesClient.updateMachineAgent === "function";
+  const canUpdateOperationPolicy =
+    typeof machinesClient.updateMachineOperationPolicy === "function";
 
   const attach = useCallback(
     async (sandboxId: string): Promise<boolean> => {
@@ -246,6 +303,41 @@ export function useMachines(options: UseMachinesOptions = {}): UseMachinesResult
     [machinesClient, workspaceId, identityKey, run, refresh],
   );
 
+  const updateAgent = useCallback(
+    async (enrollmentId: string): Promise<UpdateMachineAgentResponse | null> => {
+      if (!machinesClient.updateMachineAgent) return null;
+      const ownedIdentity = identityKey;
+      setUpdateState({ identity: ownedIdentity, enrollmentId });
+      const result = await run(() => machinesClient.updateMachineAgent!(workspaceId, enrollmentId));
+      if (identityRef.current === ownedIdentity) {
+        setUpdateState({ identity: ownedIdentity, enrollmentId: null });
+        await refresh();
+      }
+      return result;
+    },
+    [machinesClient, workspaceId, identityKey, run, refresh],
+  );
+
+  const updateOperationPolicy = useCallback(
+    async (
+      enrollmentId: string,
+      request: UpdateMachineOperationPolicyRequest,
+    ): Promise<MachineOperationPolicy | null> => {
+      if (!machinesClient.updateMachineOperationPolicy) return null;
+      const ownedIdentity = identityKey;
+      setOperationPolicyState({ identity: ownedIdentity, enrollmentId });
+      const result = await run(() =>
+        machinesClient.updateMachineOperationPolicy!(workspaceId, enrollmentId, request),
+      );
+      if (identityRef.current === ownedIdentity) {
+        setOperationPolicyState({ identity: ownedIdentity, enrollmentId: null });
+        if (result) await refresh();
+      }
+      return result;
+    },
+    [machinesClient, workspaceId, identityKey, run, refresh],
+  );
+
   return {
     machines: data.machines,
     activeSandboxId: data.activeSandboxId,
@@ -261,6 +353,12 @@ export function useMachines(options: UseMachinesOptions = {}): UseMachinesResult
     remove,
     canRemove,
     removingEnrollmentId,
+    updateAgent,
+    canUpdateAgent,
+    updatingEnrollmentId,
+    updateOperationPolicy,
+    canUpdateOperationPolicy,
+    updatingOperationPolicyEnrollmentId,
     mutationError,
     clearMutationError,
   };