feat: added check for running project-id

src/components/AddContentModal/index.tsx

@@ -15,6 +15,7 @@ import {
   YOUTUBE_CHANNEL,
 } from '~/constants'
 import { api } from '~/network/api'
+import { useDataStore } from '~/stores/useDataStore'
 import { useModal } from '~/stores/useModalStore'
 import { useUserStore } from '~/stores/useUserStore'
 import { sphinxBridge } from '~/testSphinxBridge'
@@ -39,6 +40,7 @@ const handleSubmitForm = async (
   data: FieldValues,
   sourceType: string,
   setBudget: (value: number | null) => void,
+  setRunningProjectId: (value: string) => void,
 ): Promise<void> => {
   const endPoint = isSource(sourceType) ? 'radar' : 'add_node'
 
@@ -111,6 +113,10 @@ const handleSubmitForm = async (
       Authorization: lsatToken,
     })
 
+    if (res.data.project_id) {
+      setRunningProjectId(res.data.project_id)
+    }
+
     if (res.error) {
       const { message } = res.error
 
@@ -122,7 +128,7 @@ const handleSubmitForm = async (
     if (err.status === 402) {
       await payLsat(setBudget)
       await updateBudget(setBudget)
-      await handleSubmitForm(data, sourceType, setBudget)
+      await handleSubmitForm(data, sourceType, setBudget, setRunningProjectId)
     } else {
       let errorMessage = NODE_ADD_ERROR
 
@@ -147,6 +153,7 @@ export const AddContentModal = () => {
   const [currentStep, setCurrentStep] = useState(0)
   const { close, visible } = useModal('addContent')
   const { setBudget } = useUserStore((s) => s)
+  const { setRunningProjectId } = useDataStore((s) => s)
   const form = useForm<FormData>({ mode: 'onChange' })
   const { watch, setValue, reset } = form
   const [loading, setLoading] = useState(false)
@@ -198,7 +205,7 @@ export const AddContentModal = () => {
     setLoading(true)
 
     try {
-      await handleSubmitForm(data, type, setBudget)
+      await handleSubmitForm(data, type, setBudget, setRunningProjectId)
       SuccessNotify('Content Added')
       handleClose()
       // eslint-disable-next-line  @typescript-eslint/no-explicit-any

src/components/App/index.tsx

@@ -69,7 +69,8 @@
 
   const setTeachMeAnswer = useTeachStore((s) => s.setTeachMeAnswer)
 
-  const { fetchData, setCategoryFilter, setAbortRequests, addNewNode, splashDataLoading } = useDataStore((s) => s)
+  const { fetchData, setCategoryFilter, setAbortRequests, addNewNode, splashDataLoading, runningProjectId } =
+    useDataStore((s) => s)
 
   const { setAiSummaryAnswer, getKeyExist, aiRefId } = useAiSummaryStore((s) => s)
 
@@ -238,18 +239,16 @@
   ])
 
   useEffect(() => {
-    const ws = new WebSocket('wss://staging.stakwork.com/cable?channel=ProjectLogChannel')
-
-    ws.onopen = () => {
-      console.log('WebSocket connection established')
+    if (!runningProjectId) {
+      return
     }
 
-    ws.onmessage = (event) => {
-      console.log('Message from server:', event.data)
+    const ws = new WebSocket('wss://jobs.stakwork.com/cable?channel=ProjectLogChannel')
 
+    ws.onopen = () => {
       let id = 'a'
 
-      id = '51998781'
+      id = runningProjectId
 
       const command = {
         command: 'subscribe',
@@ -259,7 +258,13 @@
       // Send the command as a JSON string
       ws.send(JSON.stringify(command))
 
       console.log('Subscription command sent:', command)
+      console.log('WebSocket connection established')
+    }
+
+    ws.onmessage = (event) => {
+      console.log('Message from server:', event.data)
+
       // Handle the message from the server here
     }
 
@@ -268,14 +273,9 @@
     }
 
     ws.onclose = () => {
       console.log('WebSocket connection closed')
     }
-
-    // Cleanup when the component is unmounted
-    return () => {
-      // ws.close()
-    }
-  }, [])
+  }, [runningProjectId])
 
   useEffect(() => {
     if (!splashDataLoading) {

src/stores/useDataStore/index.ts

@@ -45,6 +45,7 @@
   stats: TStats | null
   nodeTypes: string[]
   seedQuestions: string[] | null
+  runningProjectId: string
 
   setTrendingTopics: (trendingTopics: Trending[]) => void
   setDataNew: (data: GraphData) => void
@@ -62,6 +63,7 @@
   setSources: (sources: Sources[] | null) => void
   setQueuedSources: (sources: Sources[] | null) => void
   setIsFetching: (_: boolean) => void
+  setRunningProjectId: (runningProjectId: string) => void
   setHideNodeDetails: (_: boolean) => void
   addNewNode: (data: FetchDataResponse) => void
   updateNode: (updatedNode: NodeExtended) => void
@@ -129,6 +131,7 @@
   abortRequest: false,
   dataNew: null,
   seedQuestions: null,
+  runningProjectId: '',
 }
 
 let abortController: AbortController | null = null
@@ -244,7 +247,7 @@
           sidebarFilterCounts,
         })
       } catch (error) {
         console.log(error)
 
         if (error !== 'abort') {
           set({ isLoadingNew: false, isFetching: false })
@@ -296,7 +299,7 @@
     setHideNodeDetails: (hideNodeDetails) => set({ hideNodeDetails }),
     setSeedQuestions: (questions) => set({ seedQuestions: questions }),
     updateNode: (updatedNode) => {
       console.log(updatedNode)
     },
     addNewNode: (data) => {
       const { dataInitial: existingData } = get()
@@ -340,8 +343,10 @@
       })
     },
     removeNode: (id) => {
       console.log(id)
     },
+
+    setRunningProjectId: (runningProjectId) => set({ runningProjectId }),
     setAbortRequests: (abortRequest) => set({ abortRequest }),
   })),
 )

src/types/index.ts

@@ -226,6 +226,7 @@ export type SubmitErrRes = {
   error?: { message?: string }
   data: {
     ref_id: string
+    project_id?: string
   }
 }