ChatKit-server/src/lib/server/client_management/client/mod.rs

// pub mod client_profile;
// pub mod client_v3;
pub mod traits;

use std::collections::HashMap;
use std::cmp::Ordering;
use std::net::TcpStream;
use std::sync::Mutex;
use std::sync::Arc;
use std::io::{BufReader, BufWriter};
use std::io::BufRead;

use uuid::Uuid;
use serde::{Serialize, Deserialize};
use crossbeam_channel::{Sender, Receiver, unbounded};

use traits::IClient;
use crate::lib::Foundation::{ICooperative, IMessagable};
use crate::lib::server::ServerMessages;

/// # ClientMessage
/// This enum defined the message that a client can receive from the server
/// This uses the serde library to transform to and from json. 
#[derive(Serialize, Deserialize)]
pub enum ClientMessage {
  Disconnect {id: String},
  Update {id: String},

  ServerMessage {id: String, msg: String},

  NewMessage {id: String, from_user_id: String, msg: String},
  NewgroupMessage {id: String, from_group_id: String, from_user_id: String, msg: String},
}

/// # ClientSocketMessage
/// This enum defines a message that can be sent from a client to the server once connected
/// This uses the serde library to transform to and from json. 
#[derive(Serialize, Deserialize)]
pub enum ClientSocketMessage {
  Disconnect {id: String},
  SendMessage {id: String, to_user_id: String, msg: String}
}

/// # Client
/// This struct represents a connected user.
/// 
/// ## Attrubutes
/// - uuid: The id of the connected user.
/// - username: The username of the connected user.
/// - address: The the address of the connected client.
/// 
/// - stream: The socket for the connected client.
/// - owner: An optional reference to the owning object.
#[derive(Debug, Serialize)]
pub struct Client {
  pub uuid: Uuid,
  username: String,
  address: String,

	// non serializable
	#[serde(skip)]
  server_channel: Option<Sender<ServerMessages>>,

  #[serde(skip)]
  input: Sender<ClientMessage>,

  #[serde(skip)]
  output: Receiver<ClientMessage>,

	#[serde(skip)]
  stream: Mutex<Option<TcpStream>>,

  #[serde(skip)]
  stream_reader: Mutex<Option<BufReader<TcpStream>>>,

  #[serde(skip)]
  stream_writer: Mutex<Option<BufWriter<TcpStream>>>,

}

// client funciton implmentations
impl IClient<ClientMessage> for Client {
  fn new(
		uuid: String,
		username: String,
		address: String,
		stream: TcpStream,
		server_channel: Sender<ServerMessages>
	) -> Arc<Client> {
    let (sender, receiver) = unbounded();

    let out_stream = stream.try_clone().unwrap();
    let in_stream = stream.try_clone().unwrap();

    Arc::new(Client {
      username,
      uuid: Uuid::parse_str(&uuid).expect("invalid id"),
      address,

      server_channel: Some(server_channel),

      input: sender,
      output: receiver,

      stream: Mutex::new(Some(stream)),

      stream_reader: Mutex::new(Some(BufReader::new(in_stream))),
      stream_writer: Mutex::new(Some(BufWriter::new(out_stream))),
    })
  }

	// MARK: - removeable
  fn send(&self, _bytes: Vec<u8>) -> Result<(), &str> { todo!() }
  fn recv(&self) -> Option<Vec<u8>> { todo!() }
	// Mark: end -
}

impl IMessagable<ClientMessage> for Client{
	fn send_message(&self, msg: ClientMessage) {
		self.input.send(msg).expect("failed to send message to client.");
	}
}

// cooperative multitasking implementation
impl ICooperative for Client {
	fn tick(&self) {
    // aquire locks (so value isn't dropped)
    let mut reader_lock = self.stream_reader.lock().unwrap();
    let mut writer_lock = self.stream_writer.lock().unwrap();

    // aquiring mutable buffers
    let reader = reader_lock.as_mut().unwrap();
    let _writer = writer_lock.as_mut().unwrap();

    // create buffer
    let mut buffer = String::new();

    // loop over all lines that have been sent.
    while let Ok(_size) = reader.read_line(&mut buffer) {
      let command = serde_json::from_str::<ClientSocketMessage>(buffer.as_str()).unwrap();

      match command {
        ClientSocketMessage::Disconnect {id} => println!("got Disconnect from id: {:?}", id),
        _ => println!("New command found"),
      }
    }


    // handle incomming messages

	}
}

// default value implementation
impl Default for Client {
	fn default() -> Self {
		let (sender, reciever) = unbounded();
		Client {
			username: "generic_client".to_string(),
      uuid: Uuid::new_v4(),
      address: "127.0.0.1".to_string(),

		  output: reciever,
			input: sender,

      server_channel: None,

      stream: Mutex::new(None),

      stream_reader: Mutex::new(None),
      stream_writer: Mutex::new(None),
		}
	}
}

// MARK: - used for sorting.
impl PartialEq for Client {
      fn eq(&self, other: &Self) -> bool {
        self.uuid == other.uuid
    }
}

impl Eq for Client {
}

impl PartialOrd for Client {
	fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
			Some(self.cmp(other))
	}
}

impl Ord for Client {
      fn cmp(&self, other: &Self) -> Ordering {
        self.uuid.cmp(&other.uuid)
    }
}