| // Copyright 2016 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| import 'dart:async'; |
| import 'dart:convert' show BASE64, UTF8; |
| |
| import 'package:path/path.dart' as path; |
| |
| import 'base/context.dart'; |
| import 'base/file_system.dart'; |
| import 'base/io.dart'; |
| import 'build_info.dart'; |
| import 'dart/package_map.dart'; |
| import 'asset.dart'; |
| import 'globals.dart'; |
| import 'vmservice.dart'; |
| |
| typedef void DevFSProgressReporter(int progress, int max); |
| |
| class DevFSConfig { |
| /// Should DevFS assume that symlink targets are stable? |
| bool cacheSymlinks = false; |
| /// Should DevFS assume that there are no symlinks to directories? |
| bool noDirectorySymlinks = false; |
| } |
| |
| DevFSConfig get devFSConfig => context[DevFSConfig]; |
| |
| /// Common superclass for content copied to the device. |
| abstract class DevFSContent { |
| bool _exists = true; |
| |
| /// Return `true` if this is the first time this method is called |
| /// or if the entry has been modified since this method was last called. |
| bool get isModified; |
| |
| int get size; |
| |
| Future<List<int>> contentsAsBytes(); |
| |
| Stream<List<int>> contentsAsStream(); |
| |
| Stream<List<int>> contentsAsCompressedStream() { |
| return contentsAsStream().transform(GZIP.encoder); |
| } |
| } |
| |
| // File content to be copied to the device. |
| class DevFSFileContent extends DevFSContent { |
| DevFSFileContent(this.file); |
| |
| final FileSystemEntity file; |
| FileSystemEntity _linkTarget; |
| FileStat _fileStat; |
| |
| File _getFile() { |
| if (_linkTarget != null) { |
| return _linkTarget; |
| } |
| if (file is Link) { |
| // The link target. |
| return fs.file(file.resolveSymbolicLinksSync()); |
| } |
| return file; |
| } |
| |
| void _stat() { |
| if (_linkTarget != null) { |
| // Stat the cached symlink target. |
| _fileStat = _linkTarget.statSync(); |
| return; |
| } |
| _fileStat = file.statSync(); |
| if (_fileStat.type == FileSystemEntityType.LINK) { |
| // Resolve, stat, and maybe cache the symlink target. |
| String resolved = file.resolveSymbolicLinksSync(); |
| FileSystemEntity linkTarget = fs.file(resolved); |
| // Stat the link target. |
| _fileStat = linkTarget.statSync(); |
| if (devFSConfig.cacheSymlinks) { |
| _linkTarget = linkTarget; |
| } |
| } |
| } |
| |
| @override |
| bool get isModified { |
| FileStat _oldFileStat = _fileStat; |
| _stat(); |
| return _oldFileStat == null || _fileStat.modified.isAfter(_oldFileStat.modified); |
| } |
| |
| @override |
| int get size { |
| if (_fileStat == null) |
| _stat(); |
| return _fileStat.size; |
| } |
| |
| @override |
| Future<List<int>> contentsAsBytes() => _getFile().readAsBytes(); |
| |
| @override |
| Stream<List<int>> contentsAsStream() => _getFile().openRead(); |
| } |
| |
| /// Byte content to be copied to the device. |
| class DevFSByteContent extends DevFSContent { |
| DevFSByteContent(this._bytes); |
| |
| List<int> _bytes; |
| |
| bool _isModified = true; |
| |
| List<int> get bytes => _bytes; |
| |
| set bytes(List<int> newBytes) { |
| _bytes = newBytes; |
| _isModified = true; |
| } |
| |
| /// Return `true` only once so that the content is written to the device only once. |
| @override |
| bool get isModified { |
| bool modified = _isModified; |
| _isModified = false; |
| return modified; |
| } |
| |
| @override |
| int get size => _bytes.length; |
| |
| @override |
| Future<List<int>> contentsAsBytes() async => _bytes; |
| |
| @override |
| Stream<List<int>> contentsAsStream() => |
| new Stream<List<int>>.fromIterable(<List<int>>[_bytes]); |
| } |
| |
| /// String content to be copied to the device. |
| class DevFSStringContent extends DevFSByteContent { |
| DevFSStringContent(String string) : _string = string, super(UTF8.encode(string)); |
| |
| String _string; |
| |
| String get string => _string; |
| |
| set string(String newString) { |
| _string = newString; |
| super.bytes = UTF8.encode(_string); |
| } |
| |
| @override |
| set bytes(List<int> newBytes) { |
| string = UTF8.decode(newBytes); |
| } |
| } |
| |
| /// Abstract DevFS operations interface. |
| abstract class DevFSOperations { |
| Future<Uri> create(String fsName); |
| Future<dynamic> destroy(String fsName); |
| Future<dynamic> writeFile(String fsName, String devicePath, DevFSContent content); |
| Future<dynamic> deleteFile(String fsName, String devicePath); |
| } |
| |
| /// An implementation of [DevFSOperations] that speaks to the |
| /// vm service. |
| class ServiceProtocolDevFSOperations implements DevFSOperations { |
| final VMService vmService; |
| |
| ServiceProtocolDevFSOperations(this.vmService); |
| |
| @override |
| Future<Uri> create(String fsName) async { |
| Map<String, dynamic> response = await vmService.vm.createDevFS(fsName); |
| return Uri.parse(response['uri']); |
| } |
| |
| @override |
| Future<dynamic> destroy(String fsName) async { |
| await vmService.vm.invokeRpcRaw('_deleteDevFS', |
| <String, dynamic> { 'fsName': fsName }); |
| } |
| |
| @override |
| Future<dynamic> writeFile(String fsName, String devicePath, DevFSContent content) async { |
| List<int> bytes; |
| try { |
| bytes = await content.contentsAsBytes(); |
| } catch (e) { |
| return e; |
| } |
| String fileContents = BASE64.encode(bytes); |
| try { |
| return await vmService.vm.invokeRpcRaw('_writeDevFSFile', |
| <String, dynamic> { |
| 'fsName': fsName, |
| 'path': devicePath, |
| 'fileContents': fileContents |
| }); |
| } catch (e) { |
| printTrace('DevFS: Failed to write $devicePath: $e'); |
| } |
| } |
| |
| @override |
| Future<dynamic> deleteFile(String fsName, String devicePath) async { |
| // TODO(johnmccutchan): Add file deletion to the devFS protocol. |
| } |
| } |
| |
| class _DevFSHttpWriter { |
| _DevFSHttpWriter(this.fsName, VMService serviceProtocol) |
| : httpAddress = serviceProtocol.httpAddress; |
| |
| final String fsName; |
| final Uri httpAddress; |
| |
| static const int kMaxInFlight = 6; |
| |
| int _inFlight = 0; |
| Map<String, DevFSContent> _outstanding; |
| Completer<Null> _completer; |
| HttpClient _client; |
| int _done; |
| int _max; |
| |
| Future<Null> write(Map<String, DevFSContent> entries, |
| {DevFSProgressReporter progressReporter}) async { |
| _client = new HttpClient(); |
| _client.maxConnectionsPerHost = kMaxInFlight; |
| _completer = new Completer<Null>(); |
| _outstanding = new Map<String, DevFSContent>.from(entries); |
| _done = 0; |
| _max = _outstanding.length; |
| _scheduleWrites(progressReporter); |
| await _completer.future; |
| _client.close(); |
| } |
| |
| void _scheduleWrites(DevFSProgressReporter progressReporter) { |
| while (_inFlight < kMaxInFlight) { |
| if (_outstanding.length == 0) { |
| // Finished. |
| break; |
| } |
| String devicePath = _outstanding.keys.first; |
| DevFSContent content = _outstanding.remove(devicePath); |
| _scheduleWrite(devicePath, content, progressReporter); |
| _inFlight++; |
| } |
| } |
| |
| Future<Null> _scheduleWrite(String devicePath, |
| DevFSContent content, |
| DevFSProgressReporter progressReporter) async { |
| try { |
| HttpClientRequest request = await _client.putUrl(httpAddress); |
| request.headers.removeAll(HttpHeaders.ACCEPT_ENCODING); |
| request.headers.add('dev_fs_name', fsName); |
| request.headers.add('dev_fs_path_b64', |
| BASE64.encode(UTF8.encode(devicePath))); |
| Stream<List<int>> contents = content.contentsAsCompressedStream(); |
| await request.addStream(contents); |
| HttpClientResponse response = await request.close(); |
| await response.drain(); |
| } catch (e) { |
| printError('Error writing "$devicePath" to DevFS: $e'); |
| } |
| if (progressReporter != null) { |
| _done++; |
| progressReporter(_done, _max); |
| } |
| _inFlight--; |
| if ((_outstanding.length == 0) && (_inFlight == 0)) { |
| _completer.complete(null); |
| } else { |
| _scheduleWrites(progressReporter); |
| } |
| } |
| } |
| |
| class DevFS { |
| /// Create a [DevFS] named [fsName] for the local files in [directory]. |
| DevFS(VMService serviceProtocol, |
| String fsName, |
| this.rootDirectory, { |
| String packagesFilePath |
| }) |
| : _operations = new ServiceProtocolDevFSOperations(serviceProtocol), |
| _httpWriter = new _DevFSHttpWriter(fsName, serviceProtocol), |
| fsName = fsName { |
| _packagesFilePath = |
| packagesFilePath ?? path.join(rootDirectory.path, kPackagesFileName); |
| } |
| |
| DevFS.operations(this._operations, |
| this.fsName, |
| this.rootDirectory, { |
| String packagesFilePath, |
| }) |
| : _httpWriter = null { |
| _packagesFilePath = |
| packagesFilePath ?? path.join(rootDirectory.path, kPackagesFileName); |
| } |
| |
| final DevFSOperations _operations; |
| final _DevFSHttpWriter _httpWriter; |
| final String fsName; |
| final Directory rootDirectory; |
| String _packagesFilePath; |
| final Map<String, DevFSContent> _entries = <String, DevFSContent>{}; |
| final Set<String> assetPathsToEvict = new Set<String>(); |
| |
| final List<Future<Map<String, dynamic>>> _pendingOperations = |
| new List<Future<Map<String, dynamic>>>(); |
| |
| Uri _baseUri; |
| Uri get baseUri => _baseUri; |
| |
| Future<Uri> create() async { |
| _baseUri = await _operations.create(fsName); |
| printTrace('DevFS: Created new filesystem on the device ($_baseUri)'); |
| return _baseUri; |
| } |
| |
| Future<dynamic> destroy() { |
| printTrace('DevFS: Deleted filesystem on the device ($_baseUri)'); |
| return _operations.destroy(fsName); |
| } |
| |
| /// Update files on the device and return the number of bytes sync'd |
| Future<int> update({ DevFSProgressReporter progressReporter, |
| AssetBundle bundle, |
| bool bundleDirty: false, |
| Set<String> fileFilter}) async { |
| // Mark all entries as possibly deleted. |
| for (DevFSContent content in _entries.values) { |
| content._exists = false; |
| } |
| |
| // Scan workspace, packages, and assets |
| printTrace('DevFS: Starting sync from $rootDirectory'); |
| logger.printTrace('Scanning project files'); |
| await _scanDirectory(rootDirectory, |
| recursive: true, |
| fileFilter: fileFilter); |
| if (fs.isFileSync(_packagesFilePath)) { |
| printTrace('Scanning package files'); |
| await _scanPackages(fileFilter); |
| } |
| if (bundle != null) { |
| printTrace('Scanning asset files'); |
| bundle.entries.forEach((String archivePath, DevFSContent content) { |
| _scanBundleEntry(archivePath, content, bundleDirty); |
| }); |
| } |
| |
| // Handle deletions. |
| printTrace('Scanning for deleted files'); |
| String assetBuildDirPrefix = getAssetBuildDirectory() + path.separator; |
| final List<String> toRemove = new List<String>(); |
| _entries.forEach((String devicePath, DevFSContent content) { |
| if (!content._exists) { |
| Future<Map<String, dynamic>> operation = |
| _operations.deleteFile(fsName, devicePath); |
| if (operation != null) |
| _pendingOperations.add(operation); |
| toRemove.add(devicePath); |
| if (devicePath.startsWith(assetBuildDirPrefix)) { |
| String archivePath = devicePath.substring(assetBuildDirPrefix.length); |
| assetPathsToEvict.add(archivePath); |
| } |
| } |
| }); |
| if (toRemove.isNotEmpty) { |
| printTrace('Removing deleted files'); |
| toRemove.forEach(_entries.remove); |
| await Future.wait(_pendingOperations); |
| _pendingOperations.clear(); |
| } |
| |
| // Update modified files |
| int numBytes = 0; |
| Map<String, DevFSContent> dirtyEntries = <String, DevFSContent>{}; |
| _entries.forEach((String devicePath, DevFSContent content) { |
| String archivePath; |
| if (devicePath.startsWith(assetBuildDirPrefix)) |
| archivePath = devicePath.substring(assetBuildDirPrefix.length); |
| if (content.isModified || (bundleDirty && archivePath != null)) { |
| dirtyEntries[devicePath] = content; |
| numBytes += content.size; |
| if (archivePath != null) |
| assetPathsToEvict.add(archivePath); |
| } |
| }); |
| if (dirtyEntries.length > 0) { |
| printTrace('Updating files'); |
| if (_httpWriter != null) { |
| try { |
| await _httpWriter.write(dirtyEntries, |
| progressReporter: progressReporter); |
| } catch (e) { |
| printError("Could not update files on device: $e"); |
| } |
| } else { |
| // Make service protocol requests for each. |
| dirtyEntries.forEach((String devicePath, DevFSContent content) { |
| Future<Map<String, dynamic>> operation = |
| _operations.writeFile(fsName, devicePath, content); |
| if (operation != null) |
| _pendingOperations.add(operation); |
| }); |
| if (progressReporter != null) { |
| final int max = _pendingOperations.length; |
| int complete = 0; |
| _pendingOperations.forEach((Future<dynamic> f) => f.then((dynamic v) { |
| complete += 1; |
| progressReporter(complete, max); |
| })); |
| } |
| await Future.wait(_pendingOperations, eagerError: true); |
| _pendingOperations.clear(); |
| } |
| } |
| |
| printTrace('DevFS: Sync finished'); |
| return numBytes; |
| } |
| |
| void _scanFile(String devicePath, FileSystemEntity file) { |
| DevFSContent content = _entries.putIfAbsent(devicePath, () => new DevFSFileContent(file)); |
| content._exists = true; |
| } |
| |
| void _scanBundleEntry(String archivePath, DevFSContent content, bool bundleDirty) { |
| // We write the assets into the AssetBundle working dir so that they |
| // are in the same location in DevFS and the iOS simulator. |
| final String devicePath = path.join(getAssetBuildDirectory(), archivePath); |
| |
| _entries[devicePath] = content; |
| content._exists = true; |
| } |
| |
| bool _shouldIgnore(String devicePath) { |
| List<String> ignoredPrefixes = <String>['android/', |
| getBuildDirectory(), |
| 'ios/', |
| '.pub/']; |
| for (String ignoredPrefix in ignoredPrefixes) { |
| if (devicePath.startsWith(ignoredPrefix)) |
| return true; |
| } |
| return false; |
| } |
| |
| Future<bool> _scanDirectory(Directory directory, |
| {String directoryName, |
| bool recursive: false, |
| bool ignoreDotFiles: true, |
| String packagesDirectoryName, |
| Set<String> fileFilter}) async { |
| String prefix = directoryName; |
| if (prefix == null) { |
| prefix = path.relative(directory.path, from: rootDirectory.path); |
| if (prefix == '.') |
| prefix = ''; |
| } |
| try { |
| Stream<FileSystemEntity> files = |
| directory.list(recursive: recursive, followLinks: false); |
| await for (FileSystemEntity file in files) { |
| if (!devFSConfig.noDirectorySymlinks && (file is Link)) { |
| // Check if this is a symlink to a directory and skip it. |
| final String linkPath = file.resolveSymbolicLinksSync(); |
| final FileSystemEntityType linkType = |
| fs.statSync(linkPath).type; |
| if (linkType == FileSystemEntityType.DIRECTORY) { |
| continue; |
| } |
| } |
| if (file is Directory) { |
| // Skip non-files. |
| continue; |
| } |
| assert((file is Link) || (file is File)); |
| if (ignoreDotFiles && path.basename(file.path).startsWith('.')) { |
| // Skip dot files. |
| continue; |
| } |
| final String relativePath = |
| path.relative(file.path, from: directory.path); |
| final String devicePath = path.join(prefix, relativePath); |
| bool filtered = false; |
| if ((fileFilter != null) && |
| !fileFilter.contains(devicePath)) { |
| if (packagesDirectoryName != null) { |
| // Double check the filter for packages/packagename/ |
| final String packagesDevicePath = |
| path.join(packagesDirectoryName, relativePath); |
| if (!fileFilter.contains(packagesDevicePath)) { |
| // File was not in the filter set. |
| filtered = true; |
| } |
| } else { |
| // File was not in the filter set. |
| filtered = true; |
| } |
| } |
| if (filtered) { |
| // Skip files that are not included in the filter. |
| continue; |
| } |
| if (ignoreDotFiles && devicePath.startsWith('.')) { |
| // Skip directories that start with a dot. |
| continue; |
| } |
| if (!_shouldIgnore(devicePath)) |
| _scanFile(devicePath, file); |
| } |
| } catch (e) { |
| // Ignore directory and error. |
| return false; |
| } |
| return true; |
| } |
| |
| Future<Null> _scanPackages(Set<String> fileFilter) async { |
| StringBuffer sb; |
| PackageMap packageMap = new PackageMap(_packagesFilePath); |
| |
| for (String packageName in packageMap.map.keys) { |
| Uri uri = packageMap.map[packageName]; |
| // This project's own package. |
| final bool isProjectPackage = uri.toString() == 'lib/'; |
| final String directoryName = |
| isProjectPackage ? 'lib' : 'packages/$packageName'; |
| // If this is the project's package, we need to pass both |
| // package:<package_name> and lib/ as paths to be checked against |
| // the filter because we must support both package: imports and relative |
| // path imports within the project's own code. |
| final String packagesDirectoryName = |
| isProjectPackage ? 'packages/$packageName' : null; |
| Directory directory = fs.directory(uri); |
| bool packageExists = |
| await _scanDirectory(directory, |
| directoryName: directoryName, |
| recursive: true, |
| packagesDirectoryName: packagesDirectoryName, |
| fileFilter: fileFilter); |
| if (packageExists) { |
| sb ??= new StringBuffer(); |
| sb.writeln('$packageName:$directoryName'); |
| } |
| } |
| if (sb != null) { |
| DevFSContent content = _entries['.packages']; |
| if (content is DevFSStringContent && content.string == sb.toString()) { |
| content._exists = true; |
| return; |
| } |
| _entries['.packages'] = new DevFSStringContent(sb.toString()); |
| } |
| } |
| } |