pkg/kernel/lib/src/tool/find_referenced_libraries.dart

// Copyright (c) 2021, the Dart project authors.  Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

// @dart = 2.9

import 'package:kernel/ast.dart';
import 'package:kernel/visitor.dart';

Set<Library> findAllReferencedLibraries(List<Library> from) {
  _LibraryCollector collector = new _LibraryCollector();
  for (Library library in from) {
    collector.visitLibrary(library);
  }
  return collector.allSeenLibraries;
}

bool duplicateLibrariesReachable(List<Library> from) {
  Set<Uri> seenUris = {};
  for (Library lib in findAllReferencedLibraries(from)) {
    if (!seenUris.add(lib.importUri)) return true;
  }
  return false;
}

class _LibraryCollector extends RecursiveVisitor<Null> {
  Set<Library> allSeenLibraries = {};

  Null defaultNode(Node node) {
    if (node is NamedNode) {
      // Named nodes can be linked to.
      seen(node);
    } else if (node is Name) {
      if (node.library != null) {
        seen(node.library);
      }
    }
    super.defaultNode(node);
  }

  Null defaultMemberReference(Member node) {
    seen(node);
    super.defaultMemberReference(node);
  }

  void seen(TreeNode node) {
    TreeNode parent = node;
    while (parent != null && parent is! Library) {
      parent = parent.parent;
    }
    allSeenLibraries.add(parent);
  }
}