use lto for release, better list
This commit is contained in:
parent
73712a1242
commit
3094454ad9
4 changed files with 51 additions and 19 deletions
|
@ -11,3 +11,6 @@ actix-web = "1.0"
|
||||||
serde = "1.0"
|
serde = "1.0"
|
||||||
serde_yaml = "0.8"
|
serde_yaml = "0.8"
|
||||||
handlebars = "2.0"
|
handlebars = "2.0"
|
||||||
|
|
||||||
|
[profile.release]
|
||||||
|
lto = true
|
||||||
|
|
|
@ -13,4 +13,3 @@ routes:
|
||||||
yt: "https://www.youtube.com/results?search_query={{query}}"
|
yt: "https://www.youtube.com/results?search_query={{query}}"
|
||||||
# Searches
|
# Searches
|
||||||
na: "https://nyaa.si/?f=0&c=1_2&q={{query}}"
|
na: "https://nyaa.si/?f=0&c=1_2&q={{query}}"
|
||||||
#
|
|
||||||
|
|
52
src/main.rs
52
src/main.rs
|
@ -20,26 +20,33 @@ routes:
|
||||||
g: "https://google.com/search?q={{query}}"
|
g: "https://google.com/search?q={{query}}"
|
||||||
"#;
|
"#;
|
||||||
|
|
||||||
|
static CONFIG_FILE: &str = "bunbun.toml";
|
||||||
|
|
||||||
|
#[get("/ls")]
|
||||||
|
fn list(data: Data<Arc<State>>) -> impl Responder {
|
||||||
|
HttpResponse::Ok().body(
|
||||||
|
data.renderer
|
||||||
|
.read()
|
||||||
|
.unwrap()
|
||||||
|
.render("list", &data.routes)
|
||||||
|
.unwrap(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Deserialize)]
|
#[derive(Deserialize)]
|
||||||
struct SearchQuery {
|
struct SearchQuery {
|
||||||
to: String,
|
to: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[get("/ls")]
|
|
||||||
fn list(data: Data<Arc<State>>) -> impl Responder {
|
|
||||||
let mut resp = String::new();
|
|
||||||
for (k, v) in data.routes.iter() {
|
|
||||||
resp.push_str(&format!("{}: {}\n", k, v));
|
|
||||||
}
|
|
||||||
HttpResponse::Ok().body(resp)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[get("/hop")]
|
#[get("/hop")]
|
||||||
fn hop(data: Data<Arc<State>>, query: Query<SearchQuery>) -> impl Responder {
|
fn hop(data: Data<Arc<State>>, query: Query<SearchQuery>) -> impl Responder {
|
||||||
let reg = Handlebars::new();
|
|
||||||
let mut raw_args = query.to.split_ascii_whitespace();
|
let mut raw_args = query.to.split_ascii_whitespace();
|
||||||
let command = raw_args.next();
|
let command = raw_args.next();
|
||||||
|
|
||||||
|
if command.is_none() {
|
||||||
|
return HttpResponse::NotFound().body("not found");
|
||||||
|
}
|
||||||
|
|
||||||
// Reform args into url-safe string (probably want to go thru an actual parser)
|
// Reform args into url-safe string (probably want to go thru an actual parser)
|
||||||
let mut args = String::new();
|
let mut args = String::new();
|
||||||
if let Some(first_arg) = raw_args.next() {
|
if let Some(first_arg) = raw_args.next() {
|
||||||
|
@ -50,10 +57,6 @@ fn hop(data: Data<Arc<State>>, query: Query<SearchQuery>) -> impl Responder {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if command.is_none() {
|
|
||||||
return HttpResponse::NotFound().body("not found");
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut template_args = HashMap::new();
|
let mut template_args = HashMap::new();
|
||||||
template_args.insert("query", args);
|
template_args.insert("query", args);
|
||||||
|
|
||||||
|
@ -61,7 +64,11 @@ fn hop(data: Data<Arc<State>>, query: Query<SearchQuery>) -> impl Responder {
|
||||||
Some(template) => HttpResponse::Found()
|
Some(template) => HttpResponse::Found()
|
||||||
.header(
|
.header(
|
||||||
header::LOCATION,
|
header::LOCATION,
|
||||||
reg.render_template(template, &template_args).unwrap(),
|
data.renderer
|
||||||
|
.read()
|
||||||
|
.unwrap()
|
||||||
|
.render_template(template, &template_args)
|
||||||
|
.unwrap(),
|
||||||
)
|
)
|
||||||
.finish(),
|
.finish(),
|
||||||
None => match &data.default_route {
|
None => match &data.default_route {
|
||||||
|
@ -77,7 +84,10 @@ fn hop(data: Data<Arc<State>>, query: Query<SearchQuery>) -> impl Responder {
|
||||||
HttpResponse::Found()
|
HttpResponse::Found()
|
||||||
.header(
|
.header(
|
||||||
header::LOCATION,
|
header::LOCATION,
|
||||||
reg.render_template(data.routes.get(route).unwrap(), &template_args)
|
data.renderer
|
||||||
|
.read()
|
||||||
|
.unwrap()
|
||||||
|
.render_template(data.routes.get(route).unwrap(), &template_args)
|
||||||
.unwrap(),
|
.unwrap(),
|
||||||
)
|
)
|
||||||
.finish()
|
.finish()
|
||||||
|
@ -126,7 +136,7 @@ struct State {
|
||||||
}
|
}
|
||||||
|
|
||||||
fn main() -> Result<(), Error> {
|
fn main() -> Result<(), Error> {
|
||||||
let config_file = match File::open("bunbun.toml") {
|
let config_file = match File::open(CONFIG_FILE) {
|
||||||
Ok(file) => file,
|
Ok(file) => file,
|
||||||
Err(_) => {
|
Err(_) => {
|
||||||
eprintln!("Unable to find a bunbun.toml file. Creating default!");
|
eprintln!("Unable to find a bunbun.toml file. Creating default!");
|
||||||
|
@ -136,7 +146,7 @@ fn main() -> Result<(), Error> {
|
||||||
.open("bunbun.toml")
|
.open("bunbun.toml")
|
||||||
.expect("Unable to write to directory!");
|
.expect("Unable to write to directory!");
|
||||||
fd.write_all(DEFAULT_CONFIG)?;
|
fd.write_all(DEFAULT_CONFIG)?;
|
||||||
File::open("bunbun.toml")?
|
File::open(CONFIG_FILE)?
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
@ -176,4 +186,10 @@ fn compile_templates() -> Handlebars {
|
||||||
)
|
)
|
||||||
.unwrap();
|
.unwrap();
|
||||||
handlebars
|
handlebars
|
||||||
|
.register_template_string(
|
||||||
|
"list",
|
||||||
|
String::from_utf8_lossy(include_bytes!("templates/list.hbs")),
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
|
handlebars
|
||||||
}
|
}
|
||||||
|
|
14
src/templates/list.hbs
Normal file
14
src/templates/list.hbs
Normal file
|
@ -0,0 +1,14 @@
|
||||||
|
<!DOCTYPE html>
|
||||||
|
<html>
|
||||||
|
<body>
|
||||||
|
<table>
|
||||||
|
<tr>
|
||||||
|
<th>Shortcut</th>
|
||||||
|
<th>Target</th>
|
||||||
|
</tr>
|
||||||
|
{{#each this}}
|
||||||
|
<tr><td>{{@key}}</td><td>{{this}}</td></tr>
|
||||||
|
{{/each}}
|
||||||
|
</table>
|
||||||
|
</body>
|
||||||
|
</html>
|
Loading…
Reference in a new issue