Track licenses needed as code is being printed in order to avoid printing the license of code that ends up being entirely removed from the final bundle.

PiperOrigin-RevId: 484050870

Author: 12wrigja

src/com/google/javascript/jscomp/AbstractCommandLineRunner.java

@@ -42,6 +42,10 @@
 import com.google.gson.reflect.TypeToken;
 import com.google.gson.stream.JsonReader;
 import com.google.gson.stream.JsonWriter;
+import com.google.javascript.jscomp.CodePrinter.LicenseTracker;
+import com.google.javascript.jscomp.Compiler.ChunkGraphAwareLicenseTracker;
+import com.google.javascript.jscomp.Compiler.ScriptNodeLicensesOnlyTracker;
+import com.google.javascript.jscomp.Compiler.SingleBinaryLicenseTracker;
 import com.google.javascript.jscomp.CompilerOptions.JsonStreamMode;
 import com.google.javascript.jscomp.CompilerOptions.OutputJs;
 import com.google.javascript.jscomp.CompilerOptions.TweakProcessing;
@@ -1036,7 +1040,8 @@ String getModuleOutputFileName(JSChunk m) {
 
   @VisibleForTesting
   @GwtIncompatible("Unnecessary")
-  void writeModuleOutput(String fileName, Appendable out, JSChunk m) throws IOException {
+  void writeModuleOutput(String fileName, Appendable out, LicenseTracker lt, JSChunk m)
+      throws IOException {
     if (parsedModuleWrappers == null) {
       parsedModuleWrappers =
           parseModuleWrappers(
@@ -1050,6 +1055,7 @@ void writeModuleOutput(String fileName, Appendable out, JSChunk m) throws IOExce
     writeOutput(
         out,
         compiler,
+        lt,
         m,
         parsedModuleWrappers.get(m.getName()).replace("%basename%", baseName),
         "%s",
@@ -1067,6 +1073,7 @@ void writeModuleOutput(String fileName, Appendable out, JSChunk m) throws IOExce
   void writeOutput(
       Appendable out,
       Compiler compiler,
+      LicenseTracker licenseTracker,
       @Nullable JSChunk module,
       String wrapper,
       String codePlaceholder,
@@ -1079,7 +1086,7 @@ void writeOutput(
     }
     checkState(compiler.getOptions().outputJs == OutputJs.NORMAL);
 
-    String code = module == null ? compiler.toSource() : compiler.toSource(module);
+    String code = module == null ? compiler.toSource() : compiler.toSource(licenseTracker, module);
     writeOutput(out, compiler, code, wrapper, codePlaceholder, escaper, filename);
   }
 
@@ -1654,7 +1661,7 @@ int processResults(Result result, List<JSChunk> modules, B options) throws IOExc
           outputSourceMap(options, config.jsOutputFile);
         }
       } else {
-        DiagnosticType error = outputModuleBinaryAndSourceMaps(compiler.getModules(), options);
+        DiagnosticType error = outputModuleBinaryAndSourceMaps(compiler.getModuleGraph(), options);
         if (error != null) {
           compiler.report(JSError.make(error));
           return 1;
@@ -1719,6 +1726,9 @@ void outputSingleBinary(B options) throws IOException {
       writeOutput(
           jsOutput,
           compiler,
+          // So long as the JSChunk arg is null the compiler will write all sources to jsOutput
+          // Use single-binary license tracking to dedupe licenses among all the inputs
+          new SingleBinaryLicenseTracker(compiler),
           (JSChunk) null,
           config.outputWrapper,
           marker,
@@ -1736,6 +1746,9 @@ JsonFileSpec createJsonFile(B options, String outputMarker, Function<String, Str
     writeOutput(
         jsOutput,
         compiler,
+        // So long as the JSChunk arg is null the compiler will write all sources to jsOutput
+        // Use single-binary license tracking to dedupe licenses among all the inputs
+        new SingleBinaryLicenseTracker(compiler),
         (JSChunk) null,
         config.outputWrapper,
         outputMarker,
@@ -1768,7 +1781,8 @@ void outputJsonStream() throws IOException {
 
   @GwtIncompatible("Unnecessary")
   private @Nullable DiagnosticType outputModuleBinaryAndSourceMaps(
-      Iterable<JSChunk> modules, B options) throws IOException {
+      JSChunkGraph moduleGraph, B options) throws IOException {
+    Iterable<JSChunk> modules = moduleGraph.getAllChunks();
     parsedModuleWrappers = parseModuleWrappers(config.moduleWrapper, modules);
     if (!isOutputInJson()) {
       // make sure the method generates all dirs up to the latest /
@@ -1790,6 +1804,7 @@ void outputJsonStream() throws IOException {
       return INVALID_MODULE_SOURCEMAP_PATTERN;
     }
 
+    ChunkGraphAwareLicenseTracker mlicenseTracker = new ChunkGraphAwareLicenseTracker(compiler);
     for (JSChunk m : modules) {
       if (m.getName().equals(JSChunk.WEAK_CHUNK_NAME)) {
         // Skip the weak module, which is always empty.
@@ -1808,7 +1823,8 @@ void outputJsonStream() throws IOException {
           if (options.shouldGatherSourceMapInfo()) {
             compiler.resetAndIntitializeSourceMap();
           }
-          writeModuleOutput(moduleFilename, writer, m);
+          mlicenseTracker.setCurrentChunkContext(m);
+          writeModuleOutput(moduleFilename, writer, mlicenseTracker, m);
           if (options.shouldGatherSourceMapInfo()) {
             compiler.getSourceMap().appendTo(mapFileOut, moduleFilename);
           }
@@ -1834,7 +1850,7 @@ private JsonFileSpec createJsonFileFromModule(JSChunk module) throws IOException
 
     String filename = getModuleOutputFileName(module);
     StringBuilder output = new StringBuilder();
-    writeModuleOutput(filename, output, module);
+    writeModuleOutput(filename, output, new ScriptNodeLicensesOnlyTracker(compiler), module);
 
     JsonFileSpec jsonFile = new JsonFileSpec(output.toString(), filename);
 

src/com/google/javascript/jscomp/CodeConsumer.java

@@ -55,8 +55,14 @@ void endSourceMapping(Node node) {
   }
 
   /**
-   * Provides a means of interrupting the CodeGenerator. Derived classes should
-   * return false to stop further processing.
+   * Indicates to the CodeConsumer that this Node might carry licensing information, and allows the
+   * code consumer to manage licenses as it sees fit.
+   */
+  void trackLicenses(Node node) {}
+
+  /**
+   * Provides a means of interrupting the CodeGenerator. Derived classes should return false to stop
+   * further processing.
    */
   boolean continueProcessing() {
     return true;

src/com/google/javascript/jscomp/CodeGenerator.java

@@ -204,6 +204,7 @@ protected void add(Node node, Context context, boolean printComments) {
     if (printComments) {
       printLeadingCommentsInOrder(node);
     }
+    cc.trackLicenses(node);
 
     Token type = node.getToken();
     String opstr = NodeUtil.opToStr(type);