generate method

String generate({ required String name, required List<RepoMethod> methods, })

Implementation

String generate({required String name, required List<RepoMethod> methods}) {
  final interfaceName = '${name}Repository';
  final implName = '${name}RepositoryImpl';
  final datasourceName = '\$${name}Datasource';

  final library = cb.Library((b) {
    b.directives.add(cb.Directive.import('package:zuraffa/zuraffa.dart'));

    // Interface
    b.body.add(
      cb.Class((c) {
        c
          ..name = interfaceName
          ..abstract = true;

        for (final method in methods) {
          final returnTypeStr = method.kind == RepoMethodKind.subscription
              ? 'Stream<SignalResult<${zorphyType(typeMapper, method.returnType)}>>'
              : 'Future<SignalResult<${zorphyType(typeMapper, method.returnType)}>>';

          c.methods.add(
            cb.Method((m) {
              m
                ..name = method.name
                ..returns = cb.refer(returnTypeStr)
                ..requiredParameters.addAll(
                  method.args.map(
                    (arg) => cb.Parameter((p) {
                      p
                        ..name = TypeMapper.fieldName(arg.name)
                        ..type = cb.refer(typeMapper.mapType(arg.type));
                    }),
                  ),
                );
            }),
          );
        }
      }),
    );

    // Implementation
    b.body.add(
      cb.Class((c) {
        c
          ..name = implName
          ..implements.add(cb.refer(interfaceName));

        c.fields.add(
          cb.Field((f) {
            f
              ..name = '_datasource'
              ..modifier = cb.FieldModifier.final$
              ..type = cb.refer(datasourceName);
          }),
        );

        c.constructors.add(
          cb.Constructor((ctor) {
            ctor.requiredParameters.add(
              cb.Parameter((p) {
                p
                  ..name = 'datasource'
                  ..toThis = true;
              }),
            );
          }),
        );

        for (final method in methods) {
          final returnTypeStr = method.kind == RepoMethodKind.subscription
              ? 'Stream<SignalResult<${zorphyType(typeMapper, method.returnType)}>>'
              : 'Future<SignalResult<${zorphyType(typeMapper, method.returnType)}>>';

          c.methods.add(
            cb.Method((m) {
              m
                ..name = method.name
                ..annotations.add(cb.refer('override'))
                ..returns = cb.refer(returnTypeStr)
                ..requiredParameters.addAll(
                  method.args.map(
                    (arg) => cb.Parameter((p) {
                      p
                        ..name = TypeMapper.fieldName(arg.name)
                        ..type = cb.refer(typeMapper.mapType(arg.type));
                    }),
                  ),
                )
                ..body = cb.Block((bl) {
                  final argNames = method.args
                      .map((a) => TypeMapper.fieldName(a.name))
                      .join(', ');
                  bl.statements.add(
                    cb.Code('return _datasource.${method.name}($argNames);'),
                  );
                });
            }),
          );
        }
      }),
    );
  });

  final emitter = cb.DartEmitter();
  final raw = library.accept(emitter).toString();
  var formatted = raw;
  try {
    formatted = _formatter.format(raw);
  } on FormatterException {
    // Fallback: unformatted code is better than a crash.
  }
  return formatted;
}