try fixing state machine

telemetry/packages/constants/src/pods/states.ts

@@ -1,4 +1,4 @@
-export type PodStateType = keyof typeof ALL_POD_STATES;
+export type PodStateType = (typeof ALL_POD_STATES)[keyof typeof ALL_POD_STATES];
 
 export const FAILURE_STATES = {
   FAILURE_BRAKING: 'kFailureBrake',
@@ -47,11 +47,11 @@ export const ALL_POD_STATE_TYPES = [
 export type PodStateCategoryType = (typeof ALL_POD_STATE_TYPES)[number];
 
 export const getStateType = (
-  state: string,
+  state: any,
 ): (typeof ALL_POD_STATE_TYPES)[number] => {
-  if (FAILURE_STATES[state as keyof typeof FAILURE_STATES]) return 'FAILURE';
-  if (PASSIVE_STATES[state as keyof typeof PASSIVE_STATES]) return 'PASSIVE';
-  if (ACTIVE_STATES[state as keyof typeof ACTIVE_STATES]) return 'ACTIVE';
-  if (NULL_STATES[state as keyof typeof NULL_STATES]) return 'NULL';
+  if (Object.values(FAILURE_STATES).includes(state)) return 'FAILURE';
+  if (Object.values(PASSIVE_STATES).includes(state)) return 'PASSIVE';
+  if (Object.values(ACTIVE_STATES).includes(state)) return 'ACTIVE';
+  if (Object.values(NULL_STATES).includes(state)) return 'NULL';
   throw new Error(`Unknown state: ${state}`);
 };

telemetry/packages/server/src/modules/common/utils/zodEnumFromObjKeys.ts

@@ -6,3 +6,10 @@ export function zodEnumFromObjKeys<K extends string>(
   const [firstKey, ...otherKeys] = Object.keys(obj) as K[];
   return z.enum([firstKey, ...otherKeys]);
 }
+
+export function zodEnumFromObjValues<V extends string>(
+  obj: Record<string, V>,
+): z.ZodEnum<[V, ...V[]]> {
+  const [firstValue, ...otherValues] = Object.values(obj);
+  return z.enum([firstValue, ...otherValues]);
+}

telemetry/packages/server/src/modules/state/StateUpdate.types.ts

@@ -1,11 +1,14 @@
 import { ALL_POD_STATES, pods } from '@hyped/telemetry-constants';
-import { zodEnumFromObjKeys } from '@/modules/common/utils/zodEnumFromObjKeys';
+import {
+  zodEnumFromObjKeys,
+  zodEnumFromObjValues,
+} from '@/modules/common/utils/zodEnumFromObjKeys';
 import { z } from 'zod';
 
 export const StateUpdateSchema = z.object({
   podId: zodEnumFromObjKeys(pods),
   timestamp: z.string(), // to handle nanoseconds timestamp
-  value: zodEnumFromObjKeys(ALL_POD_STATES),
+  value: zodEnumFromObjValues(ALL_POD_STATES),
 });
 
 export type StateUpdate = z.infer<typeof StateUpdateSchema>;

telemetry/packages/ui/app/components/views/state-machine/index.tsx

@@ -527,7 +527,11 @@ export function StateMachine() {
       nodes.filter((node) =>
         isEnabledState(
           currentMode,
-          node.id.replace(/-/g, '_').toUpperCase() as PodStateType,
+          ALL_POD_STATES[
+            node.id
+              .replace(/-/g, '_')
+              .toUpperCase() as keyof typeof ALL_POD_STATES
+          ],
         ),
       ),
     );

telemetry/packages/ui/app/context/pods.tsx

@@ -207,8 +207,9 @@ export const PodsProvider = ({ children }: { children: React.ReactNode }) => {
     function subscribeToMqttAndLatency() {
       if (!client) return;
       const processMessage = (podId: PodId, topic: string, message: Buffer) => {
-        if (topic === getTopic('state', podId)) {
-          const newPodState = message.toString();
+        if (topic === getTopic('state/state', podId)) {
+          const newPodState = JSON.parse(message.toString())
+            .state as PodStateType;
           const allowedStates = Object.values(ALL_POD_STATES);
           if ((allowedStates as string[]).includes(newPodState)) {
             setPodsState((prevState) => ({
@@ -275,7 +276,7 @@ export const PodsProvider = ({ children }: { children: React.ReactNode }) => {
       // subscribe to latency messages and add MQTT message callback for each pod
       POD_IDS.map((podId) => {
         subscribe('latency/response', podId);
-        subscribe('state', podId);
+        subscribe('state/state', podId);
         client.on('message', (topic, message) =>
           processMessage(podId, topic, message),
         );