Fix passing jvm options

Author: BarkingBad

compiler/src/dotty/tools/dotc/config/ScalaSettings.scala

@@ -73,6 +73,9 @@ trait AllScalaSettings extends CommonScalaSettings, PluginSettings, VerboseSetti
   )
 
   val wikiSyntax: Setting[Boolean] = BooleanSetting("-Xwiki-syntax", "Retains the Scala2 behavior of using Wiki Syntax in Scaladoc.")
+
+  val jvmargs  = PrefixSetting("-J<flag>", "-J", "Pass <flag> directly to the runtime system.")
+  val defines  = PrefixSetting("-Dproperty=value", "-D", "Pass -Dproperty=value directly to the runtime system.")
 end AllScalaSettings
 
 /** Settings shared by compiler and scaladoc */

compiler/test-coursier/dotty/tools/coursier/CoursierScalaTests.scala

@@ -60,6 +60,13 @@ class CoursierScalaTests:
       assertTrue(output.mkString("\n").endsWith("scriptPath.sc"))
     scriptPath()
 
+    def scriptEnvDashJDashD() =
+      val scriptPath = scripts("/scripting").find(_.getName == "envtest.sc").get.absPath
+      val args = scriptPath
+      val output = CoursierScalaTests.csScalaCmd("-J-Dkey=World", args)
+      assertEquals(output.mkString("\n"), "Hello World")
+    scriptEnvDashJDashD()
+
     def version() =
       val output = CoursierScalaTests.csScalaCmd("-version")
       assertTrue(output.mkString("\n").contains(sys.env("DOTTY_BOOTSTRAPPED_VERSION")))
@@ -75,6 +82,11 @@ class CoursierScalaTests:
       assertEquals(output.mkString("\n"), "Hello")
     run()
 
+    def runDashJDashD() =
+      val output = CoursierScalaTests.csScalaCmd("-J-Dkey=World", "-classpath", scripts("/run").head.getParentFile.getParent, "-run", "run.envtest")
+      assertEquals(output.mkString("\n"), "Hello World")
+    runDashJDashD()
+
     def notOnlyOptionsEqualsRun() =
       val output = CoursierScalaTests.csScalaCmd("-classpath", scripts("/run").head.getParentFile.getParent, "run.myfile")
       assertEquals(output.mkString("\n"), "Hello")
@@ -147,10 +159,12 @@ object CoursierScalaTests:
     csCmd("dotty.tools.dotc.Main", options*)
 
   private def csCmd(entry: String, options: String*): List[String] =
