Merge pull request #515 from sinking-point/sinkingpoint/add-spa-index-option-474

Add --spa-index option

1 files changed, 26 insertions, 10 deletions

diff --git a/src/main.rs b/src/main.rs
index c43ac95..205fac5 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,9 +1,11 @@
 use std::io;
 use std::io::Write;
 use std::net::{IpAddr, SocketAddr, TcpListener};
+use std::path::{Path, PathBuf};
 use std::thread;
 use std::time::Duration;
 
+use actix_files::NamedFile;
 use actix_web::web;
 use actix_web::{http::header::ContentType, Responder};
 use actix_web::{middleware, App, HttpRequest, HttpResponse};
@@ -95,15 +97,9 @@ async fn run(miniserve_config: MiniserveConfig) -> Result<(), ContextualError> {
         ContextualError::IoError("Failed to resolve path to be served".to_string(), e)
     })?;
 
-    if let Some(index_path) = &miniserve_config.index {
-        let has_index: std::path::PathBuf = [&canon_path, index_path].iter().collect();
-        if !has_index.exists() {
-            error!(
-                "The file '{}' provided for option --index could not be found.",
-                index_path.to_string_lossy()
-            );
-        }
-    }
+    check_file_exists(&canon_path, &miniserve_config.index, "index");
+    check_file_exists(&canon_path, &miniserve_config.spa_index, "spa-index");
+
     let path_string = canon_path.to_string_lossy();
 
     println!(
@@ -268,6 +264,20 @@ async fn run(miniserve_config: MiniserveConfig) -> Result<(), ContextualError> {
         .map_err(|e| ContextualError::IoError("".to_owned(), e))
 }
 
+fn check_file_exists(canon_path: &Path, file_option: &Option<PathBuf>, option_name: &str) {
+    if let Some(file_path) = file_option {
+        let file_path: &Path = file_path.as_ref();
+        let has_file: std::path::PathBuf = [canon_path, file_path].iter().collect();
+        if !has_file.exists() {
+            error!(
+                "The file '{}' provided for option --{} could not be found.",
+                file_path.to_string_lossy(),
+                option_name,
+            );
+        }
+    }
+}
+
 /// Allows us to set low-level socket options
 ///
 /// This mainly used to set `set_only_v6` socket option
@@ -300,6 +310,13 @@ fn configure_app(app: &mut web::ServiceConfig, conf: &MiniserveConfig) {
             Some(index_file) => files.index_file(index_file.to_string_lossy()),
             None => files,
         };
+        let files = match &conf.spa_index {
+            Some(spa_index_file) => files.default_handler(
+                NamedFile::open(&conf.path.join(spa_index_file))
+                    .expect("Cant open SPA index file."),
+            ),
+            None => files.default_handler(web::to(error_404)),
+        };
         let files = match conf.show_hidden {
             true => files.use_hidden_files(),
             false => files,
@@ -309,7 +326,6 @@ fn configure_app(app: &mut web::ServiceConfig, conf: &MiniserveConfig) {
             .files_listing_renderer(listing::directory_listing)
             .prefer_utf8(true)
             .redirect_to_slash_directory()
-            .default_handler(web::to(error_404))
     };
 
     if !conf.path.is_file() {