update_packages.dart 67.1 KB
Newer Older
Ian Hickson's avatar
Ian Hickson committed
1
// Copyright 2014 The Flutter Authors. All rights reserved.
2 3 4
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

5
import 'dart:async';
6
import 'dart:collection';
7

8 9
import 'package:meta/meta.dart';

10
import '../base/common.dart';
11
import '../base/context.dart';
12
import '../base/file_system.dart';
13 14
import '../base/logger.dart';
import '../base/net.dart';
15
import '../base/task_queue.dart';
16
import '../cache.dart';
17
import '../dart/pub.dart';
18
import '../globals.dart' as globals;
19
import '../project.dart';
20 21
import '../runner/flutter_command.dart';

22 23
/// Map from package name to package version, used to artificially pin a pub
/// package version in cases when upgrading to the latest breaks Flutter.
24 25 26 27 28
///
/// These version pins must be pins, not ranges! Allowing these to be ranges
/// defeats the whole purpose of pinning all our dependencies, which is to
/// prevent upstream changes from causing our CI to fail randomly in ways
/// unrelated to the commits. It also, more importantly, risks breaking users
29
/// in ways that prevent them from ever upgrading Flutter again!
30
const Map<String, String> kManuallyPinnedDependencies = <String, String>{
Ian Hickson's avatar
Ian Hickson committed
31
  // Add pinned packages here. Please leave a comment explaining why.
32
  'flutter_gallery_assets': '1.0.2', // Tests depend on the exact version.
33
  'flutter_template_images': '4.2.0', // Must always exactly match flutter_tools template.
34
  'video_player': '2.2.11',
35 36
  // Could potentially break color scheme tests on upgrade,
  // so pin and manually update as needed.
37
  'material_color_utilities': '0.2.0',
38 39
  // https://github.com/flutter/flutter/issues/111304
  'url_launcher_android': '6.0.17',
40 41
  // https://github.com/flutter/flutter/issues/115660
  'archive': '3.3.2',
42 43
  // https://github.com/flutter/flutter/issues/116376
  'path_provider_android': '2.0.21',
44 45
  // https://github.com/flutter/flutter/issues/117163
  'intl': '0.17.0',
46 47
};

48
class UpdatePackagesCommand extends FlutterCommand {
49
  UpdatePackagesCommand() {
50 51 52 53 54
    argParser
      ..addFlag(
        'force-upgrade',
        help: 'Attempt to update all the dependencies to their latest versions.\n'
              'This will actually modify the pubspec.yaml files in your checkout.',
55
        negatable: false,
56 57 58 59
      )
      ..addFlag(
        'paths',
        help: 'Finds paths in the dependency chain leading from package specified '
60
              'in "--from" to package specified in "--to".',
61
        negatable: false,
62 63 64
      )
      ..addOption(
        'from',
65
        help: 'Used with "--dependency-path". Specifies the package to begin '
66 67 68 69
              'searching dependency path from.',
      )
      ..addOption(
        'to',
70 71
        help: 'Used with "--dependency-path". Specifies the package that the '
              'sought-after dependency path leads to.',
72 73 74 75 76
      )
      ..addFlag(
        'transitive-closure',
        help: 'Prints the dependency graph that is the transitive closure of '
              'packages the Flutter SDK depends on.',
77
        negatable: false,
78
      )
79 80
      ..addFlag(
        'consumer-only',
81
        help: 'Only prints the dependency graph that is the transitive closure '
82 83
              'that a consumer of the Flutter SDK will observe (when combined '
              'with transitive-closure).',
84 85
        negatable: false,
      )
86 87
      ..addFlag(
        'verify-only',
88
        help: 'Verifies the package checksum without changing or updating deps.',
89
        negatable: false,
90 91 92
      )
      ..addFlag(
        'offline',
93
        help: 'Use cached packages instead of accessing the network.',
94
        negatable: false,
95 96 97 98 99
      )
      ..addFlag(
        'crash',
        help: 'For Flutter CLI testing only, forces this command to throw an unhandled exception.',
        negatable: false,
100 101 102 103 104 105
      )
      ..addOption(
        'jobs',
        abbr: 'j',
        help: 'Causes the "pub get" runs to happen concurrently on this many '
              'CPUs. Defaults to the number of CPUs that this machine has.',
106 107 108 109 110 111 112 113
      )
      ..addOption(
        'synthetic-package-path',
        help: 'Write the synthetic monolithic pub package generated to do '
              'version solving to a persistent path. By default, a temporary '
              'directory that is deleted before the command exits. By '
              'providing this path, a Flutter maintainer can inspect further '
              'exactly how version solving was achieved.',
114
      );
115 116
  }

117
  @override
118
  final String name = 'update-packages';
119 120

  @override
121 122 123 124
  final String description = 'Update the packages inside the Flutter repo. '
                             'This is intended for CI and repo maintainers. '
                             'Normal Flutter developers should not have to '
                             'use this command.';
125

126 127 128
  @override
  final List<String> aliases = <String>['upgrade-packages'];

129
  @override
130
  final bool hidden = true;
131

132 133

  // Lazy-initialize the net utilities with values from the context.
134
  late final Net _net = Net(
135
    httpClientFactory: context.get<HttpClientFactory>(),
136 137 138 139
    logger: globals.logger,
    platform: globals.platform,
  );

140
  Future<void> _downloadCoverageData() async {
141
    final String urlBase = globals.platform.environment['FLUTTER_STORAGE_BASE_URL'] ?? 'https://storage.googleapis.com';
142
    final Uri coverageUri = Uri.parse('$urlBase/flutter_infra_release/flutter/coverage/lcov.info');
143 144 145 146 147 148 149
    final List<int>? data = await _net.fetchUrl(
      coverageUri,
      maxAttempts: 3,
    );
    if (data == null) {
      throwToolExit('Failed to fetch coverage data from $coverageUri');
    }
150
    final String coverageDir = globals.fs.path.join(
151
      Cache.flutterRoot!,
152 153
      'packages/flutter/coverage',
    );
154
    globals.fs.file(globals.fs.path.join(coverageDir, 'lcov.base.info'))
155 156
      ..createSync(recursive: true)
      ..writeAsBytesSync(data, flush: true);
157
    globals.fs.file(globals.fs.path.join(coverageDir, 'lcov.info'))
158 159 160 161
      ..createSync(recursive: true)
      ..writeAsBytesSync(data, flush: true);
  }

162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177
  late final Directory _syntheticPackageDir = (() {
    final String? optionPath = stringArg('synthetic-package-path');
    if (optionPath == null) {
      return globals.fs.systemTempDirectory.createTempSync('flutter_update_packages.');
    }
    final Directory syntheticPackageDir = globals.fs.directory(optionPath);
    if (!syntheticPackageDir.existsSync()) {
      syntheticPackageDir.createSync(recursive: true);
    }
    globals.printStatus(
      'The synthetic package with all pub dependencies across the repo will '
      'be written to ${syntheticPackageDir.absolute.path}.',
    );
    return syntheticPackageDir;
  })();

178
  @override
179
  Future<FlutterCommandResult> runCommand() async {
180
    final List<Directory> packages = runner!.getRepoPackages();
181

182 183 184 185 186 187
    final bool forceUpgrade = boolArgDeprecated('force-upgrade');
    final bool isPrintPaths = boolArgDeprecated('paths');
    final bool isPrintTransitiveClosure = boolArgDeprecated('transitive-closure');
    final bool isVerifyOnly = boolArgDeprecated('verify-only');
    final bool isConsumerOnly = boolArgDeprecated('consumer-only');
    final bool offline = boolArgDeprecated('offline');
188
    final bool doUpgrade = forceUpgrade || isPrintPaths || isPrintTransitiveClosure;
189

190
    if (boolArgDeprecated('crash')) {
191 192
      throw StateError('test crash please ignore.');
    }
193

194
    if (forceUpgrade && offline) {
195 196 197 198
      throwToolExit(
          '--force-upgrade cannot be used with the --offline flag'
      );
    }
199

200
    // "consumer" packages are those that constitute our public API (e.g. flutter, flutter_test, flutter_driver, flutter_localizations, integration_test).
201 202 203 204
    if (isConsumerOnly) {
      if (!isPrintTransitiveClosure) {
        throwToolExit(
          '--consumer-only can only be used with the --transitive-closure flag'
205 206
        );
      }
207 208 209 210 211 212 213
      // Only retain flutter, flutter_test, flutter_driver, and flutter_localizations.
      const List<String> consumerPackages = <String>['flutter', 'flutter_test', 'flutter_driver', 'flutter_localizations', 'integration_test'];
      // ensure we only get flutter/packages
      packages.retainWhere((Directory directory) {
        return consumerPackages.any((String package) {
          return directory.path.endsWith('packages${globals.fs.path.separator}$package');
        });
214 215
      });
    }
216 217

    if (isVerifyOnly) {
218
      _verifyPubspecs(packages);
219
      return FlutterCommandResult.success();
220 221
    }

222
    if (doUpgrade) {
223 224 225 226
      // This feature attempts to collect all the packages used across all the
      // pubspec.yamls in the repo (including via transitive dependencies), and
      // find the latest version of each that can be used while keeping each
      // such package fixed at a single version across all the pubspec.yamls.
227 228 229
      globals.printStatus('Upgrading packages...');
    }

230
    // First, collect the dependencies:
231
    final List<PubspecYaml> pubspecs = <PubspecYaml>[];
232 233
    final Map<String, PubspecDependency> explicitDependencies = <String, PubspecDependency>{};
    final Map<String, PubspecDependency> allDependencies = <String, PubspecDependency>{};
234
    final Set<String> specialDependencies = <String>{};
235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252
    _collectDependencies(
      packages: packages,
      pubspecs: pubspecs,
      explicitDependencies: explicitDependencies,
      allDependencies: allDependencies,
      specialDependencies: specialDependencies,
      doUpgrade: doUpgrade,
    );

    // Now that we have all the dependencies we care about, we are going to
    // create a fake package and then run either "pub upgrade", if requested,
    // followed by "pub get" on it. If upgrading, the pub tool will attempt to
    // bring these dependencies up to the most recent possible versions while
    // honoring all their constraints. If not upgrading the pub tool will only
    // attempt to download any necessary package versions to the pub cache to
    // warm the cache.
    final PubDependencyTree tree = PubDependencyTree(); // object to collect results
    await _generateFakePackage(
253
      tempDir: _syntheticPackageDir,
254 255 256 257 258 259
      dependencies: doUpgrade ? explicitDependencies.values : allDependencies.values,
      pubspecs: pubspecs,
      tree: tree,
      doUpgrade: doUpgrade,
    );

260 261 262 263 264
    // Only delete the synthetic package if it was done in a temp directory
    if (stringArg('synthetic-package-path') == null) {
      _syntheticPackageDir.deleteSync(recursive: true);
    }

265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287
    if (doUpgrade) {
      final bool done = _upgradePubspecs(
        tree: tree,
        pubspecs: pubspecs,
        explicitDependencies: explicitDependencies,
        specialDependencies: specialDependencies,
      );

      if (done) {
        // Complete early if we were just printing data.
        return FlutterCommandResult.success();
      }
    }

    await _runPubGetOnPackages(packages);

    return FlutterCommandResult.success();
  }

