From 9473ebe7a0d95100dad89a759f482494d6047021 Mon Sep 17 00:00:00 2001
From: BitHeaven <bit.heaven@bk.ru>
Date: Sat, 16 Mar 2024 13:34:13 +0500
Subject: [PATCH] More struct

---
 src/main.rs                           | 52 +++++++++++++++++----------
 src/types.rs                          |  2 ++
 src/{type_sites.rs => types/sites.rs} |  0
 src/{type_users.rs => types/users.rs} |  0
 4 files changed, 35 insertions(+), 19 deletions(-)
 create mode 100644 src/types.rs
 rename src/{type_sites.rs => types/sites.rs} (100%)
 rename src/{type_users.rs => types/users.rs} (100%)

diff --git a/src/main.rs b/src/main.rs
index 8cb12ab..ab28e6e 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,5 +1,4 @@
-mod type_users;
-mod type_sites;
+mod types;
 
 use {
 	std::{
@@ -80,8 +79,8 @@ use {
 		Pool,
 	},
 	crate::{
-		type_users::Users,
-		type_sites::Sites,
+		types::users::Users,
+		types::sites::Sites,
 	},
 };
 
@@ -159,6 +158,7 @@ const CSS3: &str = r#"<style>
 	footer { text-align: right; }
 </style>"#;
 
+
 #[tokio::main]
 async fn main() -> Result<()> {
 	let addr = SocketAddr::from(([0, 0, 0, 0], PORT));
@@ -214,17 +214,22 @@ async fn handle_connection(req: Request<Incoming>, pool: DBPool, ip: String) ->
 		_ => ""
 	};
 
-	println!("{}", token);
+	match <str as AsRef<str>>::as_ref(req.uri().path()) {
+		x if x.starts_with("/api/") => {}
+		_ => {
+			println!("{}", token);
 
-	if token != "" && jwt_verify(pool.clone(), token)
-		.await?.claims.as_object().unwrap().len() == 0
-	{
-		println!("Invalid suka");
-		parts.status = StatusCode::FOUND;
-		set_cookie(&mut headers, "token", "");
-		set_location(&mut headers, "/");
-		parts.headers = headers;
-		return Ok(Response::from_parts(parts, Full::new(Bytes::new())));
+			if token != "" && jwt_verify(pool.clone(), token)
+				.await?.claims.as_object().unwrap().len() == 0
+			{
+				println!("Invalid suka");
+				parts.status = StatusCode::FOUND;
+				set_cookie(&mut headers, "token", "");
+				set_location(&mut headers, "/");
+				parts.headers = headers;
+				return Ok(Response::from_parts(parts, Full::new(Bytes::new())));
+			}
+		}
 	}
 
 	(body, parts.status, restype) = match req.uri().path().as_ref() {
@@ -234,7 +239,7 @@ async fn handle_connection(req: Request<Incoming>, pool: DBPool, ip: String) ->
 		"/register" => uri_register(req, pool.clone(), &mut headers).await?,
 		"/recover" => uri_recover(),
 		x if x.starts_with("/@") => uri_user(req, pool.clone()).await?,
-		x if x.starts_with("/api") => uri_api(req),
+		x if x.starts_with("/api/") => uri_api(req),
 		_ => uri_404()
 	};
 
@@ -337,6 +342,7 @@ fn uri_api(req: Request<Incoming>) -> (String, StatusCode, HeaderValue) {
 	let uri: &str = req.uri().path().as_ref();
 	let res: Json = match &uri[4..uri.len()] {
 		"/test" => json!({"error": false, "msg": "test"}),
+		x if x.starts_with("/v0/") => uri_api_v0(req),
 		_ => json!({"error": true, "msg": "No endpoint"})
 	};
 	
@@ -344,6 +350,14 @@ fn uri_api(req: Request<Incoming>) -> (String, StatusCode, HeaderValue) {
 	(res.to_string(), StatusCode::IM_A_TEAPOT, restype)
 }
 
+fn uri_api_v0(req: Request<Incoming>) -> Json {
+	let uri: &str = req.uri().path().as_ref();
+	match &uri[7..uri.len()] {
+		"/test" => json!({"error": false, "msg": "test endpoint v0"}),
+		_ => json!({"error": true, "msg": "No endpoint"})
+	}	
+}
+
 async fn uri_user(req: Request<Incoming>, pool: DBPool) -> Res<(String, StatusCode, HeaderValue), SkyError> {
 	let uri: &str = req.uri().path().as_ref();
 	let login = &uri[2..uri.len()];
@@ -542,10 +556,10 @@ fn rsa_gen() -> (Vec<u8>, Vec<u8>) {
 	let skey = RsaPrivateKey::new(&mut rng, bits).expect("RSA err");
 	let pkey = RsaPublicKey::from(&skey);
 
-	let skey = skey.to_pkcs1_der().unwrap().as_bytes().to_vec();
-	let pkey = pkey.to_pkcs1_der().unwrap().as_bytes().to_vec();
-
-	(skey, pkey)
+	(
+		skey.to_pkcs1_der().unwrap().as_bytes().to_vec(),
+		pkey.to_pkcs1_der().unwrap().as_bytes().to_vec()
+	)
 }
 
 async fn jwt_sign(pool: DBPool, data: Json) -> Result<String> {
diff --git a/src/types.rs b/src/types.rs
new file mode 100644
index 0000000..ac618f9
--- /dev/null
+++ b/src/types.rs
@@ -0,0 +1,2 @@
+pub mod users;
+pub mod sites;
diff --git a/src/type_sites.rs b/src/types/sites.rs
similarity index 100%
rename from src/type_sites.rs
rename to src/types/sites.rs
diff --git a/src/type_users.rs b/src/types/users.rs
similarity index 100%
rename from src/type_users.rs
rename to src/types/users.rs