summaryrefslogtreecommitdiff
path: root/contrib/scoverage/worker/1.4.0/src/ScoverageReportWorkerImpl.scala
blob: 63fdd6931545015042337528ff7e5b3fbc06a0b4 (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
package mill.contrib.scoverage.worker

import mill.contrib.scoverage.api.ScoverageReportWorkerApi
import mill.eval.PathRef
import _root_.scoverage.Serializer.{ coverageFile, deserialize }
import _root_.scoverage.IOUtils.{ findMeasurementFiles, invoked }
import _root_.scoverage.report.ScoverageHtmlWriter
import _root_.scoverage.report.ScoverageXmlWriter

private sealed trait ReportType { def folderName: String }
private case object Html extends ReportType { val folderName: String = "htmlReport" }
private case object Xml extends ReportType { val folderName: String = "xmlReport" }

class ScoverageReportWorkerImpl extends ScoverageReportWorkerApi {
  private def buildReport(sources: Seq[PathRef], dataDir: String, selfDir: String, reportType: ReportType) = {
    val coverageFileObj = coverageFile(dataDir)
    val coverage = deserialize(coverageFileObj)
    coverage(invoked(findMeasurementFiles(dataDir)))
    val sourceFolders = sources.map(_.path.toIO)
    val folder = new java.io.File(s"${selfDir}/${reportType.folderName}")
    folder.mkdir()
    reportType match {
      case Html =>
        new ScoverageHtmlWriter(sourceFolders, folder, None)
          .write(coverage)
      case Xml =>
        new ScoverageXmlWriter(sourceFolders, folder, false)
            .write(coverage)
    }
  }
  def htmlReport(sources: Seq[PathRef], dataDir: String, selfDir: String) = {
    buildReport(sources, dataDir, selfDir, Html)
  }
  def xmlReport(sources: Seq[PathRef], dataDir: String, selfDir: String) = {
    buildReport(sources, dataDir, selfDir, Xml)
  }
}