From d3138e6617f8da606c1f2a712fe9a00e65d3c966 Mon Sep 17 00:00:00 2001
From: Joeri Exelmans <joeri.exelmans@gmail.com>
Date: Wed, 14 May 2025 10:36:49 +0200
Subject: [PATCH] add 'filter' property to CallBlock to prepare to filter
 suggestions on output type

---
 src/CallBlock.tsx | 11 ++++++-----
 src/Editor.tsx    |  1 +
 2 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/src/CallBlock.tsx b/src/CallBlock.tsx
index 5ddd789..f032d22 100644
--- a/src/CallBlock.tsx
+++ b/src/CallBlock.tsx
@@ -22,9 +22,10 @@ interface CallBlockProps<
   FnState=EditorState,
   InputState=EditorState,
 > extends State2Props<CallBlockState<FnState,InputState>,EditorState> {
+  filter: (suggestion: SuggestionType) => boolean;
 }
 
-function headlessCallBlock({setState}: CallBlockProps) {
+function headlessCallBlock(setState: (callback: SetStateFn<CallBlockState,EditorState>) => void) {
   const setFn = (callback: SetStateFn) => {
     setState(state => ({...state, fn: callback(state.fn)}));
   }
@@ -40,9 +41,9 @@ function headlessCallBlock({setState}: CallBlockProps) {
   return {setFn, setInput, onFnCancel, onInputCancel};
 }
 
-export function CallBlock({ state, setState }: CallBlockProps) {
+export function CallBlock({ state, setState, filter }: CallBlockProps) {
   const {setFn, setInput, onFnCancel, onInputCancel}
-    = headlessCallBlock({ state, setState });
+    = headlessCallBlock(setState);
   const env = useContext(EnvContext);
   const resolved = evalEditorBlock(state, env);
   return <span className={"functionBlock" + ((resolved instanceof DeepError) ? " unifyError" : "")}>
@@ -84,7 +85,7 @@ function FunctionHeader({ fn, setFn, input, onFnCancel }) {
     const {
       setFn         : setFnFn,
       onFnCancel    : onFnFnCancel,
-    } = headlessCallBlock({state: fn, setState: setFn});
+    } = headlessCallBlock(setFn);
 
     return <FunctionHeader
       fn={fn.fn}
@@ -130,7 +131,7 @@ function NestedParams({fn, setFn, depth, errorDepth}) {
   const {
       setFn         : setFnFn,
       setInput      : setFnInput,
-    } = headlessCallBlock({state: fn, setState: setFn});
+    } = headlessCallBlock(setFn);
   return <InputParams
     fn={fn.fn}
     setFn={setFnFn}
diff --git a/src/Editor.tsx b/src/Editor.tsx
index 6f54503..57dbd25 100644
--- a/src/Editor.tsx
+++ b/src/Editor.tsx
@@ -151,6 +151,7 @@ export function Editor({state, setState, onCancel, filter}: EditorProps) {
         return <CallBlock
           state={state}
           setState={setState as (callback:(p:CallBlockState)=>EditorState)=>void}
+          filter={filter}
         />;
       case "let":
         return <LetInBlock