All files / client/src/hooks use-simulation-controls.ts

100% Statements 16/16
87.5% Branches 7/8
75% Functions 3/4
100% Lines 13/13

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124                                                                                                                                      64x             64x                               64x 6x 5x 4x 4x 4x 4x 4x 4x 2x         64x                                      
import { useCallback } from "react";
import type { MutableRefObject } from "react";
import type { UseMutationResult } from "@tanstack/react-query";
import type { IncomingArduinoMessage } from "@/types/websocket";
import type { SimulationStatus } from "@shared/types/arduino.types";
import { useSimulationController } from "./use-simulation-controller";
import { useUiFeedbackAdapter } from "./use-ui-feedback-adapter";
import type { ServerCapabilities } from "@/lib/server-capabilities";
export type { SimulationStatus } from "@shared/types/arduino.types";
 
export type SetState<T> = (value: T | ((prev: T) => T)) => void;
 
export type DebugMessageParams = {
  source: "frontend" | "server";
  type: string;
  data: string;
  protocol?: "websocket" | "http";
};
 
export type UseSimulationControlsParams = {
  readonly capabilities?: ServerCapabilities;
  ensureBackendConnected: (reason: string) => boolean;
  sendMessage: (message: IncomingArduinoMessage) => void;
  /** Optional immediate sender for time-critical commands (stop) */
  // return value indicates whether the message was actually sent (socket open)
  sendMessageImmediate?: (message: IncomingArduinoMessage) => boolean;
  resetPinUI: (opts?: { keepDetected?: boolean }) => void;
  clearOutputs: () => void;
  addDebugMessage: (params: DebugMessageParams) => void;
  serialEventQueueRef: MutableRefObject<
    Array<{ payload: IncomingArduinoMessage; receivedAt: number }>
  >;
  toast: (args: {
    title: string;
    description?: string;
    variant?: "destructive";
  }) => void;
  pendingPinConflicts: number[];
  setPendingPinConflicts: SetState<number[]>;
  setCliOutput: SetState<string>;
  isModified: boolean;
  handleCompileAndStart: () => void;
  startSimulationRef: MutableRefObject<(() => void) | null>;
};
 
type UseSimulationControlsResult = {
  simulationStatus: SimulationStatus;
  setSimulationStatus: SetState<SimulationStatus>;
  hasCompiledOnce: boolean;
  setHasCompiledOnce: SetState<boolean>;
  simulationTimeout: number;
  setSimulationTimeout: SetState<number>;
  startMutation: UseMutationResult<{ success: boolean }, unknown, void, unknown>;
  stopMutation: UseMutationResult<{ success: boolean }, unknown, void, unknown>;
  pauseMutation: UseMutationResult<{ success: boolean }, unknown, void, unknown>;
  resumeMutation: UseMutationResult<{ success: boolean }, unknown, void, unknown>;
  handleStart: () => void;
  handleStop: () => void;
  handlePause: () => void;
  handleResume: () => void;
  handleReset: () => void;
  suppressAutoStopOnce: () => void;
};
 
export function useSimulationControls(
  params: UseSimulationControlsParams,
): UseSimulationControlsResult {
  const uiFeedback = useUiFeedbackAdapter({
    toast: params.toast,
    addDebugMessage: params.addDebugMessage,
    triggerErrorGlitch: () => {},
    setCliOutput: params.setCliOutput,
    setPendingPinConflicts: params.setPendingPinConflicts,
  });
  const controller = useSimulationController({
    code: "",
    hasCompilationErrors: false,
    isModified: params.isModified,
    capabilities: params.capabilities,
    ensureBackendConnected: params.ensureBackendConnected,
    sendMessage: params.sendMessage,
    sendMessageImmediate: params.sendMessageImmediate,
    resetPinUI: params.resetPinUI,
    clearOutputs: params.clearOutputs,
    serialEventQueueRef: params.serialEventQueueRef,
    pendingPinConflicts: params.pendingPinConflicts,
    startSimulationRef: params.startSimulationRef,
    uiFeedback,
  });
 
  const handleReset = useCallback(() => {
    if (params.capabilities && !params.capabilities.canSimulate) return;
    if (!params.ensureBackendConnected("Reset simulation")) return;
    params.clearOutputs();
    Eif (controller.simulationStatus === "running") controller.stopSimulationImmediately();
    controller.setSimulationStatus("idle");
    params.resetPinUI({ keepDetected: true });
    uiFeedback.showResettingToast();
    setTimeout(() => {
      params.handleCompileAndStart();
    }, 100);
  }, [controller, params, uiFeedback]);
 
  // mirror original return shape exactly
  return {
    simulationStatus: controller.simulationStatus,
    setSimulationStatus: controller.setSimulationStatus,
    hasCompiledOnce: controller.hasCompiledOnce,
    setHasCompiledOnce: controller.setHasCompiledOnce,
    simulationTimeout: controller.simulationTimeout,
    setSimulationTimeout: controller.setSimulationTimeout,
    startMutation: controller.startMutation,
    stopMutation: controller.stopMutation,
    pauseMutation: controller.pauseMutation,
    resumeMutation: controller.resumeMutation,
    handleStart: controller.handleStart,
    handleStop: controller.handleStop,
    handlePause: controller.handlePause,
    handleResume: controller.handleResume,
    handleReset,
    suppressAutoStopOnce: controller.suppressAutoStopOnce,
  };
}