  void _verifyPubspecs(List<Directory> packages) {
    bool needsUpdate = false;
    globals.printStatus('Verifying pubspecs...');
    for (final Directory directory in packages) {
288
      final PubspecYaml pubspec = PubspecYaml(directory);
289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305
      globals.printTrace('Reading pubspec.yaml from ${directory.path}');
      if (pubspec.checksum.value == null) {
        // If the checksum is invalid or missing, we can just ask them run to run
        // upgrade again to compute it.
        globals.printWarning(
            'Warning: pubspec in ${directory.path} has out of date dependencies. '
            'Please run "flutter update-packages --force-upgrade" to update them correctly.'
        );
        needsUpdate = true;
      }
      // all dependencies in the pubspec sorted lexically.
      final Map<String, String> checksumDependencies = <String, String>{};
      for (final PubspecLine data in pubspec.inputData) {
        if (data is PubspecDependency && data.kind == DependencyKind.normal) {
          checksumDependencies[data.name] = data.version;
        }
      }
306
      final String checksum = _computeChecksum(checksumDependencies.keys, (String name) => checksumDependencies[name]!);
307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331
      if (checksum != pubspec.checksum.value) {
        // If the checksum doesn't match, they may have added or removed some dependencies.
        // we need to run update-packages to recapture the transitive deps.
        globals.printWarning(
            'Warning: pubspec in ${directory.path} has updated or new dependencies. '
            'Please run "flutter update-packages --force-upgrade" to update them correctly '
            '(checksum ${pubspec.checksum.value} != $checksum).'
        );
        needsUpdate = true;
      } else {
        // everything is correct in the pubspec.
        globals.printTrace('pubspec in ${directory.path} is up to date!');
      }
    }
    if (needsUpdate) {
      throwToolExit(
        'Warning: one or more pubspecs have invalid dependencies. '
        'Please run "flutter update-packages --force-upgrade" to update them correctly.',
        exitCode: 1,
      );
    }
    globals.printStatus('All pubspecs were up to date.');
  }

