add subdir prefix from config

jdub233 · Jun 26, 2019 · b5b299d · b5b299d
1 parent d53411b
commit b5b299d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/config.example.yml b/config.example.yml
@@ -1,5 +1,7 @@
 CAPTURE_URL: "https://www.bu.edu"
 
+SUBDIR_PREFIX: ""
+
 #S3 Bucket details
 S3_BUCKET_NAME: "static-page-bucket"
 S3_PATH: "page"
diff --git a/handler.js b/handler.js
@@ -5,16 +5,19 @@ const Url = require('url-parse');
 
 const captureURL = new Url(process.env.CAPTURE_URL);
 
+// Allow for a prefix to the subdirectory, and add a slash if it is set.
+const subDirPrefix = (process.env.SUBDIR_PREFIX !== '') ? `${process.env.SUBDIR_PREFIX}/` : '';
+
 const scrapeOptions = {
     urls: [`${process.env.CAPTURE_URL}?cachebust=${Date.now()}`],
     urlFilter: function(url) {
         return url.indexOf(`${captureURL.protocol}//${captureURL.host}`) === 0;
     },
     directory: `/tmp/page-capture/capture-${Date.now()}`,
     subdirectories: [
-        {directory: 'img', extensions: ['.jpg', '.png', '.svg', '.gif', '.mp4', '.webm', '.mov']},
-        {directory: 'js', extensions: ['.js']},
-        {directory: 'css', extensions: ['.css']}
+        {directory: `${subDirPrefix}img`, extensions: ['.jpg', '.png', '.svg', '.gif', '.mp4', '.webm', '.mov']},
+        {directory: `${subDirPrefix}js`, extensions: ['.js']},
+        {directory: `${subDirPrefix}css`, extensions: ['.css']}
     ],
 };