#!/usr/bin/env ruby -wKU

module Scout
  class Command
    class Run < Command
      def run
        key = @args.first
        # TODO: this is an awkward way to force creation of the config directory. Could use a little refactoring.
        configuration_directory = config_dir
        log.debug("Running Scout [#{Scout::VERSION}] with server_metrics [#{ServerMetrics::VERSION}] on #{@hostname}") if log
        log.debug("Configuration directory is #{configuration_directory} ") if log
        # TODO: too much external logic of command doing things TO server. This should be moved into the server class.
        @scout = Scout::Server.new(server, key, history, log, server_name, @http_proxy, @https_proxy, @roles, @hostname, @environment)
        @scout.load_history
        
        unless $stdin.tty?
          log.info "Sleeping #{@scout.sleep_interval} sec" if log
          sleep @scout.sleep_interval
        end

        begin
          @scout.fetch_plan
        rescue SystemExit => e
          puts "Failure. Run with '-v -ldebug' for more information" if $stdin.tty?
          raise e
        end

        # Spawn or stop streamer as needed
        if @scout.streamer_command.is_a?(String)
          if @scout.streamer_command.start_with?("start")
            log.info "streamer command: start"
            Scout::StreamerDaemon.start_daemon(history, @scout.streamer_command, @hostname, @http_proxy)
          elsif @scout.streamer_command == "stop"
            log.info "streamer command: stop"
            Scout::StreamerDaemon.stop_daemon(history)
          end
        end

        # Check in if appropriate
        if @scout.new_plan || @scout.time_to_checkin?  || @force
          if @scout.new_plan
            log.info("Now checking in with new plugin plan") if log
          elsif @scout.time_to_checkin?
            log.info("It is time to checkin") if log
          elsif @force
            log.info("overriding checkin schedule with --force and checking in now.") if log
          end

          begin
            create_pid_file_or_exit
            @scout.run_plugins_by_plan
            @scout.save_history
            puts "Successfully reported to #{server}" if $stdin.tty?
          rescue SystemExit => e
            puts "Failure. Run with '-v -ldebug' for more information" if $stdin.tty?
            raise e
          end

          begin
            # Since this is a new checkin, overwrite the existing log
            File.open(log_path, "w") do|log_file|
              log_file.puts log.messages # log.messages is an array of every message logged during this run
            end
          rescue
            log.info "Could not write to #{log_path}."
          end
        else
          log.info "Not time to checkin yet. Next checkin in #{@scout.next_checkin}. Override by passing --force to the scout command" if log
          begin
            # Since this a ping, append to the existing log
            File.open(log_path, "a") do|log_file|
              log_file.puts log.messages
            end
          rescue
            log.info "Could not write to #{log_path}."
          end
        end
      end
    end
  end
end