generateForAnnotatedElement method
String
generateForAnnotatedElement(
- Element element,
- ConstantReader annotation,
- BuildStep buildStep
Implement to return source code to generate for element
.
This method is invoked based on finding elements annotated with an
instance of T
. The annotation
is provided as a ConstantReader
.
Supported return values include a single String or multiple String instances within an Iterable or Stream. It is also valid to return a Future of String, Iterable, or Stream.
Implementations should return null
when no content is generated. Empty
or whitespace-only String instances are also ignored.
Implementation
@override
String generateForAnnotatedElement(
Element element,
ConstantReader annotation,
BuildStep buildStep,
) {
if (element is! ClassElement) {
throw InvalidGenerationSourceError(
'Luthor can only be applied to classes.',
element: element,
);
}
if (element.constructors.isEmpty) {
throw InvalidGenerationSourceError(
'Luthor can only be applied to classes with at least one constructor.',
element: element,
);
}
final name = element.name;
final constructor =
element.constructors.firstWhereOrNull((c) => c.isFactory);
if (constructor == null) {
throw InvalidGenerationSourceError(
'Luthor can only be applied to classes with a factory constructor.',
element: element,
);
}
final params = constructor.parameters;
final buffer = StringBuffer();
buffer.write('Validator \$${name}Schema = l.schema({\n');
for (final param in params) {
var name = param.name;
final jsonKeyName = jsonKeyChecker
.firstAnnotationOf(param)
?.getField('name')
?.toStringValue();
if (jsonKeyName != null) {
name = jsonKeyName;
}
buffer.write("'$name': ");
buffer.write(getValidations(param));
buffer.write(',\n');
}
buffer.write('});\n\n');
_writeValidateMethod(buffer, name);
_writeExtension(buffer, name);
return buffer.toString();
}