From 36b4393866128755569aa0ad3b6eb3bad6a5e9ba Mon Sep 17 00:00:00 2001
From: myron <myronliu347@gmail.com>
Date: Mon, 26 Aug 2024 19:57:04 +0800
Subject: [PATCH] fix(TimePicker): fixed only support hh:mm format (#3280)

---
 src/time-picker/__tests__/index.test.jsx |  8 ++++
 src/time-picker/panel/single-panel.tsx   | 51 +++++++++++++++---------
 2 files changed, 41 insertions(+), 18 deletions(-)

diff --git a/src/time-picker/__tests__/index.test.jsx b/src/time-picker/__tests__/index.test.jsx
index 256afd20f..6f40a87b8 100644
--- a/src/time-picker/__tests__/index.test.jsx
+++ b/src/time-picker/__tests__/index.test.jsx
@@ -76,6 +76,14 @@ describe('TimePicker', () => {
       const panelNode = document.querySelector('.t-time-picker__panel');
       // format为HH:mm 只展示两列 即时分
       expect(panelNode.querySelectorAll('.t-time-picker__panel-body-scroll').length).toBe(2);
+
+      await wrapper.setProps({
+        popupProps: {
+          visible: true,
+        },
+        format: 'HH时mm分',
+      });
+      expect(panelNode.querySelectorAll('.t-time-picker__panel-body-scroll').length).toBe(2);
       panelNode.parentNode.removeChild(panelNode);
     });
 
diff --git a/src/time-picker/panel/single-panel.tsx b/src/time-picker/panel/single-panel.tsx
index 5dacb513d..f600248cb 100644
--- a/src/time-picker/panel/single-panel.tsx
+++ b/src/time-picker/panel/single-panel.tsx
@@ -9,12 +9,7 @@ import customParseFormat from 'dayjs/plugin/customParseFormat';
 
 import { panelColProps } from './props';
 import {
-  EPickerCols,
-  TWELVE_HOUR_FORMAT,
-  TIME_FORMAT,
-  AM,
-  PM,
-  MERIDIEM_LIST,
+  EPickerCols, TWELVE_HOUR_FORMAT, AM, PM, MERIDIEM_LIST,
 } from '../../_common/js/time-picker/const';
 import { closestLookup } from '../../_common/js/time-picker/utils';
 import { useConfig } from '../../hooks/useConfig';
@@ -28,6 +23,8 @@ const panelOffset = {
   bottom: 21,
 };
 
+export const REGEX_FORMAT = /\[([^\]]+)]|Y{1,4}|M{1,4}|D{1,2}|d{1,4}|H{1,2}|h{1,2}|a|A|m{1,2}|s{1,2}|Z{1,2}|SSS/g;
+
 export default defineComponent({
   name: 'TTimePickerPanelCol',
   props: {
@@ -82,22 +79,40 @@ export default defineComponent({
     );
 
     onMounted(() => {
-      const match = format.value.match(TIME_FORMAT);
-
-      const [, startCol, hourCol, minuteCol, secondCol, milliSecondCol, endCol] = match;
+      const match = format.value.match(REGEX_FORMAT);
       const {
         meridiem, hour, minute, second, milliSecond,
       } = EPickerCols;
 
-      const renderCol = [
-        startCol && meridiem,
-        hourCol && hour,
-        minuteCol && minute,
-        secondCol && second,
-        milliSecondCol && milliSecond,
-        endCol && meridiem,
-      ].filter((v) => !!v);
-
+      const renderCol: EPickerCols[] = [];
+
+      match.forEach((m) => {
+        switch (m) {
+          case 'H':
+          case 'HH':
+          case 'h':
+          case 'hh':
+            renderCol.push(hour);
+            break;
+          case 'a':
+          case 'A':
+            renderCol.push(meridiem);
+            break;
+          case 'm':
+          case 'mm':
+            renderCol.push(minute);
+            break;
+          case 's':
+          case 'ss':
+            renderCol.push(second);
+            break;
+          case 'SSS':
+            renderCol.push(milliSecond);
+            break;
+          default:
+            break;
+        }
+      });
       cols.value = renderCol;
     });