Merge pull request #128 from MasseGuillaume/roughly-apply-infix

Add ApplyInfix syntax for Roughly

Author: julienrf

scalafix/input/src/main/scala/fix/RoughlySrc.scala

@@ -20,6 +20,8 @@ object RoughlySrc {
   val d2 = 1L -> 1
   val multi = new m.HashMap[Int, m.Set[Int]] with m.MultiMap[Int, Int]
 
+  Map(d) mapValues (id)          : i.Map[Int, Int]
+  Map(d).mapValues(id)           : i.Map[Int, Int]
   // i.SortedMap(d).mapValues(id): i.SortedMap[Int, Int]
   // m.SortedMap(d).mapValues(id): c.SortedMap[Int, Int]
   i.IntMap(d).mapValues(f2)      : i.Map[Int, Int]
@@ -31,6 +33,8 @@ object RoughlySrc {
   multi.mapValues(f2)            : c.Map[Int, Int]
   props.mapValues(id)            : c.Map[String, String]
 
+  Map(d) filterKeys (f)          : i.Map[Int, Int]
+  Map(d).filterKeys(f)           : i.Map[Int, Int]
   // i.SortedMap(d).filterKeys(f): i.SortedMap[Int, Int]
   // m.SortedMap(d).filterKeys(f): c.SortedMap[Int, Int]
   i.IntMap(d).filterKeys(f)    : i.Map[Int, Int]

scalafix/output/src/main/scala/fix/RoughlySrc.scala

@@ -16,6 +16,8 @@ object RoughlySrc {
   val d2 = 1L -> 1
   val multi = new m.HashMap[Int, m.Set[Int]] with m.MultiMap[Int, Int]
 
+  (Map(d) mapValues (id)).toMap          : i.Map[Int, Int]
+  Map(d).mapValues(id).toMap           : i.Map[Int, Int]
   // i.SortedMap(d).mapValues(id): i.SortedMap[Int, Int]
   // m.SortedMap(d).mapValues(id): c.SortedMap[Int, Int]
   i.IntMap(d).mapValues(f2).toMap      : i.Map[Int, Int]
@@ -27,6 +29,8 @@ object RoughlySrc {
   multi.mapValues(f2).toMap            : c.Map[Int, Int]
   props.mapValues(id).toMap            : c.Map[String, String]
 
+  (Map(d) filterKeys (f)).toMap          : i.Map[Int, Int]
+  Map(d).filterKeys(f).toMap           : i.Map[Int, Int]
   // i.SortedMap(d).filterKeys(f): i.SortedMap[Int, Int]
   // m.SortedMap(d).filterKeys(f): c.SortedMap[Int, Int]
   i.IntMap(d).filterKeys(f).toMap    : i.Map[Int, Int]

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

@@ -69,9 +69,17 @@ final case class Roughly(index: SemanticdbIndex, config: RoughlyConfig)
 
     val collectFixes =
       ctx.tree.collect {
+        case ap @ Term.ApplyInfix(_, mapValues(_), _, _) if strictMapValues =>
+          ctx.addLeft(ap, "(") +
+          ctx.addRight(ap, ").toMap")
+
         case ap @ Term.Apply(Term.Select(_, mapValues(_)), List(_)) if strictMapValues =>
           ctx.addRight(ap, ".toMap")
 
+        case ap @ Term.ApplyInfix(_, filterKeys(_), _, _) if strictFilterKeys =>
+          ctx.addLeft(ap, "(") +
+          ctx.addRight(ap, ").toMap")
+
         case ap @ Term.Apply(Term.Select(_, filterKeys(_)), List(_)) if strictFilterKeys =>
           ctx.addRight(ap, ".toMap")