|
| 1 | +import 'dart:io'; |
| 2 | +import 'dart:isolate'; |
| 3 | + |
| 4 | +import 'package:build_daemon/client.dart'; |
| 5 | +import 'package:build_daemon/constants.dart'; |
| 6 | +import 'package:build_daemon/data/build_status.dart'; |
| 7 | +import 'package:build_daemon/data/build_target.dart'; |
| 8 | +import 'package:collection/collection.dart'; |
| 9 | +import 'package:package_config/package_config.dart'; |
| 10 | +import 'package:path/path.dart' as p; |
| 11 | +import 'package:shelf/shelf_io.dart'; |
| 12 | +import 'package:shelf_proxy/shelf_proxy.dart'; |
| 13 | + |
| 14 | +/// Serves example/web under localhost. |
| 15 | +void main() async { |
| 16 | + await TestAssetServer.start(); |
| 17 | + print('Serving on http://localhost:8080/'); |
| 18 | +} |
| 19 | + |
| 20 | +class TestAssetServer { |
| 21 | + final BuildDaemonClient buildRunner; |
| 22 | + late final HttpServer server; |
| 23 | + |
| 24 | + TestAssetServer(this.buildRunner); |
| 25 | + |
| 26 | + Future<void> close() async { |
| 27 | + await server.close(force: true); |
| 28 | + await buildRunner.close(); |
| 29 | + } |
| 30 | + |
| 31 | + static Future<TestAssetServer> start() async { |
| 32 | + final packageConfig = await loadPackageConfigUri( |
| 33 | + (await Isolate.packageConfig)!, |
| 34 | + ); |
| 35 | + final ownPackage = packageConfig['sqlite3']!.root; |
| 36 | + var packageDir = ownPackage.toFilePath(windows: Platform.isWindows); |
| 37 | + if (packageDir.endsWith('/')) { |
| 38 | + packageDir = packageDir.substring(0, packageDir.length - 1); |
| 39 | + } |
| 40 | + |
| 41 | + final buildRunner = await BuildDaemonClient.connect(packageDir, [ |
| 42 | + Platform.executable, // dart |
| 43 | + 'run', |
| 44 | + 'build_runner', |
| 45 | + 'daemon', |
| 46 | + ], logHandler: (log) => print(log.message)); |
| 47 | + |
| 48 | + buildRunner |
| 49 | + ..registerBuildTarget(DefaultBuildTarget((b) => b.target = 'example')) |
| 50 | + ..startBuild(); |
| 51 | + |
| 52 | + // Wait for the build to complete, so that the server we return is ready to |
| 53 | + // go. |
| 54 | + await buildRunner.buildResults.firstWhere((b) { |
| 55 | + final buildResult = b.results.firstWhereOrNull( |
| 56 | + (r) => r.target == 'example', |
| 57 | + ); |
| 58 | + return buildResult != null && buildResult.status != BuildStatus.started; |
| 59 | + }); |
| 60 | + |
| 61 | + final assetServerPortFile = File( |
| 62 | + p.join(daemonWorkspace(packageDir), '.asset_server_port'), |
| 63 | + ); |
| 64 | + final assetServerPort = int.parse(await assetServerPortFile.readAsString()); |
| 65 | + |
| 66 | + final server = TestAssetServer(buildRunner); |
| 67 | + |
| 68 | + final proxy = proxyHandler( |
| 69 | + 'http://localhost:$assetServerPort/example/web/', |
| 70 | + ); |
| 71 | + server.server = await serve(proxy, 'localhost', 8080); |
| 72 | + return server; |
| 73 | + } |
| 74 | +} |
0 commit comments