From 9f777508cb8254595da58a67b45ce4946e6b4013 Mon Sep 17 00:00:00 2001
From: PizieDust <playersrebirth@gmail.com>
Date: Fri, 25 Oct 2024 16:32:37 +0200
Subject: [PATCH] return to input

---
 src/extension_commands.ml | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/src/extension_commands.ml b/src/extension_commands.ml
index 6d5366f76..e86472672 100644
--- a/src/extension_commands.ml
+++ b/src/extension_commands.ml
@@ -593,7 +593,7 @@ module Search_by_type = struct
         Type_search.request
         (Type_search.make ~uri ~position ~limit ~query ~with_doc ()))
 
-  let rec display_search_results query results text_editor position =
+  let rec display_search_results query results text_editor position client =
     let quickPickItems =
       List.map
         results
@@ -623,7 +623,7 @@ module Search_by_type = struct
       QuickPick.onDidTriggerButton
         quickPick
         ~listener:(fun _ ->
-          let _ = get_query_input ~previous_query:query () in
+          let _ = handle_search ~previous_query:query text_editor client in
           ())
         ()
     in
@@ -645,18 +645,19 @@ module Search_by_type = struct
                 ()
             in
             QuickPick.hide quickPick
-          | _ -> display_search_results query results text_editor position)
+          | _ ->
+            display_search_results query results text_editor position client)
         ()
     in
     QuickPick.show quickPick
 
-  let rec handle_search ?previous_query text_editor client =
+  and handle_search ?previous_query text_editor client =
     let open Promise.Syntax in
     let position = TextEditor.selection text_editor |> Selection.active in
     let* query_input = get_query_input ?previous_query () in
     match query_input with
     | Some query -> (
-      let* type_search_results =
+      let* query_results =
         get_search_results
           ~query
           ~with_doc:true
@@ -665,7 +666,7 @@ module Search_by_type = struct
           text_editor
           client
       in
-      match type_search_results with
+      match query_results with
       | [] -> handle_search ~previous_query:query text_editor client
       | results ->
         let _ =
@@ -680,6 +681,7 @@ module Search_by_type = struct
                results)
             text_editor
             position
+            client
         in
         Promise.return ())
     | _ -> Promise.return ()