diff --git a/src/main.rs b/src/main.rs index 49137f3..305583c 100644 --- a/src/main.rs +++ b/src/main.rs @@ -51,6 +51,8 @@ impl RootSpanBuilder for DomainRootSpanBuilder { struct Cli { #[arg(short, long)] dev: bool, + #[arg(short, long)] + site: String, } #[actix_web::main] @@ -60,12 +62,12 @@ async fn main() -> std::io::Result<()> { let logs; let db; if Cli::parse().dev { - site = "./public"; + site = "./public".to_string(); logs = "/tmp/tfcsite"; temp = "/tmp/tfcsite"; db = "sqlite://./data.db"; } else { - site = "../public"; + site = Cli::parse().site; logs = "/storage/logs/tfcsite"; temp = "/tmp/tfcsite"; db = "sqlite:///storage/data.db";