pork/tool/src/main/kotlin/gay/pizza/pork/tool/RunCommand.kt

package gay.pizza.pork.tool

import com.github.ajalt.clikt.core.CliktCommand
import com.github.ajalt.clikt.parameters.arguments.argument
import com.github.ajalt.clikt.parameters.options.flag
import com.github.ajalt.clikt.parameters.options.option
import gay.pizza.dough.fs.PlatformFsProvider
import gay.pizza.pork.evaluator.Arguments
import gay.pizza.pork.evaluator.CallableFunction
import gay.pizza.pork.evaluator.None
import gay.pizza.pork.evaluator.Scope
import gay.pizza.pork.ffi.JavaNativeProvider
import gay.pizza.pork.ffi.JnaNativeProvider

class RunCommand : CliktCommand(help = "Run Program", name = "run") {
  val loop by option("--loop", help = "Loop Program").flag()
  val measure by option("--measure", help = "Measure Time").flag()
  val quiet by option("--quiet", help = "Silence Prints").flag()
  val path by argument("file")

  override fun run() {
    val tool = FileTool(PlatformFsProvider.resolve(path))
    val scope = Scope()
    scope.define("println", CallableFunction { arguments ->
      if (quiet) {
        return@CallableFunction None
      }
      for (argument in arguments.values) {
        println(argument)
      }
      None
    })

    val main = tool.loadMainFunction(scope, setupEvaluator = {
      addNativeFunctionProvider("ffi", JnaNativeProvider())
      addNativeFunctionProvider("java", JavaNativeProvider())
    })

    maybeLoopAndMeasure(loop, measure) {
      main.call(Arguments(emptyList()))
    }
  }
}