From 6a47bfac02f34b3caf489c3138faf11b2a0f0397 Mon Sep 17 00:00:00 2001
From: Hongbo Wu <hongbo@omnivore.app>
Date: Thu, 21 Mar 2024 12:19:12 +0800
Subject: [PATCH] add long polling to check exporter state

---
 .../mutations/exportToIntegrationMutation.ts  | 10 +++++-
 .../pages/settings/integrations/notion.tsx    | 36 ++++++++++++++++---
 2 files changed, 40 insertions(+), 6 deletions(-)

diff --git a/packages/web/lib/networking/mutations/exportToIntegrationMutation.ts b/packages/web/lib/networking/mutations/exportToIntegrationMutation.ts
index ac8ec9591..96b375c6b 100644
--- a/packages/web/lib/networking/mutations/exportToIntegrationMutation.ts
+++ b/packages/web/lib/networking/mutations/exportToIntegrationMutation.ts
@@ -1,9 +1,17 @@
 import { gql } from 'graphql-request'
 import { gqlFetcher } from '../networkHelpers'
 
+export enum TaskState {
+  Cancelled = 'CANCELLED',
+  Failed = 'FAILED',
+  Pending = 'PENDING',
+  Running = 'RUNNING',
+  Succeeded = 'SUCCEEDED'
+}
+
 export interface Task {
   id: string
-  state: string
+  state: TaskState
   createdAt: Date
   name: string
   runningTime: number
diff --git a/packages/web/pages/settings/integrations/notion.tsx b/packages/web/pages/settings/integrations/notion.tsx
index 294a3b8fb..1514349b0 100644
--- a/packages/web/pages/settings/integrations/notion.tsx
+++ b/packages/web/pages/settings/integrations/notion.tsx
@@ -12,15 +12,20 @@ import 'antd/dist/antd.compact.css'
 import { CheckboxValueType } from 'antd/lib/checkbox/Group'
 import Image from 'next/image'
 import { useRouter } from 'next/router'
-import { useEffect } from 'react'
+import { useEffect, useState } from 'react'
 import { HStack, VStack } from '../../../components/elements/LayoutPrimitives'
 import { PageMetaData } from '../../../components/patterns/PageMetaData'
 import { Beta } from '../../../components/templates/Beta'
 import { Header } from '../../../components/templates/settings/SettingsTable'
 import { SettingsLayout } from '../../../components/templates/SettingsLayout'
 import { deleteIntegrationMutation } from '../../../lib/networking/mutations/deleteIntegrationMutation'
-import { exportToIntegrationMutation } from '../../../lib/networking/mutations/exportToIntegrationMutation'
+import {
+  exportToIntegrationMutation,
+  Task,
+  TaskState,
+} from '../../../lib/networking/mutations/exportToIntegrationMutation'
 import { setIntegrationMutation } from '../../../lib/networking/mutations/setIntegrationMutation'
+import { apiFetcher } from '../../../lib/networking/networkHelpers'
 import { useGetIntegrationQuery } from '../../../lib/networking/queries/useGetIntegrationQuery'
 import { applyStoredTheme } from '../../../lib/themeUpdater'
 import { showSuccessToast } from '../../../lib/toastHelpers'
@@ -40,6 +45,7 @@ export default function Notion(): JSX.Element {
 
   const [form] = Form.useForm<FieldType>()
   const [messageApi, contextHolder] = message.useMessage()
+  const [exporting, setExporting] = useState(false)
 
   useEffect(() => {
     form.setFieldsValue({
@@ -93,7 +99,23 @@ export default function Notion(): JSX.Element {
   const exportToNotion = async () => {
     try {
       const task = await exportToIntegrationMutation(notion.id)
-      console.log('task', task)
+      // long polling to check the status of the task in every 10 seconds
+      setExporting(true)
+      const interval = setInterval(async () => {
+        const updatedTask = (await apiFetcher(`/api/tasks/${task.id}`)) as Task
+        if (updatedTask.state === TaskState.Succeeded) {
+          clearInterval(interval)
+          setExporting(false)
+          showSuccessToast('Exported to Notion successfully.')
+          return
+        }
+        if (updatedTask.state === TaskState.Failed) {
+          clearInterval(interval)
+          setExporting(false)
+          messageApi.error('There was an error exporting to Notion.')
+          return
+        }
+      }, 10000)
       showSuccessToast('Exporting to Notion...')
     } catch (error) {
       messageApi.error('There was an error exporting to Notion.')
@@ -193,8 +215,12 @@ export default function Notion(): JSX.Element {
               </Form.Item>
             </Form>
 
-            <Button type="primary" onClick={exportToNotion}>
-              Export most recent items to Notion
+            <Button
+              type="primary"
+              onClick={exportToNotion}
+              disabled={exporting}
+            >
+              {exporting ? 'Exporting' : 'Export most recent items to Notion'}
             </Button>
           </div>
         </VStack>