  void _collectDependencies({
332 333 334 335 336 337
    required List<Directory> packages,
    required List<PubspecYaml> pubspecs,
    required Set<String> specialDependencies,
    required Map<String, PubspecDependency> explicitDependencies,
    required Map<String, PubspecDependency> allDependencies,
    required bool doUpgrade,
338
  }) {
339 340 341
    // Visit all the directories with pubspec.yamls we care about.
    for (final Directory directory in packages) {
      if (doUpgrade) {
342
        globals.printTrace('Reading pubspec.yaml from: ${directory.path}');
343
      }
344
      final PubspecYaml pubspec = PubspecYaml(directory); // this parses the pubspec.yaml
345
      pubspecs.add(pubspec); // remember it for later
346 347
      for (final PubspecDependency dependency in pubspec.allDependencies) {
        if (allDependencies.containsKey(dependency.name)) {
348 349 350 351 352 353 354 355 356 357
          // If we've seen the dependency before, make sure that we are
          // importing it the same way. There's several ways to import a
          // dependency. Hosted (from pub via version number), by path (e.g.
          // pointing at the version of a package we get from the Dart SDK
          // that we download with Flutter), by SDK (e.g. the "flutter"
          // package is explicitly from "sdk: flutter").
          //
          // This makes sure that we don't import a package in two different
          // ways, e.g. by saying "sdk: flutter" in one pubspec.yaml and
          // saying "path: ../../..." in another.
358 359
          final PubspecDependency previous = allDependencies[dependency.name]!;
          if (dependency.kind != previous.kind || dependency._lockTarget != previous._lockTarget) {
360
            throwToolExit(
361
                'Inconsistent requirements around ${dependency.name}; '
362 363
                    'saw ${dependency.kind} (${dependency._lockTarget}) in "${dependency.sourcePath}" '
                    'and ${previous.kind} (${previous._lockTarget}) in "${previous.sourcePath}".'
364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387
            );
          }
          if (dependency.version != previous.version) {
            globals.printError(
                'Requiring multiple versions: multiple versions required by ${dependency.name}; '
                    'saw ${dependency.version} in "${dependency.sourcePath}" '
                    'and ${previous.version} in "${previous.sourcePath}".'
            );
          }
        }
        allDependencies[dependency.name] = dependency;
      }
      for (final PubspecDependency dependency in pubspec.allExplicitDependencies) {
        if (explicitDependencies.containsKey(dependency.name)) {
          // If we've seen the dependency before, make sure that we are
          // importing it the same way. There's several ways to import a
          // dependency. Hosted (from pub via version number), by path (e.g.
          // pointing at the version of a package we get from the Dart SDK
          // that we download with Flutter), by SDK (e.g. the "flutter"
          // package is explicitly from "sdk: flutter").
          //
          // This makes sure that we don't import a package in two different
          // ways, e.g. by saying "sdk: flutter" in one pubspec.yaml and
          // saying "path: ../../..." in another.
388 389
          final PubspecDependency previous = explicitDependencies[dependency.name]!;
          if (dependency.kind != previous.kind || dependency._lockTarget != previous._lockTarget) {
390 391
            throwToolExit(
                'Inconsistent requirements around ${dependency.name}; '
392 393
                'saw ${dependency.kind} (${dependency._lockTarget}) in "${dependency.sourcePath}" '
                'and ${previous.kind} (${previous._lockTarget}) in "${previous.sourcePath}".'
394 395 396 397
            );
          }
        }
        // Remember this dependency by name so we can look it up again.
398
        explicitDependencies[dependency.name] = dependency;
399 400 401 402 403 404
        // Normal dependencies are those we get from pub. The others we
        // already implicitly pin since we pull down one version of the
        // Flutter and Dart SDKs, so we track which those are here so that we
        // can omit them from our list of pinned dependencies later.
        if (dependency.kind != DependencyKind.normal) {
          specialDependencies.add(dependency.name);
405
        }
406 407
      }
    }
408
  }
409

410
  Future<void> _generateFakePackage({
411 412 413 414 415
    required Directory tempDir,
    required Iterable<PubspecDependency> dependencies,
    required List<PubspecYaml> pubspecs,
    required PubDependencyTree tree,
    required bool doUpgrade,
416
  }) async {
417
    Directory? temporaryFlutterSdk;
418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435
    final Directory syntheticPackageDir = tempDir.childDirectory('synthetic_package');
    final File fakePackage = _pubspecFor(syntheticPackageDir);
    fakePackage.createSync(recursive: true);
    fakePackage.writeAsStringSync(
      generateFakePubspec(
        dependencies,
        doUpgrade: doUpgrade,
      ),
    );
    // Create a synthetic flutter SDK so that transitive flutter SDK
    // constraints are not affected by this upgrade.
    if (doUpgrade) {
      temporaryFlutterSdk = createTemporaryFlutterSdk(
        globals.logger,
        globals.fs,
        globals.fs.directory(Cache.flutterRoot),
        pubspecs,
        tempDir,
436
      );
437
    }
438

439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456
    // Next we run "pub get" on it in order to force the download of any
    // needed packages to the pub cache, upgrading if requested.
    await pub.get(
      context: PubContext.updatePackages,
      project: FlutterProject.fromDirectory(syntheticPackageDir),
      upgrade: doUpgrade,
      offline: boolArgDeprecated('offline'),
      flutterRootOverride: temporaryFlutterSdk?.path,
    );

    if (doUpgrade) {
      // If upgrading, we run "pub deps --style=compact" on the result. We
      // pipe all the output to tree.fill(), which parses it so that it can
      // create a graph of all the dependencies so that we can figure out the
      // transitive dependencies later. It also remembers which version was
      // selected for each package.
      await pub.batch(
        <String>['deps', '--style=compact'],
457
        context: PubContext.updatePackages,
458 459
        directory: syntheticPackageDir.path,
        filter: tree.fill,
460 461
      );
    }
462
  }
463

464
  bool _upgradePubspecs({
465 466 467 468
    required PubDependencyTree tree,
    required List<PubspecYaml> pubspecs,
    required Set<String> specialDependencies,
    required Map<String, PubspecDependency> explicitDependencies,
469 470 471 472 473 474 475 476 477 478 479 480
  }) {
    // The transitive dependency tree for the fake package does not contain
    // dependencies between Flutter SDK packages and pub packages. We add them
    // here.
    for (final PubspecYaml pubspec in pubspecs) {
      final String package = pubspec.name;
      specialDependencies.add(package);
      tree._versions[package] = pubspec.version;
      assert(!tree._dependencyTree.containsKey(package));
      tree._dependencyTree[package] = <String>{};
      for (final PubspecDependency dependency in pubspec.dependencies) {
        if (dependency.kind == DependencyKind.normal) {
481
          tree._dependencyTree[package]!.add(dependency.name);
482 483
        }
      }
484
    }
485

486
    if (boolArgDeprecated('transitive-closure')) {
487 488 489 490 491
      tree._dependencyTree.forEach((String from, Set<String> to) {
        globals.printStatus('$from -> $to');
      });
      return true;
    }
492

493
    if (boolArgDeprecated('paths')) {
494
      showDependencyPaths(from: stringArgDeprecated('from')!, to: stringArgDeprecated('to')!, tree: tree);
495 496
      return true;
    }
497

498 499 500 501 502 503 504 505 506 507
    // Now that we have collected all the data, we can apply our dependency
    // versions to each pubspec.yaml that we collected. This mutates the
    // pubspec.yaml files.
    //
    // The specialDependencies argument is the set of package names to not pin
    // to specific versions because they are explicitly pinned by their
    // constraints. Here we list the names we earlier established we didn't
    // need to pin because they come from the Dart or Flutter SDKs.
    for (final PubspecYaml pubspec in pubspecs) {
      pubspec.apply(tree, specialDependencies);
508
    }
509 510
    return false;
  }
511

512
  Future<void> _runPubGetOnPackages(List<Directory> packages) async {
513
    final Stopwatch timer = Stopwatch()..start();
514
    int count = 0;
515

516 517 518 519 520 521 522 523 524 525 526
    // Now we run pub get on each of the affected packages to update their
    // pubspec.lock files with the right transitive dependencies.
    //
    // This can be expensive, so we run them in parallel. If we hadn't already
    // warmed the cache above, running them in parallel could be dangerous due
    // to contention when unpacking downloaded dependencies, but since we have
    // downloaded all that we need, it is safe to run them in parallel.
    final Status status = globals.logger.startProgress(
      'Running "flutter pub get" in affected packages...',
    );
    try {
527
      // int.tryParse will not accept null, but will convert empty string to null
528
      final int? maxJobs = int.tryParse(stringArgDeprecated('jobs') ?? '');
529
      final TaskQueue<void> queue = TaskQueue<void>(maxJobs: maxJobs);
530 531 532 533 534 535
      for (final Directory dir in packages) {
        unawaited(queue.add(() async {
          final Stopwatch stopwatch = Stopwatch();
          stopwatch.start();
          await pub.get(
            context: PubContext.updatePackages,
536
            project: FlutterProject.fromDirectory(dir),
537 538 539
            // All dependencies should already have been downloaded by the fake
            // package, so the concurrent checks can all happen offline.
            offline: true,
540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556
            printProgress: false,
          );
          stopwatch.stop();
          final double seconds = stopwatch.elapsedMilliseconds / 1000.0;
          final String relativeDir = globals.fs.path.relative(dir.path, from: Cache.flutterRoot);
          globals.printStatus('Ran pub get in $relativeDir in ${seconds.toStringAsFixed(1)}s...');
        }));
        count += 1;
      }
      unawaited(queue.add(() async {
        final Stopwatch stopwatch = Stopwatch();
        await _downloadCoverageData();
        stopwatch.stop();
        final double seconds = stopwatch.elapsedMilliseconds / 1000.0;
        globals.printStatus('Downloaded lcov data for package:flutter in ${seconds.toStringAsFixed(1)}s...');
      }));
      await queue.tasksComplete;
557
      status.stop();
558 559
      // The exception is rethrown, so don't catch only Exceptions.
    } catch (exception) { // ignore: avoid_catches_without_on_clauses
560
      status.cancel();
561
      rethrow;
562
    }
563

564
    final double seconds = timer.elapsedMilliseconds / 1000.0;
565
    globals.printStatus("\nRan 'pub get' $count time${count == 1 ? "" : "s"} and fetched coverage data in ${seconds.toStringAsFixed(1)}s.");
566
  }
567 568

  void showDependencyPaths({
569 570 571
    required String from,
    required String to,
    required PubDependencyTree tree,
572
  }) {
573
    if (!tree.contains(from)) {
574
      throwToolExit('Package $from not found in the dependency tree.');
575 576
    }
    if (!tree.contains(to)) {
577
      throwToolExit('Package $to not found in the dependency tree.');
578
    }
579

580
    final Queue<_DependencyLink> traversalQueue = Queue<_DependencyLink>();
581
    final Set<String> visited = <String>{};
582 583
    final List<_DependencyLink> paths = <_DependencyLink>[];

584
    traversalQueue.addFirst(_DependencyLink(from: null, to: from));
585
    while (traversalQueue.isNotEmpty) {
586
      final _DependencyLink link = traversalQueue.removeLast();
587
      if (link.to == to) {
588
        paths.add(link);
589 590
      }
      if (link.from != null) {
591
        visited.add(link.from!.to);
592
      }
593
      for (final String dependency in tree._dependencyTree[link.to]!) {
594
        if (!visited.contains(dependency)) {
595
          traversalQueue.addFirst(_DependencyLink(from: link, to: dependency));
596 597 598 599 600
        }
      }
    }

    for (_DependencyLink path in paths) {
601
      final StringBuffer buf = StringBuffer();
602
      while (path != null) {
603
        buf.write(path.to);
604
        path = path.from!;
605
        if (path != null) {
606
          buf.write(' <- ');
607
        }
608
      }
609
      globals.printStatus(buf.toString(), wrap: false);
610 611 612
    }

    if (paths.isEmpty) {
613
      globals.printStatus('No paths found from $from to $to');
614 615 616 617 618 619
    }
  }
}

class _DependencyLink {
  _DependencyLink({
620 621
    required this.from,
    required this.to,
622 623
  });

624
  final _DependencyLink? from;
625 626 627 628
  final String to;

  @override
  String toString() => '${from?.to} -> $to';
629
}
630

631 632 633 634 635 636
/// The various sections of a pubspec.yaml file.
///
/// We care about the "dependencies", "dev_dependencies", and
/// "dependency_overrides" sections, as well as the "name" and "version" fields
/// in the pubspec header bucketed into [header]. The others are all bucketed
/// into [other].
637
enum Section { header, dependencies, devDependencies, dependencyOverrides, builders, other }
638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656

