From de00a92b88d7e173c5007e42179e94ae8251978c Mon Sep 17 00:00:00 2001
From: clerie <git@clerie.de>
Date: Sun, 8 Jan 2023 18:07:05 +0100
Subject: [PATCH] Fail on unspecified run mode

---
 src/main.rs | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 511ccad..0207a77 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -12,6 +12,7 @@ use std::str::FromStr;
 
 #[derive(PartialEq)]
 enum OperationMode {
+    None,
     Exporter,
     Validator,
 }
@@ -29,9 +30,9 @@ fn parse_nix_store_path(path: std::path::PathBuf) -> Result<(String, String), St
 #[tokio::main]
 async fn main() {
     let mut listen = String::from("[::]:9152");
-    let mut operationmode = OperationMode::Exporter;
+    let mut operationmode = OperationMode::None;
     let mut args = std::env::args();
-    args.next();
+    let name = args.next().unwrap();
     loop {
         let arg = if let Some(arg) = args.next() {
             arg
@@ -69,7 +70,10 @@ async fn main() {
     } else if operationmode == OperationMode::Validator {
         println!("Running NixOS Exporter in Validator mode");
         app = app.route("/metrics", get(check));
-    }
+    } else {
+        println!("Run mode not specified, do {} --help", name);
+        std::process::exit(1);
+    };
 
     let addr = SocketAddr::from_str(&listen).unwrap();
     println!("listening on http://{}", addr);