Remove SQL migrations, postgres ORM
This commit is contained in:
Tobe O
parent
db6de4bd8a
commit
d377f5a481
2 changed files with 0 additions and 1214 deletions
File diff suppressed because it is too large
Load diff
|
|
@ -1,165 +0,0 @@
|
||||||
import 'dart:async';
|
|
||||||
import 'package:analyzer/dart/ast/ast.dart';
|
|
||||||
import 'package:analyzer/dart/element/element.dart';
|
|
||||||
import 'package:angel_orm/angel_orm.dart';
|
|
||||||
import 'package:build/build.dart';
|
|
||||||
import 'build_context.dart';
|
|
||||||
import 'package:source_gen/source_gen.dart';
|
|
||||||
import 'postgres_build_context.dart';
|
|
||||||
|
|
||||||
class SqlMigrationBuilder implements Builder {
|
|
||||||
/// If `true` (default), then field names will automatically be (de)serialized as snake_case.
|
|
||||||
final bool autoSnakeCaseNames;
|
|
||||||
|
|
||||||
/// If `true` (default), then the schema will automatically add id, created_at and updated_at fields.
|
|
||||||
final bool autoIdAndDateFields;
|
|
||||||
|
|
||||||
/// If `true` (default: `false`), then the resulting schema will generate a `TEMPORARY` table.
|
|
||||||
final bool temporary;
|
|
||||||
|
|
||||||
const SqlMigrationBuilder(
|
|
||||||
{this.autoSnakeCaseNames: true,
|
|
||||||
this.autoIdAndDateFields: true,
|
|
||||||
this.temporary: false});
|
|
||||||
|
|
||||||
@override
|
|
||||||
Map<String, List<String>> get buildExtensions => {
|
|
||||||
'.dart': ['.up.g.sql', '.down.g.sql']
|
|
||||||
};
|
|
||||||
|
|
||||||
@override
|
|
||||||
Future build(BuildStep buildStep) async {
|
|
||||||
var resolver = await buildStep.resolver;
|
|
||||||
var up = new StringBuffer();
|
|
||||||
var down = new StringBuffer();
|
|
||||||
|
|
||||||
if (!await resolver.isLibrary(buildStep.inputId)) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
var lib = await resolver.libraryFor(buildStep.inputId);
|
|
||||||
var elements = lib.definingCompilationUnit.unit.declarations;
|
|
||||||
|
|
||||||
if (!elements.any(
|
|
||||||
(el) => ormTypeChecker.firstAnnotationOf(el.element) != null)) return;
|
|
||||||
|
|
||||||
await generateSqlMigrations(lib, resolver, buildStep, up, down);
|
|
||||||
buildStep.writeAsString(
|
|
||||||
buildStep.inputId.changeExtension('.up.g.sql'), up.toString());
|
|
||||||
buildStep.writeAsString(
|
|
||||||
buildStep.inputId.changeExtension('.down.g.sql'), down.toString());
|
|
||||||
}
|
|
||||||
|
|
||||||
Future generateSqlMigrations(LibraryElement libraryElement, Resolver resolver,
|
|
||||||
BuildStep buildStep, StringBuffer up, StringBuffer down) async {
|
|
||||||
List<String> done = [];
|
|
||||||
for (var element
|
|
||||||
in libraryElement.definingCompilationUnit.unit.declarations) {
|
|
||||||
if (element is ClassDeclaration && !done.contains(element.name)) {
|
|
||||||
var ann = ormTypeChecker.firstAnnotationOf(element.element);
|
|
||||||
if (ann != null) {
|
|
||||||
var ctx = await buildContext(
|
|
||||||
element.element,
|
|
||||||
reviveOrm(new ConstantReader(ann)),
|
|
||||||
buildStep,
|
|
||||||
resolver,
|
|
||||||
autoSnakeCaseNames != false,
|
|
||||||
autoIdAndDateFields != false);
|
|
||||||
buildUpMigration(ctx, up);
|
|
||||||
buildDownMigration(ctx, down);
|
|
||||||
done.add(element.name.name);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void buildUpMigration(PostgresBuildContext ctx, StringBuffer buf) {
|
|
||||||
if (temporary == true)
|
|
||||||
buf.writeln('CREATE TEMPORARY TABLE "${ctx.tableName}" (');
|
|
||||||
else
|
|
||||||
buf.writeln('CREATE TABLE "${ctx.tableName}" (');
|
|
||||||
|
|
||||||
List<String> dup = [];
|
|
||||||
int i = 0;
|
|
||||||
ctx.columnInfo.forEach((name, col) {
|
|
||||||
var key = ctx.resolveFieldName(name);
|
|
||||||
|
|
||||||
if (dup.contains(key))
|
|
||||||
return;
|
|
||||||
else {
|
|
||||||
if (key != 'id' || autoIdAndDateFields == false) {
|
|
||||||
// Check for relationships that might duplicate
|
|
||||||
for (var rName in ctx.relationships.keys) {
|
|
||||||
var relationship = ctx.populateRelationship(rName);
|
|
||||||
if (relationship.localKey == key) return;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
dup.add(key);
|
|
||||||
if (i++ > 0) buf.writeln(',');
|
|
||||||
}
|
|
||||||
|
|
||||||
buf.write(' "$key" ${col.type.name}');
|
|
||||||
|
|
||||||
if (col.index == IndexType.PRIMARY_KEY)
|
|
||||||
buf.write(' PRIMARY KEY');
|
|
||||||
else if (col.index == IndexType.UNIQUE) buf.write(' UNIQUE');
|
|
||||||
|
|
||||||
if (col.nullable != true) buf.write(' NOT NULLABLE');
|
|
||||||
});
|
|
||||||
|
|
||||||
// Relations
|
|
||||||
ctx.relationships.forEach((name, r) {
|
|
||||||
var relationship = ctx.populateRelationship(name);
|
|
||||||
|
|
||||||
if (relationship.isBelongsTo) {
|
|
||||||
var key = relationship.localKey;
|
|
||||||
|
|
||||||
if (dup.contains(key))
|
|
||||||
return;
|
|
||||||
else {
|
|
||||||
dup.add(key);
|
|
||||||
if (i++ > 0) buf.writeln(',');
|
|
||||||
}
|
|
||||||
|
|
||||||
buf.write(
|
|
||||||
' "${relationship.localKey}" int REFERENCES ${relationship.foreignTable}(${relationship.foreignKey})');
|
|
||||||
if (relationship.cascadeOnDelete != false && relationship.isSingular)
|
|
||||||
buf.write(' ON DELETE CASCADE');
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
// Primary keys, unique
|
|
||||||
bool hasPrimary = false;
|
|
||||||
ctx.fields.forEach((f) {
|
|
||||||
var col = ctx.columnInfo[f.name];
|
|
||||||
if (col != null) {
|
|
||||||
var name = ctx.resolveFieldName(f.name);
|
|
||||||
if (col.index == IndexType.UNIQUE) {
|
|
||||||
if (i++ > 0) buf.writeln(',');
|
|
||||||
buf.write(' UNIQUE($name)');
|
|
||||||
} else if (col.index == IndexType.PRIMARY_KEY) {
|
|
||||||
if (i++ > 0) buf.writeln(',');
|
|
||||||
hasPrimary = true;
|
|
||||||
buf.write(' PRIMARY KEY($name)');
|
|
||||||
}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
if (!hasPrimary) {
|
|
||||||
var idField =
|
|
||||||
ctx.fields.firstWhere((f) => f.name == 'id', orElse: () => null);
|
|
||||||
if (idField != null) {
|
|
||||||
if (i++ > 0) buf.writeln(',');
|
|
||||||
buf.write(' PRIMARY KEY(id)');
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
buf.writeln();
|
|
||||||
buf.writeln(');');
|
|
||||||
}
|
|
||||||
|
|
||||||
void buildDownMigration(PostgresBuildContext ctx, StringBuffer buf) {
|
|
||||||
buf.writeln('DROP TABLE "${ctx.tableName}";');
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Loading…
Reference in a new issue