89 lines
2.8 KiB
TypeScript
89 lines
2.8 KiB
TypeScript
/**
|
||
* 支持通过 name 路径批量/单项赋值 Pinia store 的工具方法
|
||
* @param store Pinia store 实例
|
||
* @param params { name: string, value: any }
|
||
* name 支持 a.b.c 形式嵌套赋值
|
||
*/
|
||
/**
|
||
* 用法示例：
|
||
* setStoreValue(userStore, { name: 'profile.avatar', value: 'xxx.png' })
|
||
* setStoreValueTyped(userStore, 'token', 'xxxx')
|
||
* setStoreValues(userStore, [{ name: 'token', value: 'xxx' }, { name: 'profile.avatar', value: 'img.png' }])
|
||
* getStoreValue(userStore, 'profile.avatar')
|
||
* resetStore(userStore, { token: '', profile: { avatar: '' } })
|
||
*/
|
||
export function setStoreValue<T extends object>(store: T, params: { name: string; value: any }): void {
|
||
const nameArr = params.name.split('.');
|
||
let obj: any = store;
|
||
if (nameArr.length >= 2) {
|
||
for (let i = 0; i < nameArr.length - 1; i++) {
|
||
if (!(nameArr[i] in obj)) {
|
||
obj[nameArr[i]] = {};
|
||
}
|
||
obj = obj[nameArr[i]];
|
||
}
|
||
obj[nameArr[nameArr.length - 1]] = params.value;
|
||
} else {
|
||
(store as any)[params.name] = params.value;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 类型安全的嵌套属性赋值工具
|
||
* @param store Pinia store 实例
|
||
* @param path 属性路径（如 profile.avatar）
|
||
* @param value 赋值内容
|
||
*/
|
||
export function setStoreValueTyped<T, K extends keyof T>(store: T, path: K, value: T[K]): void;
|
||
export function setStoreValueTyped<T>(store: T, path: string, value: any): void {
|
||
const nameArr = path.split('.');
|
||
let obj: any = store;
|
||
if (nameArr.length >= 2) {
|
||
for (let i = 0; i < nameArr.length - 1; i++) {
|
||
if (!(nameArr[i] in obj)) {
|
||
obj[nameArr[i]] = {};
|
||
}
|
||
obj = obj[nameArr[i]];
|
||
}
|
||
obj[nameArr[nameArr.length - 1]] = value;
|
||
} else {
|
||
(store as any)[path] = value;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 批量赋值 Pinia store 工具方法
|
||
* @param store Pinia store 实例
|
||
* @param values { name: string, value: any }[]
|
||
*/
|
||
export function setStoreValues<T extends object>(store: T, values: Array<{ name: string; value: any }>): void {
|
||
values.forEach(item => setStoreValue(store, item));
|
||
}
|
||
|
||
/**
|
||
* 获取嵌套属性值
|
||
* @param store Pinia store 实例
|
||
* @param path 属性路径（如 profile.avatar）
|
||
* @returns 属性值
|
||
*/
|
||
export function getStoreValue<T>(store: T, path: string): any {
|
||
const nameArr = path.split('.');
|
||
let obj: any = store;
|
||
for (let i = 0; i < nameArr.length; i++) {
|
||
if (obj == null) return undefined;
|
||
obj = obj[nameArr[i]];
|
||
}
|
||
return obj;
|
||
}
|
||
|
||
/**
|
||
* 重置 Pinia store 为初始值
|
||
* @param store Pinia store 实例
|
||
* @param initial 初始值对象
|
||
*/
|
||
export function resetStore<T extends object>(store: T, initial: Partial<T>): void {
|
||
Object.keys(initial).forEach(key => {
|
||
(store as any)[key] = (initial as any)[key];
|
||
});
|
||
}
|