require 'json'
require 'net/http'
require 'uri'

module Haste

  DEFAULT_URL = 'http://hastebin.com'

  class CLI

    attr_reader :input

    # Pull all of the data from STDIN
    def initialize
      if STDIN.tty?
        if ARGV.empty?
          puts "No input file given"
          exit
        end
        
        file = ARGV[0]
        abort "#{file} doesn't exist" unless File.exists?(file)
        
        @input = open(file).read.strip
      else
        @input = STDIN.readlines.join
        @input.strip!
      end
    end

    # Upload the and output the URL we get back
    def start
      uri = URI.parse server
      http = Net::HTTP.new uri.host, uri.port
      response = http.post '/documents', input
      if response.is_a?(Net::HTTPOK)
        data = JSON.parse(response.body)
        STDOUT.puts "#{server}/#{data['key']}"
      else
        STDERR.puts "failure uploading: #{res.status}"
      end
    rescue RuntimeError, JSON::ParserError => e
      STDERR.puts "failure uploading: #{response.code}"
    rescue Errno::ECONNREFUSED => e
      STDERR.puts "failure connecting: #{e.message}"
    end

    private

    def server
      return @server if @server
      @server = ENV['HASTE_SERVER'] || Haste::DEFAULT_URL
      @server.chop! if server.end_with?('/')
      @server
    end

  end

end