Add a "terminal" callback to TastyInspector

This interface allows performing arbitrary operations on all compilation units.

Author: abgruszecki

tasty-inspector/src/scala/tasty/inspector/TastyInspector.scala

@@ -20,6 +20,9 @@ trait TastyInspector:
   /** Process a TASTy file using TASTy reflect */
   protected def processCompilationUnit(using QuoteContext)(root: qctx.reflect.Tree): Unit
 
+  /** Called after all compilation units are processed */
+  protected def finishProcessingCompilationUnits(using QuoteContext): Unit = ()
+
   /** Load and process TASTy files using TASTy reflect
    *
    *  @param tastyFiles List of paths of `.tasty` files
@@ -70,7 +73,8 @@ trait TastyInspector:
       override protected def transformPhases: List[List[Phase]] = Nil
 
       override protected def backendPhases: List[List[Phase]] =
-        List(new TastyInspectorPhase) ::  // Print all loaded classes
+        List(new TastyInspectorPhase) ::  // Perform a callback for each compilation unit
+        List(new TastyInspectorFinishPhase) :: // Perform a final callback
         Nil
 
       override def newRun(implicit ctx: Context): Run =
@@ -89,6 +93,19 @@ trait TastyInspector:
 
     end TastyInspectorPhase
 
+    class TastyInspectorFinishPhase extends Phase:
+
+      override def phaseName: String = "tastyInspectorFinish"
+
+      override def run(implicit ctx: Context): Unit =
+        if !alreadyRan then
+          val qctx = QuoteContextImpl()
+          self.finishProcessingCompilationUnits(using qctx)
+          alreadyRan = true
+
+      var alreadyRan = false
+    end TastyInspectorFinishPhase
+
     val currentClasspath = ClasspathFromClassloader(getClass.getClassLoader)
     val fullClasspath = (classpath :+ currentClasspath).mkString(pathSeparator)
     val args = "-from-tasty" :: "-Yretain-trees" :: "-classpath" :: fullClasspath :: classes