mirror of https://github.com/AMT-Cheif/drift.git
96 lines
2.4 KiB
Dart
96 lines
2.4 KiB
Dart
import 'package:build/build.dart';
|
|
import 'package:build/experiments.dart';
|
|
import 'package:build_resolvers/build_resolvers.dart';
|
|
import 'package:build_test/build_test.dart';
|
|
import 'package:drift_dev/integrations/build.dart';
|
|
import 'package:logging/logging.dart';
|
|
import 'package:test/test.dart';
|
|
import 'package:yaml/yaml.dart';
|
|
|
|
final _resolvers =
|
|
withEnabledExperiments(() => AnalyzerResolvers(), ['records']);
|
|
|
|
BuilderOptions builderOptionsFromYaml(String yaml) {
|
|
final map = loadYaml(yaml);
|
|
return BuilderOptions((map as YamlMap).cast());
|
|
}
|
|
|
|
Logger loggerThat(dynamic expectedLogs) {
|
|
final logger = Logger.detached('drift_dev_test');
|
|
|
|
expect(logger.onRecord, expectedLogs);
|
|
return logger;
|
|
}
|
|
|
|
Future<RecordingAssetWriter> emulateDriftBuild({
|
|
required Map<String, String> inputs,
|
|
BuilderOptions options = const BuilderOptions({}),
|
|
Logger? logger,
|
|
bool modularBuild = false,
|
|
}) async {
|
|
_resolvers.reset();
|
|
logger ??= Logger.detached('emulateDriftBuild');
|
|
|
|
final writer = InMemoryAssetWriter();
|
|
final reader = MultiAssetReader([
|
|
WrittenAssetReader(writer),
|
|
InMemoryAssetReader(
|
|
rootPackage: 'a',
|
|
sourceAssets: {
|
|
for (final entry in inputs.entries) makeAssetId(entry.key): entry.value,
|
|
},
|
|
),
|
|
await PackageAssetReader.currentIsolate(),
|
|
]);
|
|
|
|
final stages = [
|
|
preparingBuilder(options),
|
|
analyzer(options),
|
|
modularBuild ? modular(options) : driftBuilderNotShared(options),
|
|
driftCleanup(options),
|
|
];
|
|
|
|
for (final stage in stages) {
|
|
if (stage is Builder) {
|
|
await runBuilder(
|
|
stage,
|
|
inputs.keys.map(makeAssetId),
|
|
reader,
|
|
writer,
|
|
_resolvers,
|
|
logger: logger,
|
|
);
|
|
} else if (stage is PostProcessBuilder) {
|
|
final deleted = <AssetId>[];
|
|
|
|
for (final assetId in writer.assets.keys) {
|
|
final shouldBuild =
|
|
stage.inputExtensions.any((e) => assetId.path.endsWith(e));
|
|
if (shouldBuild) {
|
|
await runPostProcessBuilder(
|
|
stage,
|
|
assetId,
|
|
reader,
|
|
writer,
|
|
logger,
|
|
addAsset: (_) {},
|
|
deleteAsset: deleted.add,
|
|
);
|
|
}
|
|
}
|
|
deleted.forEach(writer.assets.remove);
|
|
}
|
|
}
|
|
|
|
logger.clearListeners();
|
|
return writer;
|
|
}
|
|
|
|
extension OnlyDartOutputs on RecordingAssetWriter {
|
|
Iterable<AssetId> get dartOutputs {
|
|
return assets.keys.where((e) {
|
|
return e.extension == '.dart';
|
|
});
|
|
}
|
|
}
|