WildcardTypeRewrite

xuwei-k · Dec 28, 2023 · d4eee69 · d4eee69
1 parent 13b9f52
commit d4eee69
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/rules/src/main/scala/fix/WildcardTypeRewrite.scala b/rules/src/main/scala/fix/WildcardTypeRewrite.scala
@@ -0,0 +1,15 @@
+package fix
+
+import scala.meta.Type
+import scalafix.Patch
+import scalafix.v1.SyntacticDocument
+import scalafix.v1.SyntacticRule
+import scala.meta.tokens.Token
+
+class WildcardTypeRewrite extends SyntacticRule("WildcardTypeRewrite") {
+  override def fix(implicit doc: SyntacticDocument): Patch = {
+    doc.tree.collect { case t: Type.Wildcard =>
+      t.tokens.find(_.is[Token.Underscore]).map(x => Patch.replaceToken(x, "?")).asPatch
+    }.asPatch
+  }
+}