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.*
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 dumpScope by option("--dump-scope", help = "Dump Scope").flag()
  val path by argument("file")

  override fun run() {
    val tool = FileTool(PlatformFsProvider.resolve(path))
    val scope = Scope()
    val main = tool.loadMainFunction(scope, setupEvaluator = {
      addNativeProvider("internal", InternalNativeProvider(quiet = quiet))
      addNativeProvider("ffi", JnaNativeProvider())
      addNativeProvider("java", JavaNativeProvider())
    })

    if (dumpScope) {
      val functionContext = main as FunctionContext
      val internalScope = functionContext.compilationUnitContext.internalScope
      internalScope.crawlScopePath { key, path ->
        println("[scope] $key [${path.joinToString(" -> ")}]")
      }
    }

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