1 files changed, 13 insertions, 0 deletions
diff --git a/stacks/uri-rewrite.in b/stacks/uri-rewrite.in
new file mode 100644
index 0000000..6df707c
--- /dev/null
+++ b/stacks/uri-rewrite.in
@@ -0,0 +1,13 @@
+[+ autogen5 template -*- mode: json -*- +]
+{"Fn::Join": ["\n", [
+    "'use strict';",
+    "exports.handler = (event, context, callback) => {",
+    "    var whitelist = ['css', 'html', 'jpg', 'svg', 'png', 'txt', 'xml', 'pdf'];",
+    "    var request = event.Records[0].cf.request;",
+    "    var extension = request.uri.split('.').pop();",
+    "    if (typeof extension == 'undefined' || !whitelist.includes(extension)) {",
+    "        request.uri = request.uri.replace(/\\\/?$/, '\/index.html');",
+    "    }",
+    "    return callback(null, request);",
+    "};"
+]]}