mirror of
https://github.com/ivabus/urouter
synced 2024-11-21 16:05:12 +03:00
0.2.0: Add support to route requests with curl user-agent
Signed-off-by: Ivan Bushchik <ivabus@ivabus.dev>
This commit is contained in:
parent
26fa8d9aa4
commit
fc1ebcbe60
3 changed files with 56 additions and 13 deletions
|
@ -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]
|
||||
|
|
|
@ -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).%
|
57
src/main.rs
57
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(())
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue