Custom serve directory

Author: daxpedda

src/main.rs

@@ -19,12 +19,14 @@ fn main() -> Result<(), anyhow::Error> {
     let title = std::env::var("CARGO_PKG_NAME").unwrap_or_else(|_| "".to_string());
     let address =
         std::env::var("WASM_SERVER_RUNNER_ADDRESS").unwrap_or_else(|_| "127.0.0.1".to_string());
+    let directory =
+        std::env::var("WASM_SERVER_RUNNER_DIRECTORY").unwrap_or_else(|_| String::from("."));
     let https =
         std::env::var("WASM_SERVER_RUNNER_HTTPS").unwrap_or_else(|_| String::from("0")) == "1";
     let no_module =
         std::env::var("WASM_SERVER_RUNNER_NO_MODULE").unwrap_or_else(|_| String::from("0")) == "1";
 
-    let options = Options { title, address, https, no_module };
+    let options = Options { title, address, directory, https, no_module };
 
     let wasm_file = std::env::args()
         .nth(1)

src/server.rs

@@ -23,6 +23,7 @@ fn generate_version() -> String {
 pub struct Options {
     pub title: String,
     pub address: String,
+    pub directory: String,
     pub https: bool,
     pub no_module: bool,
 }
@@ -51,7 +52,8 @@ pub async fn run_server(options: Options, output: WasmBindgenOutput) -> Result<(
     };
     let html = html.replace("{{ TITLE }}", &options.title);
 
-    let serve_dir = get_service(ServeDir::new(".")).handle_error(internal_server_error);
+    let serve_dir =
+        get_service(ServeDir::new(options.directory)).handle_error(internal_server_error);
 
     let serve_wasm = || async move {
         ([("content-encoding", "br")], WithContentType("application/wasm", compressed_wasm))