-    val newOptions = options match
-      case Nil => options
-      case _ => "--" +: options
-    execCmd("./cs", (s"""launch "org.scala-lang:scala3-compiler_3:${sys.env("DOTTY_BOOTSTRAPPED_VERSION")}" --main-class "$entry" --property "scala.usejavacp=true"""" +: newOptions)*)
+    val (jOpts, args) = options.partition(_.startsWith("-J"))
+    val newOptions = args match
+      case Nil => args
+      case _ => "--" +: args
+    val newJOpts = jOpts.map(s => s"--java-opt ${s.stripPrefix("-J")}").mkString(" ")
+    execCmd("./cs", (s"""launch "org.scala-lang:scala3-compiler_3:${sys.env("DOTTY_BOOTSTRAPPED_VERSION")}" $newJOpts --main-class "$entry" --property "scala.usejavacp=true"""" +: newOptions)*)
 
   /** Get coursier script */
   @BeforeClass def setup(): Unit =

compiler/test-coursier/run/envtest.scala

@@ -0,0 +1,4 @@
+package run
+
+object envtest extends App:
+  println("Hello " + sys.props("key"))

compiler/test-resources/scripting/envtest.sc

@@ -0,0 +1,2 @@
+def main(args: Array[String]): Unit =
+  println("Hello " + util.Properties.propOrNull("key"))

compiler/test-resources/scripting/envtest.scala

@@ -0,0 +1,4 @@
+package run
+
+object envtest extends App:
+  println("Hello " + sys.props("key"))

compiler/test/dotty/tools/repl/EnvPropsTest.scala

@@ -0,0 +1,17 @@
+package dotty.tools.repl
+
+import org.junit.Assert._
+import org.junit.Test
+
+class EnvPropsTest extends ReplTest(options = Array("-J-Dkey=value", "-Dkey2=value2")) {
+  @Test def fromDashJDashD = fromInitialState { implicit s =>
+    run("util.Properties.propOrNull(\"key\")")
+    assertTrue(storedOutput().contains("val res0: String = value"))
+  }
+
+  @Test def fromDashD = fromInitialState { implicit s =>
+    run("util.Properties.propOrNull(\"key2\")")
+    assertTrue(storedOutput().contains("val res0: String = value2"))
+  }
+}
+

compiler/test/dotty/tools/scripting/BashScriptsTests.scala

@@ -7,6 +7,7 @@ import java.nio.file.{Path, Paths, Files}
 import scala.sys.process._
 
 import org.junit.Test
+import org.junit.Assert.assertEquals
 
 import vulpix.TestConfiguration
 
@@ -28,13 +29,13 @@ class BashScriptsTests:
   printf("scalac path: [%s]\n", scalacPath)
 
   lazy val expectedOutput = List(
-    "arg  0:[a]", 
-    "arg  1:[b]", 
-    "arg  2:[c]", 
-    "arg  3:[-repl]", 
-    "arg  4:[-run]", 
-    "arg  5:[-script]", 
-    "arg  6:[-debug]", 
+    "arg  0:[a]",
+    "arg  1:[b]",
+    "arg  2:[c]",
+    "arg  3:[-repl]",
+    "arg  4:[-run]",
+    "arg  5:[-script]",
+    "arg  6:[-debug]",
   )
   lazy val testScriptArgs = Seq(
     "a", "b", "c", "-repl", "-run", "-script", "-debug"
@@ -56,6 +57,38 @@ class BashScriptsTests:
       if fail then
         assert(stdout == expectedOutput)
 
+  /* verify `dist/bin/scala` with -J setting */
+  @Test def verifyScJProperty =
+    val commandline = Seq(scalaPath, "-J-Dkey=World", testFiles.find(_.getName == "envtest.sc").get.absPath).mkString(" ")
+    val (validTest, exitCode, stdout, stderr) = bashCommand(commandline)
+    assertEquals(stdout.mkString("/n"), "Hello World")
+
+  /* verify `dist/bin/scala` with -J setting */
+  @Test def verifyScalaJProperty =
+    val commandline = Seq(scalaPath, "-J-Dkey=World3", testFiles.find(_.getName == "envtest.scala").get.absPath).mkString(" ")
+    val (validTest, exitCode, stdout, stderr) = bashCommand(commandline)
+    assertEquals(stdout.mkString("/n"), "Hello World3")
+
+  /* verify `dist/bin/scala` with -D setting */
+  @Test def verifyScDProperty =
+    val commandline = Seq(scalaPath, "-Dkey=World3", testFiles.find(_.getName == "envtest.sc").get.absPath).mkString(" ")
+    val (validTest, exitCode, stdout, stderr) = bashCommand(commandline)
+    assertEquals(stdout.mkString("/n"), "Hello World3")
+
+  /* verify `dist/bin/scala` with -D setting */
+  @Test def verifyScalaDProperty =
+    val commandline = Seq(scalaPath, "-Dkey=World4", testFiles.find(_.getName == "envtest.scala").get.absPath).mkString(" ")
+    val (validTest, exitCode, stdout, stderr) = bashCommand(commandline)
+    assertEquals(stdout.mkString("/n"), "Hello World4")
+
+  /* verify `dist/bin/scala` with -D setting */
+  @Test def saveAndRunWithDProperty =
+    val commandline = Seq(scalaPath, "-save", testFiles.find(_.getName == "envtest.scala").get.absPath).mkString(" ")
+    val (_, _, _, _) = bashCommand(commandline)
+    val commandline2 = Seq(scalaPath, "-Dkey=World5", testFiles.find(_.getName == "envtest.jar").get.absPath).mkString(" ")
+    val (validTest, exitCode, stdout, stderr) = bashCommand(commandline2)
+    assertEquals(stdout.mkString("/n"), "Hello World5")
+
   /* verify `dist/bin/scala` non-interference with command line args following script name */
   @Test def verifyScalaArgs =
     val commandline = (Seq("SCALA_OPTS= ", scalaPath, showArgsScript) ++ testScriptArgs).mkString(" ")
@@ -73,7 +106,7 @@ class BashScriptsTests:
         assert(stdout == expectedOutput)
 
   /*
-   * verify that scriptPath.sc sees a valid script.path property, 
+   * verify that scriptPath.sc sees a valid script.path property,
    * and that it's value is the path to "scriptPath.sc".
    */
   @Test def verifyScriptPathProperty =
@@ -192,7 +225,7 @@ class BashScriptsTests:
       val envPairs = testEnvPairs ++ additionalEnvPairs
       val proc = Process(cmd, None, envPairs *)
       val exitVal = proc ! ProcessLogger (
-        (out: String) => stdout ::= out, 
+        (out: String) => stdout ::= out,
         (err: String) => stderr ::= err
       )
       val validTest = exitVal == 0 && ! stderr.exists(_.contains("Permission denied"))

dist/bin/scala

@@ -28,10 +28,34 @@ fi
 
 source "$PROG_HOME/bin/common"
 
+while [[ $# -gt 0 ]]; do
+  case "$1" in
+    -D*)
+      # pass to scala as well: otherwise we lose it sometimes when we
+      # need it, e.g. communicating with a server compiler.
+      addJava "$1"
+      addScala "$1"
+      # respect user-supplied -Dscala.usejavacp
+      shift
+      ;;
+    -J*)
+      # as with -D, pass to scala even though it will almost
+      # never be used.
+      addJava "${1:2}"
+      addScala "$1"
+      shift
+      ;;
+    *)
+      addScala "$1"
+      shift
+      ;;
+  esac
+done
+
 # exec here would prevent onExit from being called, leaving terminal in unusable state
 compilerJavaClasspathArgs
 [ -z "${ConEmuPID-}" -o -n "${cygwin-}" ] && export MSYSTEM= PWD= # workaround for #12405
-eval "\"$JAVACMD\"" "-Dscala.home=$PROG_HOME" "-classpath \"$jvm_cp_args\"" "dotty.tools.MainGenericRunner" "-classpath \"$jvm_cp_args\"" "$@"
+eval "\"$JAVACMD\"" "${java_args[@]}" "-Dscala.home=$PROG_HOME" "-classpath \"$jvm_cp_args\"" "dotty.tools.MainGenericRunner" "-classpath \"$jvm_cp_args\"" "${scala_args[@]}"
 scala_exit_status=$?