From cd66f79518229d7ccad8312f530c7bc7189073b1 Mon Sep 17 00:00:00 2001
From: adids1221 <adids1221@gmail.com>
Date: Wed, 4 Sep 2024 16:09:42 +0300
Subject: [PATCH] Picker dailog validation

---
 src/components/picker/PickerItemsList.tsx     |  6 +-
 .../picker/helpers/usePickerDialogProps.tsx   | 67 +++++++++++++++++++
 .../picker/helpers/usePickerSelection.tsx     | 55 +++++++++++++--
 src/components/picker/index.tsx               | 46 +++++++------
 src/components/picker/types.ts                | 26 ++++++-
 5 files changed, 172 insertions(+), 28 deletions(-)
 create mode 100644 src/components/picker/helpers/usePickerDialogProps.tsx

diff --git a/src/components/picker/PickerItemsList.tsx b/src/components/picker/PickerItemsList.tsx
index 26db3b6e97..c3ba9a22ba 100644
--- a/src/components/picker/PickerItemsList.tsx
+++ b/src/components/picker/PickerItemsList.tsx
@@ -32,7 +32,8 @@ const PickerItemsList = (props: PickerItemsListProps) => {
     useSafeArea,
     useDialog,
     mode,
-    testID
+    testID,
+    migrateDialog
   } = props;
   const context = useContext(PickerContext);
 
@@ -138,7 +139,8 @@ const PickerItemsList = (props: PickerItemsListProps) => {
           </Text>
         </View>
       );
-    } else if (!useDialog || mode === PickerModes.MULTI) {
+      //} else if (!migrateDialog && (!useDialog || mode === PickerModes.MULTI)) {
+    } else if (!migrateDialog && (!useDialog || mode === PickerModes.MULTI)) {
       return <Modal.TopBar testID={`${props.testID}.topBar`} {...topBarProps}/>;
     }
   };
diff --git a/src/components/picker/helpers/usePickerDialogProps.tsx b/src/components/picker/helpers/usePickerDialogProps.tsx
new file mode 100644
index 0000000000..a8158f8d91
--- /dev/null
+++ b/src/components/picker/helpers/usePickerDialogProps.tsx
@@ -0,0 +1,67 @@
+import React from 'react';
+import {StyleSheet} from 'react-native';
+import {PickerProps, PickerModes} from '../types';
+import Button from '../../button';
+import {Colors} from '../../../style';
+
+const DIALOG_PROPS = {
+  bottom: true,
+  width: '100%',
+  height: 250
+};
+
+const NEW_DIALOG_PROPS = {
+  bottom: true,
+  useSafeArea: true,
+  height: '60%',
+  width: '95%'
+};
+
+type HookProps = PickerProps & {
+  shouldDisableDoneButton?: boolean;
+};
+
+const usePickerDialogProps = (props: HookProps, onDone: any) => {
+  const {customPickerProps, mode, testID, selectionOptions, shouldDisableDoneButton} = props;
+  const migrateDialog = customPickerProps?.migrateDialog;
+  const defaultProps = migrateDialog ? NEW_DIALOG_PROPS : DIALOG_PROPS;
+  const {validationMessage, validationMessageStyle} = selectionOptions || {};
+
+  console.log(`shouldDisableDoneButton?`, shouldDisableDoneButton);
+
+  const modifiedHeaderProps = migrateDialog && {
+    headerProps: {
+      trailingAccessory: (
+        <Button
+          label="Save"
+          link
+          style={{height: 30}}
+          onPress={mode === PickerModes.MULTI ? onDone : undefined}
+          testID={`${testID}.dialog.header.save`}
+          disabled={!shouldDisableDoneButton}
+        />
+      ),
+      subtitle: !shouldDisableDoneButton && validationMessage,
+      subtitleStyle: [styles.validationMessage, validationMessageStyle],
+      ...customPickerProps?.dialogProps?.headerProps
+    }
+  };
+
+  const dialogProps: PickerProps['customPickerProps'] = {
+    dialogProps: {
+      ...defaultProps,
+      ...customPickerProps?.dialogProps,
+      ...modifiedHeaderProps
+    }
+  };
+
+  return dialogProps;
+};
+
+export default usePickerDialogProps;
+
+const styles = StyleSheet.create({
+  validationMessage: {
+    color: Colors.red30
+  }
+});
diff --git a/src/components/picker/helpers/usePickerSelection.tsx b/src/components/picker/helpers/usePickerSelection.tsx
index 7ab850c086..2277c63015 100644
--- a/src/components/picker/helpers/usePickerSelection.tsx
+++ b/src/components/picker/helpers/usePickerSelection.tsx
@@ -3,15 +3,40 @@ import _ from 'lodash';
 import {PickerProps, PickerValue, PickerSingleValue, PickerMultiValue, PickerModes} from '../types';
 
 interface UsePickerSelectionProps
