feat(angular): support context.component

Ergosign · Mar 4, 2022 · c1a4642 · c1a4642
1 parent e0dd3be
commit c1a4642
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/angular/angular.ts b/src/angular/angular.ts
@@ -85,8 +85,11 @@ export const withPseudo = makeDecorator({
   ) => {
     const story = getStory(context);
     const channel = addons.getChannel();
-    const compInternal = (story.component || context.parameters.component)
-      ?.__annotations__[0];
+    const compInternal = (
+      story?.component ??
+      context?.component ??
+      context?.parameters?.component
+    )?.__annotations__[0];
 
     // are options set by user
     const options: OptionsParameter = settings?.options;