penkovsky_sa/cherrypick
1
0
Fork 0
mirror of https://github.com/pese-git/cherrypick.git synced 2026-01-24 13:47:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: fix generator for singletone annotation

Browse Source
This commit is contained in:
Sergey Penkovsky
2025-05-17 11:27:30 +03:00
parent d1e726aaec
commit 3d071626e5
1 changed files with 15 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
cherrypick_generator/lib/module_generator.dart
View File

@@ -7,7 +7,10 @@ import 'package:cherrypick_annotations/cherrypick_annotations.dart' as ann;
class ModuleGenerator extends GeneratorForAnnotation<ann.module> { class ModuleGenerator extends GeneratorForAnnotation<ann.module> {
@override @override
String generateForAnnotatedElement( String generateForAnnotatedElement(
Element element, ConstantReader annotation, BuildStep buildStep) { Element element,
ConstantReader annotation,
BuildStep buildStep,
) {
if (element is! ClassElement) { if (element is! ClassElement) {
throw InvalidGenerationSourceError( throw InvalidGenerationSourceError(
'@module() может быть применён только к классам.', '@module() может быть применён только к классам.',
@@ -20,10 +23,9 @@ class ModuleGenerator extends GeneratorForAnnotation<ann.module> {
final generatedClassName = r'$' + className; final generatedClassName = r'$' + className;
final buffer = StringBuffer(); final buffer = StringBuffer();
//buffer.writeln("part of '${buildStep.inputId.uri.pathSegments.last}';\n");
buffer.writeln('final class $generatedClassName extends $className {'); buffer.writeln('final class $generatedClassName extends $className {');
buffer.writeln(' @override'); buffer.writeln('@override');
buffer.writeln(' void builder(Scope currentScope) {'); buffer.writeln('void builder(Scope currentScope) {');
for (final method in classElement.methods.where((m) => !m.isAbstract)) { for (final method in classElement.methods.where((m) => !m.isAbstract)) {
final hasSingleton = method.metadata.any( final hasSingleton = method.metadata.any(
@@ -36,7 +38,6 @@ class ModuleGenerator extends GeneratorForAnnotation<ann.module> {
.contains('singleton') ?? .contains('singleton') ??
false, false,
); );
if (!hasSingleton) continue;
final returnType = final returnType =
method.returnType.getDisplayString(withNullability: false); method.returnType.getDisplayString(withNullability: false);
@@ -46,12 +47,17 @@ class ModuleGenerator extends GeneratorForAnnotation<ann.module> {
"currentScope.resolve<${p.type.getDisplayString(withNullability: false)}>()") "currentScope.resolve<${p.type.getDisplayString(withNullability: false)}>()")
.join(', '); .join(', ');
buffer.write(' bind<$returnType>()' buffer.write('bind<$returnType>()'
'.toProvide(() => $methodName($args))' '.toProvide(() => $methodName($args))');
'.singleton();\n'); if (hasSingleton) {
buffer.write('.singleton()');
}
buffer.write(';\n');
} }
buffer.writeln(' }\n}'); buffer.writeln('}');
buffer.writeln('}');
return buffer.toString(); return buffer.toString();
} }
} }