support nested types

eure · Feb 22, 2024 · 65ab543 · 65ab543
1 parent 178ee53
commit 65ab543
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/Sources/StorybookMacrosPlugin/StorybookPageMacro.swift b/Sources/StorybookMacrosPlugin/StorybookPageMacro.swift
@@ -68,8 +68,11 @@ public struct StorybookPageMacro: DeclarationMacro {
     closure: ClosureExprSyntax
   ) {
     var argumentsIterator = node.argumentList.makeIterator()
-    var title: ExprSyntax? = node.genericArgumentClause?.arguments.first?.argument
-      .as(IdentifierTypeSyntax.self)
+    var title: ExprSyntax? = (node.genericArgumentClause?.arguments.first?.argument)
+      .flatMap { genericType -> TypeSyntaxProtocol? in
+        genericType.as(IdentifierTypeSyntax.self)
+        ?? genericType.as(MemberTypeSyntax.self)
+      }
       .map({ .init(stringLiteral: "_typeName(\($0).self)") })
 
     var closure: ClosureExprSyntax? = node.trailingClosure