gree/observed

require 'logger'
require 'optparse'
require 'pathname'

require 'observed/config'
require 'observed/context'

module Observed

  module Application
    # The application which is usually ran from CLI to run health-checks and write the results to a log file, and then exit.
    # An "Oneshot" application is the opposite of a "Daemon" or "Resident" application.
    class Oneshot

      class InvalidArgumentError < RuntimeError; end

      # @param [Observed::Config] config
      def initialize(config, sys)
        @config = config
        @system = sys
      end

      def config
        @config || fail('Missing configuration for Application::Oneshot')
      end

      def run(observation_name=nil)
        @system.run(observation_name)
      end

      class << self
        def parse_argv!(argv)
          command_line_args = argv

          args = {}

          opts = OptionParser.new
          opts.accept(Pathname) do |s,|
            Pathname.new(s)
          end
          opts.on('-d', '--debug') do
            args[:debug] = true
          end
          opts.on('-l LOG_FILE', '--l LOG_FILE', Pathname) do |log_file|
            args[:log_file] = log_file
          end

          opts.parse!(command_line_args)

          unless command_line_args.size == 1 || command_line_args.size == 2
            fail InvalidArgumentError, "Invalid number of arguments #{command_line_args.size} where arguments are #{command_line_args}"
          end

          args[:config_file] = command_line_args.shift

          args
        end

        def from_argv(argv)
          args = parse_argv!(argv.dup)
          create(args)
        end

        # @param [Hash<Symbol,String>] args
        # @option args [Array<String>] :argv The Ruby's `ARGV` like object which is treated as intialization parameters for Oneshoft application.
        def create(args)
          ctx = Observed::Context.new(args)
          sys = ctx.system
          config = if args[:yaml_file]
                     YAML.load_file(args[:yaml_file])
                   elsif args[:config_file]
                     sys.config
                   elsif args[:config]
                     c = args[:config]
                     c
                   else
                     fail 'No configuration provided'
                   end
          config = if config.is_a? Hash
                     Observed::Config.create(config)
                   else
                     config
                   end
          sys.config = config
          new(config, sys)
        end
      end

    end
  end
end