dify

dify/web/app/components/goto-anything/context.tsx

@@ -0,0 +1,67 @@
+'use client'
+
+import type { ReactNode } from 'react'
+import React, { createContext, useContext, useEffect, useState } from 'react'
+import { usePathname } from 'next/navigation'
+import { isInWorkflowPage } from '../workflow/constants'
+
+/**
+ * Interface for the GotoAnything context
+ */
+type GotoAnythingContextType = {
+  /**
+   * Whether the current page is a workflow page
+   */
+  isWorkflowPage: boolean
+  /**
+   * Whether the current page is a RAG pipeline page
+   */
+  isRagPipelinePage: boolean
+}
+
+// Create context with default values
+const GotoAnythingContext = createContext<GotoAnythingContextType>({
+  isWorkflowPage: false,
+  isRagPipelinePage: false,
+})
+
+/**
+ * Hook to use the GotoAnything context
+ */
+export const useGotoAnythingContext = () => useContext(GotoAnythingContext)
+
+type GotoAnythingProviderProps = {
+  children: ReactNode
+}
+
+/**
+ * Provider component for GotoAnything context
+ */
+export const GotoAnythingProvider: React.FC<GotoAnythingProviderProps> = ({ children }) => {
+  const [isWorkflowPage, setIsWorkflowPage] = useState(false)
+  const [isRagPipelinePage, setIsRagPipelinePage] = useState(false)
+  const pathname = usePathname()
+
+  // Update context based on current pathname using more robust route matching
+  useEffect(() => {
+    if (!pathname) {
+      setIsWorkflowPage(false)
+      setIsRagPipelinePage(false)
+      return
+    }
+
+    // Workflow pages: /app/[appId]/workflow or /workflow/[token] (shared)
+    const isWorkflow = isInWorkflowPage()
+    // RAG Pipeline pages: /datasets/[datasetId]/pipeline
+    const isRagPipeline = /^\/datasets\/[^/]+\/pipeline$/.test(pathname)
+
+    setIsWorkflowPage(isWorkflow)
+    setIsRagPipelinePage(isRagPipeline)
+  }, [pathname])
+
+  return (
+    <GotoAnythingContext.Provider value={{ isWorkflowPage, isRagPipelinePage }}>
+      {children}
+    </GotoAnythingContext.Provider>
+  )
+}