/// The various kinds of dependencies we know and care about.
enum DependencyKind {
  // Dependencies that will be path or sdk dependencies but
  // for which we haven't yet parsed the data.
  unknown,

  // Regular dependencies with a specified version range.
  normal,

  // Dependency that uses an explicit path, e.g. into the Dart SDK.
  path,

  // Dependency defined as coming from an SDK (typically "sdk: flutter").
  sdk,

  // A dependency that was "normal", but for which we later found a "path" or
  // "sdk" dependency in the dependency_overrides section.
  overridden,
657

658
  // A dependency that uses git.
659
  git,
660 661 662 663 664
}

/// This is the string we output next to each of our autogenerated transitive
/// dependencies so that we can ignore them the next time we parse the
/// pubspec.yaml file.
665 666 667 668 669
const String kTransitiveMagicString= '# THIS LINE IS AUTOGENERATED - TO UPDATE USE "flutter update-packages --force-upgrade"';


/// This is the string output before a checksum of the packages used.
const String kDependencyChecksum = '# PUBSPEC CHECKSUM: ';
670 671 672 673 674 675 676 677

/// This class represents a pubspec.yaml file for the purposes of upgrading the
/// dependencies as done by this file.
class PubspecYaml {
  /// You create one of these by providing a directory, from which we obtain the
  /// pubspec.yaml and parse it into a line-by-line form.
  factory PubspecYaml(Directory directory) {
    final File file = _pubspecFor(directory);
678
    return _parse(file, file.readAsLinesSync());
679 680
  }

681
  PubspecYaml._(this.file, this.name, this.version, this.inputData, this.checksum);
682 683 684

  final File file; // The actual pubspec.yaml file.

685 686 687 688
  /// The package name.
  final String name;

  /// The package version.
689
  final String? version;
690

691 692
  final List<PubspecLine> inputData; // Each line of the pubspec.yaml file, parsed(ish).

693
  /// The package checksum.
694
  ///
695 696 697 698
  /// If this was not found in the pubspec, a synthetic checksum is created
  /// with a value of `-1`.
  final PubspecChecksum checksum;

699 700 701 702 703
  /// This parses each line of a pubspec.yaml file (a list of lines) into
  /// slightly more structured data (in the form of a list of PubspecLine
  /// objects). We don't just use a YAML parser because we care about comments
  /// and also because we can just define the style of pubspec.yaml files we care
  /// about (since they're all under our control).
704 705
  static PubspecYaml _parse(File file, List<String> lines) {
    final String filename = file.path;
706 707 708
    String? packageName;
    String? packageVersion;
    PubspecChecksum? checksum; // the checksum value used to verify that dependencies haven't changed.
709 710 711 712 713 714 715 716 717 718 719 720 721
    final List<PubspecLine> result = <PubspecLine>[]; // The output buffer.
    Section section = Section.other; // Which section we're currently reading from.
    bool seenMain = false; // Whether we've seen the "dependencies:" section.
    bool seenDev = false; // Whether we've seen the "dev_dependencies:" section.
    // The masterDependencies map is used to keep track of the objects
    // representing actual dependencies we've seen so far in this file so that
    // if we see dependency overrides we can update the actual dependency so it
    // knows that it's not really a dependency.
    final Map<String, PubspecDependency> masterDependencies = <String, PubspecDependency>{};
    // The "special" dependencies (the ones that use git: or path: or sdk: or
    // whatnot) have the style of having extra data after the line that declares
    // the dependency. So we track what is the "current" (or "last") dependency
    // that we are dealing with using this variable.
722
    PubspecDependency? lastDependency;
723 724
    for (int index = 0; index < lines.length; index += 1) {
      String line = lines[index];
725 726 727
      if (lastDependency == null) {
        // First we look to see if we're transitioning to a new top-level section.
        // The PubspecHeader.parse static method can recognize those headers.
728
        final PubspecHeader? header = PubspecHeader.parse(line); // See if it's a header.
729 730
        if (header != null) { // It is!
          section = header.section; // The parser determined what kind of section it is.
731
          if (section == Section.header) {
732
            if (header.name == 'name') {
733
              packageName = header.value;
734
            } else if (header.name == 'version') {
735
              packageVersion = header.value;
736
            }
737
          } else if (section == Section.dependencies) {
738 739
            // If we're entering the "dependencies" section, we want to make sure that
            // it's the first section (of those we care about) that we've seen so far.
740
            if (seenMain) {
741
              throwToolExit('Two dependencies sections found in $filename. There should only be one.');
742
            }
743
            if (seenDev) {
744
              throwToolExit('The dependencies section was after the dev_dependencies section in $filename. '
745
                    'To enable one-pass processing, the dependencies section must come before the '
746
                    'dev_dependencies section.');
747 748 749 750 751
            }
            seenMain = true;
          } else if (section == Section.devDependencies) {
            // Similarly, if we're entering the dev_dependencies section, we should verify
            // that we've not seen one already.
752
            if (seenDev) {
753
              throwToolExit('Two dev_dependencies sections found in $filename. There should only be one.');
754
            }
755 756 757
            seenDev = true;
          }
          result.add(header);
758 759 760 761 762
        } else if (section == Section.builders) {
          // Do nothing.
          // This line isn't a section header, and we're not in a section we care about.
          // We just stick the line into the output unmodified.
          result.add(PubspecLine(line));
763
        } else if (section == Section.other) {
764 765 766 767 768 769 770
          if (line.contains(kDependencyChecksum)) {
            // This is the pubspec checksum. After computing it, we remove it from the output data
            // since it will be recomputed later.
            checksum = PubspecChecksum.parse(line);
          } else {
            // This line isn't a section header, and we're not in a section we care about.
            // We just stick the line into the output unmodified.
771
            result.add(PubspecLine(line));
772
          }
773 774
        } else {
          // We're in a section we care about. Try to parse out the dependency:
775
          final PubspecDependency? dependency = PubspecDependency.parse(line, filename: filename);
776
          if (dependency != null) { // We got one!
777 778
            // Track whether or not this a dev dependency.
            dependency.isDevDependency = seenDev;
779 780 781 782 783 784 785 786 787 788 789 790 791 792
            result.add(dependency);
            if (dependency.kind == DependencyKind.unknown) {
              // If we didn't get a version number, then we need to be ready to
              // read the next line as part of this dependency, so keep track of
              // this dependency object.
              lastDependency = dependency;
            }
            if (section != Section.dependencyOverrides) {
              // If we're not in the overrides section, then just remember the
              // dependency, in case it comes up again later in the overrides
              // section.
              //
              // First, make sure it's a unique dependency. Listing dependencies
              // twice doesn't make sense.
793
              if (masterDependencies.containsKey(dependency.name)) {
794
                throwToolExit('$filename contains two dependencies on ${dependency.name}.');
795
              }
796 797 798 799 800 801 802 803 804
              masterDependencies[dependency.name] = dependency;
            } else {
              // If we _are_ in the overrides section, then go tell the version
              // we saw earlier (if any -- there might not be, we might be
              // overriding a transitive dependency) that we have overridden it,
              // so that later when we output the dependencies we can leave
              // the line unmodified.
              masterDependencies[dependency.name]?.markOverridden(dependency);
            }
805 806 807 808
          } else if (line.contains(kDependencyChecksum)) {
            // This is the pubspec checksum. After computing it, we remove it from the output data
            // since it will be recomputed later.
            checksum = PubspecChecksum.parse(line);
809 810 811 812
          } else {
            // We're in a section we care about but got a line we didn't
            // recognize. Maybe it's a comment or a blank line or something.
            // Just pass it through.
813
            result.add(PubspecLine(line));
814 815 816 817 818 819 820 821
          }
        }
      } else {
        // If we're here it means the last line was a dependency that needed
        // extra information to be parsed from the next line.
        //
        // Try to parse the line by giving it to the last PubspecDependency
        // object we created. If parseLock fails to recognize the line, it will
822 823
        // throw. If it does recognize the line and needs the following lines in
        // its lockLine, it'll return false.
824 825 826 827 828
        // Otherwise it returns true.
        //
        // If it returns true, then it will have updated itself internally to
        // store the information from this line.
        if (!lastDependency.parseLock(line, filename, lockIsOverride: section == Section.dependencyOverrides)) {
829 830 831 832 833 834 835
          // Ok we're dealing with some "git:" dependency. Consume lines until
          // we are out of the git dependency, and stuff them into the lock
          // line.
          lastDependency._lockLine = line;
          lastDependency._lockIsOverride = section == Section.dependencyOverrides;
          do {
            index += 1;
836 837 838
            if (index == lines.length) {
              throw StateError('Invalid pubspec.yaml: a "git" dependency section terminated early.');
            }
839
            line = lines[index];
840
            lastDependency._lockLine = '${lastDependency._lockLine}\n$line';
841
          } while (line.startsWith('   '));
842 843 844 845 846 847
        }
        // We're done with this special dependency, so reset back to null so
        // we'll go in the top section next time instead.
        lastDependency = null;
      }
    }
848
    return PubspecYaml._(file, packageName!, packageVersion, result, checksum ?? PubspecChecksum(null, ''));
849 850
  }

851
  /// This returns all the explicit dependencies that this pubspec.yaml lists under dependencies.
852
  Iterable<PubspecDependency> get dependencies {
853 854 855 856 857
    // It works by iterating over the parsed data from _parse above, collecting
    // all the dependencies that were found, ignoring any that are flagged as as
    // overridden by subsequent entries in the same file and any that have the
    // magic comment flagging them as auto-generated transitive dependencies
    // that we added in a previous run.
858 859 860
    return inputData
        .whereType<PubspecDependency>()
        .where((PubspecDependency data) => data.kind != DependencyKind.overridden && !data.isTransitive && !data.isDevDependency);
861 862 863
  }

