// copy from element-plus
|
|
import type { ComputedRef, ExtractPropTypes, PropType, Ref } from 'vue';
|
|
import { unref, warn } from 'vue';
|
|
import { isObject } from '@vben/utils';
|
|
import { fromPairs } from 'lodash-es';
|
|
const wrapperKey = Symbol('wrapperKey');
|
export type PropWrapper<T> = { [wrapperKey]: T };
|
|
export const propKey = Symbol('propKey');
|
|
type IfUnknown<T, V> = [unknown] extends [T] ? V : T;
|
type ResolveProp<T> = ExtractPropTypes<{
|
key: { required: true; type: T };
|
}>['key'];
|
type ResolvePropType<T> =
|
ResolveProp<T> extends { type: infer V } ? V : ResolveProp<T>;
|
type ResolvePropTypeWithReadonly<T> =
|
Readonly<T> extends Readonly<Array<infer A>>
|
? ResolvePropType<A[]>
|
: ResolvePropType<T>;
|
type _BuildPropType<T, V, C> =
|
| C
|
| (T extends PropWrapper<unknown>
|
? T[typeof wrapperKey]
|
: [V] extends [never]
|
? ResolvePropTypeWithReadonly<T>
|
: never)
|
| V;
|
|
export type BuildPropType<T, V, C> = _BuildPropType<
|
IfUnknown<T, never>,
|
IfUnknown<V, never>,
|
IfUnknown<C, never>
|
>;
|
|
export type BuildPropOption<T, D extends BuildPropType<T, V, C>, R, V, C> = {
|
default?: R extends true
|
? never
|
: D extends Array<any> | Record<string, unknown>
|
? () => D
|
: (() => D) | D;
|
required?: R;
|
type?: T;
|
validator?: ((val: any) => boolean) | ((val: any) => val is C);
|
values?: readonly V[];
|
};
|
|
type _BuildPropDefault<T, D> = [T] extends [
|
// eslint-disable-next-line @typescript-eslint/no-unsafe-function-type
|
Array<any> | Function | Record<string, unknown>,
|
]
|
? D
|
: D extends () => T
|
? ReturnType<D>
|
: D;
|
|
export type BuildPropDefault<T, D, R> = R extends true
|
? { readonly default?: undefined }
|
: {
|
readonly default: Exclude<D, undefined> extends never
|
? undefined
|
: Exclude<_BuildPropDefault<T, D>, undefined>;
|
};
|
export type BuildPropReturn<T, D, R, V, C> = BuildPropDefault<
|
BuildPropType<T, V, C>,
|
IfUnknown<D, never>,
|
IfUnknown<R, false>
|
> & {
|
[propKey]: true;
|
readonly required: IfUnknown<R, false>;
|
readonly type: PropType<BuildPropType<T, V, C>>;
|
readonly validator: ((val: unknown) => boolean) | undefined;
|
};
|
|
/**
|
* @description Build prop. It can better optimize prop types
|
* @description 生成 prop，能更好地优化类型
|
* @example
|
// limited options
|
// the type will be PropType<'light' | 'dark'>
|
buildProp({
|
type: String,
|
values: ['light', 'dark'],
|
} as const)
|
* @example
|
// limited options and other types
|
// the type will be PropType<'small' | 'medium' | number>
|
buildProp({
|
type: [String, Number],
|
values: ['small', 'medium'],
|
validator: (val: unknown): val is number => typeof val === 'number',
|
} as const)
|
@link see more: https://github.com/element-plus/element-plus/pull/3341
|
*/
|
export function buildProp<
|
T = never,
|
D extends BuildPropType<T, V, C> = never,
|
R extends boolean = false,
|
V = never,
|
C = never,
|
>(
|
option: BuildPropOption<T, D, R, V, C>,
|
key?: string,
|
): BuildPropReturn<T, D, R, V, C> {
|
// filter native prop type and nested prop, e.g `null`, `undefined` (from `buildProps`)
|
if (!isObject(option) || !!option[propKey]) return option as any;
|
|
const { default: defaultValue, required, type, validator, values } = option;
|
|
const _validator =
|
values || validator
|
? (val: unknown) => {
|
let valid = false;
|
let allowedValues: unknown[] = [];
|
|
if (values) {
|
allowedValues = [...values, defaultValue];
|
valid || (valid = allowedValues.includes(val));
|
}
|
if (validator) valid || (valid = validator(val));
|
|
if (!valid && allowedValues.length > 0) {
|
const allowValuesText = [...new Set(allowedValues)]
|
.map((value) => JSON.stringify(value))
|
.join(', ');
|
warn(
|
`Invalid prop: validation failed${key ? ` for prop "${key}"` : ''}. Expected one of [${allowValuesText}], got value ${JSON.stringify(val)}.`,
|
);
|
}
|
return valid;
|
}
|
: undefined;
|
|
return {
|
default: defaultValue,
|
[propKey]: true,
|
required: !!required,
|
type:
|
typeof type === 'object' &&
|
Object.getOwnPropertySymbols(type).includes(wrapperKey)
|
? (type as any)[wrapperKey]
|
: type,
|
validator: _validator,
|
} as unknown as BuildPropReturn<T, D, R, V, C>;
|
}
|
|
type NativePropType = [
|
((...args: any) => any) | null | undefined | { new (...args: any): any },
|
];
|
|
export const buildProps = <
|
O extends {
|
[K in keyof O]: O[K] extends BuildPropReturn<any, any, any, any, any>
|
? O[K]
|
: [O[K]] extends NativePropType
|
? O[K]
|
: O[K] extends BuildPropOption<
|
infer T,
|
infer D,
|
infer R,
|
infer V,
|
infer C
|
>
|
? D extends BuildPropType<T, V, C>
|
? BuildPropOption<T, D, R, V, C>
|
: never
|
: never;
|
},
|
>(
|
props: O,
|
) =>
|
fromPairs(
|
Object.entries(props).map(([key, option]) => [
|
key,
|
buildProp(option as any, key),
|
]),
|
) as unknown as {
|
[K in keyof O]: O[K] extends { [propKey]: boolean }
|
? O[K]
|
: [O[K]] extends NativePropType
|
? O[K]
|
: O[K] extends BuildPropOption<
|
infer T,
|
infer _D,
|
infer R,
|
infer V,
|
infer C
|
>
|
? BuildPropReturn<T, O[K]['default'], R, V, C>
|
: never;
|
};
|
|
export const definePropType = <T>(val: any) =>
|
({ [wrapperKey]: val }) as PropWrapper<T>;
|
|
export const keyOf = <T>(arr: T) => Object.keys(arr as any) as Array<keyof T>;
|
|
type Mutable<T> = { -readonly [P in keyof T]: T[P] };
|
export const mutable = <T extends readonly any[] | Record<string, unknown>>(
|
val: T,
|
) => val as Mutable<typeof val>;
|
|
export const componentSize = ['large', 'medium', 'small', 'mini'] as const;
|
|
// dynamic use hook props
|
export type DynamicProps<T> = {
|
[P in keyof T]: ComputedRef<T[P]> | Ref<T[P]> | T[P];
|
};
|
export function getDynamicProps<T extends Record<string, unknown>, U>(
|
props: T,
|
): Partial<U> {
|
const ret: Recordable = {};
|
|
Object.keys(props).map((key) => {
|
ret[key] = unref((props as Recordable)[key]);
|
return key;
|
});
|
|
return ret as Partial<U>;
|
}
|
