|
|
@@ -3,7 +3,7 @@
|
|
|
* SPDX-License-Identifier: MIT
|
|
|
*/
|
|
|
|
|
|
-import { createContext, useContext } from 'react';
|
|
|
+import React, { createContext, useContext, useMemo } from 'react';
|
|
|
|
|
|
import { BaseVariableField } from '@flowgram.ai/editor';
|
|
|
|
|
|
@@ -13,4 +13,16 @@ export const VariableSelectorContext = createContext<{
|
|
|
|
|
|
export const useVariableSelectorContext = () => useContext(VariableSelectorContext);
|
|
|
|
|
|
-export const VariableSelectorProvider = VariableSelectorContext.Provider;
|
|
|
+export const VariableSelectorProvider = ({
|
|
|
+ children,
|
|
|
+ skipVariable,
|
|
|
+}: {
|
|
|
+ skipVariable?: (variable?: BaseVariableField) => boolean;
|
|
|
+ children: React.ReactNode;
|
|
|
+}) => {
|
|
|
+ const context = useMemo(() => ({ skipVariable }), [skipVariable]);
|
|
|
+
|
|
|
+ return (
|
|
|
+ <VariableSelectorContext.Provider value={context}>{children}</VariableSelectorContext.Provider>
|
|
|
+ );
|
|
|
+};
|