summaryrefslogtreecommitdiff
path: root/contrib/playlib/worker/2.6/src/RouteCompilerWorker.scala
blob: 83de5e42b2af50a37febe02117ce0237525d3590 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
package mill
package playlib
package worker

import java.io.File

import ammonite.ops.Path
import mill.api.{Ctx, Result}
import mill.eval.PathRef
import mill.playlib.api.{RouteCompilerType, RouteCompilerWorkerApi}
import mill.scalalib.api.CompilationResult
import play.routes.compiler
import play.routes.compiler.RoutesCompiler.RoutesCompilerTask
import play.routes.compiler._


private[playlib] class RouteCompilerWorker extends RouteCompilerWorkerApi {

  override def compile(files: Seq[Path],
                       additionalImports: Seq[String],
                       forwardsRouter: Boolean,
                       reverseRouter: Boolean,
                       namespaceReverseRouter: Boolean,
                       generatorType: RouteCompilerType,
                       dest: Path)
                      (implicit ctx: mill.api.Ctx): mill.api.Result[CompilationResult] = {
    generatorType match {
      case RouteCompilerType.InjectedGenerator =>
        val result = compileWithPlay(files, additionalImports, forwardsRouter, reverseRouter,
          namespaceReverseRouter, dest, ctx, InjectedRoutesGenerator)
        asMillResult(ctx, result)
      case RouteCompilerType.StaticGenerator =>
        ctx.log.error("Static generator was deprecated in 2.6.0 and will be removed in 2.7.0")
        val result = compileWithPlay(files, additionalImports, forwardsRouter, reverseRouter,
          namespaceReverseRouter, dest, ctx, StaticRoutesGenerator)
        asMillResult(ctx, result)
      case _ => throw new Exception(s"Unrecognized generator type: $generatorType. Use injected or static")
    }
  }

  private def compileWithPlay(files: Seq[Path],
                              additionalImports: Seq[String],
                              forwardsRouter: Boolean,
                              reverseRouter: Boolean,
                              namespaceReverseRouter: Boolean,
                              dest: Path,
                              ctx: Ctx,
                              routesGenerator: RoutesGenerator): Either[Seq[compiler.RoutesCompilationError], Seq[File]] = {
    val seed: Either[Seq[compiler.RoutesCompilationError], List[File]] = Right(List.empty[File])
    files.map(file => compileWithPlay(file, additionalImports, forwardsRouter, reverseRouter,
      namespaceReverseRouter, dest, ctx, routesGenerator)).foldLeft(seed) {
      case (Right(accFiles), Right(files)) => Right(accFiles ++ files)
      case (Right(accFiles), Left(errors)) => Left(errors)
      case (left@Left(errors), _) => left
    }
  }

  private def compileWithPlay(file: Path,
                              additionalImports: Seq[String],
                              forwardsRouter: Boolean,
                              reverseRouter: Boolean,
                              namespaceReverseRouter: Boolean,
                              dest: Path,
                              ctx: Ctx,
                              routesGenerator: RoutesGenerator): Either[Seq[compiler.RoutesCompilationError], Seq[File]] = {
    ctx.log.debug(s"compiling $file with play generator $routesGenerator")
    val result =
      RoutesCompiler.compile(
        RoutesCompilerTask(file.toIO, additionalImports, forwardsRouter, reverseRouter,
          namespaceReverseRouter),
        generator = routesGenerator,
        generatedDir = dest.toIO
      )
    ctx.log.debug(s"compilation result: $result")
    result
  }

  private def asMillResult(ctx: Ctx, result: Either[Seq[RoutesCompilationError], Seq[File]]): Result[CompilationResult] = {
    result match {
      case Right(_) =>
        val zincFile = ctx.dest / 'zinc
        Result.Success(CompilationResult(zincFile, PathRef(ctx.dest)))
      case Left(errors) =>
        val errorMsg = errors.map(error =>
          s"compilation error in ${error.source.getPath} at line ${error.line.getOrElse("?")}, " +
            s"column ${error.column.getOrElse("?")}: ${error.message}")
          .mkString("\n")
        Result.Failure("Unable to compile play routes\n" + errorMsg)
    }
  }
}