Update Scalafix migration to handle explicit imports from the plugin autoImport object

Author: DavidGregory084

build.sbt

@@ -64,4 +64,6 @@ lazy val `sbt-tpolecat-scalafix` = scalafixProject("sbt-tpolecat")
     )
   )
   .inputSettings(addSbtPlugin("io.github.davidgregory084" % "sbt-tpolecat" % "0.4.0"))
+  .inputConfigure(_.enablePlugins(SbtPlugin))
   .outputConfigure(_.dependsOn(`sbt-tpolecat-plugin`))
+  .outputConfigure(_.enablePlugins(SbtPlugin))

scalafix/input/src/main/scala/org/typelevel/fix/v0_5Tests.scala

@@ -5,6 +5,8 @@ rule = v0_5
 import io.github.davidgregory084.TpolecatPlugin
 import io.github.davidgregory084.{ScalaVersion => Version}
 import io.github.davidgregory084.{DevMode => Dev}
+import io.github.davidgregory084.TpolecatPlugin.autoImport._
+import io.github.davidgregory084.TpolecatPlugin.autoImport.{ tpolecatCiModeEnvVar, scalacOptionsFor }
 import io.github.davidgregory084._
 import _root_.io.github.davidgregory084.ScalacOption
 import _root_.io.github.davidgregory084._

scalafix/output/src/main/scala/org/typelevel/fix/v0_5Tests.scala

@@ -1,5 +1,6 @@
 // format: off
 import org.typelevel.sbt.tpolecat.{ DevMode => Dev, TpolecatPlugin, _ }
+import org.typelevel.sbt.tpolecat.TpolecatPlugin.autoImport.{ scalacOptionsFor, tpolecatCiModeEnvVar, _ }
 import org.typelevel.scalacoptions.{ ScalaVersion => Version, ScalacOption, _ }
 // format: on
 

scalafix/rules/src/main/scala/fix/v0_5.scala

@@ -40,6 +40,10 @@ class v0_5 extends SemanticRule("v0_5") {
 
   val ScalacOptionsSym = JavaMajorVersionSym + ScalacOptionSym + ScalaVersionSym
 
+  // autoImport contents
+  val TpolecatPluginAutoImportSym =
+    SymbolMatcher.exact("io/github/davidgregory084/TpolecatPlugin.autoImport.")
+
   private def makeSelector(packages: String*): Term.Ref =
     packages.tail.foldLeft(Term.Name(packages.head): Term.Ref) { case (selector, pkg) =>
       Term.Select(selector, Term.Name(pkg))
@@ -50,6 +54,14 @@ class v0_5 extends SemanticRule("v0_5") {
       Importer(makeSelector("org", "typelevel", "sbt", "tpolecat"), List(importee))
     )
 
+  private def makeAutoImportObjectImport(importee: Importee) =
+    Patch.addGlobalImport(
+      Importer(
+        makeSelector("org", "typelevel", "sbt", "tpolecat", "TpolecatPlugin", "autoImport"),
+        List(importee)
+      )
+    )
+
   private def makeScalacOptionsImport(importee: Importee) =
     Patch.addGlobalImport(
       Importer(makeSelector("org", "typelevel", "scalacoptions"), List(importee))
@@ -72,5 +84,14 @@ class v0_5 extends SemanticRule("v0_5") {
               makeScalacOptionsImport(importee) +
               Patch.removeImportee(importee)
         }.asPatch
+      case importer @ Importer(ref, importees) if TpolecatPluginAutoImportSym.matches(ref) =>
+        importees.collect {
+          case importee @ Importee.Name(name) =>
+            makeAutoImportObjectImport(importee) + Patch.removeImportee(importee)
+          case rename @ Importee.Rename(name, _) =>
+            makeAutoImportObjectImport(rename) + Patch.removeImportee(rename)
+          case importee @ Importee.Wildcard() =>
+            makeAutoImportObjectImport(importee) + Patch.removeImportee(importee)
+        }.asPatch
     }.asPatch
 }