mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-07-09 01:51:50 +08:00
51 lines
1.1 KiB
TypeScript
51 lines
1.1 KiB
TypeScript
'use client'
|
|
|
|
import type { ReactNode } from 'react'
|
|
import {
|
|
useRef,
|
|
useState,
|
|
} from 'react'
|
|
import {
|
|
createContext,
|
|
useContextSelector,
|
|
} from 'use-context-selector'
|
|
|
|
export type PluginPageContextValue = {
|
|
containerRef: React.RefObject<HTMLDivElement>
|
|
scrollDisabled: boolean
|
|
setScrollDisabled: (scrollDisabled: boolean) => void
|
|
}
|
|
|
|
export const PluginPageContext = createContext<PluginPageContextValue>({
|
|
containerRef: { current: null },
|
|
scrollDisabled: false,
|
|
setScrollDisabled: () => {},
|
|
})
|
|
|
|
type PluginPageContextProviderProps = {
|
|
children: ReactNode
|
|
}
|
|
|
|
export function usePluginPageContext(selector: (value: PluginPageContextValue) => any) {
|
|
return useContextSelector(PluginPageContext, selector)
|
|
}
|
|
|
|
export const PluginPageContextProvider = ({
|
|
children,
|
|
}: PluginPageContextProviderProps) => {
|
|
const containerRef = useRef<HTMLDivElement>(null)
|
|
const [scrollDisabled, setScrollDisabled] = useState(false)
|
|
|
|
return (
|
|
<PluginPageContext.Provider
|
|
value={{
|
|
containerRef,
|
|
scrollDisabled,
|
|
setScrollDisabled,
|
|
}}
|
|
>
|
|
{children}
|
|
</PluginPageContext.Provider>
|
|
)
|
|
}
|