src/structs.rs - master - ivabus/urouter

src/structs.rs
99 строк2 KB
Ivan Bushchik
0.6.1: Add "html" alias type, fix --alias-file
21 янв 2024, 08:36
21 янв 2024, 08:36653b41a
/* SPDX-License-Identifier: MIT */

use clap::Parser;
use rocket::http::{ContentType, Status};
use rocket::request::{FromRequest, Outcome};
use rocket::response::content::{RawHtml, RawText};
use rocket::response::{Redirect, Responder};
use rocket::Request;
use serde::Deserialize;
use std::collections::HashMap;
use std::net::IpAddr;
use std::path::PathBuf;

#[derive(Parser, Debug)]
#[command(about, author)]
pub struct Args {
	#[arg(long)]
	pub alias_file: Option<PathBuf>,

	/// For internal usage
	#[arg(long, default_value = "false")]
	pub alias_file_is_set_not_a_list: bool,

	/// Dir to lookup file alias
	#[arg(long, default_value = ".")]
	pub dir: PathBuf,

	#[arg(short, long, default_value = "127.0.0.1")]
	pub address: IpAddr,

	#[arg(short, long, default_value = "8080")]
	pub port: u16,
}

// For better compatability with Nix (with set on the top of alias.json instead of a list)
#[derive(Deserialize, Clone, Debug)]
pub struct NixJson {
	pub alias: Vec<Alias>,
}

#[derive(Deserialize, Clone, Debug)]
pub struct Alias {
	pub uri: String,
	pub alias: AliasType,
	pub agent: Option<Agent>,
}

#[derive(Deserialize, Clone, Debug)]
pub enum AliasType {
	#[serde(alias = "url")]
	Url(String),
	#[serde(alias = "file")]
	File(String),
	#[serde(alias = "text")]
	Text(String),
	#[serde(alias = "html")]
	Html(String),
	#[serde(alias = "external")]
	External(External),
}

#[derive(Deserialize, Clone, Debug)]
pub struct Agent {
	pub regex: String,
	pub only_matching: Option<bool>,
}

#[derive(Deserialize, Clone, Debug)]
pub struct External {
	pub url: String,
	#[serde(default)]
	pub headers: HashMap<String, String>,
}

#[derive(Responder)]
pub enum Response {
	Text(Box<RawText<String>>),
	Html(Box<RawHtml<String>>),
	Redirect(Box<Redirect>),
	Status(Status),
	Custom(Box<(ContentType, RawText<String>)>),
}

pub struct UserAgent(pub String);

#[derive(Debug)]
pub 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())),
		}
	}
}