From e7d7c9cc048e6650d73a33d782c6bea25c7c348c Mon Sep 17 00:00:00 2001
From: kahirokunn <okinakahiro@gmail.com>
Date: Thu, 13 Aug 2020 12:19:40 +0900
Subject: [PATCH] Allowing ArgsStoryFn's Args type to be more specific.

---
 lib/addons/src/types.ts | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/lib/addons/src/types.ts b/lib/addons/src/types.ts
index e2efc16e9423..77b463d8627e 100644
--- a/lib/addons/src/types.ts
+++ b/lib/addons/src/types.ts
@@ -92,8 +92,13 @@ export interface OptionsParameter extends Object {
 export type StoryGetter = (context: StoryContext) => any;
 
 export type LegacyStoryFn<ReturnType = unknown> = (p?: StoryContext) => ReturnType;
-export type ArgsStoryFn<ReturnType = unknown> = (a?: Args, p?: StoryContext) => ReturnType;
-export type StoryFn<ReturnType = unknown> = LegacyStoryFn<ReturnType> | ArgsStoryFn<ReturnType>;
+export type ArgsStoryFn<ReturnType = unknown, ArgsType extends Args = Args> = (
+  a?: ArgsType,
+  p?: StoryContext
+) => ReturnType;
+export type StoryFn<ReturnType = unknown, ArgsType = Args> =
+  | LegacyStoryFn<ReturnType>
+  | ArgsStoryFn<ReturnType, ArgsType>;
 
 export type StoryWrapper = (
   getStory: StoryGetter,