Add reflect newMethodOverride and newValOverride

Closes #15035

Author: nicolasstucki

compiler/src/scala/quoted/runtime/impl/QuotesImpl.scala

@@ -2485,8 +2485,20 @@ class QuotesImpl private (using val ctx: Context) extends Quotes, QuoteUnpickler
         newMethod(owner, name, tpe, Flags.EmptyFlags, noSymbol)
       def newMethod(owner: Symbol, name: String, tpe: TypeRepr, flags: Flags, privateWithin: Symbol): Symbol =
         dotc.core.Symbols.newSymbol(owner, name.toTermName, flags | dotc.core.Flags.Method, tpe, privateWithin)
+
+      def newMethodOverride(owner: Symbol, sym: Symbol): Symbol =
+        assert(isMethod(sym), "not a method symbol: " + sym) // TODO check that is is a member of a class
+        val flags = sym.flags &~ dotc.core.Flags.Deferred | dotc.core.Flags.Override
+        dotc.core.Symbols.newSymbol(owner, sym.name, flags, sym.info, dotc.core.Symbols.NoSymbol)
+
       def newVal(owner: Symbol, name: String, tpe: TypeRepr, flags: Flags, privateWithin: Symbol): Symbol =
         dotc.core.Symbols.newSymbol(owner, name.toTermName, flags, tpe, privateWithin)
+
+      def newValOverride(owner: Symbol, sym: Symbol): Symbol =
+        assert(isField(sym) || isMethod(sym), "not a method or field symbol: " + sym) // TODO check that is is a member of a class
+        val flags = sym.flags &~ dotc.core.Flags.Deferred &~ dotc.core.Flags.Method | dotc.core.Flags.Override
+        dotc.core.Symbols.newSymbol(owner, sym.name, flags, sym.info, dotc.core.Symbols.NoSymbol)
+
       def newBind(owner: Symbol, name: String, flags: Flags, tpe: TypeRepr): Symbol =
         dotc.core.Symbols.newSymbol(owner, name.toTermName, flags | Case, tpe)
       def noSymbol: Symbol = dotc.core.Symbols.NoSymbol
@@ -2661,13 +2673,14 @@ class QuotesImpl private (using val ctx: Context) extends Quotes, QuoteUnpickler
       private def appliedTypeRef(sym: Symbol): TypeRepr =
         sym.typeRef.appliedTo(sym.typeParams.map(_.typeRef))
 
-      private def isMethod(sym: Symbol): Boolean =
-        sym.isTerm && sym.is(dotc.core.Flags.Method) && !sym.isConstructor
-
-      private def isField(sym: Symbol): Boolean =
-        sym.isTerm && !sym.is(dotc.core.Flags.Method)
     end SymbolMethods
 
+    private def isMethod(sym: Symbol): Boolean =
+      sym.isTerm && sym.is(dotc.core.Flags.Method) && !sym.isConstructor
+
+    private def isField(sym: Symbol): Boolean =
+      sym.isTerm && !sym.is(dotc.core.Flags.Method)
+
     type Signature = dotc.core.Signature
 
     object Signature extends SignatureModule:

library/src/scala/quoted/Quotes.scala

@@ -3673,6 +3673,9 @@ trait Quotes { self: runtime.QuoteUnpickler & runtime.QuoteMatching =>
       */
       def newMethod(parent: Symbol, name: String, tpe: TypeRepr, flags: Flags, privateWithin: Symbol): Symbol
 
+      /** TODO */
+      def newMethodOverride(parent: Symbol, sym: Symbol): Symbol
+
       /** Generates a new val/var/lazy val symbol with the given parent, name and type.
       *
       *  This symbol starts without an accompanying definition.
@@ -3691,6 +3694,9 @@ trait Quotes { self: runtime.QuoteUnpickler & runtime.QuoteMatching =>
       */
       def newVal(parent: Symbol, name: String, tpe: TypeRepr, flags: Flags, privateWithin: Symbol): Symbol
 
+      /** TODO */
+      def newValOverride(parent: Symbol, sym: Symbol): Symbol
+
       /** Generates a pattern bind symbol with the given parent, name and type.
       *
       *  This symbol starts without an accompanying definition.

tests/run-macros/i15035.check

@@ -0,0 +1,8 @@
+1
+1
+1
+1
+3
+3
+3
+3

tests/run-macros/i15035/Macro_1.scala

@@ -0,0 +1,24 @@
+import scala.quoted.*
+
+inline def make[T](inline value: Int): T = ${impl[T]('value)}
+
+def impl[T: Type](value: Expr[Int])(using Quotes): Expr[T] = {
+  import quotes.reflect.*
+
+  val className = "Foo"
+  val parents = List(TypeTree.of[Object], TypeTree.of[T])
+
+  val parentMethods = TypeRepr.of[T].typeSymbol.declaredMethods
+
+  def decls(cls: Symbol): List[Symbol] =
+    TypeRepr.of[T].typeSymbol.declaredFields.map(field => Symbol.newValOverride(cls, field)) ++
+    TypeRepr.of[T].typeSymbol.declaredMethods.map(method => Symbol.newMethodOverride(cls, method))
+
+  val cls = Symbol.newClass(Symbol.spliceOwner, className, parents = parents.map(_.tpe), decls, selfType = None)
+  val body =
+    cls.declaredFields.map { method => ValDef(method, Some(value.asTerm)) } ++
+    cls.declaredMethods.map { method => DefDef(method, argss => Some(value.asTerm)) }
+  val clsDef = ClassDef(cls, parents, body = body)
+  val newCls = Typed(Apply(Select(New(TypeIdent(cls)), cls.primaryConstructor), Nil), TypeTree.of[T])
+  Block(List(clsDef), newCls).asExprOf[T]
+}

tests/run-macros/i15035/Test_2.scala

@@ -0,0 +1,29 @@
+trait A {
+  val foo0: Int
+  def foo1(): Int
+  def foo2(foo: String): Int
+  def foo3: Int
+}
+
+trait B {
+  val foo0: Int = 2
+  def foo1(): Int = 2
+  def foo2(foo: String): Int = 2
+  def foo3: Int = 2
+}
+
+
+@main
+def Test = {
+  val a: A = make[A](1)
+  println(a.foo0)
+  println(a.foo1())
+  println(a.foo2("test"))
+  println(a.foo3)
+
+  val b: B = make[B](3)
+  println(b.foo0)
+  println(b.foo1())
+  println(b.foo2("test"))
+  println(b.foo3)
+}