-  extends Pick<PickerProps, 'migrate' | 'value' | 'onChange' | 'getItemValue' | 'topBarProps' | 'mode'> {
+  extends Pick<
+    PickerProps,
+    | 'migrate'
+    | 'value'
+    | 'onChange'
+    | 'getItemValue'
+    | 'topBarProps'
+    | 'mode'
+    | 'selectionValidation'
+    | 'selectionOptions'
+    | 'useDialog'
+  > {
   pickerExpandableRef: RefObject<any>;
   setSearchValue: (searchValue: string) => void;
 }
 
 const usePickerSelection = (props: UsePickerSelectionProps) => {
-  const {migrate, value, onChange, topBarProps, pickerExpandableRef, getItemValue, setSearchValue, mode} = props;
+  const {
+    migrate,
+    value,
+    onChange,
+    topBarProps,
+    pickerExpandableRef,
+    getItemValue,
+    setSearchValue,
+    mode,
+    selectionValidation,
+    selectionOptions,
+    useDialog
+  } = props;
+  const {onValidationFailed, validateOnStart} = selectionOptions || {};
   const [multiDraftValue, setMultiDraftValue] = useState(value as PickerMultiValue);
   const [multiFinalValue, setMultiFinalValue] = useState(value as PickerMultiValue);
+  const [isValid, setIsValid] = useState<boolean | undefined>(true);
 
   useEffect(() => {
     if (mode === PickerModes.MULTI && multiFinalValue !== value) {
@@ -20,13 +45,31 @@ const usePickerSelection = (props: UsePickerSelectionProps) => {
     }
   }, [value]);
 
+  useEffect(() => {
+    if (validateOnStart) {
+      _selectionValidation?.(value);
+    }
+  }, []);
+
+  const _selectionValidation = (item: PickerValue) => {
+    if (useDialog) {
+      const isValid = selectionValidation?.(item);
+      console.log(`isValid:`, isValid);
+      if (!isValid) {
+        onValidationFailed?.(item);
+      }
+      setIsValid(isValid);
+    }
+  };
+
   const onDoneSelecting = useCallback((item: PickerValue) => {
     setSearchValue('');
     setMultiFinalValue(item as PickerMultiValue);
     pickerExpandableRef.current?.closeExpandable?.();
+    _selectionValidation(item);
     onChange?.(item);
   },
-  [onChange]);
+  [onChange, useDialog, selectionValidation]);
 
   const toggleItemSelection = useCallback((item: PickerSingleValue) => {
     let newValue;
@@ -37,9 +80,10 @@ const usePickerSelection = (props: UsePickerSelectionProps) => {
       newValue = _.xor(multiDraftValue, itemAsArray);
     }
 
+    _selectionValidation(newValue);
     setMultiDraftValue(newValue);
   },
-  [multiDraftValue, getItemValue]);
+  [multiDraftValue, getItemValue, useDialog, selectionValidation]);
 
   const cancelSelect = useCallback(() => {
     setSearchValue('');
@@ -52,7 +96,8 @@ const usePickerSelection = (props: UsePickerSelectionProps) => {
     multiDraftValue,
     onDoneSelecting,
     toggleItemSelection,
-    cancelSelect
+    cancelSelect,
+    shouldDisableDoneButton: isValid
   };
 };
 
diff --git a/src/components/picker/index.tsx b/src/components/picker/index.tsx
index ff59edc01a..3d438b5cfe 100644
--- a/src/components/picker/index.tsx
+++ b/src/components/picker/index.tsx
@@ -17,6 +17,7 @@ import usePickerSearch from './helpers/usePickerSearch';
 import useImperativePickerHandle from './helpers/useImperativePickerHandle';
 import useFieldType from './helpers/useFieldType';
 import useNewPickerProps from './helpers/useNewPickerProps';
+import usePickerDialogProps from './helpers/usePickerDialogProps';
 // import usePickerMigrationWarnings from './helpers/usePickerMigrationWarnings';
 import {extractPickerItems} from './PickerPresenter';
 import {
@@ -31,12 +32,6 @@ import {
   PickerMethods
 } from './types';
 
-const DIALOG_PROPS = {
-  bottom: true,
-  width: '100%',
-  height: 250
-};
-
 type PickerStatics = {
   Item: typeof PickerItem;
   modes: typeof PickerModes;
@@ -76,6 +71,8 @@ const Picker = React.forwardRef((props: PickerProps, ref) => {
     accessibilityLabel,
     accessibilityHint,
     items: propItems,
+    selectionValidation,
+    selectionOptions,
     ...others
   } = themeProps;
   const {preset, placeholder, style, trailingAccessory, label: propsLabel} = others;
@@ -84,10 +81,10 @@ const Picker = React.forwardRef((props: PickerProps, ref) => {
   const [items, setItems] = useState<PickerItemProps[]>(propItems || extractPickerItems(themeProps));
   const pickerExpandable = useRef<ExpandableOverlayMethods>(null);
   const pickerRef = useImperativePickerHandle(ref, pickerExpandable);
-  
+
   // TODO: Remove this when migration is completed, starting of v8
   // usePickerMigrationWarnings({children, migrate, getItemLabel, getItemValue});
-  
+
   useEffect(() => {
     if (propItems) {
       setItems(propItems);
@@ -99,17 +96,24 @@ const Picker = React.forwardRef((props: PickerProps, ref) => {
     setSearchValue,
     onSearchChange: _onSearchChange
   } = usePickerSearch({showSearch, onSearchChange, getItemLabel, children});
-  
-  const {multiDraftValue, onDoneSelecting, toggleItemSelection, cancelSelect} = usePickerSelection({
-    migrate,
-    value,
-    onChange,
-    pickerExpandableRef: pickerExpandable,
-    getItemValue,
-    topBarProps,
-    setSearchValue,
-    mode
-  });
+
+  const {multiDraftValue, onDoneSelecting, toggleItemSelection, cancelSelect, shouldDisableDoneButton} =
+    usePickerSelection({
+      migrate,
+      value,
+      onChange,
+      pickerExpandableRef: pickerExpandable,
+      getItemValue,
+      topBarProps,
+      setSearchValue,
+      mode,
+      selectionValidation,
+      selectionOptions,
+      useDialog
+    });
+
+  const {dialogProps = {}} = usePickerDialogProps({...themeProps, shouldDisableDoneButton}, () =>
+    onDoneSelecting(multiDraftValue));
 
   const {label, accessibilityInfo} = usePickerLabel({
     value,
@@ -242,6 +246,7 @@ const Picker = React.forwardRef((props: PickerProps, ref) => {
         renderHeader={renderHeader}
         listProps={listProps}
         useSafeArea={useSafeArea}
+        migrateDialog={customPickerProps?.migrateDialog}
       >
         {filteredChildren}
       </PickerItemsList>
@@ -271,15 +276,16 @@ const Picker = React.forwardRef((props: PickerProps, ref) => {
   return (
     <PickerContext.Provider value={contextValue}>
       {
+        //@ts-ignore remove after dialog migration finished
         <ExpandableOverlay
           ref={pickerExpandable}
           useDialog={useDialog || useWheelPicker}
-          dialogProps={DIALOG_PROPS}
           expandableContent={expandableModalContent}
           renderCustomOverlay={renderOverlay ? _renderOverlay : undefined}
           onPress={onPress}
           testID={testID}
           {...customPickerProps}
+          dialogProps={dialogProps}
           disabled={themeProps.editable === false}
         >
           {renderTextField()}
diff --git a/src/components/picker/types.ts b/src/components/picker/types.ts
index 5012a03ef7..3523e66ae5 100644
--- a/src/components/picker/types.ts
+++ b/src/components/picker/types.ts
@@ -158,10 +158,32 @@ type PickerExpandableOverlayProps = {
   enableModalBlur?: boolean;
 };
 
+type PickerSelectionOptions = Pick<
+  TextFieldProps,
+  'validateOnStart' | 'onChangeValidity' | 'validationMessage' | 'validationMessageStyle'
+> & {
+  /**
+   * Callback for when field validated and failed
+   */
+  onValidationFailed?: (value: PickerValue) => void;
+};
+
+type PickerSelectionValidation = {
+  /**
+   * Callback for when selection was made
+   */
+  selectionValidation?: (value: PickerValue) => boolean;
+  /**
+   * Selection validation options
+   */
+  selectionOptions?: PickerSelectionOptions;
+};
+
 export type PickerBaseProps = Omit<TextFieldProps, 'value' | 'onChange'> &
   PickerPropsDeprecation &
   PickerExpandableOverlayProps &
-  PickerListProps & {
+  PickerListProps &
+  PickerSelectionValidation & {
     /* ...TextField.propTypes, */
     /**
      * Use dialog instead of modal picker
@@ -317,6 +339,8 @@ export type PickerItemsListProps = Pick<
 > & {
   //TODO: after finish Picker props migration, items should be taken from PickerProps
   items?: {value: any; label: any}[];
+  //Remove the migrateDialog prop when the dialog migration finished in v8
+  migrateDialog?: boolean;
 };
 
 export type PickerMethods = TextFieldMethods & ExpandableOverlayMethods;