kigster/turnstile-rb

require 'optparse'
require 'colored2'

require_relative '../version'
require_relative '../configuration'

require_relative 'launcher'
require_relative 'parser'

module Turnstile
  module CLI
    class Runner
      attr_reader :argv, :stdin, :stdout, :stderr, :kernel

      # Allow everything fun to be injected from the outside while defaulting to normal implementations.
      def initialize(argv, stdin = STDIN, stdout = STDOUT, stderr = STDERR, kernel = Kernel)
        @argv, @stdin, @stdout, @stderr, @kernel = argv, stdin, stdout, stderr, kernel
      end

      def execute!
        exit_code = begin
          Colored2.disable! unless stdout.tty?

          $stderr = stderr
          $stdin  = stdin
          $stdout = stdout

          options = Parser.new(argv, self).parse
          Configuration.from_file(options.config_file) if options && options.config_file
          Launcher.new(options).launch if options

          # Thor::Base#start does not have a return value, assume success if no exception is raised.
          0
        rescue StandardError => e
          # The ruby interpreter would pipe this to STDERR and exit 1 in the case of an unhandled exception
          if options && options[:trace]
            b = e.backtrace
            terr("#{b.shift}: #{e.message} (#{e.class})")
            terr(b.map { |s| "\tfrom #{s}" }.join("\n"))
          else
            terr(e.message)
          end
          1
        rescue SystemExit => e
          e.status
        ensure
          $stderr = STDERR
          $stdin  = STDIN
          $stdout = STDOUT
        end

        # Proxy our exit code back to the injected kernel.
        @kernel.exit(exit_code)
      end
    end
  end
end