  /// This returns all regular dependencies and all dev dependencies.
864
  Iterable<PubspecDependency> get allExplicitDependencies {
865 866 867
    return inputData
        .whereType<PubspecDependency>()
        .where((PubspecDependency data) => data.kind != DependencyKind.overridden && !data.isTransitive);
868 869
  }

870 871 872 873 874
  /// This returns all dependencies.
  Iterable<PubspecDependency> get allDependencies {
    return inputData.whereType<PubspecDependency>();
  }

875 876 877 878 879 880 881
  /// Take a dependency graph with explicit version numbers, and apply them to
  /// the pubspec.yaml, ignoring any that we know are special dependencies (those
  /// that depend on the Flutter or Dart SDK directly and are thus automatically
  /// pinned).
  void apply(PubDependencyTree versions, Set<String> specialDependencies) {
    assert(versions != null);
    final List<String> output = <String>[]; // the string data to output to the file, line by line
882 883
    final Set<String> directDependencies = <String>{}; // packages this pubspec directly depends on (i.e. not transitive)
    final Set<String> devDependencies = <String>{};
884
    Section section = Section.other; // the section we're currently handling
885 886

    // the line number where we're going to insert the transitive dependencies.
887
    int? endOfDirectDependencies;
888
    // The line number where we're going to insert the transitive dev dependencies.
889
    int? endOfDevDependencies;
890 891 892 893 894 895 896
    // Walk the pre-parsed input file, outputting it unmodified except for
    // updating version numbers, removing the old transitive dependencies lines,
    // and adding our new transitive dependencies lines. We also do a little
    // cleanup, removing trailing spaces, removing double-blank lines, leading
    // blank lines, and trailing blank lines, and ensuring the file ends with a
    // newline. This cleanup lets us be a little more aggressive while building
    // the output.
897
    for (final PubspecLine data in inputData) {
898 899 900 901 902 903
      if (data is PubspecHeader) {
        // This line was a header of some sort.
        //
        // If we're leaving one of the sections in which we can list transitive
        // dependencies, then remember this as the current last known valid
        // place to insert our transitive dependencies.
904
        if (section == Section.dependencies) {
905
          endOfDirectDependencies = output.length;
906 907
        }
        if (section == Section.devDependencies) {
908
          endOfDevDependencies = output.length;
909
        }
910 911 912 913 914 915 916
        section = data.section; // track which section we're now in.
        output.add(data.line); // insert the header into the output
      } else if (data is PubspecDependency) {
        // This was a dependency of some sort.
        // How we handle this depends on the section.
        switch (section) {
          case Section.devDependencies:
917
          case Section.dependencies:
918 919 920 921
            // For the dependencies and dev_dependencies sections, we reinsert
            // the dependency if it wasn't one of our autogenerated transitive
            // dependency lines.
            if (!data.isTransitive) {
922
              // Assert that we haven't seen it in this file already.
923
              assert(!directDependencies.contains(data.name) && !devDependencies.contains(data.name));
924 925 926 927 928 929 930
              if (data.kind == DependencyKind.normal) {
                // This is a regular dependency, so we need to update the
                // version number.
                //
                // We output data that matches the format that
                // PubspecDependency.parse can handle. The data.suffix is any
                // previously-specified trailing comment.
931 932
                assert(versions.contains(data.name),
                       "versions doesn't contain ${data.name}");
933 934 935 936 937 938
                output.add('  ${data.name}: ${versions.versionFor(data.name)}${data.suffix}');
              } else {
                // If it wasn't a regular dependency, then we output the line
                // unmodified. If there was an additional line (e.g. an "sdk:
                // flutter" line) then we output that too.
                output.add(data.line);
939
                if (data.lockLine != null) {
940
                  output.add(data.lockLine!);
941
                }
942 943 944
              }
              // Remember that we've dealt with this dependency so we don't
              // mention it again when doing the transitive dependencies.
945 946 947 948
              if (section == Section.dependencies) {
                directDependencies.add(data.name);
              } else {
                devDependencies.add(data.name);
949
              }
950 951 952
            }
            // Since we're in one of the places where we can list dependencies,
            // remember this as the current last known valid place to insert our
953
            // transitive dev dependencies. If the section is for regular dependencies,
954
            // then also remember the line for the end of direct dependencies.
955 956 957 958
            if (section == Section.dependencies) {
              endOfDirectDependencies = output.length;
            }
            endOfDevDependencies = output.length;
959
            break;
960 961 962 963
          case Section.builders:
          case Section.dependencyOverrides:
          case Section.header:
          case Section.other:
964 965
            // In other sections, pass everything through in its original form.
            output.add(data.line);
966
            if (data.lockLine != null) {
967
              output.add(data.lockLine!);
968
            }
969 970 971 972 973 974 975 976
            break;
        }
      } else {
        // Not a header, not a dependency, just pass that through unmodified.
        output.add(data.line);
      }
    }

977 978 979 980
    // If there are no dependencies or dev_dependencies sections, these will be
    // null. We have such files in our tests, so account for them here.
    endOfDirectDependencies ??= output.length;
    endOfDevDependencies ??= output.length;
981 982 983 984 985 986 987

    // Now include all the transitive dependencies and transitive dev dependencies.
    // The blocks of text to insert for each dependency section.
    final List<String> transitiveDependencyOutput = <String>[];
    final List<String> transitiveDevDependencyOutput = <String>[];

    // Which dependencies we need to handle for the transitive and dev dependency sections.
988 989
    final Set<String> transitiveDependencies = <String>{};
    final Set<String> transitiveDevDependencies = <String>{};
990

991
    // Merge the lists of dependencies we've seen in this file from dependencies, dev dependencies,
992
    // and the dependencies we know this file mentions that are already pinned
993
    // (and which didn't get special processing above).
994 995 996 997 998
    final Set<String> implied = <String>{
      ...directDependencies,
      ...specialDependencies,
      ...devDependencies,
    };
999

1000 1001
    // Create a new set to hold the list of packages we've already processed, so
    // that we don't redundantly process them multiple times.
1002
    final Set<String> done = <String>{};
1003
    for (final String package in directDependencies) {
1004
      transitiveDependencies.addAll(versions.getTransitiveDependenciesFor(package, seen: done, exclude: implied));
1005
    }
1006
    for (final String package in devDependencies) {
1007
      transitiveDevDependencies.addAll(versions.getTransitiveDependenciesFor(package, seen: done, exclude: implied));
1008
    }
1009

1010
    // Sort each dependency block lexically so that we don't get noisy diffs when upgrading.
1011
    final List<String> transitiveDependenciesAsList = transitiveDependencies.toList()..sort();
1012 1013
    final List<String> transitiveDevDependenciesAsList = transitiveDevDependencies.toList()..sort();

1014 1015 1016 1017
    String computeTransitiveDependencyLineFor(String package) {
      return '  $package: ${versions.versionFor(package)} $kTransitiveMagicString';
    }

1018
    // Add a line for each transitive dependency and transitive dev dependency using our magic string to recognize them later.
1019
    for (final String package in transitiveDependenciesAsList) {
1020
      transitiveDependencyOutput.add(computeTransitiveDependencyLineFor(package));
1021
    }
1022
    for (final String package in transitiveDevDependenciesAsList) {
1023
      transitiveDevDependencyOutput.add(computeTransitiveDependencyLineFor(package));
1024
    }
1025 1026

    // Build a sorted list of all dependencies for the checksum.
1027 1028 1029 1030 1031 1032
    final Set<String> checksumDependencies = <String>{
      ...directDependencies,
      ...devDependencies,
      ...transitiveDependenciesAsList,
      ...transitiveDevDependenciesAsList,
    }..removeAll(specialDependencies);
1033 1034 1035 1036 1037 1038 1039 1040 1041 1042

    // Add a blank line before and after each section to keep the resulting output clean.
    transitiveDependencyOutput
      ..insert(0, '')
      ..add('');
    transitiveDevDependencyOutput
      ..insert(0, '')
      ..add('');

    // Compute a new checksum from all sorted dependencies and their version and convert to a hex string.
1043
    final String checksumString = _computeChecksum(checksumDependencies, versions.versionFor);
1044 1045 1046 1047 1048 1049 1050 1051 1052

    // Insert the block of transitive dependency declarations into the output after [endOfDirectDependencies],
    // and the blocks of transitive dev dependency declarations into the output after [lastPossiblePlace]. Finally,
    // insert the [checksumString] at the very end.
    output
      ..insertAll(endOfDevDependencies, transitiveDevDependencyOutput)
      ..insertAll(endOfDirectDependencies, transitiveDependencyOutput)
      ..add('')
      ..add('$kDependencyChecksum$checksumString');
1053

1054
    // Remove trailing lines.
1055
    while (output.last.isEmpty) {
1056
      output.removeLast();
1057
    }
1058 1059 1060

    // Output the result to the pubspec.yaml file, skipping leading and
    // duplicate blank lines and removing trailing spaces.
1061
    final StringBuffer contents = StringBuffer();
1062 1063 1064 1065
    bool hadBlankLine = true;
    for (String line in output) {
      line = line.trimRight();
      if (line == '') {
1066
        if (!hadBlankLine) {
1067
          contents.writeln();
1068
        }
1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088
        hadBlankLine = true;
      } else {
        contents.writeln(line);
        hadBlankLine = false;
      }
    }
    file.writeAsStringSync(contents.toString());
  }
}

