fix: add set selection to the searchbar (#4708)

Author: lukewalczak

src/components/Searchbar.tsx

@@ -149,7 +149,7 @@ export type Props = React.ComponentPropsWithRef<typeof TextInput> & {
 
 type TextInputHandles = Pick<
   TextInput,
-  'setNativeProps' | 'isFocused' | 'clear' | 'blur' | 'focus'
+  'setNativeProps' | 'isFocused' | 'clear' | 'blur' | 'focus' | 'setSelection'
 >;
 
 /**
@@ -210,31 +210,16 @@ const Searchbar = forwardRef<TextInputHandles, Props>(
     const theme = useInternalTheme(themeOverrides);
     const root = React.useRef<TextInput>(null);
 
-    React.useImperativeHandle(ref, () => {
-      const input = root.current;
-
-      if (input) {
-        return {
-          focus: () => input.focus(),
-          clear: () => input.clear(),
-          setNativeProps: (args: TextInputProps) => input.setNativeProps(args),
-          isFocused: () => input.isFocused(),
-          blur: () => input.blur(),
-        };
-      }
-
-      const noop = () => {
-        throw new Error('TextInput is not available');
-      };
-
-      return {
-        focus: noop,
-        clear: noop,
-        setNativeProps: noop,
-        isFocused: noop,
-        blur: noop,
-      };
-    });
+    React.useImperativeHandle(ref, () => ({
+      focus: () => root.current?.focus(),
+      clear: () => root.current?.clear(),
+      setNativeProps: (args: TextInputProps) =>
+        root.current?.setNativeProps(args),
+      isFocused: () => root.current?.isFocused() || false,
+      blur: () => root.current?.blur(),
+      setSelection: (start: number, end: number) =>
+        root.current?.setSelection(start, end),
+    }));
 
     const handleClearPress = (e: any) => {
       root.current?.clear();