json works
This commit is contained in:
parent
2e3c3f9fff
commit
cbf63fabd5
5 changed files with 22 additions and 96 deletions
|
@ -9,30 +9,19 @@ use crossterm::{
|
|||
},
|
||||
event::Event,
|
||||
};
|
||||
use std::io::{self, Write};
|
||||
|
||||
#[tokio::main]
|
||||
pub async fn main() -> anyhow::Result<()> {
|
||||
let mut client = Client::connect("127.0.0.1", 8080).await?;
|
||||
let local_address = &client.stream.local_addr()?;
|
||||
|
||||
enable_raw_mode()?;
|
||||
execute!(
|
||||
io::stdout(),
|
||||
std::io::stdout(),
|
||||
Clear(ClearType::All),
|
||||
)?;
|
||||
|
||||
'get_instruction: loop {
|
||||
match TODO {
|
||||
todo!()
|
||||
}
|
||||
}
|
||||
|
||||
for i in 1..11 {
|
||||
let message = Message::new(lib::message::MessageKind::Move, format!("Hello toto x {}!!!", i));
|
||||
client.send_message(message.unwrap().clone()).await?;
|
||||
}
|
||||
|
||||
client.send_message(Message::new(MessageKind::Move, "EXIT").unwrap()).await?;
|
||||
client.send_message(Message::new(&local_address.to_string(), MessageKind::Test, "test json").unwrap()).await?;
|
||||
|
||||
disable_raw_mode()?;
|
||||
Ok(())
|
||||
|
|
|
@ -22,7 +22,8 @@ impl Client {
|
|||
}
|
||||
|
||||
pub async fn send_message(&mut self, message: Message) -> anyhow::Result<()> {
|
||||
self.stream.write_all(&message.encode()).await?;
|
||||
let message_string = message.encode()?;
|
||||
self.stream.write_all(&message_string.as_bytes()).await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
|
|
|
@ -1,10 +1,5 @@
|
|||
use anyhow::Ok;
|
||||
use serde::{Deserialize, Serialize};
|
||||
|
||||
// use crate::constant::{MESSAGE_KIND_CREATE_LOBBY, MESSAGE_KIND_CREATE_PLAYER, MESSAGE_KIND_MOVE, METADATA_SIZE};
|
||||
|
||||
// everything here needs to be adjusted for the deserialized json
|
||||
//
|
||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||
pub struct Message {
|
||||
pub from: String,
|
||||
|
@ -13,9 +8,9 @@ pub struct Message {
|
|||
pub content: String,
|
||||
}
|
||||
|
||||
// idk i just want to layout something, really didn't think much
|
||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||
pub enum MessageKind {
|
||||
Test,
|
||||
Move,
|
||||
CreatePlayer,
|
||||
CreateLobby,
|
||||
|
@ -36,47 +31,14 @@ impl Message {
|
|||
}
|
||||
|
||||
pub fn encode(&self) -> anyhow::Result<String> {
|
||||
// let mut buffer = Vec::new();
|
||||
// buffer.extend(self.from.as_bytes());
|
||||
// buffer.extend()
|
||||
// buffer.extend(&self.length.to_be_bytes());
|
||||
// buffer.extend(self.content.as_bytes());
|
||||
|
||||
let message_json: String = serde_json::to_string(self)?;
|
||||
|
||||
Ok(message_json)
|
||||
}
|
||||
|
||||
pub fn decode(buffer: &[u8]) -> anyhow::Result<Self> {
|
||||
// if buffer.len() < METADATA_SIZE {
|
||||
// return Err(anyhow::anyhow!("Invalid message length"));
|
||||
// }
|
||||
|
||||
// let message_kind = MessageKind::try_from(buffer[0]).expect("Error parsing the MessageKind");
|
||||
// let length = u16::from_be_bytes([buffer[1], buffer[2]]);
|
||||
// let content = String::from_utf8(buffer[3..3 + length as usize].to_vec())?;
|
||||
|
||||
// Ok(Self {
|
||||
// message_kind,
|
||||
// length,
|
||||
// content,
|
||||
// })
|
||||
|
||||
let message: Message = serde_json::from_slice(buffer)?;
|
||||
|
||||
Ok(message)
|
||||
}
|
||||
}
|
||||
|
||||
// impl TryFrom<u8> for MessageKind {
|
||||
// type Error = &'static str;
|
||||
|
||||
// fn try_from(value: u8) -> Result<Self, Self::Error> {
|
||||
// match value {
|
||||
// MESSAGE_KIND_MOVE => Ok(MessageKind::Move),
|
||||
// MESSAGE_KIND_CREATE_PLAYER => Ok(MessageKind::CreatePlayer),
|
||||
// MESSAGE_KIND_CREATE_LOBBY => Ok(MessageKind::CreateLobby),
|
||||
// _ => Err("invalid MessageKind byte")
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
|
|
|
@ -1,6 +1,7 @@
|
|||
// add message kind everywhere is needed
|
||||
use crate::{constant::METADATA_SIZE, message::Message};
|
||||
use crate::message::Message;
|
||||
|
||||
// this won't be needed in the near future, i want
|
||||
// to do everything within Message
|
||||
pub struct MessageReader {
|
||||
pub buffer: Vec<u8>,
|
||||
}
|
||||
|
@ -10,33 +11,10 @@ impl MessageReader {
|
|||
Self { buffer: Vec::new() }
|
||||
}
|
||||
|
||||
fn can_parse(&self) -> bool {
|
||||
if self.buffer.len() < METADATA_SIZE {
|
||||
return false;
|
||||
}
|
||||
|
||||
let length = u16::from_be_bytes([self.buffer[0], self.buffer[1]]);
|
||||
self.buffer.len() >= METADATA_SIZE + length as usize
|
||||
}
|
||||
|
||||
fn parse_first(&mut self) -> anyhow::Result<Message> {
|
||||
let length = u16::from_be_bytes([self.buffer[0], self.buffer[1]]);
|
||||
let message_length = METADATA_SIZE + length as usize;
|
||||
let message = self.buffer[..message_length].to_vec();
|
||||
self.buffer = self.buffer[message_length..].to_vec();
|
||||
|
||||
Message::decode(&message)
|
||||
}
|
||||
|
||||
pub fn read(&mut self, data: &[u8]) -> anyhow::Result<Vec<Message>> {
|
||||
pub fn read(&mut self, data: &[u8]) -> anyhow::Result<Message> {
|
||||
self.buffer.extend_from_slice(data);
|
||||
let mut data = vec![];
|
||||
let message = serde_json::from_slice(&self.buffer)?;
|
||||
|
||||
while self.can_parse() {
|
||||
let message = self.parse_first()?;
|
||||
data.push(message);
|
||||
}
|
||||
|
||||
Ok(data)
|
||||
Ok(message)
|
||||
}
|
||||
}
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
use tokio::{io::AsyncReadExt, net::TcpListener};
|
||||
use tokio::{io::{AsyncReadExt, AsyncWriteExt}, net::TcpListener};
|
||||
use std::net::SocketAddr;
|
||||
use crate::{message::Message, message_read::MessageReader};
|
||||
|
||||
|
@ -31,22 +31,18 @@ impl Server {
|
|||
tokio::task::spawn(async move {
|
||||
let mut message_reader = MessageReader::new();
|
||||
|
||||
let (mut read, mut write) = socket.split();
|
||||
let (mut rd, mut wr) = socket.split();
|
||||
|
||||
'handler: loop {
|
||||
let mut buffer = vec![0; 256];
|
||||
let bytes_read = read.read(&mut buffer).await?;
|
||||
|
||||
let messages = message_reader.read(&buffer[..bytes_read])?;
|
||||
|
||||
for message in messages {
|
||||
if message.content == "EXIT" {
|
||||
println!("Connection closed by the client");
|
||||
break 'handler;
|
||||
}
|
||||
|
||||
println!("Message:\n{:?}", message);
|
||||
let mut buffer = vec![];
|
||||
if tokio::io::copy(&mut rd, &mut buffer).await.is_err() {
|
||||
eprintln!("failed to copy");
|
||||
}
|
||||
|
||||
let message = message_reader.read(&buffer)?;
|
||||
println!("{:?}", message.content);
|
||||
|
||||
break 'handler;
|
||||
}
|
||||
|
||||
Ok::<(), anyhow::Error>(())
|
||||
|
|
Loading…
Add table
Reference in a new issue