Render initialization status

frontend/components/Layout/InitializationStatusChecker.tsx

@@ -1,44 +1,52 @@
 import { FC, useContext, useEffect, useState } from "react"
 
-import { NotificationContext } from "@/context/NotificationContext"
+import { Notification, NotificationContext } from "@/context/NotificationContext"
+import { useLocalStorage } from "@/hooks/useLocalStorage"
 import { useInitializationStatus } from "@/repositories/initializationStatusRepository"
+import { usePid } from "@/repositories/pidRepository"
+
+const INITIAL_KEY = 'InitializationStatusChecker-closed'
 
 export const InitializationStatusChecker: FC = () => {
   const { setNotification } = useContext(NotificationContext)
 
-  const [closed, setClosed] = useState<boolean>(false)
+  const { pid } = usePid()
+  const key = pid ? `InitializationStatusChecker-closed-${pid}` : INITIAL_KEY
+  const [closed, setClosed] = useLocalStorage<boolean>(key, false)
   const [initialized, setInitialized] = useState<boolean>(false)
 
   // Stop loading if initialization process is finished
   const { initializationStatus } = useInitializationStatus((initialized || closed) ? 0 : 100)
 
   useEffect(() => {
-    if (!initializationStatus) return;
+    if (!initializationStatus || !pid) return;
+
+    let notification: Notification | undefined = undefined
 
     if (initializationStatus.total === initializationStatus.loaded) {
       setInitialized(true)
 
-      if (!closed) {
-        setNotification(
-          {
-            type: 'success',
-            message: `Successfully loaded ${initializationStatus.loaded} definitions!`,
-            onClose: () => setClosed(true)
-          }
-        )
+      notification = {
+        type: 'success',
+        message: `Successfully loaded ${initializationStatus.loaded} definitions!`,
+        onClose: () => setClosed(true)
       }
     } else if (!closed) {
       const progress = Math.round(initializationStatus.loaded / initializationStatus.total * 100)
 
-      setNotification(
-        {
-          type: 'info',
-          message: `Loading definitions... ${progress}% (${initializationStatus.loaded}/${initializationStatus.total})`,
-          onClose: () => setClosed(true)
-        }
-      )
+      notification = {
+        type: 'info',
+        message: `Loading definitions... ${progress}% (${initializationStatus.loaded}/${initializationStatus.total})`,
+        onClose: () => setClosed(true)
+      }
+    }
+
+    if (closed) {
+      setNotification(null)
+    } else if (notification) {
+      setNotification(notification)
     }
-  }, [closed, initializationStatus, setNotification])
+  }, [pid, closed, setClosed, initializationStatus, setNotification])
 
   return null
 }

frontend/constants/path.ts

@@ -13,6 +13,7 @@ export const path = {
     index: () => '/licenses',
   },
   api: {
+    pid: () => `/api/pid.json`,
     initializationStatus: () => `/api/initialization_status.json`,
     definitions: {
       index: () => '/api/definitions.json',

frontend/repositories/pidRepository.ts

@@ -0,0 +1,18 @@
+import useSWR from 'swr'
+
+import { path } from '@/constants/path'
+
+import { get } from './httpRequest'
+
+type PidReponse = {
+  pid: number
+}
+
+export const usePid = () => {
+  const { data } = useSWR(path.api.pid(), async (): Promise<number> => {
+    const response = await get<PidReponse>(path.api.pid())
+    return response.pid
+  })
+
+  return { pid: data }
+}

lib/diver_down/web.rb

@@ -50,6 +50,8 @@ def call(env)
       in ['GET', %r{\A/api/sources/(?<source>.+)\.json\z}]
         source = Regexp.last_match[:source]
         action.source(source)
+      in ['GET', %r{\A/api/pid\.json\z}]
+        action.pid
       in ['GET', %r{\A/api/initialization_status\.json\z}]
         action.initialization_status(@total_definition_files_size)
       in ['GET', %r{\A/assets/}]

lib/diver_down/web/action.rb

@@ -70,6 +70,13 @@ def initialization_status(total)
         )
       end
 
+      # GET /api/pid.json
+      def pid
+        json(
+          pid: Process.pid
+        )
+      end
+
       # GET /api/definitions/:bit_id.json
       #
       # @param bit_id [Integer]

spec/diver_down/web_spec.rb

@@ -213,6 +213,17 @@ def assert_source(source, expected_ids)
     end
   end
 
+  describe 'GET /api/pid.json' do
+    it 'returns pid' do
+      get '/api/pid.json'
+
+      expect(last_response.status).to eq(200)
+      expect(JSON.parse(last_response.body)).to eq({
+        'pid' => Process.pid,
+      })
+    end
+  end
+
   describe 'GET /api/sources.json' do
     it 'returns [] if store is blank' do
       get '/api/sources.json'