add: implement basic server

src/main.rs

@@ -1,10 +1,14 @@
+use std::sync::Mutex;
+
 use clap::Parser;
 
-mod rewriter;
+use skubelb::Rewriter;
+use skubelb::Server;
 
 use env_logger::Env;
-use log::info;
-use rewriter::Rewriter;
+use log::{info, warn};
+use anyhow::Result;
+use rouille::{router, Request, Response};
 
 /// Implements a HTTP server which allows clients to 'register'
 /// themselves. Their IP address will be used to replace a
@@ -29,11 +33,20 @@ struct Args {
     /// The folder which contains the templates that
     /// will be be searched for the needle.
     #[arg(short, long)]
-    source_folder: String,
+    template_dir: String,
 
-    /// Where to write the replaced lines.
+    /// The symlink that should be updated each time the config changes.
+    /// 
+    /// Symlinks are used because file updates are not atomic.
     #[arg(short, long)]
-    dest_folder: String,
+    config_symlink: String,
+
+    /// Where to actually store the generated configs.
+    #[arg(short, long)]
+    workspace_dir: String,
+
+    #[arg(short, long, default_value_t = String::from("0.0.0.0:8080"))]
+    listen: String,
 }
 
 fn main() {
@@ -41,11 +54,31 @@ fn main() {
     env_logger::Builder::from_env(Env::default().default_filter_or("info")).init();
     let args = Args::parse();
 
-    let mut rewriter = Rewriter::new(args.rewrite_string);
+    let rewriter = Rewriter::new(args.rewrite_string);
+    let server_impl = Mutex::new(Server::new(rewriter, args.workspace_dir, args.template_dir, args.config_symlink));
 
-    rewriter.add_replacement(String::from("abc"));
-    rewriter
-        .rewrite_folder(&args.source_folder, &args.dest_folder)
-        .unwrap();
-    info!("Finished writing new config to {}", args.dest_folder);
+    rouille::start_server(args.listen, move |request| {
+        info!("Processing request: {:?}", request);
+        match handle(request, &server_impl) {
+            Ok(resp) => resp,
+            Err(e) => {
+                warn!("{:?}", e);
+                Response{status_code: 500, ..Response::empty_400()}
+            }
+        }
+    });
 }
+
+fn handle(request: &Request, server_impl: &Mutex<Server>) -> Result<Response> {
+    router!(request,
+        (POST) (/register) => {
+            server_impl.lock().unwrap().register(request)?;
+            Ok(Response{status_code: 200, ..Response::empty_204()})
+        },
+        (DELETE) (/register) => {
+            server_impl.lock().unwrap().unregister(request)?;
+            Ok(Response{status_code: 200, ..Response::empty_204()})
+        },
+        _ => Ok(Response::empty_404()),
+    )
+}