mirror of https://github.com/AMT-Cheif/drift.git
52 lines
1.6 KiB
Dart
52 lines
1.6 KiB
Dart
import 'package:sqlparser/sqlparser.dart';
|
|
import 'package:sqlparser/src/engine/autocomplete/engine.dart';
|
|
import 'package:test/test.dart';
|
|
|
|
/// Parses the [driftFile] and computes available autocomplete suggestions at
|
|
/// the position of a `^` character in the source.
|
|
ComputedSuggestions completionsFor(String driftFile,
|
|
{void Function(SqlEngine)? setup}) {
|
|
final position = driftFile.indexOf('^');
|
|
final engine =
|
|
SqlEngine(EngineOptions(driftOptions: const DriftSqlOptions()));
|
|
setup?.call(engine);
|
|
|
|
final result = engine.parseDriftFile(driftFile.replaceFirst('^', ''));
|
|
return result.autoCompleteEngine!.suggestCompletions(position - 1);
|
|
}
|
|
|
|
Matcher hasCode(String code) => _SuggestionWithCode(code);
|
|
Matcher suggests(String code) => _SuggestsMatcher(contains(hasCode(code)));
|
|
Matcher suggestsAll(List<String> codes) {
|
|
return _SuggestsMatcher(containsAll(codes.map(hasCode)));
|
|
}
|
|
|
|
class _SuggestionWithCode extends Matcher {
|
|
final Matcher codeMatcher;
|
|
|
|
_SuggestionWithCode(dynamic code) : codeMatcher = wrapMatcher(code);
|
|
|
|
@override
|
|
Description describe(Description description) {
|
|
return description.add('suggests ').addDescriptionOf(codeMatcher);
|
|
}
|
|
|
|
@override
|
|
bool matches(dynamic item, Map matchState) {
|
|
return item is Suggestion && codeMatcher.matches(item.code, matchState);
|
|
}
|
|
}
|
|
|
|
class _SuggestsMatcher extends CustomMatcher {
|
|
_SuggestsMatcher(matcher)
|
|
: super('Suggestions containing', 'suggestions', matcher);
|
|
|
|
@override
|
|
List<Suggestion>? featureValueOf(dynamic actual) {
|
|
if (actual is ComputedSuggestions) {
|
|
return actual.suggestions;
|
|
}
|
|
return null;
|
|
}
|
|
}
|