feat(generator): ✨ add LocalizationAddon to generator

packages/widgetbook_generator/CHANGELOG.md

@@ -3,6 +3,7 @@
 - **BREAKING**: refactor: remove `DeviceFrameBuilderResolver`, `LocalizationBuilderResolver`, `ScaffoldBuilderResolver`, `ThemeBuilderResolver`, `UseCaseBuilderResolver`.
 - feat: add `MaterialThemeAddon` to generator
 - feat: add `TextScaleAddon` to generator
+- feat: add `LocalizationAddon` to generator
 
 ## 2.4.1
 

packages/widgetbook_generator/lib/code_generators/instances/addons/addons.dart

@@ -1,2 +1,4 @@
+export './addon_instance.dart';
+export './localization_addon/localization_addon.dart';
 export './material_theme_addon/material_theme_addon.dart';
 export './text_scale_addon/text_scale_addon.dart';

packages/widgetbook_generator/lib/code_generators/instances/addons/localization_addon/localization_addon.dart

@@ -0,0 +1,2 @@
+export 'localization_addon_instance.dart';
+export 'localization_setting_instance.dart';

packages/widgetbook_generator/lib/code_generators/instances/addons/localization_addon/localization_addon_instance.dart

@@ -0,0 +1,22 @@
+import 'package:widgetbook_generator/code_generators/instances/instances.dart';
+import 'package:widgetbook_generator/code_generators/properties/property.dart';
+import 'package:widgetbook_generator/models/widgetbook_locales_data.dart';
+import 'package:widgetbook_generator/models/widgetbook_localizations_delegates_data.dart';
+
+class LocalizationAddonInstance extends AddOnInstance {
+  LocalizationAddonInstance({
+    required WidgetbookLocalesData localesData,
+    required WidgetbookLocalizationsDelegatesData localizationDelegatesData,
+  }) : super(
+          name: 'LocalizationAddon',
+          properties: [
+            Property(
+              key: 'settings',
+              instance: LocalizationSettingInstance(
+                localesData: localesData,
+                localizationDelegatesData: localizationDelegatesData,
+              ),
+            ),
+          ],
+        );
+}

packages/widgetbook_generator/lib/code_generators/instances/addons/localization_addon/localization_setting_instance.dart

@@ -0,0 +1,33 @@
+import 'package:widgetbook_generator/code_generators/instances/instance.dart';
+import 'package:widgetbook_generator/code_generators/instances/variable_instance.dart';
+import 'package:widgetbook_generator/code_generators/properties/property.dart';
+import 'package:widgetbook_generator/models/widgetbook_locales_data.dart';
+import 'package:widgetbook_generator/models/widgetbook_localizations_delegates_data.dart';
+
+class LocalizationSettingInstance extends Instance {
+  LocalizationSettingInstance({
+    required WidgetbookLocalesData localesData,
+    required WidgetbookLocalizationsDelegatesData localizationDelegatesData,
+  }) : super(
+          name: 'LocalizationSetting',
+          properties: [
+            Property(
+              key: 'locales',
+              instance: VariableInstance(variableIdentifier: localesData.name),
+            ),
+            // TODO this is a big hacky
+            Property(
+              key: 'activeLocale',
+              instance: VariableInstance(
+                variableIdentifier: '${localesData.name}.first',
+              ),
+            ),
+            Property(
+              key: 'localizationsDelegates',
+              instance: VariableInstance(
+                variableIdentifier: localizationDelegatesData.name,
+              ),
+            ),
+          ],
+        );
+}

packages/widgetbook_generator/lib/generators/app_generator.dart

@@ -1,5 +1,4 @@
 import 'package:widgetbook_annotation/widgetbook_annotation.dart';
-import 'package:widgetbook_generator/code_generators/instances/addons/addon_instance.dart';
 import 'package:widgetbook_generator/code_generators/instances/addons/addons.dart';
 import 'package:widgetbook_generator/code_generators/instances/app_info_instance.dart';
 import 'package:widgetbook_generator/code_generators/instances/variable_instance.dart';
@@ -39,6 +38,14 @@ String generateWidgetbook({
   }
 
   addons.add(TextScaleAddonInstance(textScales: textScaleFactors));
+  if (localesData != null && localizationDelegatesData != null) {
+    addons.add(
+      LocalizationAddonInstance(
+        localesData: localesData,
+        localizationDelegatesData: localizationDelegatesData,
+      ),
+    );
+  }
 
   final widgetbookInstanceCode = WidgetbookInstance(
     constructor: constructor,

packages/widgetbook_generator/test/src/code_generators/instances/addons/localization_addon/localization_addon_instance_test.dart

@@ -0,0 +1,28 @@
+import 'package:test/test.dart';
+import 'package:widgetbook_generator/code_generators/instances/instances.dart';
+import 'package:widgetbook_generator/models/widgetbook_locales_data.dart';
+import 'package:widgetbook_generator/models/widgetbook_localizations_delegates_data.dart';
+
+void main() {
+  group(
+    '$LocalizationAddonInstance',
+    () {
+      test('.name is "LocalizationAddon"', () {
+        final sut = LocalizationAddonInstance(
+          localesData: WidgetbookLocalesData(
+            name: 'locales',
+            importStatement: '',
+            dependencies: [],
+            locales: [],
+          ),
+          localizationDelegatesData: WidgetbookLocalizationsDelegatesData(
+            name: 'delegates',
+            importStatement: '',
+            dependencies: [],
+          ),
+        );
+        expect(sut.name, equals('LocalizationAddon'));
+      });
+    },
+  );
+}

packages/widgetbook_generator/test/src/code_generators/instances/addons/localization_addon/localization_setting_instance_test.dart

@@ -0,0 +1,30 @@
+import 'package:test/test.dart';
+import 'package:widgetbook_generator/code_generators/instances/instances.dart';
+import 'package:widgetbook_generator/models/widgetbook_locales_data.dart';
+import 'package:widgetbook_generator/models/widgetbook_localizations_delegates_data.dart';
+
+void main() {
+  group('$LocalizationSettingInstance', () {
+    test(
+      '.name returns "LocalizationSetting"',
+      () {
+        expect(
+          LocalizationSettingInstance(
+            localesData: WidgetbookLocalesData(
+              name: 'locales',
+              importStatement: '',
+              dependencies: [],
+              locales: [],
+            ),
+            localizationDelegatesData: WidgetbookLocalizationsDelegatesData(
+              name: 'delegates',
+              importStatement: '',
+              dependencies: [],
+            ),
+          ).name,
+          equals('LocalizationSetting'),
+        );
+      },
+    );
+  });
+}