com.github.chengpohi.ELKInterpreter.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of elasticdsl_2.12 Show documentation

elasticdsl

The newest version!

package com.github.chengpohi

import com.github.chengpohi.parser.ELKParser
import com.github.chengpohi.registry.ELKCommandRegistry

import scala.io.Source

/**
  * Created by chengpohi on 1/3/16.
  */
class ELKInterpreter(env: {val elkParser: ELKParser}) {

  import env.elkParser._

  def interceptDefinitions(instructions: Seq[Instruction]): String = {
    val res = for {
      instruction <- instructions
    } yield instruction.f.apply(instruction.params).json
    res.mkString("\n")
  }

  def run(source: String): String = {
    val parsed = elkParser.parse(source)
    val instruments = generateDefinitions(parsed)
    interceptDefinitions(instruments)
  }

  def run(str: String, parameters: String*): String = {
    val s = String.format(str, parameters: _*)
    run(s)
  }
}

object ELKInterpreter {
  private val runEngine: ELKInterpreter = new ELKInterpreter(ELKCommandRegistry)

  def main(args: Array[String]): Unit = {
    if (args.length != 1) {
      println("usage elk file.elk")
      System.exit(0)
    }
    val parseFile: String = Source.fromFile(args(0)).getLines().filter(!_.trim().startsWith("//")).toList.mkString("")
    println(runEngine.run(parseFile))
  }
}