fix: added error when WidgetbookTheme is used on non-function code el…

…ements
widgetbook · Mar 23, 2022 · b7e65c2 · b7e65c2
1 parent 4f9f488
commit b7e65c2
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/packages/widgetbook_generator/lib/resolvers/theme_resolver.dart b/packages/widgetbook_generator/lib/resolvers/theme_resolver.dart
@@ -20,11 +20,20 @@ class ThemeResolver extends GeneratorForAnnotation<WidgetbookTheme> {
       );
     }
 
+    if (element.kind != ElementKind.FUNCTION ||
+        !(element as FunctionElement).isStatic) {
+      throw InvalidGenerationSourceError(
+        'The $WidgetbookTheme annotation cannot be applied to static '
+        'functions.',
+        element: element,
+      );
+    }
+
     final isDefault = annotation.read('isDefault').boolValue;
     final name = annotation.read('name').stringValue;
 
     final themeData = WidgetbookThemeData(
-      name: element.name!,
+      name: element.name,
       importStatement: element.importStatement,
       dependencies: element.dependencies,
       isDefault: isDefault,