feat: implement generator di module

This commit is contained in:
Sergey Penkovsky
2025-05-17 00:34:56 +03:00
parent b906e927c3
commit d1e726aaec
18 changed files with 118 additions and 129 deletions

View File

@@ -1,8 +1,8 @@
builders:
injectable:
import: "package:cherrypick_generator/injectable_generator.dart"
builder_factories: ["injectableBuilder"]
build_extensions: {".dart": [".cherrypick_injectable.g.dart"]}
module_generator:
import: "package:cherrypick_generator/module_generator.dart"
builder_factories: ["moduleBuilder"]
build_extensions: {".dart": [".cherrypick.g.dart"]}
auto_apply: dependents
required_inputs: ["lib/**"]
runs_before: []
@@ -11,6 +11,6 @@ builders:
targets:
$default:
builders:
cherrypick_generator|injectable:
cherrypick_generator|module_generator:
generate_for:
- lib/**.dart

View File

@@ -1,3 +1,3 @@
library;
export 'inject_generator.dart';
export 'module_generator.dart';

View File

@@ -1,38 +0,0 @@
import 'dart:async';
import 'package:build/build.dart';
import 'package:source_gen/source_gen.dart';
import 'package:analyzer/dart/element/element.dart';
import 'package:cherrypick_annotations/cherrypick_annotations.dart';
class InjectGenerator extends GeneratorForAnnotation<Injectable> {
@override
FutureOr<String> generateForAnnotatedElement(
Element element,
ConstantReader annotation,
BuildStep buildStep,
) {
print('[TRACE] Processing element: ${element.name}');
if (element is! FieldElement) {
throw InvalidGenerationSourceError(
'Inject can only be used on fields.',
element: element,
);
}
print('[TRACE] Starting code generation for element: ${element.name}');
final className = element.enclosingElement.name;
final fieldName = element.name;
final fieldType = element.type.getDisplayString(withNullability: false);
final annotationName = annotation.read('named').stringValue;
return '''
extension \$${className}Inject on $className {
void init$fieldName() {
print("Injected $fieldType named '$annotationName' into $fieldName");
}
}
''';
}
}

View File

@@ -1,33 +0,0 @@
import 'package:source_gen/source_gen.dart';
import 'package:analyzer/dart/element/element.dart';
import 'package:build/build.dart';
import 'package:cherrypick_annotations/cherrypick_annotations.dart';
class InjectableGenerator extends GeneratorForAnnotation<Injectable> {
@override
generateForAnnotatedElement(
Element element, ConstantReader annotation, BuildStep buildStep) {
if (element is! ClassElement) return null;
final className = element.name;
// Используйте уникальное имя функции (например, привязанное к файлу/классу)
return '''
void \$initCherrypickGenerated() {
print("Generate code success $className");
}
''';
}
}
Builder injectableBuilder(BuilderOptions options) =>
PartBuilder([InjectableGenerator()], '.cherrypick_injectable.g.dart');
/*
Builder injectableBuilder(BuilderOptions options) => SharedPartBuilder(
[InjectableGenerator()],
'injectable',
allowSyntaxErrors: true,
writeDescriptions: true,
);
*/

View File

@@ -0,0 +1,60 @@
// ... остальные импорты ...
import 'package:analyzer/dart/element/element.dart';
import 'package:build/build.dart';
import 'package:source_gen/source_gen.dart';
import 'package:cherrypick_annotations/cherrypick_annotations.dart' as ann;
class ModuleGenerator extends GeneratorForAnnotation<ann.module> {
@override
String generateForAnnotatedElement(
Element element, ConstantReader annotation, BuildStep buildStep) {
if (element is! ClassElement) {
throw InvalidGenerationSourceError(
'@module() может быть применён только к классам.',
element: element,
);
}
final classElement = element;
final className = classElement.displayName;
final generatedClassName = r'$' + className;
final buffer = StringBuffer();
//buffer.writeln("part of '${buildStep.inputId.uri.pathSegments.last}';\n");
buffer.writeln('final class $generatedClassName extends $className {');
buffer.writeln(' @override');
buffer.writeln(' void builder(Scope currentScope) {');
for (final method in classElement.methods.where((m) => !m.isAbstract)) {
final hasSingleton = method.metadata.any(
(m) =>
m
.computeConstantValue()
?.type
?.getDisplayString(withNullability: false)
.toLowerCase()
.contains('singleton') ??
false,
);
if (!hasSingleton) continue;
final returnType =
method.returnType.getDisplayString(withNullability: false);
final methodName = method.displayName;
final args = method.parameters
.map((p) =>
"currentScope.resolve<${p.type.getDisplayString(withNullability: false)}>()")
.join(', ');
buffer.write(' bind<$returnType>()'
'.toProvide(() => $methodName($args))'
'.singleton();\n');
}
buffer.writeln(' }\n}');
return buffer.toString();
}
}
Builder moduleBuilder(BuilderOptions options) =>
PartBuilder([ModuleGenerator()], '.cherrypick.g.dart');