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

100% Statements 14/14
75% Branches 3/4
75% Functions 3/4
100% Lines 12/12

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                                                                                                                                  63x             63x                             63x 5x 4x 4x 4x 4x 4x 4x 2x         63x                                      
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";
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 = {
  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,
    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.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,
  };
}