1 files changed, 236 insertions, 0 deletions
diff --git a/src/main.rs b/src/main.rs
new file mode 100644
index 0000000..56cd695
--- /dev/null
+++ b/src/main.rs
@@ -0,0 +1,236 @@
+#[macro_use] extern crate rocket;
+use rocket::serde::json::Json;
+use std::collections::HashMap;
+use rocket::State;
+use std::sync::Mutex;
+use std::time::SystemTime;
+
+
+#[derive(serde::Serialize)]
+struct Response<T, Y> (Result<T, Y>);
+
+
+#[get("/get_message/<id>")]
+fn get_message(id: usize, db: &State<SharedDB>) -> Json<Response<Message, &'static str>> {
+    let lock = db.sdb.lock().unwrap();
+    match lock.get_message(id) {
+        Some(n) => Json(Response(Ok(n))),
+        None => Json(Response(Err("Message not found"))),
+    }
+}
+#[get("/get_message_id_list/<id_start>/<id_end>")]
+fn rget_message_id_list(id_start: usize, id_end: usize, db: &State<SharedDB>) -> Json<Response<Vec<usize>, &'static str>> {
+    let lock = db.sdb.lock().unwrap();
+    let list = lock.get_message_id_list(id_start, id_end);
+
+    if list.len() == 0 {
+        Json(Response(Err("No messages")))
+    } else {
+        Json(Response(Ok(list)))
+    }
+}
+#[get("/send_message/<user_id>/<msg>")]
+fn send_message(user_id: usize, msg: String, db: &State<SharedDB>) -> Json<Response<Message, &'static str>> {
+    let mut lock = db.sdb.lock().unwrap();
+
+    let user = match lock.get_user(user_id) {
+        Some(n) => n,
+        None => return Json(Response(Err("User not found")))
+    };
+    let msg = Message{
+        sender: user.clone(),
+        message: msg,
+        date: Date::now(),
+        id: lock.get_next_message_id(),
+    };
+    lock.send_message(msg.clone());
+    Json(Response(Ok(msg)))
+}
+
+#[get("/get_user/<id>")]
+fn get_user(id: usize, db: &State<SharedDB>) -> Json<Response<User, &'static str>> {
+    let lock = db.sdb.lock().unwrap();
+    match lock.get_user(id) {
+        Some(n) => Json(Response(Ok(n))),
+        None => Json(Response(Err("User not found")))
+    }
+}
+#[get("/get_user_by_name/<name>")]
+fn get_user_by_name(name: String, db: &State<SharedDB>) -> Json<Response<Vec<User>, &'static str>> {
+    let lock = db.sdb.lock().unwrap();
+    match lock.get_user_by_name(name) {
+        Some(n) => Json(Response(Ok(n))),
+        None => Json(Response(Err("User not found")))
+    }
+}
+#[get("/create_user/<name>")]
+fn create_user(name: String, db: &State<SharedDB>) -> Json<Response<User, &'static str>> {
+    let mut lock = db.sdb.lock().unwrap();
+
+    Json(Response(lock.create_user(name)))
+}
+
+struct SharedDB {
+    sdb: Mutex<Database>
+}
+
+struct Database {
+    usermap: HashMap<usize, User>,
+    messagemap: HashMap<usize, Message>,
+}
+impl Database {
+    fn get_message(&self, id: usize) -> Option<Message> {
+        match self.messagemap.get(&id) {
+            Some(n) => Some(n.to_owned()),
+            None => None,
+        }
+    }
+    fn get_message_id_list(&self, id_start: usize, id_end: usize) -> Vec<usize> {
+        let mut message_id_vec: Vec<usize> = vec![];
+
+        for x in id_start..=id_end {
+            match self.messagemap.get(&x) {
+                Some(_) => message_id_vec.push(x),
+                None => (),
+            }
+        }
+
+        return message_id_vec;
+    }
+    fn get_next_message_id(&self) -> usize {
+        self.messagemap.len()
+    }
+    fn send_message(&mut self, msg: Message) {
+        self.messagemap.insert(msg.id, msg);
+        println!("{:?}", self.messagemap.get(&0));
+    }
+
+    fn get_user(&self, id: usize) -> Option<User> {
+        match self.usermap.get(&id) {
+            Some(n) => Some(n.to_owned()),
+            None => None
+        }
+    }
+    fn get_user_by_name(&self, name: String) -> Option<Vec<User>> {
+        let list: Vec<Option<User>> = self.usermap.values().map(|x| if x.username == name {Some(x.clone())} else {None}).collect();
+
+        let mut users = vec![];
+        for x in list {
+            match x {
+                Some(n) => users.push(n),
+                None => ()
+            }
+        }
+
+        if users.len() == 0 {
+            None
+        } else {
+            Some(users)
+        }
+    }
+
+    fn create_user(&mut self, name: String) -> Result<User, &'static str> {
+        let id = self.usermap.len();
+        let user = User::new(name.clone(), id);
+
+        match self.get_user_by_name(name) {
+            Some(_) => return Err("ID taken"),
+            None => (),
+        }
+
+        match self.usermap.get(&id) {
+            Some(_) => Err("ID taken"),
+            None => {self.usermap.insert(id, user.clone()); Ok(user)},
+        }
+    }
+
+    pub fn new() -> Database {
+        Database {
+            usermap: HashMap::new(),
+            messagemap: HashMap::new(),
+
+        }
+    }
+}
+
+#[derive(serde::Serialize, Clone, Debug)]
+struct Date {
+    date: u64,
+}
+impl Date {
+    pub fn now() -> Date {
+        Date{
+            date: SystemTime::now().duration_since(SystemTime::UNIX_EPOCH).unwrap().as_secs()
+        }
+    }
+}
+
+#[derive(serde::Serialize, Clone, Debug)]
+struct User {
+    username: String,
+    id: usize
+}
+impl User {
+    pub fn new(name: String, id: usize) -> User {
+        User{
+            username: name,
+            id: id
+        }
+    }
+}
+
+#[derive(serde::Serialize, Clone, Debug)]
+struct Message {
+    date: Date,
+    message: String,
+    sender: User,
+    id: usize,
+}
+
+#[derive(serde::Serialize)]
+struct Info {
+    name: String,
+    url: String,
+    users: usize,
+}
+impl Info {
+    pub fn get() -> Info {
+        Info{
+            name: String::from("Infernal Garden"),
+            url: String::from("https://wf.infernal.garden"),
+            users: 1,
+        }
+    }
+}
+
+#[get("/ping")]
+fn ping() -> Json<Response<&'static str, &'static str>> {
+    Json(Response(Ok("pong")))
+}
+
+#[get("/api")]
+fn api_index() -> Json<Info> {
+    Json(Info::get())
+}
+
+
+#[derive(serde::Serialize)]
+struct HitCount {
+    c: usize
+}
+
+#[launch]
+fn rocket() -> _ {
+    rocket::build()
+    .mount("/api", routes![
+        get_message,
+        send_message,
+        rget_message_id_list,
+        ping,
+        get_user,
+        create_user,
+        get_user_by_name
+    ])
+    .mount("/", routes![api_index])
+    .manage(SharedDB{sdb: Mutex::new(Database::new())})
+}