rustfmt
parent
64aec75659
commit
be505480c5
GPG Key ID:
F350507060ED6C90
|
|
@ -0,0 +1,2 @@
|
|||
tab_spaces = 2
|
||||
use_field_init_shorthand = true
|
||||
292
src/main.rs
292
src/main.rs
|
|
@ -1,8 +1,8 @@
|
|||
use actix_web::{
|
||||
get,
|
||||
http::header,
|
||||
web::{Data, Query},
|
||||
App, HttpResponse, HttpServer, Responder,
|
||||
get,
|
||||
http::header,
|
||||
web::{Data, Query},
|
||||
App, HttpResponse, HttpServer, Responder,
|
||||
};
|
||||
use handlebars::Handlebars;
|
||||
use hotwatch::{Event, Hotwatch};
|
||||
|
|
@ -18,184 +18,186 @@ static CONFIG_FILE: &str = "bunbun.toml";
|
|||
|
||||
#[get("/ls")]
|
||||
fn list(data: Data<Arc<RwLock<State>>>) -> impl Responder {
|
||||
let data = data.read().unwrap();
|
||||
HttpResponse::Ok().body(data.renderer.render("list", &data.routes).unwrap())
|
||||
let data = data.read().unwrap();
|
||||
HttpResponse::Ok().body(data.renderer.render("list", &data.routes).unwrap())
|
||||
}
|
||||
|
||||
#[derive(Deserialize)]
|
||||
struct SearchQuery {
|
||||
to: String,
|
||||
to: String,
|
||||
}
|
||||
|
||||
#[get("/hop")]
|
||||
fn hop(data: Data<Arc<RwLock<State>>>, query: Query<SearchQuery>) -> impl Responder {
|
||||
let data = data.read().unwrap();
|
||||
let mut raw_args = query.to.split_ascii_whitespace();
|
||||
let command = raw_args.next();
|
||||
let data = data.read().unwrap();
|
||||
let mut raw_args = query.to.split_ascii_whitespace();
|
||||
let command = raw_args.next();
|
||||
|
||||
if command.is_none() {
|
||||
return HttpResponse::NotFound().body("not found");
|
||||
if command.is_none() {
|
||||
return HttpResponse::NotFound().body("not found");
|
||||
}
|
||||
|
||||
// Reform args into url-safe string (probably want to go thru an actual parser)
|
||||
let mut args = String::new();
|
||||
if let Some(first_arg) = raw_args.next() {
|
||||
args.push_str(first_arg);
|
||||
for arg in raw_args {
|
||||
args.push_str("+");
|
||||
args.push_str(arg);
|
||||
}
|
||||
}
|
||||
|
||||
// Reform args into url-safe string (probably want to go thru an actual parser)
|
||||
let mut args = String::new();
|
||||
if let Some(first_arg) = raw_args.next() {
|
||||
args.push_str(first_arg);
|
||||
for arg in raw_args {
|
||||
args.push_str("+");
|
||||
args.push_str(arg);
|
||||
}
|
||||
}
|
||||
let mut template_args = HashMap::new();
|
||||
template_args.insert("query", args);
|
||||
|
||||
let mut template_args = HashMap::new();
|
||||
template_args.insert("query", args);
|
||||
|
||||
match data.routes.get(command.unwrap()) {
|
||||
Some(template) => HttpResponse::Found()
|
||||
.header(
|
||||
header::LOCATION,
|
||||
data.renderer
|
||||
.render_template(template, &template_args)
|
||||
.unwrap(),
|
||||
)
|
||||
.finish(),
|
||||
None => match &data.default_route {
|
||||
Some(route) => {
|
||||
template_args.insert(
|
||||
"query",
|
||||
format!(
|
||||
"{}+{}",
|
||||
command.unwrap(),
|
||||
template_args.get("query").unwrap()
|
||||
),
|
||||
);
|
||||
HttpResponse::Found()
|
||||
.header(
|
||||
header::LOCATION,
|
||||
data.renderer
|
||||
.render_template(data.routes.get(route).unwrap(), &template_args)
|
||||
.unwrap(),
|
||||
)
|
||||
.finish()
|
||||
}
|
||||
None => HttpResponse::NotFound().body("not found"),
|
||||
},
|
||||
}
|
||||
match data.routes.get(command.unwrap()) {
|
||||
Some(template) => HttpResponse::Found()
|
||||
.header(
|
||||
header::LOCATION,
|
||||
data
|
||||
.renderer
|
||||
.render_template(template, &template_args)
|
||||
.unwrap(),
|
||||
)
|
||||
.finish(),
|
||||
None => match &data.default_route {
|
||||
Some(route) => {
|
||||
template_args.insert(
|
||||
"query",
|
||||
format!(
|
||||
"{}+{}",
|
||||
command.unwrap(),
|
||||
template_args.get("query").unwrap()
|
||||
),
|
||||
);
|
||||
HttpResponse::Found()
|
||||
.header(
|
||||
header::LOCATION,
|
||||
data
|
||||
.renderer
|
||||
.render_template(data.routes.get(route).unwrap(), &template_args)
|
||||
.unwrap(),
|
||||
)
|
||||
.finish()
|
||||
}
|
||||
None => HttpResponse::NotFound().body("not found"),
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
#[get("/")]
|
||||
fn index(data: Data<Arc<RwLock<State>>>) -> impl Responder {
|
||||
let data = data.read().unwrap();
|
||||
let mut template_args = HashMap::new();
|
||||
template_args.insert("hostname", &data.public_address);
|
||||
HttpResponse::Ok().body(data.renderer.render("index", &template_args).unwrap())
|
||||
let data = data.read().unwrap();
|
||||
let mut template_args = HashMap::new();
|
||||
template_args.insert("hostname", &data.public_address);
|
||||
HttpResponse::Ok().body(data.renderer.render("index", &template_args).unwrap())
|
||||
}
|
||||
|
||||
#[get("/bunbunsearch.xml")]
|
||||
fn opensearch(data: Data<Arc<RwLock<State>>>) -> impl Responder {
|
||||
let data = data.read().unwrap();
|
||||
let mut template_args = HashMap::new();
|
||||
template_args.insert("hostname", &data.public_address);
|
||||
HttpResponse::Ok()
|
||||
.header(
|
||||
header::CONTENT_TYPE,
|
||||
"application/opensearchdescription+xml",
|
||||
)
|
||||
.body(data.renderer.render("opensearch", &template_args).unwrap())
|
||||
let data = data.read().unwrap();
|
||||
let mut template_args = HashMap::new();
|
||||
template_args.insert("hostname", &data.public_address);
|
||||
HttpResponse::Ok()
|
||||
.header(
|
||||
header::CONTENT_TYPE,
|
||||
"application/opensearchdescription+xml",
|
||||
)
|
||||
.body(data.renderer.render("opensearch", &template_args).unwrap())
|
||||
}
|
||||
|
||||
#[derive(Deserialize)]
|
||||
struct Config {
|
||||
bind_address: String,
|
||||
public_address: String,
|
||||
default_route: Option<String>,
|
||||
routes: BTreeMap<String, String>,
|
||||
bind_address: String,
|
||||
public_address: String,
|
||||
default_route: Option<String>,
|
||||
routes: BTreeMap<String, String>,
|
||||
}
|
||||
|
||||
struct State {
|
||||
public_address: String,
|
||||
default_route: Option<String>,
|
||||
routes: BTreeMap<String, String>,
|
||||
renderer: Handlebars,
|
||||
public_address: String,
|
||||
default_route: Option<String>,
|
||||
routes: BTreeMap<String, String>,
|
||||
renderer: Handlebars,
|
||||
}
|
||||
|
||||
fn main() -> Result<(), Error> {
|
||||
let conf = read_config(CONFIG_FILE)?;
|
||||
let renderer = compile_templates();
|
||||
let state = Arc::from(RwLock::new(State {
|
||||
public_address: conf.public_address,
|
||||
default_route: conf.default_route,
|
||||
routes: conf.routes,
|
||||
renderer,
|
||||
}));
|
||||
let state_ref = state.clone();
|
||||
let conf = read_config(CONFIG_FILE)?;
|
||||
let renderer = compile_templates();
|
||||
let state = Arc::from(RwLock::new(State {
|
||||
public_address: conf.public_address,
|
||||
default_route: conf.default_route,
|
||||
routes: conf.routes,
|
||||
renderer,
|
||||
}));
|
||||
let state_ref = state.clone();
|
||||
|
||||
let mut watch =
|
||||
Hotwatch::new_with_custom_delay(Duration::from_millis(500)).expect("Failed to init watch");
|
||||
let mut watch =
|
||||
Hotwatch::new_with_custom_delay(Duration::from_millis(500)).expect("Failed to init watch");
|
||||
|
||||
watch
|
||||
.watch(CONFIG_FILE, move |e: Event| {
|
||||
if let Event::Write(_) = e {
|
||||
let mut state = state.write().unwrap();
|
||||
let conf = read_config(CONFIG_FILE).unwrap();
|
||||
state.public_address = conf.public_address;
|
||||
state.default_route = conf.default_route;
|
||||
state.routes = conf.routes;
|
||||
}
|
||||
})
|
||||
.expect("failed to watch");
|
||||
|
||||
HttpServer::new(move || {
|
||||
App::new()
|
||||
.data(state_ref.clone())
|
||||
.service(hop)
|
||||
.service(list)
|
||||
.service(index)
|
||||
.service(opensearch)
|
||||
watch
|
||||
.watch(CONFIG_FILE, move |e: Event| {
|
||||
if let Event::Write(_) = e {
|
||||
let mut state = state.write().unwrap();
|
||||
let conf = read_config(CONFIG_FILE).unwrap();
|
||||
state.public_address = conf.public_address;
|
||||
state.default_route = conf.default_route;
|
||||
state.routes = conf.routes;
|
||||
}
|
||||
})
|
||||
.bind(&conf.bind_address)?
|
||||
.run()
|
||||
.expect("failed to watch");
|
||||
|
||||
HttpServer::new(move || {
|
||||
App::new()
|
||||
.data(state_ref.clone())
|
||||
.service(hop)
|
||||
.service(list)
|
||||
.service(index)
|
||||
.service(opensearch)
|
||||
})
|
||||
.bind(&conf.bind_address)?
|
||||
.run()
|
||||
}
|
||||
|
||||
fn read_config(config_file_path: &str) -> Result<Config, Error> {
|
||||
let config_file = match File::open(config_file_path) {
|
||||
Ok(file) => file,
|
||||
Err(_) => {
|
||||
eprintln!(
|
||||
"Unable to find a {} file. Creating default!",
|
||||
config_file_path
|
||||
);
|
||||
let mut fd = OpenOptions::new()
|
||||
.write(true)
|
||||
.create_new(true)
|
||||
.open(config_file_path)
|
||||
.expect("Unable to write to directory!");
|
||||
fd.write_all(DEFAULT_CONFIG)?;
|
||||
File::open(config_file_path)?
|
||||
}
|
||||
};
|
||||
Ok(serde_yaml::from_reader(config_file).unwrap())
|
||||
let config_file = match File::open(config_file_path) {
|
||||
Ok(file) => file,
|
||||
Err(_) => {
|
||||
eprintln!(
|
||||
"Unable to find a {} file. Creating default!",
|
||||
config_file_path
|
||||
);
|
||||
let mut fd = OpenOptions::new()
|
||||
.write(true)
|
||||
.create_new(true)
|
||||
.open(config_file_path)
|
||||
.expect("Unable to write to directory!");
|
||||
fd.write_all(DEFAULT_CONFIG)?;
|
||||
File::open(config_file_path)?
|
||||
}
|
||||
};
|
||||
Ok(serde_yaml::from_reader(config_file).unwrap())
|
||||
}
|
||||
|
||||
fn compile_templates() -> Handlebars {
|
||||
let mut handlebars = Handlebars::new();
|
||||
handlebars
|
||||
.register_template_string(
|
||||
"index",
|
||||
String::from_utf8_lossy(include_bytes!("templates/index.hbs")),
|
||||
)
|
||||
.unwrap();
|
||||
handlebars
|
||||
.register_template_string(
|
||||
"opensearch",
|
||||
String::from_utf8_lossy(include_bytes!("templates/bunbunsearch.xml")),
|
||||
)
|
||||
.unwrap();
|
||||
handlebars
|
||||
.register_template_string(
|
||||
"list",
|
||||
String::from_utf8_lossy(include_bytes!("templates/list.hbs")),
|
||||
)
|
||||
.unwrap();
|
||||
handlebars
|
||||
let mut handlebars = Handlebars::new();
|
||||
handlebars
|
||||
.register_template_string(
|
||||
"index",
|
||||
String::from_utf8_lossy(include_bytes!("templates/index.hbs")),
|
||||
)
|
||||
.unwrap();
|
||||
handlebars
|
||||
.register_template_string(
|
||||
"opensearch",
|
||||
String::from_utf8_lossy(include_bytes!("templates/bunbunsearch.xml")),
|
||||
)
|
||||
.unwrap();
|
||||
handlebars
|
||||
.register_template_string(
|
||||
"list",
|
||||
String::from_utf8_lossy(include_bytes!("templates/list.hbs")),
|
||||
)
|
||||
.unwrap();
|
||||
handlebars
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue