Update service generator

This commit is contained in:
Tobe O 2018-07-14 18:02:56 -04:00
parent 4c9e1c78ce
commit c676bb18fd
2 changed files with 77 additions and 67 deletions

View file

@ -1,16 +1,17 @@
import 'dart:io';
import 'package:args/command_runner.dart';
import 'package:code_builder/code_builder.dart';
import 'package:console/console.dart';
import 'package:dart_style/dart_style.dart';
import 'package:inflection/inflection.dart';
import 'package:io/ansi.dart';
import 'package:prompts/prompts.dart' as prompts;
import 'package:pubspec_parse/pubspec_parse.dart';
import 'package:recase/recase.dart';
import '../service_generators/service_generators.dart';
import '../../util.dart';
import 'maker.dart';
class ServiceCommand extends Command {
final TextPen _pen = new TextPen();
@override
String get name => 'service';
@ -32,39 +33,30 @@ class ServiceCommand extends Command {
@override
run() async {
var pubspec = await Pubspec.load(Directory.current);
var pubspec = await loadPubspec();
String name;
if (argResults.wasParsed('name')) name = argResults['name'];
if (argResults.wasParsed('name')) name = argResults['name'] as String;
if (name?.isNotEmpty != true) {
var p = new Prompter('Name of Service: ');
name = await p.prompt(checker: (s) => s.isNotEmpty);
name = prompts.get('Name of service');
}
List<MakerDependency> deps = [
const MakerDependency('angel_framework', '^1.0.0')
];
// '${pubspec.name}.src.services.${rc.snakeCase}'
var rc = new ReCase(name);
var serviceLib =
new LibraryBuilder('${pubspec.name}.src.services.${rc.snakeCase}');
var serviceLib = new Library((serviceLib) {
var generator = prompts.choose(
'Choose which type of service to create', serviceGenerators);
ServiceGenerator generator;
var chooser = new Chooser<String>(
serviceGenerators.map<String>((g) => g.name).toList(),
message: 'What type of service would you like to create? ');
var type = await chooser.choose();
generator =
serviceGenerators.firstWhere((g) => g.name == type, orElse: () => null);
if (generator == null) {
_pen.red();
_pen('${Icon.BALLOT_X} \'$type\' services are not yet implemented. :(');
_pen();
throw 'Unrecognized service type: "$type".';
}
// if (generator == null) {
// _pen.red();
// _pen('${Icon.BALLOT_X} \'$type\' services are not yet implemented. :(');
// _pen();
// throw 'Unrecognized service type: "$type".';
// }
for (var dep in generator.dependencies) {
if (!deps.any((d) => d.name == dep.name)) deps.add(dep);
@ -72,54 +64,72 @@ class ServiceCommand extends Command {
if (generator.goesFirst) {
generator.applyToLibrary(serviceLib, name, rc.snakeCase);
serviceLib.addMember(
new ImportBuilder('package:angel_framework/angel_framework.dart'));
serviceLib.directives.add(new Directive.import(
'package:angel_framework/angel_framework.dart'));
} else {
serviceLib.addMember(
new ImportBuilder('package:angel_framework/angel_framework.dart'));
serviceLib.directives.add(new Directive.import(
'package:angel_framework/angel_framework.dart'));
generator.applyToLibrary(serviceLib, name, rc.snakeCase);
}
if (argResults['typed']) {
serviceLib
..addMember(new ImportBuilder('../models/${rc.snakeCase}.dart'));
if (argResults['typed'] as bool) {
serviceLib.directives
.add(new Directive.import('../models/${rc.snakeCase}.dart'));
}
// configureServer() {}
var configureServer = new MethodBuilder('configureServer',
returnType: new TypeBuilder('AngelConfigurer'));
generator.applyToConfigureServer(configureServer, name, rc.snakeCase);
serviceLib.body.add(new Method((configureServer) {
configureServer
..name = 'configureServer'
..returns = refer('AngelConfigurer');
configureServer.body = new Block((block) {
generator.applyToConfigureServer(
configureServer, block, name, rc.snakeCase);
// return (Angel app) async {}
var closure = new MethodBuilder.closure(modifier: MethodModifier.asAsync)
..addPositional(parameter('app', [new TypeBuilder('Angel')]));
generator.beforeService(closure, name, rc.snakeCase);
var closure = new Method((closure) {
closure
..modifier = MethodModifier.async
..requiredParameters.add(new Parameter((b) => b
..name = 'app'
..type = refer('Angel')));
closure.body = new Block((block) {
generator.beforeService(block, name, rc.snakeCase);
// app.use('/api/todos', new MapService());
var service = generator.createInstance(closure, name, rc.snakeCase);
var service =
generator.createInstance(closure, name, rc.snakeCase);
if (argResults['typed']) {
service = new TypeBuilder('TypedService',
genericTypes: [new TypeBuilder(rc.pascalCase)])
.newInstance([service]);
if (argResults['typed'] as bool) {
var tb = new TypeReference((b) => b
..symbol = 'TypedService'
..types.add(refer(rc.pascalCase)));
service = tb.newInstance([service]);
}
closure.addStatement(reference('app')
.invoke('use', [literal('/api/${pluralize(rc.snakeCase)}'), service]));
configureServer.addStatement(closure.asReturn());
serviceLib.addMember(configureServer);
block.addExpression(refer('app').property('use').call([
literal('/api/${pluralize(rc.snakeCase)}'),
service,
]));
});
});
block.addExpression(closure.closure.returned);
});
}));
});
final outputDir = new Directory.fromUri(
Directory.current.uri.resolve(argResults['output-dir']));
Directory.current.uri.resolve(argResults['output-dir'] as String));
final serviceFile =
new File.fromUri(outputDir.uri.resolve("${rc.snakeCase}.dart"));
if (!await serviceFile.exists()) await serviceFile.create(recursive: true);
await serviceFile.writeAsString(prettyToSource(serviceLib.buildAst()));
await serviceFile.writeAsString(new DartFormatter()
.format(serviceLib.accept(new DartEmitter()).toString()));
_pen.green();
_pen(
'${Icon.CHECKMARK} Successfully generated service file "${serviceFile.absolute.path}".');
_pen();
print(green.wrap(
'$checkmark Successfully generated service file "${serviceFile.absolute.path}".'));
if (deps.isNotEmpty) await depend(deps);
}

View file

@ -27,12 +27,12 @@ class ServiceGenerator {
void applyToLibrary(LibraryBuilder library, String name, String lower) {}
void beforeService(MethodBuilder methodBuilder, String name, String lower) {}
void beforeService(BlockBuilder builder, String name, String lower) {}
void applyToConfigureServer(
MethodBuilder configureServer, String name, String lower) {}
MethodBuilder configureServer, BlockBuilder block, String name, String lower) {}
ExpressionBuilder createInstance(
Expression createInstance(
MethodBuilder methodBuilder, String name, String lower) =>
literal(null);
}