Provide preliminary state machine model (#39)

- Create basic `StateMachine` which periodically emits sample data
  - Handle basic transition logic, to be improved later
  - An extra global state variable still needs to be cleaned up
- Send socket messages from pod action buttons in control station

---------

Co-authored-by: Sam Der <[email protected]>

Author: ryescholin

control-station/src/components/ControlPanel/ControlPanel.tsx

@@ -1,12 +1,22 @@
 import "./ControlPanel.css";
+import usePodData from "@/services/usePodData";
 
 function ControlPanel() {
+	const { podSocketClient } = usePodData();
 	return (
 		<div className="controlpanel">
-			<button className="button start">Start</button>
-			<button className="button stop">Stop</button>
-			<button className="button force">Force Stop</button>
-			<button className="button load">Load</button>
+			<button className="button start" onClick={() => podSocketClient.sendStart()}>
+				Start
+			</button>
+			<button className="button stop" onClick={() => podSocketClient.sendStop()}>
+				Stop
+			</button>
+			<button className="button force" onClick={() => podSocketClient.sendForcestop()}>
+				Force Stop
+			</button>
+			<button className="button load" onClick={() => podSocketClient.sendLoad()}>
+				Load
+			</button>
 		</div>
 	);
 }

control-station/src/services/PodSocketClient.ts

@@ -6,10 +6,20 @@ interface ServerToClientEvents {
 	connect: () => void;
 	disconnect: (reason: Socket.DisconnectReason) => void;
 	pong: (data: string) => void;
+	greet: (data: string) => void;
+	stop: (data: string) => void;
+	forcestop: (data: string) => void;
+	load: (data: string) => void;
+	start: (data: string) => void;
 }
 
 interface ClientToServerEvents {
 	ping: (data: string, ack: (data: string) => void) => void;
+	greet: (data: string, ack: (data: string) => void) => void;
+	stop: (data: string, ack: (data: string) => void) => void;
+	forcestop: (data: string, ack: (data: string) => void) => void;
+	load: (data: string, ack: (data: string) => void) => void;
+	start: (data: string, ack: (data: string) => void) => void;
 }
 
 export interface PodData {
@@ -34,7 +44,12 @@ class PodSocketClient {
 		this.serverEvents = {
 			connect: this.onConnect.bind(this),
 			disconnect: this.onDisconnect.bind(this),
-			pong: this.onPong.bind(this),
+			pong: this.onData.bind(this),
+			greet: this.onData.bind(this),
+			stop: this.onData.bind(this),
+			forcestop: this.onData.bind(this),
+			load: this.onData.bind(this),
+			start: this.onData.bind(this),
 		} as const;
 		this.setPodData = setPodData;
 	}
@@ -64,6 +79,36 @@ class PodSocketClient {
 		});
 	}
 
+	sendGreet(): void {
+		this.socket.emit("greet", "Hello from client", (ack: string) => {
+			console.log(`Server responds to greet with ${ack}`);
+		});
+	}
+
+	sendStop(): void {
+		this.socket.emit("stop", "Hello from client", (ack: string) => {
+			console.log(`Server responds to stop with ${ack}`);
+		});
+	}
+
+	sendForcestop(): void {
+		this.socket.emit("forcestop", "Hello from client", (ack: string) => {
+			console.log(`Server responds to stop with ${ack}`);
+		});
+	}
+
+	sendLoad(): void {
+		this.socket.emit("load", "Hello from client", (ack: string) => {
+			console.log(`Server responds to stop with ${ack}`);
+		});
+	}
+
+	sendStart(): void {
+		this.socket.emit("start", "Hello from client", (ack: string) => {
+			console.log(`Server responds to stop with ${ack}`);
+		});
+	}
+
 	private onConnect(): void {
 		console.log("Connected to server as", this.socket.id);
 		this.setPodData((d) => ({ ...d, connected: true }));
@@ -74,7 +119,7 @@ class PodSocketClient {
 		this.setPodData((d) => ({ ...d, connected: false }));
 	}
 
-	private onPong(data: string): void {
+	private onData(data: string): void {
 		console.log("server says", data);
 	}
 }

control-station/src/views/Dashboard/Dashboard.tsx

@@ -9,7 +9,10 @@ function Dashboard() {
 			<h1>Dashboard</h1>
 			<Status />
 			<p>{podData.connected ? "connected" : "disconnected"}</p>
-			<button onClick={() => podSocketClient.sendPing()}>Send Ping</button>
+			<button onClick={() => podSocketClient.sendStop()}>Send Stop</button>
+			<button onClick={() => podSocketClient.sendForcestop()}>Send Forcestop</button>
+			<button onClick={() => podSocketClient.sendLoad()}>Send Load</button>
+			<button onClick={() => podSocketClient.sendStart()}>Send Start</button>
 		</div>
 	);
 }

pod-operation/Cargo.lock

@@ -13,5 +13,6 @@ serde = { version = "1.0.192", features = ["derive"] }
 serde_json = "1.0.108"
 tracing = "0.1.40"
 tracing-subscriber = { version = "0.3.17", features = ["env-filter"] }
+lazy_static = "1.4.0"
 rppal = { version = "0.17.1", features = ["hal"] }
 ina219 = "0.1.0"

pod-operation/Cargo.toml

@@ -3,7 +3,7 @@ use tracing::info;
 use crate::components::pressure_transducer::PressureTransducer;
 use crate::components::signal_light::SignalLight;
 
-pub async fn blink(mut signal_light: SignalLight) {
+pub async fn _blink(mut signal_light: SignalLight) {
 	let mut i = 0;
 
 	info!("Starting blink demo.");

pod-operation/src/demo.rs

@@ -1,35 +1,34 @@
+use crate::components::pressure_transducer::PressureTransducer;
 use axum::Server;
-use socketioxide::{extract::SocketRef, SocketIo};
+use socketioxide::SocketIo;
 use tracing::{error, info};
 use tracing_subscriber::FmtSubscriber;
-
 mod components;
 mod demo;
-mod handlers;
-
-use crate::components::pressure_transducer::PressureTransducer;
-use crate::components::signal_light::SignalLight;
+mod state_machine;
+use crate::state_machine::StateMachine;
+use std::{thread, time::Duration};
 
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn std::error::Error>> {
 	tracing::subscriber::set_global_default(FmtSubscriber::default())?;
 
 	let (layer, io) = SocketIo::new_layer();
 
-	io.ns("/control-station", |socket: SocketRef| {
-		info!("Socket.IO connected: {:?} {:?}", socket.ns(), socket.id);
-		socket.on("ping", handlers::handle_ping);
-	});
-
-	let signal_light = SignalLight::new();
-	tokio::spawn(demo::blink(signal_light));
+	thread::sleep(Duration::from_millis(1));
 
 	let pressure_transducer = PressureTransducer::new(0x40);
 	tokio::spawn(demo::read_pressure_transducer(pressure_transducer));
 
+	thread::spawn(move || {
+		let mut state_machine = StateMachine::new(io);
+		state_machine.run();
+	});
+
 	let app = axum::Router::new().layer(layer);
 
 	info!("Starting server on port 5000");
+
 	let server = Server::bind(&"127.0.0.1:5000".parse().unwrap()).serve(app.into_make_service());
 
 	if let Err(e) = server.await {