From 3a1c8a9d16bc01cb947c6aa2afd1545863fe3e91 Mon Sep 17 00:00:00 2001
From: neverland <chenjiahan@youzan.com>
Date: Fri, 6 Aug 2021 18:30:07 +0800
Subject: [PATCH] types(Area): add AreaInstance type (#9195)

---
 src/area/Area.tsx        | 60 ++++++++++++++++++----------------------
 src/area/README.md       | 13 +++++++++
 src/area/README.zh-CN.md | 13 +++++++++
 src/area/index.ts        |  2 +-
 src/area/types.ts        | 31 +++++++++++++++++++++
 5 files changed, 85 insertions(+), 34 deletions(-)
 create mode 100644 src/area/types.ts

diff --git a/src/area/Area.tsx b/src/area/Area.tsx
index 03a1b1dcd..45a0dcc00 100644
--- a/src/area/Area.tsx
+++ b/src/area/Area.tsx
@@ -1,4 +1,3 @@
-/* eslint-disable camelcase */
 import {
   ref,
   watch,
@@ -8,6 +7,7 @@ import {
   PropType,
   onMounted,
   defineComponent,
+  ExtractPropTypes,
 } from 'vue';
 
 // Utils
@@ -21,6 +21,9 @@ import { useExpose } from '../composables/use-expose';
 // Components
 import { Picker } from '../picker';
 
+// Types
+import type { AreaList, AreaColumnType, AreaColumnOption } from './types';
+
 const [name, bem] = createNamespace('area');
 
 const EMPTY_CODE = '000000';
@@ -47,41 +50,32 @@ function isOverseaCode(code: string) {
   return code[0] === '9';
 }
 
-export type AreaList = {
-  city_list: Record<string, string>;
-  county_list: Record<string, string>;
-  province_list: Record<string, string>;
-};
+const props = extend({}, pickerProps, {
+  value: String,
+  areaList: {
+    type: Object as PropType<AreaList>,
+    default: () => ({}),
+  },
+  columnsNum: {
+    type: [Number, String],
+    default: 3,
+  },
+  isOverseaCode: {
+    type: Function as PropType<(code: string) => boolean>,
+    default: isOverseaCode,
+  },
+  columnsPlaceholder: {
+    type: Array as PropType<string[]>,
+    default: () => [],
+  },
+});
 
-export type AreaColumnOption = {
-  name: string;
-  code: string;
-};
-
-type ColumnType = 'province' | 'county' | 'city';
+export type AreaProps = ExtractPropTypes<typeof props>;
 
 export default defineComponent({
   name,
 
-  props: extend({}, pickerProps, {
-    value: String,
-    areaList: {
-      type: Object as PropType<AreaList>,
-      default: () => ({}),
-    },
-    columnsNum: {
-      type: [Number, String],
-      default: 3,
-    },
-    isOverseaCode: {
-      type: Function as PropType<(code: string) => boolean>,
-      default: isOverseaCode,
-    },
-    columnsPlaceholder: {
-      type: Array as PropType<string[]>,
-      default: () => [],
-    },
-  }),
+  props,
 
   emits: ['change', 'confirm'],
 
@@ -131,7 +125,7 @@ export default defineComponent({
       return '';
     };
 
-    const getColumnValues = (type: ColumnType, code?: string) => {
+    const getColumnValues = (type: AreaColumnType, code?: string) => {
       let column: AreaColumnOption[] = [];
       if (type !== 'province' && !code) {
         return column;
@@ -170,7 +164,7 @@ export default defineComponent({
     };
 
     // get index by code
-    const getIndex = (type: ColumnType, code: string) => {
+    const getIndex = (type: AreaColumnType, code: string) => {
       let compareNum = code.length;
       if (type === 'province') {
         compareNum = props.isOverseaCode(code) ? 1 : 2;
diff --git a/src/area/README.md b/src/area/README.md
index c04c2f9c3..e3ef848a7 100644
--- a/src/area/README.md
+++ b/src/area/README.md
@@ -163,3 +163,16 @@ Use [ref](https://v3.vuejs.org/guide/component-template-refs.html) to get Area i
 | Name  | Description               | Attribute       | Return value |
 | ----- | ------------------------- | --------------- | ------------ |
 | reset | Reset all options by code | _code?: string_ | -            |
+
+### Types
+
+Get the type definition of the Area instance through `AreaInstance`.
+
+```ts
+import { ref } from 'vue';
+import type { AreaInstance } from 'vant';
+
+const areaRef = ref<AreaInstance>();
+
+areaRef.value?.reset();
+```
diff --git a/src/area/README.zh-CN.md b/src/area/README.zh-CN.md
index 3f64c2c44..490023305 100644
--- a/src/area/README.zh-CN.md
+++ b/src/area/README.zh-CN.md
@@ -166,6 +166,19 @@ confirm 事件返回的数据整体为一个数组，数组每一项对应一列
 | --- | --- | --- | --- |
 | reset | 根据地区码重置所有选项，若不传地区码，则重置到第一项 | _code?: string_ | - |
 
+### 类型定义
+
+通过 `AreaInstance` 获取 Area 实例的类型定义。
+
+```ts
+import { ref } from 'vue';
+import type { AreaInstance } from 'vant';
+
+const areaRef = ref<AreaInstance>();
+
+areaRef.value?.reset();
+```
+
 ## 常见问题
 
 ### 在桌面端无法操作组件？
diff --git a/src/area/index.ts b/src/area/index.ts
index 2e877b7f1..01266b9f7 100644
--- a/src/area/index.ts
+++ b/src/area/index.ts
@@ -5,4 +5,4 @@ const Area = withInstall<typeof _Area>(_Area);
 
 export default Area;
 export { Area };
-export type { AreaList, AreaColumnOption } from './Area';
+export type { AreaList, AreaInstance, AreaColumnOption } from './types';
diff --git a/src/area/types.ts b/src/area/types.ts
new file mode 100644
index 000000000..02eeff5b5
--- /dev/null
+++ b/src/area/types.ts
@@ -0,0 +1,31 @@
+/* eslint-disable camelcase */
+
+import { ComponentPublicInstance } from 'vue';
+import { AreaProps } from './Area';
+
+export type AreaList = {
+  city_list: Record<string, string>;
+  county_list: Record<string, string>;
+  province_list: Record<string, string>;
+};
+
+export type AreaColumnOption = {
+  name: string;
+  code: string;
+};
+
+export type AreaColumnType = 'province' | 'county' | 'city';
+
+export type AreaExpose = {
+  reset: (newCode?: string) => void;
+  getArea: () => {
+    code: string;
+    country: string;
+    province: string;
+    city: string;
+    county: string;
+  };
+  getValues: () => AreaColumnOption[];
+};
+
+export type AreaInstance = ComponentPublicInstance<AreaProps, AreaExpose>;