feat: Add InitializableSingletonDependency (#9)

Add InitializableSingletonDependency
Stacked-Org · Jun 7, 2023 · 5b34ec0 · 5b34ec0
1 parent 0cac536
commit 5b34ec0
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 0 deletions.
diff --git a/lib/src/generators/getit/dependency_config/initializable_singleton_dependency.dart b/lib/src/generators/getit/dependency_config/initializable_singleton_dependency.dart
@@ -0,0 +1,30 @@
+import 'dependency_config.dart';
+import 'package:stacked_generator/src/generators/extensions/environments_extract_extension.dart';
+import 'package:stacked_generator/src/generators/extensions/string_utils_extension.dart';
+
+class InitializableSingletonDependency extends DependencyConfig {
+  const InitializableSingletonDependency({
+    required String import,
+    required String className,
+    String? abstractedImport,
+    String? abstractedTypeClassName,
+    Set<String>? environments,
+    String? instanceName,
+  }) : super(
+          instanceName: instanceName,
+          import: import,
+          className: className,
+          abstractedImport: abstractedImport,
+          abstractedTypeClassName: abstractedTypeClassName,
+          environments: environments,
+        );
+
+  @override
+  String registerDependencies(String locatorName) {
+    return '''
+        final $camelCaseClassName = $className();
+        await $camelCaseClassName.init();
+        $locatorName.registerSingleton${abstractedTypeClassName.surroundWithAngleBracketsOrReturnEmptyIfNull}($camelCaseClassName  ${environments.getFromatedEnvs}${instanceName.addInstanceNameIfNotNull});
+        ''';
+  }
+}
diff --git a/lib/src/generators/getit/dependency_config_factory.dart b/lib/src/generators/getit/dependency_config_factory.dart
@@ -12,6 +12,7 @@ import 'package:stacked_generator/src/generators/getit/stacked_locator_parameter
 import '../../../utils.dart';
 import 'dependency_config/dependency_config.dart';
 import 'dependency_config/factory_param_dependency.dart';
+import 'dependency_config/initializable_singleton_dependency.dart';
 import 'dependency_config/lazy_singleton.dart';
 
 class DependencyConfigFactory {
@@ -113,6 +114,16 @@ class DependencyConfigFactory {
           abstractedImport: abstractedImport,
           environments: environments,
           presolveFunction: presolveObject?.displayName);
+    } else if (dependencyReader
+        .instanceOf(const TypeChecker.fromRuntime(InitializableSingleton))) {
+      return InitializableSingletonDependency(
+        instanceName: instanceName,
+        import: import!,
+        className: className,
+        abstractedTypeClassName: abstractedTypeClassName,
+        abstractedImport: abstractedImport,
+        environments: environments,
+      );
     } else {
       final Set<FactoryParameter> clazzParams = {};
       var params = constructor?.parameters;