rustfmt

rustfmt.toml

@@ -0,0 +1,2 @@
+tab_spaces = 2
+use_field_init_shorthand = true

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
 }