Serve up the background image via the websocket

This commit is contained in:
Savanni D'Gerinel 2024-11-18 23:32:54 -05:00
parent 54162d0072
commit 45275be11b
6 changed files with 38 additions and 15 deletions

View File

@ -8,7 +8,7 @@ use std::{
use tokio::sync::mpsc::{unbounded_channel, UnboundedReceiver, UnboundedSender};
use uuid::Uuid;
use crate::types::Message;
use crate::types::{AppError, Message, PlayArea};
#[derive(Debug)]
struct WebsocketClient {
@ -19,6 +19,8 @@ struct WebsocketClient {
pub struct AppState {
pub image_base: PathBuf,
pub playfield_background: String,
pub clients: HashMap<String, WebsocketClient>,
}
@ -29,6 +31,7 @@ impl Core {
pub fn new() -> Self {
Self(Arc::new(RwLock::new(AppState {
image_base: PathBuf::from("/home/savanni/Pictures"),
playfield_background: "moon.jpg".to_owned(),
clients: HashMap::new(),
})))
}
@ -95,6 +98,15 @@ impl Core {
.collect()
}
pub fn set_playfield_background(&self, path: String) -> Result<(), AppError> {
{
let mut state = self.0.write().unwrap();
state.playfield_background = path.clone();
}
self.publish(Message::PlayArea(PlayArea{ background_image: path }));
Ok(())
}
pub fn publish(&self, message: Message) {
let state = self.0.read().unwrap();

View File

@ -98,13 +98,14 @@ pub async fn handle_connect_websocket(
ws.on_upgrade(move |socket| {
let core = core.clone();
async move {
let (mut ws_sender, mut ws_recv) = socket.split();
let (mut ws_sender, _) = socket.split();
let mut receiver = core.connect_client(client_id);
tokio::task::spawn(async move {
let background_image = core.0.read().unwrap().playfield_background.clone();
let _ = ws_sender
.send(Message::text(
serde_json::to_string(&crate::types::Message::Count(0)).unwrap(),
serde_json::to_string(&crate::types::Message::PlayArea(PlayArea{ background_image })).unwrap(),
))
.await;
while let Some(msg) = receiver.recv().await {

View File

@ -134,14 +134,13 @@ pub async fn main() {
move |ws, client_id| handle_connect_websocket(core.clone(), ws, client_id)
});
let route_publish = warp::path!("api" / "v1" / "message")
.and(warp::post())
let route_set_playfield_bg = warp::path!("api" / "v1" / "playfield" / "bg")
.and(warp::put())
.and(warp::body::json())
.map({
let core = core.clone();
move |body| {
println!("route_publish: {:?}", body);
core.publish(body);
core.set_playfield_background(body);
warp::reply()
}
});
@ -152,7 +151,7 @@ pub async fn main() {
.or(route_playing_field)
.or(route_image)
.or(route_available_images)
.or(route_publish)
.or(route_set_playfield_bg)
.recover(handle_rejection);
let server = warp::serve(filter);

View File

@ -12,9 +12,9 @@ pub enum AppError {
}
#[derive(Clone, Debug, Deserialize, Serialize)]
#[serde(tag = "type")]
pub enum Message {
Count(u32),
// PlayArea(PlayArea),
PlayArea(PlayArea),
}

View File

@ -26,7 +26,7 @@ const App = ({ client }: AppProps) => {
},
{
path: "/",
element: websocketUrl ? <WebsocketPlayingFieldComponent websocketUrl={websocketUrl} /> : <div> </div>
element: websocketUrl ? <WebsocketPlayingFieldComponent client={client} websocketUrl={websocketUrl} /> : <div> </div>
}
]);
return (

View File

@ -26,15 +26,26 @@ export const PlayingFieldComponent = ({ client }: PlayingFieldProps) => {
*/
interface WebsocketPlayingFieldProps {
client: Client;
websocketUrl: string;
}
export const WebsocketPlayingFieldComponent = ({ websocketUrl }: WebsocketPlayingFieldProps) => {
type Message = |
{
type: "PlayArea";
background_image: string;
}
export const WebsocketPlayingFieldComponent = ({ client, websocketUrl }: WebsocketPlayingFieldProps) => {
const { lastMessage, readyState } = useWebSocket(websocketUrl);
const [backgroundUrl, setBackgroundUrl] = useState<URL | undefined>(undefined);
useEffect(() => {
if (lastMessage !== null) {
console.log("last message: ", lastMessage);
const message: Message = JSON.parse(lastMessage.data);
console.log("playing area: ", message);
console.log("playing area: ", message.background_image);
setBackgroundUrl(client.imageUrl(message.background_image));
}
}, [lastMessage]);
@ -46,11 +57,11 @@ export const WebsocketPlayingFieldComponent = ({ websocketUrl }: WebsocketPlayin
[ReadyState.UNINSTANTIATED]: 'Uninstantiated',
}[readyState];
return <PlayingFieldComponent backgroundUrl={undefined} connectionStatus={connectionStatus} />;
return <PlayingFieldComponent backgroundUrl={backgroundUrl} connectionStatus={connectionStatus} />;
}
interface PlayingFieldProps {
backgroundUrl: string | undefined;
backgroundUrl: URL | undefined;
connectionStatus: string;
}