diff options
author | Duncan Mak <duncanmak@gmail.com> | 2017-12-28 01:27:41 -0500 |
---|---|---|
committer | Li Haoyi <haoyi.sg@gmail.com> | 2017-12-27 22:27:41 -0800 |
commit | b2cca82786a5fdf0818b4e6be5885855a95f637e (patch) | |
tree | bf164ef396d46bddb1b6f7d5e9fd5cd6205f69ad /build.sbt | |
parent | cb3c5f06b17de55a3d64faab79c081642b0c5a59 (diff) | |
download | mill-b2cca82786a5fdf0818b4e6be5885855a95f637e.tar.gz mill-b2cca82786a5fdf0818b4e6be5885855a95f637e.tar.bz2 mill-b2cca82786a5fdf0818b4e6be5885855a95f637e.zip |
Generate zipmap definitions for higher arities (#81)
* Remove method defns for zipMap/zip that will be generated.
The trait Applyer will now extend from ApplyerGenerated which will contain the code-generated definitions.
* Script to generate code for Applicative
* Generate the zip methods in Target
* Generate zip methods in ApplicativeTests
* Make sure the full 22-arities are generated
* Move code generator into build.sc.
Remove generate.sc from directories where code will be generated.
* Generate code as part of the SBT also
* Properly wire up the test sources
* Generate all the code in one place
Diffstat (limited to 'build.sbt')
-rw-r--r-- | build.sbt | 112 |
1 files changed, 112 insertions, 0 deletions
@@ -22,6 +22,117 @@ val sharedSettings = Seq( }.taskValue ) +val coreSettings = Seq( + sourceGenerators in Compile += Def.task { + object CodeGenerator { + private def generateLetters(n: Int) = { + val base = 'A'.toInt + (0 until n).map(i => (i + base).toChar) + } + + def generateApplyer(dir: File) = { + def generate(n: Int) = { + val uppercases = generateLetters(n) + val lowercases = uppercases.map(Character.toLowerCase) + val typeArgs = uppercases.mkString(", ") + val zipArgs = lowercases.mkString(", ") + val parameters = lowercases.zip(uppercases).map { case (lower, upper) => s"$lower: TT[$upper]" }.mkString(", ") + + val body = s"mapCtx(zip($zipArgs)) { case (($zipArgs), z) => cb($zipArgs, z) }" + val zipmap = s"def zipMap[$typeArgs, Res]($parameters)(cb: ($typeArgs, Ctx) => Z[Res]) = $body" + val zip = s"def zip[$typeArgs]($parameters): TT[($typeArgs)]" + + if (n < 22) List(zipmap, zip).mkString(System.lineSeparator) else zip + } + val output = List( + "package mill.define", + "import scala.language.higherKinds", + "trait ApplyerGenerated[TT[_], Z[_], Ctx] {", + "def mapCtx[A, B](a: TT[A])(f: (A, Ctx) => Z[B]): TT[B]", + (2 to 22).map(generate).mkString(System.lineSeparator), + "}").mkString(System.lineSeparator) + + val file = dir / "ApplicativeGenerated.scala" + IO.write(file, output) + file + } + + def generateTarget(dir: File) = { + def generate(n: Int) = { + val uppercases = generateLetters(n) + val lowercases = uppercases.map(Character.toLowerCase) + val typeArgs = uppercases.mkString(", ") + val args = lowercases.mkString(", ") + val parameters = lowercases.zip(uppercases).map { case (lower, upper) => s"$lower: TT[$upper]" }.mkString(", ") + val body = uppercases.zipWithIndex.map { case (t, i) => s"args[$t]($i)" }.mkString(", ") + + s"def zip[$typeArgs]($parameters) = makeT[($typeArgs)](Seq($args), (args: Ctx) => ($body))" + } + + val output = List( + "package mill.define", + "import scala.language.higherKinds", + "import mill.eval.Result", + "import mill.util.Ctx", + "trait TargetGenerated {", + "type TT[+X]", + "def makeT[X](inputs: Seq[TT[_]], evaluate: Ctx => Result[X]): TT[X]", + (3 to 22).map(generate).mkString(System.lineSeparator), + "}").mkString(System.lineSeparator) + + val file = dir / "TaskGenerated.scala" + IO.write(file, output) + file + } + + def generateSources(dir: File) = { + Seq(generateApplyer(dir), generateTarget(dir)) + } + } + val dir = (sourceManaged in Compile).value + CodeGenerator.generateSources(dir) + }.taskValue, + + sourceGenerators in Test += Def.task { + object CodeGenerator { + private def generateLetters(n: Int) = { + val base = 'A'.toInt + (0 until n).map(i => (i + base).toChar) + } + + def generateApplicativeTest(dir: File) = { + def generate(n: Int): String = { + val uppercases = generateLetters(n) + val lowercases = uppercases.map(Character.toLowerCase) + val typeArgs = uppercases.mkString(", ") + val parameters = lowercases.zip(uppercases).map { case (lower, upper) => s"$lower: Option[$upper]" }.mkString(", ") + val result = lowercases.mkString(", ") + val forArgs = lowercases.map(i => s"$i <- $i").mkString("; ") + s"def zip[$typeArgs]($parameters) = { for ($forArgs) yield ($result) }" + } + + val output = List( + "package mill.define", + "trait OptGenerated {", + (2 to 22).map(generate).mkString(System.lineSeparator), + "}" + ).mkString(System.lineSeparator) + + val file = dir / "ApplicativeTestsGenerated.scala" + IO.write(file, output) + file + } + + def generateTests(dir: File) = { + Seq(generateApplicativeTest(dir)) + } + } + + val dir = (sourceManaged in Test).value + CodeGenerator.generateTests(dir) + }.taskValue +) + val pluginSettings = Seq( scalacOptions in Test ++= { val jarFile = (packageBin in (plugin, Compile)).value @@ -86,6 +197,7 @@ lazy val core = project .dependsOn(plugin) .settings( sharedSettings, + coreSettings, pluginSettings, name := "mill-core", libraryDependencies ++= Seq( |