Skip to content

Commit 70dcc34

Browse files
nicolasstuckinicolasstucki
committed
Use infix extension to indicate right associativity with natural order
1 parent 248c1a2 commit 70dcc34

19 files changed

+120
-117
lines changed

‎compiler/src/dotty/tools/dotc/ast/Desugar.scala

+12-8
Original file line numberDiff line numberDiff line change
@@ -996,11 +996,18 @@ object desugar {
996996

997997
def badRightAssoc(problem: String) =
998998
report.error(em"right-associative extension method $problem", mdef.srcPos)
999-
() // extParamss ++ mdef.paramss
999+
extParamss ++ mdef.paramss
10001000

10011001
rightParam match
10021002
case ValDefs(vparam :: Nil) =>
1003-
if !vparam.mods.is(Given) then
1003+
if vparam.mods.is(Given) then
1004+
badRightAssoc("cannot start with using clause")
1005+
else if mdef.mods.is(Infix) then
1006+
// New encoding:
1007+
// we keep the extension method as is and rely on the swap of arguments at call site
1008+
extParamss ++ mdef.paramss
1009+
else
1010+
// Old encoding:
10041011
// we merge the extension parameters with the method parameters,
10051012
// swapping the operator arguments:
10061013
// e.g.
@@ -1010,16 +1017,13 @@ object desugar {
10101017
// def %:[A](using B)[E](f: F)(c: C)(using D)(g: G)(using H): Res = ???
10111018
//
10121019
// If you change the names of the clauses below, also change them in right-associative-extension-methods.md
1013-
// val (leftTyParamsAndLeadingUsing, leftParamAndTrailingUsing) = extParamss.span(isUsingOrTypeParamClause)
1014-
() // leftTyParamsAndLeadingUsing ::: rightTyParams ::: rightParam :: leftParamAndTrailingUsing ::: paramss1
1015-
else
1016-
badRightAssoc("cannot start with using clause")
1020+
val (leftTyParamsAndLeadingUsing, leftParamAndTrailingUsing) = extParamss.span(isUsingOrTypeParamClause)
1021+
leftTyParamsAndLeadingUsing ::: rightTyParams ::: rightParam :: leftParamAndTrailingUsing ::: paramss1
10171022
case _ =>
10181023
badRightAssoc("must start with a single parameter")
10191024
case _ =>
10201025
// no value parameters, so not an infix operator.
1021-
() // extParamss ++ mdef.paramss
1022-
extParamss ++ mdef.paramss
1026+
extParamss ++ mdef.paramss
10231027
else
10241028
extParamss ++ mdef.paramss
10251029
).withMods(mdef.mods | ExtensionMethod)

‎compiler/src/dotty/tools/dotc/ast/Positioned.scala

+3-3
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ import util.{SourceFile, SourcePosition, SrcPos}
77
import core.Contexts.*
88
import core.Decorators.*
99
import core.NameOps.*
10-
import core.Flags.{JavaDefined, ExtensionMethod}
10+
import core.Flags.{JavaDefined, ExtensionMethod, Infix}
1111
import core.StdNames.nme
1212
import ast.Trees.mods
1313
import annotation.constructorOnly
@@ -215,8 +215,8 @@ abstract class Positioned(implicit @constructorOnly src: SourceFile) extends Src
215215
check(tree.trailingParamss)
216216
case tree: DefDef if tree.mods.is(ExtensionMethod) =>
217217
tree.paramss match
218-
// case vparams1 :: vparams2 :: rest if tree.name.isRightAssocOperatorName =>
219-
// // omit check for right-associatiove extension methods; their parameters were swapped
218+
case vparams1 :: vparams2 :: rest if tree.name.isRightAssocOperatorName && !tree.mods.is(Infix) =>
219+
// omit check for right-associatiove extension methods; their parameters were swapped
220220
case _ =>
221221
check(tree.paramss)
222222
check(tree.tpt)

‎compiler/src/dotty/tools/dotc/printing/RefinedPrinter.scala

+29-29
Original file line numberDiff line numberDiff line change
@@ -933,35 +933,35 @@ class RefinedPrinter(_ctx: Context) extends PlainPrinter(_ctx) {
933933
val coreSig =
934934
if isExtension then
935935
val paramss =
936-
// if tree.name.isRightAssocOperatorName then
937-
// // If you change the names of the clauses below, also change them in right-associative-extension-methods.md
938-
// // we have the following encoding of tree.paramss:
939-
// // (leftTyParams ++ leadingUsing
940-
// // ++ rightTyParams ++ rightParam
941-
// // ++ leftParam ++ trailingUsing ++ rest)
942-
// // e.g.
943-
// // extension [A](using B)(c: C)(using D)
944-
// // def %:[E](f: F)(g: G)(using H): Res = ???
945-
// // will have the following values:
946-
// // - leftTyParams = List(`[A]`)
947-
// // - leadingUsing = List(`(using B)`)
948-
// // - rightTyParams = List(`[E]`)
949-
// // - rightParam = List(`(f: F)`)
950-
// // - leftParam = List(`(c: C)`)
951-
// // - trailingUsing = List(`(using D)`)
952-
// // - rest = List(`(g: G)`, `(using H)`)
953-
// // we need to swap (rightTyParams ++ rightParam) with (leftParam ++ trailingUsing)
954-
// val (leftTyParams, rest1) = tree.paramss.span(isTypeParamClause)
955-
// val (leadingUsing, rest2) = rest1.span(isUsingClause)
956-
// val (rightTyParams, rest3) = rest2.span(isTypeParamClause)
957-
// val (rightParam, rest4) = rest3.splitAt(1)
958-
// val (leftParam, rest5) = rest4.splitAt(1)
959-
// val (trailingUsing, rest6) = rest5.span(isUsingClause)
960-
// if leftParam.nonEmpty then
961-
// leftTyParams ::: leadingUsing ::: leftParam ::: trailingUsing ::: rightTyParams ::: rightParam ::: rest6
962-
// else
963-
// tree.paramss // it wasn't a binary operator, after all.
964-
// else
936+
if tree.name.isRightAssocOperatorName && !tree.mods.is(Infix) && !tree.symbol.is(Infix) then
937+
// If you change the names of the clauses below, also change them in right-associative-extension-methods.md
938+
// we have the following encoding of tree.paramss:
939+
// (leftTyParams ++ leadingUsing
940+
// ++ rightTyParams ++ rightParam
941+
// ++ leftParam ++ trailingUsing ++ rest)
942+
// e.g.
943+
// extension [A](using B)(c: C)(using D)
944+
// def %:[E](f: F)(g: G)(using H): Res = ???
945+
// will have the following values:
946+
// - leftTyParams = List(`[A]`)
947+
// - leadingUsing = List(`(using B)`)
948+
// - rightTyParams = List(`[E]`)
949+
// - rightParam = List(`(f: F)`)
950+
// - leftParam = List(`(c: C)`)
951+
// - trailingUsing = List(`(using D)`)
952+
// - rest = List(`(g: G)`, `(using H)`)
953+
// we need to swap (rightTyParams ++ rightParam) with (leftParam ++ trailingUsing)
954+
val (leftTyParams, rest1) = tree.paramss.span(isTypeParamClause)
955+
val (leadingUsing, rest2) = rest1.span(isUsingClause)
956+
val (rightTyParams, rest3) = rest2.span(isTypeParamClause)
957+
val (rightParam, rest4) = rest3.splitAt(1)
958+
val (leftParam, rest5) = rest4.splitAt(1)
959+
val (trailingUsing, rest6) = rest5.span(isUsingClause)
960+
if leftParam.nonEmpty then
961+
leftTyParams ::: leadingUsing ::: leftParam ::: trailingUsing ::: rightTyParams ::: rightParam ::: rest6
962+
else
963+
tree.paramss // it wasn't a binary operator, after all.
964+
else
965965
tree.paramss
966966
val trailingParamss = paramss
967967
.dropWhile(isUsingOrTypeParamClause)

‎library/src/scala/IArray.scala

+8-4
Original file line numberDiff line numberDiff line change
@@ -319,10 +319,14 @@ object IArray:
319319
def zipAll[T1 >: T, U](that: Iterable[U], thisElem: T1, thatElem: U): IArray[(T1, U)] = genericArrayOps(arr).zipAll(that, thisElem, thatElem)
320320
def zipWithIndex: IArray[(T, Int)] = genericArrayOps(arr).zipWithIndex
321321

322-
extension [T, U >: T: ClassTag](arr: IArray[U])
323-
def ++:(prefix: IterableOnce[T]): IArray[U] = genericArrayOps(arr).prependedAll(prefix)
324-
def ++:(prefix: IArray[T]): IArray[U] = genericArrayOps(arr).prependedAll(prefix)
325-
def +:(x: T): IArray[U] = genericArrayOps(arr).prepended(x)
322+
extension [T, U >: T: ClassTag](prefix: IterableOnce[T])
323+
def ++:(arr: IArray[U]): IArray[U] = genericArrayOps(arr).prependedAll(prefix)
324+
325+
extension [T, U >: T: ClassTag](prefix: IArray[T])
326+
def ++:(arr: IArray[U]): IArray[U] = genericArrayOps(arr).prependedAll(prefix)
327+
328+
extension [T, U >: T: ClassTag](x: T)
329+
def +:(arr: IArray[U]): IArray[U] = genericArrayOps(arr).prepended(x)
326330

327331
// For backwards compatibility with code compiled without -Yexplicit-nulls
328332
private inline def mapNull[A, B](a: A, inline f: B): B =

‎tests/neg-custom-args/captures/lazylists-exceptions.scala

+2-2
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,8 @@ final class LazyCons[+T](val x: T, val xs: () => LazyList[T]^) extends LazyList[
2020
def tail: LazyList[T]^{this} = xs()
2121
end LazyCons
2222

23-
extension [A](xs1: => LazyList[A]^)
24-
def #:(x: A): LazyList[A]^{xs1} =
23+
extension [A](x: A)
24+
def #:(xs1: => LazyList[A]^): LazyList[A]^{xs1} =
2525
LazyCons(x, () => xs1)
2626

2727
def tabulate[A](n: Int)(gen: Int => A): LazyList[A]^{gen} =

‎tests/neg/i13075.scala

+2-2
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,8 @@ object Implementing_Tuples:
88
type *:[H, T <: Tup] = ConsTup[H, T] // for type matching
99
type EmptyTup = EmptyTup.type // for type matching
1010

11-
extension [T <: Tup](tail: T)
12-
def *:[H](head: H) = ConsTup(head, tail)
11+
extension [H](head: H)
12+
def *:[T <: Tup](tail: T) = ConsTup(head, tail)
1313

1414
type Fold[T <: Tup, Seed, F[_,_]] = T match
1515
case EmptyTup => Seed

‎tests/neg/i9562.scala

+1-2
Original file line numberDiff line numberDiff line change
@@ -9,5 +9,4 @@ object Unrelated:
99
def h1: Int = foo // error
1010
def h2: Int = h1 + 1 // OK
1111
def h3: Int = g // error
12-
extension (x: Int)
13-
def ++:(f: Foo): Int = f.h1 + x // OK
12+
def ++: (x: Int): Int = h1 + x // OK

‎tests/pos-custom-args/captures/lazylists-exceptions.scala

+2-2
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,8 @@ extension [A](xs: LzyList[A]^)
4242
if n == 0 then xs else xs.tail.drop(n - 1)
4343
end extension
4444

45-
extension [A](xs1: => LzyList[A]^)
46-
def #:(x: A): LzyList[A]^{xs1} =
45+
extension [A](x: A)
46+
def #:(xs1: => LzyList[A]^): LzyList[A]^{xs1} =
4747
LzyCons(x, () => xs1)
4848

4949
def lazyCons[A](x: A, xs1: => LzyList[A]^): LzyList[A]^{xs1} =

‎tests/pos-custom-args/captures/logger.scala

+2-2
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,8 @@ final class LazyCons[+T](val x: T, val xs: () => LazyList[T]^) extends LazyList[
3232
def tail: LazyList[T]^{this} = xs()
3333
end LazyCons
3434

35-
extension [A](xs1: => LazyList[A]^)
36-
def #::(x: A): LazyList[A]^{xs1} =
35+
extension [A](x: A)
36+
def #::(xs1: => LazyList[A]^): LazyList[A]^{xs1} =
3737
LazyCons(x, () => xs1)
3838

3939
extension [A](xs: LazyList[A]^)

‎tests/pos-custom-args/captures/strictlists.scala

+4-3
Original file line numberDiff line numberDiff line change
@@ -28,11 +28,12 @@ extension [A](xs: StrictList[A])
2828
def concat(ys: StrictList[A]): StrictList[A] =
2929
if xs.isEmpty then ys
3030
else xs.head #: xs.tail.concat(ys)
31-
32-
def #:(x: A): StrictList[A] =
33-
StrictCons(x, xs)
3431
end extension
3532

33+
extension [A](x: A)
34+
def #:(xs1: StrictList[A]): StrictList[A] =
35+
StrictCons(x, xs1)
36+
3637
def tabulate[A](n: Int)(gen: Int => A) =
3738
def recur(i: Int): StrictList[A] =
3839
if i == n then StrictNil

‎tests/pos/i19197.scala

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
extension (tuple: Tuple)
2-
def **:[T >: tuple.type <: Tuple, H](x: H): H *: T = ???
2+
infix def **:[T >: tuple.type <: Tuple, H](x: H): H *: T = ???
33

44
def test1: (Int, String, Char) = 1 **: ("a", 'b')
55
def test2: (Int, String, Char) = ("a", 'b').**:(1)

‎tests/pos/i9562.scala

+1-3
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,4 @@ object Unrelated:
88
extension (f: Foo)
99
def h1: Int = 0
1010
def h2: Int = h1 + 1 // OK
11-
12-
extension (x: Int)
13-
def ++: (f: Foo): Int = f.h2 + x // OK
11+
def ++: (x: Int): Int = h2 + x // OK

‎tests/pos/reference/extension-methods.scala

+1-1
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ object ExtMethods:
1010
assert(circle.circumference == circumference(circle))
1111

1212
extension (x: String) def < (y: String) = x.compareTo(y) < 0
13-
extension [Elem](xs: Seq[Elem]) def #: (x: Elem) = x +: xs
13+
extension [Elem](x: Elem) def #: (xs: Seq[Elem]) = x +: xs
1414
extension (x: Number) infix def min (y: Number) = x
1515

1616
assert("a" < "bb")

‎tests/run/errorhandling/Result.scala

+1-4
Original file line numberDiff line numberDiff line change
@@ -35,15 +35,12 @@ object Result:
3535
case (Err(e), Ok(_)) => Err(e :: Nil)
3636
case (Err(e1), Err(e2)) => Err(e1 :: e2 :: Nil)
3737

38-
end extension
39-
40-
extension [U <: Tuple, E](other: Result[U, List[E]])
4138
/** Validate both `r` and `other`; return a tuple of successes or a list of failures.
4239
* Unlike with `zip`, the right hand side `other` must be a `Result` returning a `Tuple`,
4340
* and the left hand side is added to it. See `Result.empty` for a convenient
4441
* right unit of chains of `*:`s.
4542
*/
46-
def *: [T](r: Result[T, E]): Result[T *: U, List[E]] = (r, other) match
43+
def *: [U <: Tuple](other: Result[U, List[E]]): Result[T *: U, List[E]] = (r, other) match
4744
case (Ok(x), Ok(ys)) => Ok(x *: ys)
4845
case (Ok(_), es: Err[?]) => es
4946
case (Err(e), Ok(_)) => Err(e :: Nil)

‎tests/run/export-in-extension.scala

+2-2
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ object O:
1111
export cm.*
1212
def succ: Int = x + 1
1313
def succ2: Int = succ + 1
14-
def ::: (y: Int) = y - x
14+
def ::: (y: Int) = x - y
1515

1616
object O2:
1717
import O.C
@@ -20,7 +20,7 @@ object O2:
2020
export cm.{bar, baz, bam, ::}
2121
def succ: Int = x + 1
2222
def succ2: Int = succ + 1
23-
def ::: (y: Int) = y - x
23+
def ::: (y: Int) = x - y
2424

2525
@main def Test =
2626
import O.*

‎tests/run/i11583.scala

+4-4
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,11 @@ class Env:
1212
// */
1313
// def &&:[T <: ctx.Term](trm: T)(ext: env.Extra): (ctx.Type, T, env.Extra) = (tpe, trm, ext)
1414

15-
extension [Ctx <: Context, T <: Boolean](using ctx: Ctx)(trm: T)(using env: Env)
16-
def :#:(tpe: String)(ext: env.Extra): (String, T, env.Extra) = (tpe, trm, ext)
15+
extension [Ctx <: Context](using ctx: Ctx)(tpe: String)(using env: Env)
16+
def :#:[T <: Boolean](trm: T)(ext: env.Extra): (String, T, env.Extra) = (tpe, trm, ext)
1717

18-
extension [T <: Tuple](t: T)
19-
def :*:[A](a: A): A *: T = a *: t
18+
extension [A](a: A)
19+
def :*:[T <: Tuple](t: T): A *: T = a *: t
2020

2121
@main def Test =
2222

‎tests/run/i9530.scala

+1-1
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ trait Scope:
88
extension (using s: Scope)(expr: s.Expr)
99
def show = expr.toString
1010
def eval = s.value(expr)
11-
def *: (other: s.Expr) = s.combine(other, expr)
11+
def *: (other: s.Expr) = s.combine(expr, other)
1212

1313
def f(using s: Scope)(x: s.Expr): (String, s.Value) =
1414
(x.show, x.eval)

‎tests/run/instances.scala

+1-1
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ object Test extends App {
3131
extension [T](xs: List[List[T]])
3232
def flattened = xs.foldLeft[List[T]](Nil)(_ ++ _)
3333

34-
extension [T](xs: Seq[T]) def :: (x: T) = x +: xs
34+
extension [T](x: T) def :: (xs: Seq[T]) = x +: xs
3535

3636
val ss: Seq[Int] = List(1, 2, 3)
3737
val ss1 = 0 :: ss

‎tests/run/interleaving.scala

+43-43
Original file line numberDiff line numberDiff line change
@@ -56,47 +56,47 @@ object Test extends App {
5656
extension (db: DB)
5757
def ?:(k: Key)[V >: k.Value](default: V): V = db.getOrElse(k)(default)
5858

59-
assert((db1.?:(key1))[Option[Int]](default1) == Some(4))
60-
assert((db1.?:(key2))[Option[Int]](default1) == default1)
61-
assert((db1.?:(key3))[Option[String]](default1) == default1)
62-
assert((db1.?:(key1))(default1) == Some(4))
63-
assert((db1.?:(key2))(default1) == default1)
64-
assert((db1.?:(key3))(default1) == default1)
65-
66-
assert((db1.?:(key1))[Any](default2) == Some(4))
67-
assert((db1.?:(key2))[Any](default2) == default2)
68-
assert((db1.?:(key3))[Any](default2) == default2)
69-
assert((db1.?:(key1))(default2) == Some(4))
70-
assert((db1.?:(key2))(default2) == default2)
71-
assert((db1.?:(key3))(default2) == default2)
72-
73-
74-
assert(db1.?:(key1)[Option[Int]](default1) == Some(4))
75-
assert(db1.?:(key2)[Option[Int]](default1) == default1)
76-
assert(db1.?:(key3)[Option[String]](default1) == default1)
77-
assert(db1.?:(key1)(default1) == Some(4))
78-
assert(db1.?:(key2)(default1) == default1)
79-
assert(db1.?:(key3)(default1) == default1)
80-
81-
assert(db1.?:(key1)[Any](default2) == Some(4))
82-
assert(db1.?:(key2)[Any](default2) == default2)
83-
assert(db1.?:(key3)[Any](default2) == default2)
84-
assert(db1.?:(key1)(default2) == Some(4))
85-
assert(db1.?:(key2)(default2) == default2)
86-
assert(db1.?:(key3)(default2) == default2)
87-
88-
89-
assert(?:(db1)(key1)[Option[Int]](default1) == Some(4))
90-
assert(?:(db1)(key2)[Option[Int]](default1) == default1)
91-
assert(?:(db1)(key3)[Option[String]](default1) == default1)
92-
assert(?:(db1)(key1)(default1) == Some(4))
93-
assert(?:(db1)(key2)(default1) == default1)
94-
assert(?:(db1)(key3)(default1) == default1)
95-
96-
assert(?:(db1)(key1)[Any](default2) == Some(4))
97-
assert(?:(db1)(key2)[Any](default2) == default2)
98-
assert(?:(db1)(key3)[Any](default2) == default2)
99-
assert(?:(db1)(key1)(default2) == Some(4))
100-
assert(?:(db1)(key2)(default2) == default2)
101-
assert(?:(db1)(key3)(default2) == default2)
59+
assert((db1 ?: (key1))[Option[Int]](default1) == Some(4))
60+
assert((db1 ?: (key2))[Option[Int]](default1) == default1)
61+
assert((db1 ?: (key3))[Option[String]](default1) == default1)
62+
assert((db1 ?: (key1))(default1) == Some(4))
63+
assert((db1 ?: (key2))(default1) == default1)
64+
assert((db1 ?: (key3))(default1) == default1)
65+
66+
assert((db1 ?: (key1))[Any](default2) == Some(4))
67+
assert((db1 ?: (key2))[Any](default2) == default2)
68+
assert((db1 ?: (key3))[Any](default2) == default2)
69+
assert((db1 ?: (key1))(default2) == Some(4))
70+
assert((db1 ?: (key2))(default2) == default2)
71+
assert((db1 ?: (key3))(default2) == default2)
72+
73+
74+
assert(key1.?:(db1)[Option[Int]](default1) == Some(4))
75+
assert(key2.?:(db1)[Option[Int]](default1) == default1)
76+
assert(key3.?:(db1)[Option[String]](default1) == default1)
77+
assert(key1.?:(db1)(default1) == Some(4))
78+
assert(key2.?:(db1)(default1) == default1)
79+
assert(key3.?:(db1)(default1) == default1)
80+
81+
assert(key1.?:(db1)[Any](default2) == Some(4))
82+
assert(key2.?:(db1)[Any](default2) == default2)
83+
assert(key3.?:(db1)[Any](default2) == default2)
84+
assert(key1.?:(db1)(default2) == Some(4))
85+
assert(key2.?:(db1)(default2) == default2)
86+
assert(key3.?:(db1)(default2) == default2)
87+
88+
89+
assert(?:(key1)(db1)[Option[Int]](default1) == Some(4))
90+
assert(?:(key2)(db1)[Option[Int]](default1) == default1)
91+
assert(?:(key3)(db1)[Option[String]](default1) == default1)
92+
assert(?:(key1)(db1)(default1) == Some(4))
93+
assert(?:(key2)(db1)(default1) == default1)
94+
assert(?:(key3)(db1)(default1) == default1)
95+
96+
assert(?:(key1)(db1)[Any](default2) == Some(4))
97+
assert(?:(key2)(db1)[Any](default2) == default2)
98+
assert(?:(key3)(db1)[Any](default2) == default2)
99+
assert(?:(key1)(db1)(default2) == Some(4))
100+
assert(?:(key2)(db1)(default2) == default2)
101+
assert(?:(key3)(db1)(default2) == default2)
102102
}

0 commit comments

Comments
 (0)