Add option to serve a custom index html file

crycode-de · crycode-de · commit be23ca8edf28 · 2023-09-01T14:32:54.000+02:00
diff --git a/README.md b/README.md
@@ -57,6 +57,7 @@ corresponding environment variable will be ignored.
 | `--path-regexp` | `PATH_REGEXP` | A regular expression to verify a given upload path. This should be set with care, because it may allow write access to outside the upload directory. | `/^[a-zA-Z0-9-_/]*$/` |
 | `--disable-auto-port` | `DISABLE_AUTO_PORT` | Disable automatic port increase if the port is nor available. | Not set. |
 | `--enable-folder-creation` | `ENABLE_FOLDER_CREATION` | Enable automatic folder creation when uploading file to non-existent folder. | Not set. |
+| `--index-file` | `INDEX_FILE` | Use a custom html file as index instead of the default internal index. If used, the form fields need to have the same names as in the original index. | Not set. |
 | `--help`, `-h` | | Show some help text | |
 
 Examples:
diff --git a/http-server-upload.js b/http-server-upload.js
@@ -25,6 +25,7 @@ let token = process.env.TOKEN || false;
 let pathMatchRegExp = (process.env.PATH_REGEXP) ? new RegExp(process.env.PATH_REGEXP) : /^[a-zA-Z0-9-_/]*$/;
 let maxFileSize = (parseInt(process.env.MAX_FILE_SIZE, 10) || 200) * 1024 * 1024;
 let enableFolderCreation = !!process.env.ENABLE_FOLDER_CREATION;
+let indexFile = process.env.INDEX_FILE || false;
 
 console.log('HTTP Server Upload');
 
@@ -65,6 +66,10 @@ Argument | Environmen variable
   Disable automatic port increase if the port is nor available. [Not set]
 --enable-folder-creation | ENABLE_FOLDER_CREATION
   Enable automatic folder creation when uploading file to non-existent folder. [Not set]
+--index-file | INDEX_FILE
+  Use a custom html file as index instead of the default internal index.
+  If used, the form fields need to have the same names as in
+  the original index. [Not set]
 --help or -h
   Show this help text.
 
@@ -135,6 +140,9 @@ while (myArgs.length > 0) {
       case '--max-file-size':
         maxFileSize = (parseInt(val, 10) || 200) * 1024 * 1024;
         break;
+      case '--index-file':
+        indexFile = val;
+        break;
 
       default:
         console.warn(`WANRING: Unknown command line argument: ${key}`);
@@ -277,6 +285,19 @@ server.on('request', async (req, res) => {
   } else {
     // show index page
     res.writeHead(200, { 'Content-Type': 'text/html; charset=utf-8' });
+
+    if (indexFile) {
+      // try to load custom index file
+      try {
+        const index = await fs.readFile(indexFile, { encoding: 'utf-8' });
+        res.write(index);
+        return res.end();
+      } catch (err) {
+        console.log(`Error serving custom index file! ${err}`);
+      }
+    }
+
+    // default index file
     res.write(`<!DOCTYPE html>
 <html lang="en">
 <head>
