require 'httparty'
require 'cgi'
require 'uri'
require 'json'

require 'hudson/config'

module Hudson
  module Api
    include HTTParty

    headers 'content-type' => 'application/json'
    format :json
    # http_proxy 'localhost', '8888'
    
    JobAlreadyExistsError = Class.new(Exception)

    def self.setup_base_url(options)
      # Thor's HashWithIndifferentAccess is based on string keys which URI::HTTP.build ignores
      options = options.inject({}) { |mem, (key, val)| mem[key.to_sym] = val; mem }
      options[:host] ||= ENV['HUDSON_HOST']
      options[:port] ||= ENV['HUDSON_PORT']
      options[:port] &&= options[:port].to_i
      return false unless options[:host] || Hudson::Config.config["base_uri"]
      uri = options[:host] ? URI::HTTP.build(options) : Hudson::Config.config["base_uri"]
      base_uri uri.to_s
      uri
    end

    # returns true if successfully create a new job on Hudson
    # +job_config+ is a Hudson::JobConfigBuilder instance
    # +options+ are:
    #   :override - true, will delete any existing job with same name, else error
    #
    # returns true if successful, else false
    #
    # TODO Exceptions?
    def self.create_job(name, job_config, options = {})
      delete_job(name) if options[:override]
      begin
        res = post "/createItem/api/xml?name=#{CGI.escape(name)}", {
          :body => job_config.to_xml, :format => :xml, :headers => { 'content-type' => 'application/xml' }
        }
        if res.code.to_i == 200
          cache_base_uri
          true
        else
          # TODO - what are the errors we get?
          puts "Server error:"
          p res.code
          puts res.body
          false
        end
      rescue REXML::ParseException => e
        # For some reason, if the job exists we get back half a page of HTML
        raise JobAlreadyExistsError.new(name)
      end
    end
    
    # Attempts to delete a job +name+
    def self.delete_job(name)
      res = post_plain "#{job_url name}/doDelete"
      res.code.to_i == 302
    end
    
    def self.build_job(name)
      res = get_plain "/job/#{name}/build"
      res.code.to_i == 302
    end

    def self.summary
      json = get "/api/json"
      cache_base_uri
      json
    end
    
    def self.job_names
      summary["jobs"].map {|job| job["name"]}
    end

    # Return hash of job statuses
    def self.job(name)
      begin
        json = get "/job/#{name}/api/json"
        cache_base_uri
        json
      rescue Errno::ECONNREFUSED
        false
      rescue Errno::EAFNOSUPPORT
        false
      end
    end

    def self.nodes
      json = get "/computer/api/json"
      cache_base_uri
      json
    rescue Errno::ECONNREFUSED => e
      false
    rescue Errno::EAFNOSUPPORT
      false
    end

    # Adds SSH nodes only, for now
    def self.add_node(options = {})
      default_options = Hash.new("")
      default_options.merge!(
        :slave_port  => 22,
        :master_key  => "/home/hudson/.ssh/id_rsa", # FIXME - hardcoded master username assumption
        :remote_fs   => "/data/hudson-slave/",
        :description => "Automatically created by Hudson.rb",
        :executors   => 2,
        :exclusive   => true
      )

      options = default_options.merge(options)

      name = options[:name]
      type = "hudson.slaves.DumbSlave$DescriptorImpl"

      fields = {
        "name" => name,
        "type" => type,

        "json"                => {
          "name"              => name,
          "nodeDescription"   => options[:description],
          "numExecutors"      => options[:executors],
          "remoteFS"          => options[:remote_fs],
          "labelString"       => options[:label],
          "mode"              => options[:exclusive] ? "EXCLUSIVE" : "NORMAL",
          "type"              => type,
          "retentionStrategy" => { "stapler-class"  => "hudson.slaves.RetentionStrategy$Always" },
          "nodeProperties"    => { "stapler-class-bag" => "true" },
          "launcher"          => {
            "stapler-class" => "hudson.plugins.sshslaves.SSHLauncher",
            "host"          => options[:slave_host],
            "username"      => options[:slave_user],
            "privatekey"    => options[:master_key],
            "port"          => options[:slave_port]
          }
        }.to_json
      }

      url = URI.parse("#{base_uri}/computer/doCreateItem")
      req = Net::HTTP::Post.new(url.path)
      req.set_form_data(fields)

      http = Net::HTTP.new(url.host, url.port)

      case http.request(req)
      when Net::HTTPFound
        true
      else
        false
      end
    end
    
    def self.delete_node(name)
      post_plain("#{base_uri}/computer/#{CGI::escape(name).gsub('+', '%20')}/doDelete/api/json")
    end

    # Helper for POST that don't barf at Hudson's crappy API responses
    def self.post_plain(path, options = {})
      uri = URI.parse base_uri
      res = Net::HTTP.start(uri.host, uri.port) { |http| http.post(path, options) }
    end
    
    # Helper for GET that don't barf at Hudson's crappy API responses
    def self.get_plain(path, options = {})
      uri = URI.parse base_uri
      res = Net::HTTP.start(uri.host, uri.port) { |http| http.get(path, options) }
    end
    
    private
    def self.cache_base_uri
      Hudson::Config.config["base_uri"] = base_uri
      Hudson::Config.store!
    end
    
    def self.job_url(name)
      "#{base_uri}/job/#{name}"
    end
  end
end