/// This is the base class for the objects that represent lines in the
/// pubspec.yaml files.
class PubspecLine {
  PubspecLine(this.line);

  /// The raw line as we saw it in the original file. This is used so that we can
  /// output the same line unmodified for the majority of lines.
  final String line;
}

1089 1090 1091 1092
/// A checksum of the non autogenerated dependencies.
class PubspecChecksum extends PubspecLine {
  PubspecChecksum(this.value, String line) : super(line);

1093
  /// The checksum value, computed using [Object.hash] over the direct, dev,
1094
  /// and special dependencies sorted lexically.
1095
  ///
1096
  /// If the line cannot be parsed, [value] will be null.
1097
  final String? value;
1098 1099

  /// Parses a [PubspecChecksum] from a line.
1100
  ///
1101 1102
  /// The returned PubspecChecksum will have a null [value] if no checksum could
  /// be found on this line. This is a value that [_computeChecksum] cannot return.
1103
  static PubspecChecksum parse(String line) {
1104
    final List<String> tokens = line.split(kDependencyChecksum);
1105
    if (tokens.length != 2) {
1106
      return PubspecChecksum(null, line);
1107
    }
1108
    return PubspecChecksum(tokens.last.trim(), line);
1109
  }
1110 1111
}

1112 1113
/// A header, e.g. "dependencies:".
class PubspecHeader extends PubspecLine {
1114
  PubspecHeader(
1115
    super.line,
1116 1117 1118
    this.section, {
    this.name,
    this.value,
1119
  });
1120 1121

  /// The section of the pubspec where the parse [line] appears.
1122 1123
  final Section section;

1124 1125 1126 1127 1128 1129 1130 1131 1132 1133
  /// The name in the pubspec line providing a name/value pair, such as "name"
  /// and "version".
  ///
  /// Example:
  ///
  /// The value of this field extracted from the following line is "version".
  ///
  /// ```
  /// version: 0.16.5
  /// ```
1134
  final String? name;
1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145

  /// The value in the pubspec line providing a name/value pair, such as "name"
  /// and "version".
  ///
  /// Example:
  ///
  /// The value of this field extracted from the following line is "0.16.5".
  ///
  /// ```
  /// version: 0.16.5
  /// ```
1146
  final String? value;
1147

1148
  static PubspecHeader? parse(String line) {
1149 1150 1151 1152 1153 1154
    // We recognize any line that:
    //  * doesn't start with a space (i.e. is aligned on the left edge)
    //  * ignoring trailing spaces and comments, ends with a colon
    //  * has contents before the colon
    // We also try to recognize which of the kinds of Sections it is
    // by comparing those contents against known strings.
1155
    if (line.startsWith(' ')) {
1156
      return null;
1157
    }
1158
    final String strippedLine = _stripComments(line);
1159
    if (!strippedLine.contains(':') || strippedLine.length <= 1) {
1160
      return null;
1161
    }
1162 1163 1164
    final List<String> parts = strippedLine.split(':');
    final String sectionName = parts.first;
    final String value = parts.last.trim();
1165 1166
    switch (sectionName) {
      case 'dependencies':
1167
        return PubspecHeader(line, Section.dependencies);
1168
      case 'dev_dependencies':
1169
        return PubspecHeader(line, Section.devDependencies);
1170
      case 'dependency_overrides':
1171
        return PubspecHeader(line, Section.dependencyOverrides);
1172 1173
      case 'builders':
        return PubspecHeader(line, Section.builders);
1174 1175
      case 'name':
      case 'version':
1176
        return PubspecHeader(line, Section.header, name: sectionName, value: value);
1177
      default:
1178
        return PubspecHeader(line, Section.other);
1179 1180 1181 1182 1183 1184 1185
    }
  }

  /// Returns the input after removing trailing spaces and anything after the
  /// first "#".
  static String _stripComments(String line) {
    final int hashIndex = line.indexOf('#');
1186
    if (hashIndex < 0) {
1187
      return line.trimRight();
1188
    }
1189 1190 1191 1192 1193 1194
    return line.substring(0, hashIndex).trimRight();
  }
}

/// A dependency, as represented by a line (or two) from a pubspec.yaml file.
class PubspecDependency extends PubspecLine {
1195
  PubspecDependency(
1196
    super.line,
1197 1198
    this.name,
    this.suffix, {
1199 1200 1201 1202
    required this.isTransitive,
    required DependencyKind kind,
    required this.version,
    required this.sourcePath,
1203
  }) : _kind = kind;
1204

1205
  static PubspecDependency? parse(String line, { required String filename }) {
1206 1207 1208 1209 1210 1211 1212 1213
    // We recognize any line that:
    //  * starts with exactly two spaces, no more or less
    //  * has some content, then a colon
    //
    // If we recognize the line, then we look to see if there's anything after
    // the colon, ignoring comments. If there is, then this is a normal
    // dependency, otherwise it's an unknown one.
    //
1214 1215 1216
    // We also try and save the version string, if any. This is used to verify
    // the checksum of package deps.
    //
1217 1218 1219 1220 1221 1222
    // We also look at the trailing comment, if any, to see if it is the magic
    // string that identifies the line as a transitive dependency that we
    // previously pinned, so we can ignore it.
    //
    // We remember the trailing comment, if any, so that we can reconstruct the
    // line later. We forget the specified version range, if any.
1223
    if (line.length < 4 || line.startsWith('   ') || !line.startsWith('  ')) {
1224
      return null;
1225
    }
1226 1227
    final int colonIndex = line.indexOf(':');
    final int hashIndex = line.indexOf('#');
1228
    if (colonIndex < 3) { // two spaces at 0 and 1, a character at 2
1229
      return null;
1230 1231
    }
    if (hashIndex >= 0 && hashIndex < colonIndex) {
1232
      return null;
1233
    }
1234 1235 1236 1237 1238 1239
    final String package = line.substring(2, colonIndex).trimRight();
    assert(package.isNotEmpty);
    assert(line.startsWith('  $package'));
    String suffix = '';
    bool isTransitive = false;
    String stripped;
1240
    String version = '';
1241 1242 1243 1244 1245
    if (hashIndex >= 0) {
      assert(hashIndex > colonIndex);
      final String trailingComment = line.substring(hashIndex, line.length);
      assert(line.endsWith(trailingComment));
      isTransitive = trailingComment == kTransitiveMagicString;
1246
      suffix = ' $trailingComment';
1247 1248 1249 1250
      stripped = line.substring(colonIndex + 1, hashIndex).trimRight();
    } else {
      stripped = line.substring(colonIndex + 1, line.length).trimRight();
    }
1251 1252 1253
    if (colonIndex != -1) {
      version = line.substring(colonIndex + 1, hashIndex != -1 ? hashIndex : line.length).trim();
    }
1254
    return PubspecDependency(line, package, suffix, isTransitive: isTransitive, version: version, kind: stripped.isEmpty ? DependencyKind.unknown : DependencyKind.normal, sourcePath: filename);
1255 1256 1257 1258
  }

