Collect Refaster rules across all provided files

Fixes google#552

refaster/src/main/java/com/google/errorprone/refaster/RefasterRuleCompilerAnalyzer.java

@@ -29,13 +29,15 @@
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.util.ArrayList;
+import java.util.Collections;
 import java.util.List;
 
 /**
  * TaskListener that receives compilation of a Refaster rule class and outputs a serialized analyzer
  * to the specified path.
  */
 public class RefasterRuleCompilerAnalyzer implements TaskListener {
+  private final List<CodeTransformer> allRules = Collections.synchronizedList(new ArrayList<>());
   private final Context context;
   private final Path destinationPath;
 
@@ -46,6 +48,10 @@ public class RefasterRuleCompilerAnalyzer implements TaskListener {
 
   @Override
   public void finished(TaskEvent taskEvent) {
+    if (taskEvent.getKind() == Kind.COMPILATION) {
+      persist();
+      return;
+    }
     if (taskEvent.getKind() != Kind.ANALYZE) {
       return;
     }
@@ -67,9 +73,13 @@ public Void visitClass(ClassTree node, Context context) {
     if (rules.isEmpty()) {
       throw new IllegalArgumentException("Did not find any Refaster templates");
     }
+    allRules.addAll(rules);
+  }
+
+  private void persist() {
     try (ObjectOutputStream output =
         new ObjectOutputStream(Files.newOutputStream(destinationPath))) {
-      output.writeObject(CompositeCodeTransformer.compose(rules));
+      output.writeObject(CompositeCodeTransformer.compose(allRules));
     } catch (IOException e) {
       throw new RuntimeException(e);
     }