require "trickle/version"
require "ostruct"
require "optparse"

module Trickle
  class Trickle
    def initialize(*args)
      @args = args

      @config = OpenStruct.new
      @config.mode = :now
      @config.input = []
      @config.verbose = false
    end

    def run
      if parsed_options? && arguments_valid?
        case @config.mode
        when :within
          run_schedule(@config.within)
        when :rate
          run_every(@config.rate)
        when :now
          run_all
        end
      else
        abort @opts.to_s
      end
    end

    def run_all
      @config.input.each do |line|
        display_running_cmd(line)
        puts `#{line}`
      end
    end

    def run_schedule(within)
      @threads = []

      @config.input.each do |line|
        @threads << Thread.new do
          sleep(rand(60*within))
          display_running_cmd(line)
          puts `#{line}`
        end
      end

      # wait for all threads to finish before exiting
      @threads.each(&:join)
    end

    def run_every(rate)
      @config.input.each do |line|
        display_running_cmd(line)
        puts `#{line}`
        sleep(60.0/rate)
      end
    end

    protected

    def display_running_cmd(line)
      puts "Running: #{line}" if @config.verbose
    end

    def parsed_options?
      @opts = OptionParser.new
      @opts.banner = 'Usage: trickle [options] file ...'
      @opts.on('-t', '--within x', 'Run all commands randomly within x minutes') {|within| @config.mode = :within; @config.within = within.to_i}
      @opts.on('-r', '--rate x', 'Run all commands at the rate of x commands per minute') {|rate| @config.mode = :rate; @config.rate = rate.to_i}
      @opts.on('-n', '--now', 'Run all commands now (default)') {@config.mode = :now}
      @opts.on('-v', '--verbose', 'Output command that is being ran') {@config.verbose = true}
      @opts.parse!(@args) rescue return false

      if $stdin.tty?
        @args.each {|file| @config.input += IO.readlines(file)}
      else
        $stdin.each_line {|line| @config.input << line}
      end

      true
    end

    def arguments_valid?
      true if @config.input.any?
    end
  end
end