#!/usr/bin/env ruby

require "optparse"
require_relative "../lib/template-ruby"

options = { timeout: 0, profile: false }

OptionParser
  .new do |opts|
    opts.banner = "Usage: bin/code [options]"

    opts.on(
      "-i INPUT",
      "--input=INPUT",
      "Input in the code language (String or File)"
    ) do |input|
      input = File.read(input) if File.exists?(input)

      options[:input] = input
    end

    opts.on(
      "-c CONTEXT",
      "--context=CONTEXT",
      "Context in the code language (String or File)"
    ) do |context|
      context = File.read(context) if File.exists?(context)

      options[:context] = context
    end

    opts.on("-p", "--parse", "Get parser results for input") do |parse|
      options[:parse] = parse
    end

    opts.on(
      "-t TIMEOUT",
      "--timeout=TIMEOUT",
      "Set timeout in seconds"
    ) { |timeout| options[:timeout] = timeout.to_f }

    opts.on(
      "--profile",
      "Profile Ruby code"
    ) do |timeout|
      require "ruby-prof"
      options[:profile] = true
    end
  end
  .parse!

input = options.fetch(:input, "")
context = options.fetch(:context, "")

if options[:profile]
  RubyProf.start
end

if options[:parse]
  pp ::Code::Parser.parse(input).to_raw
else
  print(
    Code.evaluate(
      input,
      context,
      io: $stdout,
      timeout: options[:timeout]
    ).to_s
  )
end

if options[:profile]
  result = RubyProf.stop
  printer = RubyProf::FlatPrinter.new(result)
  printer.print($stdout)
end