  final String name; // the package name
  final String suffix; // any trailing comment we found
1259
  final String version; // the version string if found, or blank.
1260 1261
  final bool isTransitive; // whether the suffix matched kTransitiveMagicString
  final String sourcePath; // the filename of the pubspec.yaml file, for error messages
1262
  late bool isDevDependency; // Whether this dependency is under the `dev dependencies` section.
1263 1264 1265 1266 1267

  DependencyKind get kind => _kind;
  DependencyKind _kind = DependencyKind.normal;

  /// If we're a path or sdk dependency, the path or sdk in question.
1268
  String? _lockTarget;
1269 1270 1271

  /// If we were a two-line dependency, the second line (see the inherited [line]
  /// for the first).
1272 1273
  String? get lockLine => _lockLine;
  String? _lockLine;
1274 1275 1276 1277 1278 1279

  /// If we're a path or sdk dependency, whether we were found in a
  /// dependencies/dev_dependencies section, or a dependency_overrides section.
  /// We track this so that we can put ourselves in the right section when
  /// generating the fake pubspec.yaml.
  bool get lockIsOverride => _lockIsOverride;
1280
  late bool _lockIsOverride;
1281

1282 1283 1284
  static const String _pathPrefix = '    path: ';
  static const String _sdkPrefix = '    sdk: ';
  static const String _gitPrefix = '    git:';
1285

1286 1287 1288 1289 1290 1291 1292 1293
  /// Whether the dependency points to a package in the Flutter SDK.
  ///
  /// There are two ways one can point to a Flutter package:
  ///
  /// - Using a "sdk: flutter" dependency.
  /// - Using a "path" dependency that points somewhere in the Flutter
  ///   repository other than the "bin" directory.
  bool get pointsToSdk {
1294
    if (_kind == DependencyKind.sdk) {
1295
      return true;
1296
    }
1297

1298 1299 1300 1301
    final String? lockTarget = _lockTarget;
    if (_kind == DependencyKind.path && lockTarget != null &&
        !globals.fs.path.isWithin(globals.fs.path.join(Cache.flutterRoot!, 'bin'), lockTarget) &&
        globals.fs.path.isWithin(Cache.flutterRoot!, lockTarget)) {
1302
      return true;
1303
    }
1304 1305 1306 1307

    return false;
  }

1308 1309 1310
  /// If parse decided we were a two-line dependency, this is called to parse the second line.
  /// We throw if we couldn't parse this line.
  /// We return true if we parsed it and stored the line in lockLine.
1311
  /// We return false if we parsed it and it's a git dependency that needs the next few lines.
1312
  bool parseLock(String line, String pubspecPath, { required bool lockIsOverride }) {
1313 1314
    assert(lockIsOverride != null);
    assert(kind == DependencyKind.unknown);
1315
    if (line.startsWith(_pathPrefix)) {
1316
      // We're a path dependency; remember the (absolute) path.
1317 1318
      _lockTarget = globals.fs.path.canonicalize(
          globals.fs.path.absolute(globals.fs.path.dirname(pubspecPath), line.substring(_pathPrefix.length, line.length))
1319
      );
1320
      _kind = DependencyKind.path;
1321
    } else if (line.startsWith(_sdkPrefix)) {
1322
      // We're an SDK dependency.
1323
      _lockTarget = line.substring(_sdkPrefix.length, line.length);
1324
      _kind = DependencyKind.sdk;
1325
    } else if (line.startsWith(_gitPrefix)) {
1326 1327
      // We're a git: dependency. We'll have to get the next few lines.
      _kind = DependencyKind.git;
1328 1329
      return false;
    } else {
1330
      throwToolExit('Could not parse additional details for dependency $name; line was: "$line"');
1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350
    }
    _lockIsOverride = lockIsOverride;
    _lockLine = line;
    return true;
  }

  void markOverridden(PubspecDependency sibling) {
    // This is called when we find a dependency is mentioned a second time,
    // first in dependencies/dev_dependencies, and then in dependency_overrides.
    // It is called on the one found in dependencies/dev_dependencies, so that
    // we'll later know to report our version as "any" in the fake pubspec.yaml
    // and unmodified in the official pubspec.yamls.
    assert(sibling.name == name);
    assert(sibling.sourcePath == sourcePath);
    assert(sibling.kind != DependencyKind.normal);
    _kind = DependencyKind.overridden;
  }

  /// This generates the entry for this dependency for the pubspec.yaml for the
  /// fake package that we'll use to get the version numbers figured out.
1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366
  ///
  /// When called with [doUpgrade] as [true], the version constrains will be set
  /// to >= whatever the previous version was. If [doUpgrade] is [false], then
  /// the previous version is used again as an exact pin.
  void describeForFakePubspec(StringBuffer dependencies, StringBuffer overrides, { bool doUpgrade = true }) {
    final String versionToUse;
    // This should only happen when manually adding new dependencies; otherwise
    // versions should always be pinned exactly
    if (version.isEmpty || version == 'any') {
      versionToUse = 'any';
    } else if (doUpgrade) {
      // Must wrap in quotes for Yaml parsing
      versionToUse = "'>= $version'";
    } else {
      versionToUse = version;
    }
1367 1368 1369 1370 1371 1372
    switch (kind) {
      case DependencyKind.unknown:
      case DependencyKind.overridden:
        assert(kind != DependencyKind.unknown);
        break;
      case DependencyKind.normal:
1373
        if (!kManuallyPinnedDependencies.containsKey(name)) {
1374
          dependencies.writeln('  $name: $versionToUse');
1375
        }
1376 1377
        break;
      case DependencyKind.path:
1378
        if (lockIsOverride) {
1379
          dependencies.writeln('  $name: $versionToUse');
1380
          overrides.writeln('  $name:');
1381
          overrides.writeln('    path: $_lockTarget');
1382 1383
        } else {
          dependencies.writeln('  $name:');
1384
          dependencies.writeln('    path: $_lockTarget');
1385 1386 1387
        }
        break;
      case DependencyKind.sdk:
1388
        if (lockIsOverride) {
1389
          dependencies.writeln('  $name: $versionToUse');
1390
          overrides.writeln('  $name:');
1391
          overrides.writeln('    sdk: $_lockTarget');
1392 1393
        } else {
          dependencies.writeln('  $name:');
1394
          dependencies.writeln('    sdk: $_lockTarget');
1395 1396
        }
        break;
1397
      case DependencyKind.git:
1398
        if (lockIsOverride) {
1399
          dependencies.writeln('  $name: $versionToUse');
1400 1401 1402 1403 1404 1405
          overrides.writeln('  $name:');
          overrides.writeln(lockLine);
        } else {
          dependencies.writeln('  $name:');
          dependencies.writeln(lockLine);
        }
1406
        break;
1407 1408
    }
  }
1409 1410 1411 1412 1413

  @override
  String toString() {
    return '$name: $version';
  }
1414 1415 1416 1417
}

/// Generates the File object for the pubspec.yaml file of a given Directory.
File _pubspecFor(Directory directory) {
1418 1419
  return directory.fileSystem.file(
    directory.fileSystem.path.join(directory.path, 'pubspec.yaml'));
1420 1421 1422 1423
}

/// Generates the source of a fake pubspec.yaml file given a list of
/// dependencies.
1424 1425
@visibleForTesting
String generateFakePubspec(
1426
  Iterable<PubspecDependency> dependencies, {
1427
  bool doUpgrade = false
1428
}) {
1429 1430
  final StringBuffer result = StringBuffer();
  final StringBuffer overrides = StringBuffer();
1431
  final bool verbose = doUpgrade;
1432
  result.writeln('name: flutter_update_packages');
1433
  result.writeln('environment:');
1434
  result.writeln("  sdk: '>=2.10.0 <4.0.0'");
1435 1436
  result.writeln('dependencies:');
  overrides.writeln('dependency_overrides:');
1437
  if (kManuallyPinnedDependencies.isNotEmpty) {
1438 1439 1440
    if (verbose) {
      globals.printStatus('WARNING: the following packages use hard-coded version constraints:');
    }
1441 1442 1443 1444
    final Set<String> allTransitive = <String>{
      for (final PubspecDependency dependency in dependencies)
        dependency.name,
    };
1445
    kManuallyPinnedDependencies.forEach((String package, String version) {
1446 1447 1448 1449 1450 1451 1452
      // Don't add pinned dependency if it is not in the set of all transitive dependencies.
      if (!allTransitive.contains(package)) {
        if (verbose) {
          globals.printStatus('Skipping $package because it was not transitive');
        }
        return;
      }
1453
      result.writeln('  $package: $version');
1454 1455 1456
      if (verbose) {
        globals.printStatus('  - $package: $version');
      }
1457
    });
1458
  }
1459
  for (final PubspecDependency dependency in dependencies) {
1460
    if (!dependency.pointsToSdk) {
1461
      dependency.describeForFakePubspec(result, overrides, doUpgrade: doUpgrade);
1462 1463
    }
  }
1464 1465 1466 1467 1468 1469 1470 1471 1472
  result.write(overrides.toString());
  return result.toString();
}

