class App

# File lib/rest_client.rb, line 9
def main
        valid_methods = ['GET', 'PUT', 'POST', 'DELETE']
        @opts = Trollop::options do
                version "UDP RestClient (c) 2016 @reednj"
                banner "Usage: udp-rest [options] <url>"
                opt :method, "HTTP Method (GET, POST etc)", :type => :string, :default => 'GET'
                opt :headers, "Show the response headers", :default => false
        end

        Trollop::educate if ARGV.empty?
        url = ARGV.last
        url = "uhttp://" + url unless url.start_with? 'uhttp://'

        begin
                if !valid_methods.include? @opts[:method].upcase
                        raise "Invalid REST method '#{@opts[:method]}'"
                end

                r = UDPRest::Client.uhttp(@opts[:method], url)
                print_response(r)
        rescue => e
                puts e
        end
end

# File lib/rest_client.rb, line 34
def print_response(r)
        if @opts[:headers]
                puts r.ok? ? r.status_line.green : r.status_line.red
                puts ''
        end

        puts r.text
end