0.2.0: Add support to route requests with curl user-agent

Signed-off-by: Ivan Bushchik <ivabus@ivabus.dev>

Cargo.toml

@@ -1,10 +1,10 @@
 [package]
 name = "urouter"
-version = "0.1.0"
+version = "0.2.0"
 edition = "2021"
 license = "MIT"
 repository = "https://github.com/ivabus/urouter"
-description = "Small router for (kinda) short domains (static fork of ivabus/aliurl)"
+description = "Small router for (kinda) short domains (fork of ivabus/aliurl for static routing)"
 
 
 [dependencies]

README.md

@@ -23,10 +23,18 @@ Edit `alias.json` and `cargo run`
     "uri": "uri2",
     "alias": "http://example.com",
     "is_url": true
+  },
+  {
+    "uri": "/",
+    "alias": "https://somecoolscript.sh",
+    "is_url": true,
+    "curl_only": true
   }
 ]
 ```
 
+`"curl_only"` thing for `curl https://url | sh` like scripts.
+
 ## License
 
 The project is licensed under the terms of the [MIT license](./LICENSE).%

src/main.rs

@@ -29,11 +29,12 @@ use rocket::http::Status;
 use std::cell::OnceCell;
 use std::path::PathBuf;
 
+use rocket::request::{FromRequest, Outcome};
 use rocket::response::content::RawText;
 use rocket::response::{Redirect, Responder};
+use rocket::Request;
 use serde::Deserialize;
 
-const INDEX_REDIRECT: &'static str = "https://ivabus.dev";
 const _ALIAS: &'static str = include_str!("../alias.json");
 static mut ALIAS: OnceCell<Vec<Alias>> = OnceCell::new();
 
@@ -42,6 +43,7 @@ struct Alias {
 	uri: String,
 	alias: String,
 	is_url: Option<bool>,
+	curl_only: Option<bool>,
 }
 
 #[derive(Responder)]
@@ -51,28 +53,61 @@ enum Response {
 	Status(Status),
 }
 
+struct UserAgent(String);
+
+#[derive(Debug)]
+enum UserAgentError {}
+
+#[rocket::async_trait]
+impl<'r> FromRequest<'r> for UserAgent {
+	type Error = UserAgentError;
+
+	async fn from_request(req: &'r Request<'_>) -> Outcome<Self, Self::Error> {
+		match req.headers().get_one("user-agent") {
+			Some(key) => Outcome::Success(UserAgent(key.to_string())),
+			_ => Outcome::Success(UserAgent("".to_string())),
+		}
+	}
+}
+
+fn get_return(alias: &Alias) -> Response {
+	return match alias.is_url {
+		Some(true) => Response::Redirect(Redirect::to(alias.alias.clone())),
+		_ => Response::Text(RawText(
+			smurf::io::read_file_to_str(&PathBuf::from(&alias.alias)).unwrap(),
+		)),
+	};
+}
+
 #[get("/<page>")]
-async fn get_page(page: String) -> Response {
+fn get_page(page: String, user_agent: UserAgent) -> Response {
 	let mut decoded_page = String::new();
 	url_escape::decode_to_string(page, &mut decoded_page);
 	let alias = unsafe { ALIAS.get().unwrap() };
-
+	let mut pages = Vec::new();
+	let curl_check = user_agent.0.contains("curl");
 	for i in alias {
 		if i.uri == decoded_page {
-			return match i.is_url {
-				Some(true) => Response::Redirect(Redirect::to(i.alias.clone())),
-				_ => Response::Text(RawText(
-					smurf::io::read_file_to_str(&PathBuf::from(&i.alias)).unwrap(),
-				)),
+			if (i.curl_only == Some(true) && curl_check.clone())
+				|| (i.curl_only != Some(true) && !curl_check.clone())
+			{
+				return get_return(i);
 			};
+			pages.push(i);
+		}
+	}
+	// Returning normal page (if  found) to curl users.
+	for i in pages {
+		if i.curl_only != Some(true) {
+			return get_return(i);
 		}
 	}
 	Response::Status(Status::NotFound)
 }
 
 #[get("/")]
-async fn get_index() -> Redirect {
-	Redirect::to(INDEX_REDIRECT)
+async fn index(user_agent: UserAgent) -> Response {
+	get_page("/".to_string(), user_agent)
 }
 
 #[rocket::main]
@@ -80,6 +115,6 @@ async fn main() -> Result<(), rocket::Error> {
 	unsafe {
 		ALIAS.set(serde_json::from_str(_ALIAS).unwrap()).unwrap_unchecked();
 	}
-	let _rocket = rocket::build().mount("/", routes![get_page, get_index]).launch().await?;
+	let _rocket = rocket::build().mount("/", routes![get_page, index]).launch().await?;
 	Ok(())
 }