fix: Click out of input should also trigger input focus

fix ant-design/ant-design#21149

Author: zombieJ

src/Picker.tsx

@@ -272,6 +272,18 @@ function InnerPicker<DateType>(props: PickerProps<DateType>) {
     }
   };
 
+  const onInternalMouseUp: React.MouseEventHandler<HTMLDivElement> = (
+    ...args
+  ) => {
+    if (onMouseUp) {
+      onMouseUp(...args);
+    }
+
+    if (inputRef.current) {
+      inputRef.current.focus();
+    }
+  };
+
   // ============================= Input =============================
   const [inputProps, { focused, typing }] = usePickerInput({
     blurToCancel: needConfirmButton,
@@ -428,7 +440,7 @@ function InnerPicker<DateType>(props: PickerProps<DateType>) {
           })}
           style={style}
           onMouseDown={onMouseDown}
-          onMouseUp={onMouseUp}
+          onMouseUp={onInternalMouseUp}
           onMouseEnter={onMouseEnter}
           onMouseLeave={onMouseLeave}
           onContextMenu={onContextMenu}

tests/picker.spec.tsx

@@ -546,4 +546,15 @@ describe('Picker.Basic', () => {
 
     expect(wrapper.find('input').prop('value')).toEqual('2020-1st');
   });
+
+  it('click outside should also focus', () => {
+    const wrapper = mount(<MomentPicker />);
+    const inputElement = (wrapper
+      .find('input')
+      .instance() as any) as HTMLInputElement;
+    inputElement.focus = jest.fn();
+
+    wrapper.find('.rc-picker').simulate('mouseUp');
+    expect(inputElement.focus).toHaveBeenCalled();
+  });
 });