From 72d29ab1e930bfcacd3c091508d37a79aafc8526 Mon Sep 17 00:00:00 2001
From: Peter Cai <peter@typeblog.net>
Date: Thu, 20 Feb 2020 14:30:30 +0800
Subject: [PATCH] index: use ETag to facilitate client-side caching

---
 src/index.coffee | 23 +++++++++++++++++++----
 1 file changed, 19 insertions(+), 4 deletions(-)

diff --git a/src/index.coffee b/src/index.coffee
index 56f2105..12de1dc 100644
--- a/src/index.coffee
+++ b/src/index.coffee
@@ -1,4 +1,5 @@
 import * as util from './util'
+import * as crypto from './crypto'
 import S3 from './aws/s3'
 import config from '../config.json'
 import indexHtml from '../worker/index.html'
@@ -8,6 +9,8 @@ FRONTEND_PATHS = [
   '/paste/text/', '/paste/binary/'
 ]
 
+FRONTEND_SHA256 = null
+
 s3 = new S3 config
 
 main = ->
@@ -20,18 +23,30 @@ buildInvalidResponse = (msg) ->
   new Response msg,
     status: 400
 
-buildFrontendResponse = ->
+buildFrontendResponse = (req) ->
+  if req.headers.has "if-none-match"
+    if req.headers.get("if-none-match") == "W/" + FRONTEND_SHA256
+      # Skip this response if the frontend was not updated
+      return new Response null,
+        status: 304
+
   new Response indexHtml,
     status: 200
     headers:
       'content-type': 'text/html'
+      'etag': "W/" + FRONTEND_SHA256
 
 handleRequest = (event) ->
+  # Ensure we have a SHA256 value of frontend first
+  # This will be used as ETag
+  if not FRONTEND_SHA256
+    FRONTEND_SHA256 = crypto.hex await crypto.SHA256 indexHtml
+
   # Handle request for static home page first
   if event.request.method == "GET"
     parsedURL = new URL event.request.url
     if parsedURL.pathname in FRONTEND_PATHS
-      return buildFrontendResponse()
+      return buildFrontendResponse event.request 
 
   # Validate file name first, since this is shared logic
   file = util.getFileName event.request.url
@@ -90,7 +105,7 @@ handleGET = (req, file) ->
   else if req.url.endsWith "crypt"
     # We need frontend to handle encrypted files
     # The key is passed after the hash ('#'), unavailable to server
-    return buildFrontendResponse()
+    return buildFrontendResponse req
   # The full path to the original file
   fullPath = files.Contents[0].Key
   fileName = fullPath.split '/'
@@ -113,7 +128,7 @@ handleGET = (req, file) ->
     isText = util.isText resp.headers.get 'content-type'
     isBrowser = util.isBrowser req
     if isText and isBrowser
-      return buildFrontendResponse()
+      return buildFrontendResponse req 
 
   # Build response headers
   headers =