/// This object tracks the output of a call to "pub deps --style=compact".
///
/// It ends up holding the full graph of dependencies, and the version number for
/// each one.
class PubDependencyTree {
1473
  final Map<String, String?> _versions = <String, String?>{};
1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502
  final Map<String, Set<String>> _dependencyTree = <String, Set<String>>{};

  /// Handles the output from "pub deps --style=compact".
  ///
  /// That output is of this form:
  ///
  /// ```
  /// package_name 0.0.0
  ///
  /// dependencies:
  /// - analyzer 0.31.0-alpha.0 [watcher args package_config collection]
  /// - archive 1.0.31 [crypto args path]
  /// - args 0.13.7
  /// - cli_util 0.1.2+1 [path]
  ///
  /// dev dependencies:
  /// - async 1.13.3 [collection]
  /// - barback 0.15.2+11 [stack_trace source_span pool async collection path]
  ///
  /// dependency overrides:
  /// - analyzer 0.31.0-alpha.0 [watcher args package_config collection]
  /// ```
  ///
  /// We ignore all the lines that don't start with a hyphen. For each other
  /// line, we ignore any line that mentions a package we've already seen (this
  /// happens when the overrides section mentions something that was in the
  /// dependencies section). We ignore if something is a dependency or
  /// dev_dependency (pub won't use different versions for those two).
  ///
1503
  /// We then parse out the package name, version number, and sub-dependencies for
1504 1505
  /// each entry, and store than in our _versions and _dependencyTree fields
  /// above.
1506
  String? fill(String message) {
1507 1508 1509
    if (message.startsWith('- ')) {
      final int space2 = message.indexOf(' ', 2);
      int space3 = message.indexOf(' ', space2 + 1);
1510
      if (space3 < 0) {
1511
        space3 = message.length;
1512
      }
1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527
      final String package = message.substring(2, space2);
      if (!contains(package)) {
        // Some packages get listed in the dependency overrides section too.
        // We just ignore those. The data is the same either way.
        final String version = message.substring(space2 + 1, space3);
        List<String> dependencies;
        if (space3 < message.length) {
          assert(message[space3 + 1] == '[');
          assert(message[message.length - 1] == ']');
          final String allDependencies = message.substring(space3 + 2, message.length - 1);
          dependencies = allDependencies.split(' ');
        } else {
          dependencies = const <String>[];
        }
        _versions[package] = version;
1528
        _dependencyTree[package] = Set<String>.of(dependencies);
1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539
      }
    }
    return null;
  }

  /// Whether we know about this package.
  bool contains(String package) {
    return _versions.containsKey(package);
  }

  /// The transitive closure of all the dependencies for the given package,
1540
  /// excluding any listed in `seen`.
1541 1542
  Iterable<String> getTransitiveDependenciesFor(
    String package, {
1543 1544 1545
    required Set<String> seen,
    required Set<String> exclude,
    List<String>? result,
1546
  }) {
1547 1548
    assert(seen != null);
    assert(exclude != null);
1549
    result ??= <String>[];
1550 1551
    final Set<String>? dependencies = _dependencyTree[package];
    if (dependencies == null) {
1552 1553
      // We have no transitive dependencies extracted for flutter_sdk packages
      // because they were omitted from pubspec.yaml used for 'pub upgrade' run.
1554
      return result;
1555
    }
1556
    for (final String dependency in dependencies) {
1557
      if (!seen.contains(dependency)) {
1558
        if (!exclude.contains(dependency)) {
1559
          result.add(dependency);
1560
        }
1561
        seen.add(dependency);
1562
        getTransitiveDependenciesFor(dependency, seen: seen, exclude: exclude, result: result);
1563 1564
      }
    }
1565
    return result;
1566 1567 1568 1569
  }

  /// The version that a particular package ended up with.
  String versionFor(String package) {
1570
    return _versions[package]!;
1571 1572
  }
}
1573

1574
// Produces a 16-bit checksum from the codePoints of the package name and
1575
// version strings using Fletcher's algorithm.
1576
String _computeChecksum(Iterable<String> names, String Function(String name) getVersion) {
1577 1578
  int lowerCheck = 0;
  int upperCheck = 0;
1579
  final List<String> sortedNames = names.toList()..sort();
1580
  for (final String name in sortedNames) {
1581
    final String version = getVersion(name);
1582
    if (version == null) {
1583
      continue;
1584
    }
1585
    final String value = '$name: $version';
1586
    // Each code unit is 16 bits.
1587
    for (final int codeUnit in value.codeUnits) {
1588 1589 1590 1591 1592 1593 1594 1595
      final int upper = codeUnit >> 8;
      final int lower = codeUnit & 0xFF;
      lowerCheck = (lowerCheck + upper) % 255;
      upperCheck = (upperCheck + lowerCheck) % 255;
      lowerCheck = (lowerCheck + lower) % 255;
      upperCheck = (upperCheck + lowerCheck) % 255;
    }
  }
1596
  return ((upperCheck << 8) | lowerCheck).toRadixString(16).padLeft(4, '0');
1597
}
1598 1599 1600

/// Create a synthetic Flutter SDK so that pub version solving does not get
/// stuck on the old versions.
1601 1602 1603 1604 1605 1606
@visibleForTesting
Directory createTemporaryFlutterSdk(
  Logger logger,
  FileSystem fileSystem,
  Directory realFlutter,
  List<PubspecYaml> pubspecs,
1607
  Directory tempDir,
1608
) {
1609 1610 1611 1612 1613 1614 1615
  final Set<String> currentPackages = <String>{};
  for (final FileSystemEntity entity in realFlutter.childDirectory('packages').listSync()) {
    // Verify that a pubspec.yaml exists to ensure this isn't a left over directory.
    if (entity is Directory && entity.childFile('pubspec.yaml').existsSync()) {
      currentPackages.add(fileSystem.path.basename(entity.path));
    }
  }
1616 1617 1618 1619 1620 1621

  final Map<String, PubspecYaml> pubspecsByName = <String, PubspecYaml>{};
  for (final PubspecYaml pubspec in pubspecs) {
    pubspecsByName[pubspec.name] = pubspec;
  }

1622
  final Directory directory = tempDir.childDirectory('flutter_upgrade_sdk')
1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634
    ..createSync();
  // Fill in version info.
  realFlutter.childFile('version')
    .copySync(directory.childFile('version').path);

  // Directory structure should mirror the current Flutter SDK
  final Directory packages = directory.childDirectory('packages');
  for (final String flutterPackage in currentPackages) {
    final File pubspecFile = packages
      .childDirectory(flutterPackage)
      .childFile('pubspec.yaml')
      ..createSync(recursive: true);
1635
    final PubspecYaml? pubspecYaml = pubspecsByName[flutterPackage];
1636
    if (pubspecYaml == null) {
1637
      logger.printWarning(
1638 1639 1640 1641
        "Unexpected package '$flutterPackage' found in packages directory",
      );
      continue;
    }
1642 1643 1644 1645 1646
    final StringBuffer output = StringBuffer('name: $flutterPackage\n');

    // Fill in SDK dependency constraint.
    output.write('''
environment:
1647
  sdk: ">=2.7.0 <4.0.0"
1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678
''');

    output.writeln('dependencies:');
    for (final PubspecDependency dependency in pubspecYaml.dependencies) {
      if (dependency.isTransitive || dependency.isDevDependency) {
        continue;
      }
      if (dependency.kind == DependencyKind.sdk) {
        output.writeln('  ${dependency.name}:\n    sdk: flutter');
        continue;
      }
      output.writeln('  ${dependency.name}: any');
    }
    pubspecFile.writeAsStringSync(output.toString());
  }

  // Create the sky engine pubspec.yaml
  directory
    .childDirectory('bin')
    .childDirectory('cache')
    .childDirectory('pkg')
    .childDirectory('sky_engine')
    .childFile('pubspec.yaml')
    ..createSync(recursive: true)
    ..writeAsStringSync('''
name: sky_engine
version: 0.0.99
description: Dart SDK extensions for dart:ui
homepage: http://flutter.io
# sky_engine requires sdk_ext support in the analyzer which was added in 1.11.x
environment:
1679
  sdk: '>=1.11.0 <4.0.0'
1680 1681 1682 1683
''');

  return directory;
}