From c36432152b07e0f7e8f6477c24ce3fc3f434f8cc Mon Sep 17 00:00:00 2001
From: dangered wolf <d@ngeredwolf.me>
Date: Thu, 17 Nov 2022 01:37:57 -0500
Subject: [PATCH] Apply shorter cache TTL for ongoing polls

---
 src/server.ts        | 8 +++++++-
 src/status.ts        | 9 ++++++++-
 src/types/types.d.ts | 1 +
 3 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/src/server.ts b/src/server.ts
index 55be8b1..73ebca1 100644
--- a/src/server.ts
+++ b/src/server.ts
@@ -108,9 +108,15 @@ const statusRequest = async (
         return Response.redirect(`${Constants.TWITTER_ROOT}/${handle}/status/${id}`, 302);
       }
 
+      let headers = Constants.RESPONSE_HEADERS;
+
+      if (statusResponse.cacheControl) {
+        headers = { ...headers, 'cache-control': statusResponse.cacheControl };
+      }
+
       /* Return the response containing embed information */
       return new Response(statusResponse.text, {
-        headers: Constants.RESPONSE_HEADERS,
+        headers: headers,
         status: 200
       });
     } else {
diff --git a/src/status.ts b/src/status.ts
index 5036f54..554456c 100644
--- a/src/status.ts
+++ b/src/status.ts
@@ -81,6 +81,7 @@ export const handleStatus = async (
   const engagementText = authorText.replace(/ {4}/g, ' ');
   let siteName = Constants.BRANDING_NAME;
   let newText = tweet.text;
+  let cacheControl: string | null = null;
 
   /* Base headers included in all responses */
   const headers = [
@@ -273,6 +274,11 @@ export const handleStatus = async (
     /* Finally, add the footer of the poll with # of votes and time left */
     str += `\n${poll.total_votes} votes · ${poll.time_left_en}`;
 
+    /* Check if the poll is ongoing and apply low TTL cache control */
+    if (poll.time_left_en !== 'Final results') {
+      cacheControl = Constants.POLL_TWEET_CACHE;
+    }
+
     /* And now we'll put the poll right after the Tweet text! */
     newText += `\n\n${str}`;
   }
@@ -338,6 +344,7 @@ export const handleStatus = async (
     text: Strings.BASE_HTML.format({
       lang: `lang="${lang}"`,
       headers: headers.join('')
-    })
+    }),
+    cacheControl: cacheControl
   };
 };
diff --git a/src/types/types.d.ts b/src/types/types.d.ts
index c89ea29..6cf28fc 100644
--- a/src/types/types.d.ts
+++ b/src/types/types.d.ts
@@ -11,6 +11,7 @@ type InputFlags = {
 interface StatusResponse {
   text?: string;
   response?: Response;
+  cacheControl?: string | null;
 }
 
 interface Request {