optimize types

baidu · Jul 25, 2022 · 76c778f · 76c778f
1 parent 2a7c21f
commit 76c778f
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/types/index.d.ts b/types/index.d.ts
@@ -4,7 +4,7 @@ interface DataObj {
 }
 interface ComponentContext extends Pick<Component, 'dispatch' | 'fire' | 'ref' | 'nextTick'> {
     component: Component;
-    data(name: string): DataProxy;
+    data<T extends {} = {}>(name: string): DataProxy<T>;
 }
 
 declare type Creator = (context?: ComponentContext) => void;
@@ -36,18 +36,18 @@ export declare function defineComponent<DataT extends {} = {}, OptionsT extends
 export declare function template(tpl: string): void;
 export declare function template(tpl: TemplateStringsArray, ...args: string[]): void;
 
-declare class DataProxy {
+declare class DataProxy<T extends {} = {}> {
     name: string;
     instance: {
         [key: string]: any;
     };
 
     constructor(name: string);
 
-    get(name?: string): any;
+    get(name?: keyof T extends never ? string : keyof T): any;
 
+    set(name: keyof T extends never ? string : keyof T, value?: any): void;
     set(value: any): void;
-    set(name: string, value?: any): void;
 
     merge(source: DataObj): void;
     merge(name: string, source: DataObj): void;
@@ -77,7 +77,7 @@ declare class DataProxy {
     _resolveName(name: string): string;
 }
 
-export declare function data(key: string, value: any): DataProxy | undefined;
+export declare function data<T extends {} = {}>(key: string, value: any): DataProxy<T> | undefined;
 declare class ComputedProxy {
     name: string;
     instance: {