feat(types): infer defaultValue of argtype based on arg

storybookjs · Aug 9, 2021 · bfc6632 · bfc6632
1 parent a8a41dc
commit bfc6632
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/lib/addons/src/types.ts b/lib/addons/src/types.ts
@@ -36,16 +36,20 @@ export interface Args {
   [key: string]: any;
 }
 
-export interface ArgType {
+export interface ArgType<Arg = unknown> {
   name?: string;
   description?: string;
-  defaultValue?: any;
+  defaultValue?: Arg;
   [key: string]: any;
 }
 
-export type ArgTypes<Args = Record<string, any>> = {
-  [key in keyof Partial<Args>]: ArgType;
-};
+export type ArgTypes<GenericArgs = Args> = {
+  [key in keyof Partial<GenericArgs>]: ArgType<GenericArgs[key]>;
+} &
+  {
+    // for custom defined args
+    [key in string]: ArgType<unknown>;
+  };
 
 export interface StoryIdentifier {
   id: StoryId;