Initial Commit

src/database.rs

@@ -0,0 +1,17 @@
+use mongodb::Client;
+use std::env;
+
+use once_cell::sync::OnceCell;
+static DBCONN: OnceCell<Client> = OnceCell::new();
+
+pub fn connect() {
+	let client = Client::with_uri_str(
+			&env::var("DB_URI").expect("DB_URI not in environment variables!"))
+		.expect("Failed to init db connection.");
+
+	DBCONN.set(client).unwrap();
+}
+
+pub fn get_connection() -> &'static Client {
+	DBCONN.get().unwrap()
+}

src/main.rs

@@ -0,0 +1,14 @@
+#![feature(proc_macro_hygiene, decl_macro)]
+#[macro_use] extern crate rocket;
+
+pub mod database;
+pub mod routes;
+
+use dotenv;
+
+fn main() {
+	dotenv::dotenv().ok();
+	database::connect();
+
+	routes::mount(rocket::ignite()).launch();
+}

src/routes/account.rs

@@ -0,0 +1,22 @@
+use crate::database;
+use bson::{ bson, doc, ordered::OrderedDocument };
+
+#[get("/")]
+pub fn root() -> String {
+	let client = database::get_connection();
+	let cursor = client.database("revolt").collection("users").find(None, None).unwrap();
+
+	let results: Vec<Result<OrderedDocument, mongodb::error::Error>> = cursor.collect();
+
+	format!("ok boomer, users: {}", results.len())
+}
+
+#[get("/reg")]
+pub fn reg() -> String {
+	let client = database::get_connection();
+	let col = client.database("revolt").collection("users");
+
+	col.insert_one(doc! { "username": "test" }, None).unwrap();
+
+	format!("inserted")
+}

src/routes/mod.rs

@@ -0,0 +1,8 @@
+use rocket::Rocket;
+
+mod account;
+
+pub fn mount(rocket: Rocket) -> Rocket {
+	rocket
+		.mount("/api/v1", routes![account::root, account::reg])
+}