require 'httparty'
require 'json'

module GTAPI
  class GaptoolServer
    include HTTParty

    def initialize(user, apikey, uri)
      @auth = { 'X-GAPTOOL-USER' => user, 'X-GAPTOOL-KEY' => apikey}
      GaptoolServer.base_uri uri
    end

    def redishash(hash)
      options = {:body => hash, :headers => @auth}
      JSON::parse self.class.post("/redishash", options)
    end

    def getonenode(role, environment, id)
      options = {:headers => @auth}
      JSON::parse self.class.get("/host/#{role}/#{environment}/#{id}", options)
    end

    def getenvroles(role, environment)
      options = {:headers => @auth}
      JSON::parse self.class.get("/hosts/#{role}/#{environment}", options)
    end

    def addnode(zone, itype, role, environment, mirror=0)
      if mirror == 0
        @body = {
          'zone' => zone,
          'itype' => itype,
          'role' => role,
          'environment' => environment,
        }.to_json
      else
        @body = {
          'zone' => zone,
          'itype' => itype,
          'role' => role,
          'environment' => environment,
          'mirror' => mirror,
        }.to_json
      end
      options = { :body => @body, :headers => @auth}
      JSON::parse self.class.post("/init", options)
    end

    def terminatenode(id, zone)
      @body = {'id' => id, 'zone' => zone}.to_json
      options = {:body => @body, :headers => @auth}
      JSON::parse self.class.post("/terminate", options)
    end

    def getappnodes(app, environment)
      options = {:headers => @auth}
      role = JSON::parse(self.class.get("/apps", options))["app:#{app}"]['role']
      JSON::parse self.class.get("/hosts/#{role}/#{environment}", options)
    end

    def ssh(role, environment, id)
      options = { :headers => @auth}
      JSON::parse self.class.get("/ssh/#{role}/#{environment}/#{id}", options)
    end

    def regenhosts(zone)
      @body = {'zone' => zone}.to_json
      options = {:body => @body, :headers => @auth}
      JSON::parse self.class.post("/regenhosts", options)
    end

    def addservice(role, environment, name, keys, weight, enabled)
      @body = {
        'name' => name,
        'keys' => keys,
        'weight' => weight,
        'enabled' => enabled,
        'role' => role,
        'environment' => environment
      }.to_json
      options = {:body => @body, :headers => @auth}
      # output is service count
      JSON::parse self.class.put("/service/#{role}/#{environment}", options)
    end

    def deleteservice(role, environment, name)
      options = { :headers => @auth}
      # output is service count
      JSON::parse self.class.delete("/service/#{role}/#{environment}/#{name}", options)
    end

    def balanceservices(role, environment)
      options = { :headers => @auth}
      # output is runlist
      JSON::parse self.class.get("/servicebalance/#{role}/#{environment}", options)
    end

    def getservices()
      options = { :headers => @auth}
      # output is all service entries
      JSON::parse self.class.get("/services", options)
    end

    def getallnodes()
      options = { :headers => @auth}
      JSON::parse self.class.get("/hosts", options)
    end
  end
end