1 files changed, 44 insertions, 4 deletions

diff --git a/src/main.rs b/src/main.rs
index 4a5a8b1..02fdc2e 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,12 +1,13 @@
-use crate::config::CONFIG_INSTANCE;
-use crate::error::TwinHError;
+use crate::{config::CONFIG_INSTANCE, error::TwinHError};
 use hyper::{
     service::{make_service_fn, service_fn},
     Server,
 };
+use std::env;
 
 mod config;
 mod error;
+mod import;
 mod models;
 mod repo;
 mod routes;
@@ -14,14 +15,53 @@ mod templates;
 
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn std::error::Error>> {
-    let addr = CONFIG_INSTANCE.addr;
+    // handle non-config args
+    for arg in env::args().skip(1) {
+        match arg.as_str() {
+            "--create-db" => {
+                // create a fresh database and quit
+                repo::create_new_db()?;
+                return Ok(());
+            }
+            "--import" => {
+                // import CSV data into database
+                todo!();
+            }
+            "--help" | "-h" => {
+                // print help
+                print!(
+                    "twinh: a home-grown classic car parts catalog\n\
+                    \nUsage: twinh <dir> [options]\n\
+                    <dir>           your database directory (e.g. /var/db/twinh)\n\
+                    \nOptions:\n\
+                    --help | -h     prints this message and exits\n\
+                    --addr          an ip address to bind to (e.g. 127.0.0.1)\n\
+                    --port          a port to bind to (e.g. 5353)\n\
+                    --create-db     creates a fresh empty database; <dir> cannot exist yet\n\
+                    --import-cars   imports CSV car data into the database\n\
+                    --import-parts  imports CSV parts data into the database\n\
+                    "
+                );
+                return Ok(());
+            }
+            unknown => {
+                panic!("unknown option: {}", unknown);
+            }
+        };
+    }
+
+    // gather config
+    let bind_addr = CONFIG_INSTANCE.bind_addr;
 
+    // create primary listener
     let make_svc =
         make_service_fn(move |_conn| async { Ok::<_, TwinHError>(service_fn(routes::router)) });
 
-    let server = Server::bind(&addr).serve(make_svc);
+    // bind server
+    let server = Server::bind(&bind_addr).serve(make_svc);
     let graceful = server.with_graceful_shutdown(shutdown_signal());
 
+    // start and run until signal
     if let Err(e) = graceful.await {
         eprintln!("server error: {}", e);
     }