AMT-Cheif
/
drift
mirror of https://github.com/AMT-Cheif/drift.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Allow referencing enum value for default

This commit is contained in:
Alexander Wilde 2022-12-18 22:29:44 +00:00
parent 0e395bbef3
commit bdfa8124fe
1 changed files with 30 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
drift_dev/lib/src/analysis/resolver/drift/table.dart
View File

@ -1,4 +1,5 @@
import 'package:analyzer/dart/ast/ast.dart' as dart; import 'package:analyzer/dart/ast/ast.dart' as dart;
import 'package:analyzer/dart/element/element.dart';
import 'package:collection/collection.dart'; import 'package:collection/collection.dart';
import 'package:drift/drift.dart' show DriftSqlType; import 'package:drift/drift.dart' show DriftSqlType;
import 'package:drift_dev/src/analysis/driver/driver.dart'; import 'package:drift_dev/src/analysis/driver/driver.dart';
@ -141,6 +142,35 @@ class DriftTableResolver extends LocalElementResolver<DiscoveredDriftTable> {
} else if (constraint is GeneratedAs) { } else if (constraint is GeneratedAs) {
constraints.add(ColumnGeneratedAs.fromParser(constraint)); constraints.add(ColumnGeneratedAs.fromParser(constraint));
} else if (constraint is Default) { } else if (constraint is Default) {
final element = converter?.dartType.element;
if (element is EnumElement) {
final expression = constraint.expression;
if (type == DriftSqlType.int) {
if (expression is NumericLiteral) {
final values = element
.getField('values')
?.computeConstantValue()
?.toListValue() ??
const [];
if (!expression.isInt ||
expression.value < 0 ||
expression.value > values.length) {
reportError(DriftAnalysisError.inDriftFile(constraint,
'${expression.value} is not a valid index value for ${element.name}.'));
continue;
}
}
} else {
if (expression is StringLiteral &&
element.getField(expression.value)?.enclosingElement !=
element) {
reportError(DriftAnalysisError.inDriftFile(constraint,
'${expression.value} is not a valid enum value for ${element.name}.'));
continue;
}
}
}
defaultArgument = AnnotatedDartCode.build((b) => b defaultArgument = AnnotatedDartCode.build((b) => b
..addText('const ') ..addText('const ')
..addSymbol('CustomExpression', AnnotatedDartCode.drift) ..addSymbol('CustomExpression', AnnotatedDartCode.drift)