mrpastewart/platform
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add import in ORM lib

Browse source
This commit is contained in:
Tobe O 2018-08-28 11:15:29 -04:00
parent 41f1165a1f
commit 9dffee9bc7
1 changed files with 18 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
angel_orm_generator/lib/src/orm_generator.dart
View file

@ -1,5 +1,5 @@
import 'dart:async';
import 'dart:collection';
import 'package:analyzer/dart/element/element.dart';
import 'package:angel_orm/angel_orm.dart';
import 'package:build/build.dart';
@ -44,10 +44,21 @@ class OrmGenerator extends GeneratorForAnnotation<Orm> {
Library buildOrmBaseLibrary(AssetId inputId, OrmBuildContext ctx) {
return new Library((lib) {
// Necessary imports
lib.directives.addAll([
new Directive.import('dart:async'),
new Directive.import(p.basename(inputId.uri.path)),
]);
var imports = new SplayTreeSet<String>.from(
['dart:async', p.basename(inputId.uri.path)]);
switch (ctx.ormAnnotation.type) {
case OrmType.mongoDB:
imports.add('package:mongo_dart/mongo_dart.dart');
break;
case OrmType.postgreSql:
imports.add('package:postgres/postgres.dart');
break;
default:
break;
}
lib.directives.addAll(imports.map((url) => new Directive.import(url)));
// Add the corresponding `part`
String dbExtension;
@ -69,8 +80,8 @@ class OrmGenerator extends GeneratorForAnnotation<Orm> {
throw 'Unsupported ORM type: ${ctx.ormAnnotation.type}';
}
var dbFile =
p.setExtension(p.basename(inputId.uri.path), '.$dbExtension.orm.g.dart');
var dbFile = p.setExtension(
p.basename(inputId.uri.path), '.$dbExtension.orm.g.dart');
lib.body.add(new Code("part '$dbFile';"));