mavlink-plugin/src/main/scala/com/github/jodersky/mavlink/sbt/SbtMavlink.scala


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

package com.github.jodersky.mavlink.sbt

import com.github.jodersky.mavlink.Generator
import com.github.jodersky.mavlink.Parser
import com.github.jodersky.mavlink.Reporter

import sbt._
import sbt.Keys._
import sbt.plugins._

object SbtMavlink extends AutoPlugin {

  override def requires = JvmPlugin //this is required as sourceGenerators are otherwise reset

  object autoImport {

    lazy val mavlinkDialect = settingKey[File]("Dialect definition from which to generate files.")
    lazy val mavlinkTarget = settingKey[File]("Target directory to contain all generated mavlink files.")

    lazy val mavlinkGenerate = taskKey[Seq[File]]("Generate mavlink files.")

  }
  import autoImport._

  override lazy val projectSettings: Seq[Setting[_]] = Seq(
    mavlinkDialect := baseDirectory.value / "conf" / "mavlink.xml",
    mavlinkTarget := (sourceManaged in Compile).value,
    mavlinkGenerate := generationTask.value,
    sourceGenerators in Compile += mavlinkGenerate.taskValue
  )

  lazy val generationTask = Def.task[Seq[File]] {
    val dialectDefinitionFile = mavlinkDialect.value
    val outDirectory = mavlinkTarget.value

    if (!dialectDefinitionFile.exists) sys.error(
      "Dialect definition " + dialectDefinitionFile.getAbsolutePath + " does not exist."
    )

    val reporter = new Reporter {
      def printWarning(msg: String) = streams.value.log.warn(msg)
    }

    val targetFiles = Generator.targetFiles map (outDirectory / _)

    if (targetFiles forall (_.lastModified > dialectDefinitionFile.lastModified)) {
      targetFiles map (_.getAbsoluteFile)
    } else {
      val dialect = new Parser(reporter).parseDialect(dialectDefinitionFile)
      val targets = new Generator(dialect, dialectDefinitionFile.base).targets
      for (tgt <- targets) yield {
        val file = outDirectory / tgt.path

        if (dialectDefinitionFile.lastModified > file.lastModified) {
          streams.value.log.info("Generating mavlink binding " + file)
          IO.write(file, tgt.generate())
        }
        file.getAbsoluteFile
      }
    }
  }

}