From c39799fb67aa0ccc28ad0ffe5006f473ed38bf86 Mon Sep 17 00:00:00 2001
From: Hongbo Wu <hongbo@omnivore.app>
Date: Mon, 10 Jul 2023 15:54:21 +0800
Subject: [PATCH] update subscription lastFetchedAt after fetching all items

---
 packages/rss-handler/src/index.ts | 71 ++++++++++++++++++++++++++++++-
 1 file changed, 69 insertions(+), 2 deletions(-)

diff --git a/packages/rss-handler/src/index.ts b/packages/rss-handler/src/index.ts
index 895af2855..b85fe9179 100644
--- a/packages/rss-handler/src/index.ts
+++ b/packages/rss-handler/src/index.ts
@@ -66,6 +66,66 @@ const sendSavePageMutation = async (userId: string, input: unknown) => {
   }
 }
 
+const sendUpdateSubscriptionMutation = async (
+  userId: string,
+  subscriptionId: string,
+  lastFetchedAt: Date
+) => {
+  const JWT_SECRET = process.env.JWT_SECRET
+  const REST_BACKEND_ENDPOINT = process.env.REST_BACKEND_ENDPOINT
+
+  if (!JWT_SECRET || !REST_BACKEND_ENDPOINT) {
+    throw 'Environment not configured correctly'
+  }
+
+  const data = JSON.stringify({
+    query: `mutation UpdateSubscription($input: UpdateSubscriptionInput!){
+      updateSubscription(input:$input){
+        ... on UpdateSubscriptionSuccess{
+          subscription{
+            id
+            lastFetchedAt
+          }
+        }
+        ... on UpdateSubscriptionError{
+            errorCodes
+        }
+      }
+    }`,
+    variables: {
+      input: {
+        id: subscriptionId,
+        lastFetchedAt,
+      },
+    },
+  })
+
+  const auth = (await signToken({ uid: userId }, JWT_SECRET)) as string
+  try {
+    const response = await axios.post(
+      `${REST_BACKEND_ENDPOINT}/graphql`,
+      data,
+      {
+        headers: {
+          Cookie: `auth=${auth};`,
+          'Content-Type': 'application/json',
+        },
+        timeout: 30000, // 30s
+      }
+    )
+
+    /* eslint-disable @typescript-eslint/no-unsafe-member-access */
+    return !!response.data.data.savePage
+  } catch (error) {
+    if (axios.isAxiosError(error)) {
+      console.error('update subscription mutation error', error.message)
+    } else {
+      console.error(error)
+    }
+    return false
+  }
+}
+
 dotenv.config()
 Sentry.GCPFunction.init({
   dsn: process.env.SENTRY_DSN,
@@ -91,7 +151,7 @@ export const rssHandler = Sentry.GCPFunction.wrapHttpFunction(
       const { userId, feedUrl } = req.body
       // fetch feed
       const feed = await parser.parseURL(feedUrl)
-      console.debug(feed.title)
+      console.log('Fetched feed', feed.title)
 
       // save each item in the feed
       await Promise.all(
@@ -111,6 +171,7 @@ export const rssHandler = Sentry.GCPFunction.wrapHttpFunction(
           }
 
           try {
+            console.log('Saving page', input.title)
             // save page
             return sendSavePageMutation(userId, input)
           } catch (error) {
@@ -119,7 +180,13 @@ export const rssHandler = Sentry.GCPFunction.wrapHttpFunction(
         })
       )
 
-      // TODO: update subscription lastFetchedAt
+      // update subscription lastFetchedAt
+      const updatedSubscription = await sendUpdateSubscriptionMutation(
+        userId,
+        req.body.subscriptionId,
+        new Date()
+      )
+      console.log('Updated subscription', updatedSubscription)
 
       res.send('ok')
     } catch (e) {