Skip to content

Commit 50336a2

Browse files
MasseGuillaumeMasseGuillaume
committed
Add more cases for breakout
1 parent 89cc1dd commit 50336a2

File tree

3 files changed

+244
-50
lines changed

3 files changed

+244
-50
lines changed

‎scalafix/input/src/main/scala/fix/BreakoutSrc.scala

Lines changed: 82 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -4,23 +4,90 @@ rule = "scala:fix.CrossCompat"
44
package fix
55

66
import scala.collection.breakOut
7+
import scala.collection.{immutable, mutable}
78

8-
object BreakoutSrc {
9-
val xs = List(1, 2, 3)
9+
import scala.concurrent.Future
10+
import scala.concurrent.ExecutionContext.Implicits.global
1011

11-
xs.collect{ case x => x }(breakOut): Set[Int]
12-
xs.flatMap(x => List(x))(breakOut): collection.SortedSet[Int]
13-
xs.map(_ + 1)(breakOut): Set[Int]
14-
xs.reverseMap(_ + 1)(breakOut): Set[Int]
15-
xs.scanLeft(0)((a, b) => a + b)(breakOut): Set[Int]
16-
xs.updated(0, 1)(breakOut): Set[Int]
12+
class BreakoutSrc(ts: Traversable[Int], vec: Vector[Int], list: List[Int], seq: Seq[Int]) {
1713

18-
(xs ++ xs)(breakOut): Set[Int]
19-
(1 +: xs)(breakOut): Set[Int]
20-
(xs :+ 1)(breakOut): Set[Int]
21-
(xs ++: xs)(breakOut): Set[Int]
14+
// `IndexedSeqOptimized.zip`
15+
vec.zip(vec)(breakOut): Map[Int, Int]
2216

23-
xs.union(xs)(breakOut): Set[Int]
24-
xs.zip(xs)(breakOut): Map[Int, Int]
25-
xs.zipAll(xs, 0, 0)(breakOut): Array[(Int, Int)]
17+
// `IterableLike.zip`
18+
seq.zip(seq)(breakOut): Map[Int, Int]
19+
20+
// `IterableLike.zipAll`
21+
seq.zipAll(seq, 0, 0)(breakOut): Array[(Int, Int)]
22+
23+
// `List ++`
24+
(list ++ list)(breakOut): Set[Int]
25+
26+
// `List +:`
27+
(1 +: list)(breakOut): Set[Int]
28+
29+
// `List.collect`
30+
list.collect{ case x => x}(breakOut): Set[Int]
31+
32+
// `List.flatMap`
33+
list.flatMap(x => List(x))(breakOut): Set[Int]
34+
35+
// `List.map`
36+
list.map(x => x)(breakOut): Set[Int]
37+
38+
39+
// `SeqLike.reverseMap`
40+
seq.reverseMap(_ + 1)(breakOut): Set[Int]
41+
42+
// `SeqLike +:`
43+
(1 +: seq)(breakOut): List[Int]
44+
45+
// `SeqLike :+`
46+
(seq :+ 1)(breakOut): List[Int]
47+
48+
// `SeqLike.updated`
49+
(seq.updated(0, 0))(breakOut): List[Int]
50+
51+
// `SeqLike.union`
52+
seq.union(seq)(breakOut): List[Int]
53+
54+
55+
//`SetLike.map`
56+
Set(1).map(x => x)(breakOut): List[Int]
57+
58+
59+
// `TraversableLike ++`
60+
(ts ++ ts )(breakOut): Set[Int]
61+
62+
// `TraversableLike ++:`
63+
(ts ++: ts)(breakOut): Set[Int]
64+
65+
// `TraversableLike.collect`
66+
ts.collect{ case x => x }(breakOut): Set[Int]
67+
68+
// `TraversableLike.flatMap`
69+
ts.flatMap(x => List(x))(breakOut): collection.SortedSet[Int]
70+
71+
// `TraversableLike.map`
72+
ts.map(_ + 1)(breakOut): Set[Int]
73+
74+
// `TraversableLike.scanLeft`
75+
ts.scanLeft(0)((a, b) => a + b)(breakOut): Set[Int]
76+
77+
78+
// `Vector ++`
79+
(vec ++ List(1))(breakOut): List[Int]
80+
81+
// `Vector +:`
82+
(1 +: vec)(breakOut): List[Int]
83+
84+
// `Vector :+`
85+
(vec :+ 1)(breakOut): List[Int]
86+
87+
// `Vector.updated`
88+
(vec.updated(0, 0))(breakOut): List[Int]
89+
90+
// TODO
91+
// Future.sequence(List(Future(1)))(breakOut, global): Future[Seq[Int]]
92+
// Future.traverse(List(1))(x => Future(x))(breakOut, global): Future[Seq[Int]]
2693
}

‎scalafix/output212/src/main/scala/fix/BreakoutSrc.scala

Lines changed: 82 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -3,25 +3,91 @@
33

44
package fix
55

6+
import scala.collection.{immutable, mutable}
7+
8+
import scala.concurrent.Future
9+
import scala.concurrent.ExecutionContext.Implicits.global
610
import scala.collection.compat._
711

8-
object BreakoutSrc {
9-
val xs = List(1, 2, 3)
12+
class BreakoutSrc(ts: Iterable[Int], vec: Vector[Int], list: List[Int], seq: Seq[Int]) {
1013

11-
xs.iterator.collect{ case x => x }.to(scala.collection.immutable.Set): Set[Int]
12-
xs.iterator.flatMap(x => List(x)).to(scala.collection.SortedSet): collection.SortedSet[Int]
13-
xs.iterator.map(_ + 1).to(scala.collection.immutable.Set): Set[Int]
14-
xs.reverseIterator.map(_ + 1).to(scala.collection.immutable.Set): Set[Int]
15-
xs.iterator.scanLeft(0)((a, b) => a + b).to(scala.collection.immutable.Set): Set[Int]
16-
xs.view.updated(0, 1).to(scala.collection.immutable.Set): Set[Int]
14+
// `IndexedSeqOptimized.zip`
15+
vec.iterator.zip(vec.iterator).toMap: Map[Int, Int]
1716

18-
(xs.iterator ++ xs).to(scala.collection.immutable.Set): Set[Int]
19-
(1 +: xs.view).to(scala.collection.immutable.Set): Set[Int]
20-
(xs.view :+ 1).to(scala.collection.immutable.Set): Set[Int]
21-
(xs ++: xs.view).to(scala.collection.immutable.Set): Set[Int]
17+
// `IterableLike.zip`
18+
seq.iterator.zip(seq.iterator).toMap: Map[Int, Int]
19+
20+
// `IterableLike.zipAll`
21+
seq.iterator.zipAll(seq.iterator, 0, 0).to(scala.Array): Array[(Int, Int)]
22+
23+
// `List ++`
24+
(list.iterator ++ list).to(scala.collection.immutable.Set): Set[Int]
25+
26+
// `List +:`
27+
(1 +: list.view).to(scala.collection.immutable.Set): Set[Int]
28+
29+
// `List.collect`
30+
list.iterator.collect{ case x => x}.to(scala.collection.immutable.Set): Set[Int]
31+
32+
// `List.flatMap`
33+
list.iterator.flatMap(x => List(x)).to(scala.collection.immutable.Set): Set[Int]
34+
35+
// `List.map`
36+
list.iterator.map(x => x).to(scala.collection.immutable.Set): Set[Int]
37+
38+
39+
// `SeqLike.reverseMap`
40+
seq.reverseIterator.map(_ + 1).to(scala.collection.immutable.Set): Set[Int]
41+
42+
// `SeqLike +:`
43+
(1 +: seq.view).to(scala.collection.immutable.List): List[Int]
44+
45+
// `SeqLike :+`
46+
(seq.view :+ 1).to(scala.collection.immutable.List): List[Int]
47+
48+
// `SeqLike.updated`
49+
(seq.view.updated(0, 0)).to(scala.collection.immutable.List): List[Int]
50+
51+
// `SeqLike.union`
52+
seq.iterator.concat(seq).to(scala.collection.immutable.List): List[Int]
2253

23-
xs.iterator.concat(xs).to(scala.collection.immutable.Set): Set[Int]
24-
xs.iterator.zip(xs.iterator).toMap: Map[Int, Int]
25-
xs.iterator.zipAll(xs.iterator, 0, 0).to(scala.Array): Array[(Int, Int)]
26-
}
2754

55+
//`SetLike.map`
56+
Set(1).iterator.map(x => x).to(scala.collection.immutable.List): List[Int]
57+
58+
59+
// `TraversableLike ++`
60+
(ts.iterator ++ ts ).to(scala.collection.immutable.Set): Set[Int]
61+
62+
// `TraversableLike ++:`
63+
(ts ++: ts.view).to(scala.collection.immutable.Set): Set[Int]
64+
65+
// `TraversableLike.collect`
66+
ts.iterator.collect{ case x => x }.to(scala.collection.immutable.Set): Set[Int]
67+
68+
// `TraversableLike.flatMap`
69+
ts.iterator.flatMap(x => List(x)).to(scala.collection.SortedSet): collection.SortedSet[Int]
70+
71+
// `TraversableLike.map`
72+
ts.iterator.map(_ + 1).to(scala.collection.immutable.Set): Set[Int]
73+
74+
// `TraversableLike.scanLeft`
75+
ts.iterator.scanLeft(0)((a, b) => a + b).to(scala.collection.immutable.Set): Set[Int]
76+
77+
78+
// `Vector ++`
79+
(vec.iterator ++ List(1)).to(scala.collection.immutable.List): List[Int]
80+
81+
// `Vector +:`
82+
(1 +: vec.view).to(scala.collection.immutable.List): List[Int]
83+
84+
// `Vector :+`
85+
(vec.view :+ 1).to(scala.collection.immutable.List): List[Int]
86+
87+
// `Vector.updated`
88+
(vec.view.updated(0, 0)).to(scala.collection.immutable.List): List[Int]
89+
90+
// TODO
91+
// Future.sequence(List(Future(1)))(breakOut, global): Future[Seq[Int]]
92+
// Future.traverse(List(1))(x => Future(x))(breakOut, global): Future[Seq[Int]]
93+
}

‎scalafix/rules/src/main/scala/fix/Stable212Base.scala

Lines changed: 80 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -56,31 +56,68 @@ trait Stable212Base extends CrossCompatibility { self: SemanticRule =>
5656
val breakOut = SymbolMatcher.exact(Symbol("_root_.scala.collection.package.breakOut(Lscala/collection/generic/CanBuildFrom;)Lscala/collection/generic/CanBuildFrom;."))
5757

5858
// infix operators
59+
60+
val `TraversableLike ++` = Symbol("_root_.scala.collection.TraversableLike#`++`(Lscala/collection/GenTraversableOnce;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
61+
val `Vector ++` = Symbol("_root_.scala.collection.immutable.Vector#`++`(Lscala/collection/GenTraversableOnce;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
5962
val `List ++` = Symbol("_root_.scala.collection.immutable.List#`++`(Lscala/collection/GenTraversableOnce;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
63+
64+
val `SeqLike +:` = Symbol("_root_.scala.collection.SeqLike#`+:`(Ljava/lang/Object;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
65+
val `Vector +:` = Symbol("_root_.scala.collection.immutable.Vector#`+:`(Ljava/lang/Object;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
6066
val `List +:` = Symbol("_root_.scala.collection.immutable.List#`+:`(Ljava/lang/Object;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
67+
6168
val `SeqLike :+` = Symbol("_root_.scala.collection.SeqLike#`:+`(Ljava/lang/Object;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
69+
val `Vector :+` = Symbol("_root_.scala.collection.immutable.Vector#`:+`(Ljava/lang/Object;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
70+
6271
val `TraversableLike ++:` = Symbol("_root_.scala.collection.TraversableLike#`++:`(Lscala/collection/Traversable;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
6372

64-
val operatorsIteratorSymbols = List(`List ++`)
65-
val operatorsViewSymbols = List(`List +:`, `SeqLike :+`, `TraversableLike ++:`)
73+
val operatorsIteratorSymbols = List(`TraversableLike ++`, `List ++`, `Vector ++`)
74+
val operatorsViewSymbols = List(
75+
`SeqLike +:`, `Vector +:`, `List +:`,
76+
`SeqLike :+`, `Vector :+`,
77+
`TraversableLike ++:`
78+
)
79+
6680
val operatorsSymbols = operatorsViewSymbols ++ operatorsIteratorSymbols
6781

6882
val operatorsIterator = SymbolMatcher.exact(operatorsIteratorSymbols: _*)
6983
val operatorsView = SymbolMatcher.exact(operatorsViewSymbols: _*)
7084
val operators = SymbolMatcher.exact(operatorsSymbols: _*)
7185

7286
// select
73-
val `List.collect` = Symbol("_root_.scala.collection.immutable.List#collect(Lscala/PartialFunction;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
74-
val `List.flatMap` = Symbol("_root_.scala.collection.immutable.List#flatMap(Lscala/Function1;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
75-
val `List.map` = Symbol("_root_.scala.collection.immutable.List#map(Lscala/Function1;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
76-
val `IterableLike.zip` = Symbol("_root_.scala.collection.IterableLike#zip(Lscala/collection/GenIterable;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
77-
val `IterableLike.zipAll` = Symbol("_root_.scala.collection.IterableLike#zipAll(Lscala/collection/GenIterable;Ljava/lang/Object;Ljava/lang/Object;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
78-
val `SeqLike.union` = Symbol("_root_.scala.collection.SeqLike#union(Lscala/collection/GenSeq;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
79-
val `SeqLike.updated` = Symbol("_root_.scala.collection.SeqLike#updated(ILjava/lang/Object;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
80-
val `SeqLike.reverseMap` = Symbol("_root_.scala.collection.SeqLike#reverseMap(Lscala/Function1;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
81-
82-
val functionsIteratorSymbols = List(`List.collect`, `List.flatMap`, `List.map`, `IterableLike.zip`, `IterableLike.zipAll`, `SeqLike.union`)
83-
val functionsViewSymbols = List(`SeqLike.updated`)
87+
val `List.collect` = Symbol("_root_.scala.collection.immutable.List#collect(Lscala/PartialFunction;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
88+
val `TraversableLike.collect` = Symbol("_root_.scala.collection.TraversableLike#collect(Lscala/PartialFunction;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
89+
90+
val `List.flatMap` = Symbol("_root_.scala.collection.immutable.List#flatMap(Lscala/Function1;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
91+
val `TraversableLike.flatMap` = Symbol("_root_.scala.collection.TraversableLike#flatMap(Lscala/Function1;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
92+
93+
val `List.map` = Symbol("_root_.scala.collection.immutable.List#map(Lscala/Function1;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
94+
val `SetLike.map` = Symbol("_root_.scala.collection.SetLike#map(Lscala/Function1;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
95+
val `TraversableLike.map` = Symbol("_root_.scala.collection.TraversableLike#map(Lscala/Function1;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
96+
97+
val `IterableLike.zip` = Symbol("_root_.scala.collection.IterableLike#zip(Lscala/collection/GenIterable;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
98+
99+
val `IndexedSeqOptimized.zip` = Symbol("_root_.scala.collection.IndexedSeqOptimized#zip(Lscala/collection/GenIterable;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
100+
101+
val `IterableLike.zipAll` = Symbol("_root_.scala.collection.IterableLike#zipAll(Lscala/collection/GenIterable;Ljava/lang/Object;Ljava/lang/Object;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
102+
103+
val `SeqLike.union` = Symbol("_root_.scala.collection.SeqLike#union(Lscala/collection/GenSeq;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
104+
105+
val `SeqLike.updated` = Symbol("_root_.scala.collection.SeqLike#updated(ILjava/lang/Object;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
106+
val `Vector.updated` = Symbol("_root_.scala.collection.immutable.Vector#updated(ILjava/lang/Object;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
107+
val `SeqLike.reverseMap` = Symbol("_root_.scala.collection.SeqLike#reverseMap(Lscala/Function1;Lscala/collection/generic/CanBuildFrom;)Ljava/lang/Object;.")
108+
109+
val functionsZipSymbols = List(
110+
`IterableLike.zip`,
111+
`IndexedSeqOptimized.zip`,
112+
`IterableLike.zipAll`
113+
)
114+
val functionsIteratorSymbols = List(
115+
`List.collect`, `TraversableLike.collect`,
116+
`List.flatMap`, `TraversableLike.flatMap`,
117+
`List.map`, `SetLike.map`, `TraversableLike.map`,
118+
`SeqLike.union`
119+
) ++ functionsZipSymbols
120+
val functionsViewSymbols = List(`SeqLike.updated`, `Vector.updated`)
84121
val functionsReverseIteratorSymbols = List(`SeqLike.reverseMap`)
85122
val functionsSymbols = functionsIteratorSymbols ++ functionsViewSymbols ++ functionsReverseIteratorSymbols
86123

@@ -89,7 +126,7 @@ trait Stable212Base extends CrossCompatibility { self: SemanticRule =>
89126
val functionsView = SymbolMatcher.exact(functionsViewSymbols: _*)
90127
val functions = SymbolMatcher.exact(functionsSymbols: _*)
91128

92-
val functionsZip = SymbolMatcher.exact(`IterableLike.zip`, `IterableLike.zipAll`)
129+
val functionsZip = SymbolMatcher.exact(functionsZipSymbols: _*)
93130

94131
// special select
95132

@@ -174,11 +211,10 @@ trait Stable212Base extends CrossCompatibility { self: SemanticRule =>
174211
}
175212
}
176213

214+
val breakOutImports = List.newBuilder[Importee]
215+
177216
val rewriteBreakout =
178217
ctx.tree.collect {
179-
case i: Importee if breakOut.matches(i) =>
180-
ctx.removeImportee(i)
181-
182218
case ap0 @ Term.Apply(ap @ Term.ApplyInfix(lhs, operators(op), _, List(rhs)), List(breakOut(bo))) =>
183219
val subject =
184220
if(isLeftAssociative(op)) lhs
@@ -195,7 +231,6 @@ trait Stable212Base extends CrossCompatibility { self: SemanticRule =>
195231
fixIt(intermediate, subject, ap, bo, ap0)
196232

197233
case ap0 @ Term.Apply(ap @ Term.Apply(Term.Select(lhs, functions(op)), rhs :: _), List(breakOut(bo))) =>
198-
199234
val intermediateLhs =
200235
op match {
201236
case functionsIterator(_) => "iterator"
@@ -224,10 +259,20 @@ trait Stable212Base extends CrossCompatibility { self: SemanticRule =>
224259

225260
case ap0 @ Term.Apply(ap @ Term.Apply(Term.Apply(Term.Select(lhs, `TraversableLike.scanLeft`(op)), _), _), List(breakOut(bo))) =>
226261
fixIt("iterator", lhs, ap, bo, ap0)
262+
263+
case i: Importee if breakOut.matches(i) =>
264+
breakOutImports += i
265+
Patch.empty
266+
227267
}.asPatch
228268

229269
val compatImport =
230-
if(rewriteBreakout.nonEmpty) addCompatImport(ctx)
270+
if(rewriteBreakout.nonEmpty) {
271+
val removeImports = breakOutImports.result().map(i => ctx.removeImportee(i)).asPatch
272+
273+
removeImports +
274+
addCompatImport(ctx)
275+
}
231276
else Patch.empty
232277

233278
rewriteBreakout + compatImport
@@ -438,6 +483,22 @@ trait Stable212Base extends CrossCompatibility { self: SemanticRule =>
438483
}
439484
}
440485

486+
var lastPath: Option[String] = None
487+
def trace(in: Any, ctx: RuleCtx): Unit = {
488+
val path =
489+
ctx.input match {
490+
case Input.File(path, _) => path.toString
491+
case Input.VirtualFile(path, _) => path.toString
492+
case _ => "???"
493+
}
494+
495+
if (lastPath != Some(path)) {
496+
println(s"--- $path ---")
497+
lastPath = Some(path)
498+
}
499+
println(in)
500+
}
501+
441502
override def fix(ctx: RuleCtx): Patch = {
442503
replaceSymbols0(ctx) +
443504
replaceCanBuildFrom(ctx) +

0 commit comments

Comments
 (0)