# encoding: utf-8
module Ayadn
  class CNX

    def self.download url
      begin
        RestClient.get(url) {|response, request, result| response}
      rescue => e
        Errors.global_error("cnx.rb/download", url, e)
      end
    end

    def self.get url
      working = true
      begin
        RestClient.get(url) do |response, request, result|
          Debug.http response, url
          check_nr response, url
        end
      rescue SocketError, SystemCallError, OpenSSL::SSL::SSLError, RestClient::RequestTimeout => e
        if working == true
          working = false
          sleep 0.5
          retry
        end
        Errors.nr "URL: #{url}"
        return {'meta' => {'code' => 666}, 'data' => "#{e}"}.to_json
      rescue => e
        Errors.global_error("cnx.rb/get", url, e)
      end
    end

    def self.check_nr response, url
      case response.code
      when 200
        response
      when 204
        puts "\nError: the NiceRank filter made too many requests to the server. You may either wait for a little while before scrolling the filtered Global again, or set the scroll timer to a greater value (example: `ayadn set scroll timer 5`). (see http://ayadn-app.net/doc).\n".color(:red)
        Errors.global_error("cnx.rb/get", [url, response.inspect, response.headers], "NiceRank: TOO MANY REQUESTS")
      else
        response
      end
    end

    def self.get_response_from(url)
      try_cnx = 1
      begin
        RestClient.get(url) do |response, request, result| #, :verify_ssl => OpenSSL::SSL::VERIFY_NONE
          Debug.http response, url
          check response
        end
      rescue SocketError, SystemCallError, OpenSSL::SSL::SSLError, RestClient::RequestTimeout => e
        if try_cnx < 4
          try_cnx = retry_adn 10, try_cnx
          retry
        end
        puts "\nConnection error.".color(:red)
        Errors.global_error("cnx.rb/get_response_from", url, e)
      rescue => e
        Errors.global_error("cnx.rb/get_response_from", url, e)
      end
    end

    def self.retry_adn seconds, try_cnx
      Errors.warn "Unable to connect to App.net"
      puts "\n\nUnable to connect to App.net\nRetrying in #{seconds} seconds... (#{try_cnx}/3)\n".color(:red)
      try_cnx += 1
      sleep seconds
      puts "\e[H\e[2J"
      try_cnx
    end

    def self.check response
      message = JSON.parse(response)['meta']['error_message'] if response.code != 200
      case response.code
      when 200
        response
      when 204
        puts "\n#{message}".color(:red)
        Errors.global_error("cnx.rb", [message, response.headers], "NO CONTENT")
      when 400
        puts "\n#{message}".color(:red)
        Errors.global_error("cnx.rb", [message, response.headers], "BAD REQUEST")
      when 401
        puts "\n#{message}".color(:red)
        Errors.global_error("cnx.rb", [message, response.headers], "UNAUTHORIZED")
      when 403
        puts "\n#{message}".color(:red)
        Errors.global_error("cnx.rb", [message, response.headers], "FORBIDDEN")
      when 405
        puts "\n#{message}".color(:red)
        Errors.global_error("cnx.rb", [message, response.headers], "METHOD NOT ALLOWED")
      when 429
        puts "\n#{message}".color(:red)
        puts "\n\nAyadn made too many requests to the App.net API. You should wait at least ".color(:cyan) + "#{response.headers[:retry_after]} ".color(:red) + "seconds before trying again. Maybe you launched a lot of Ayadn instances at the same time? That's no problem, but in this case you should increase the value of the scroll timer (with `ayadn set scroll timer 5` for example). App.net allows 5000 requests per hour per account maximum.".color(:cyan)
        Errors.global_error("cnx.rb", [message, response.headers], "TOO MANY REQUESTS")
      when 500
        puts "\n#{message}".color(:red)
        Errors.global_error("cnx.rb", [message, response.headers], "APP.NET SERVER ERROR")
      when 507
        puts "\n#{message}".color(:red)
        Errors.global_error("cnx.rb", [message, response.headers], "INSUFFICIENT STORAGE")
      else
        response
      end
    end

    def self.delete(url)
      begin
        #RestClient::Resource.new(url).delete
        RestClient.delete(url) do |response, request, result|
          Debug.http response, url
          check response
        end
      rescue SocketError => e
        puts "\nConnection error.".color(:red)
        Errors.global_error("cnx.rb/delete", url, e)
      rescue SystemCallError => e
        puts "\nConnection error.".color(:red)
        Errors.global_error("cnx.rb/delete", url, e)
      rescue => e
        Errors.global_error("cnx.rb/delete", url, e)
      end
    end

    def self.post(url, payload = nil)
      begin
        RestClient.post(url, payload.to_json, :content_type => :json, :accept => :json) do |response, request, result|
          Debug.http response, url
          check response
        end
      rescue SocketError => e
        puts "\nConnection error.".color(:red)
        Errors.global_error("cnx.rb/post", url, e)
      rescue SystemCallError => e
        puts "\nConnection error.".color(:red)
        Errors.global_error("cnx.rb/post", url, e)
      rescue => e
        Errors.global_error("cnx.rb/post", [url, payload], e)
      end
    end

    def self.put(url, payload)
      begin
        RestClient.put(url, payload.to_json, :content_type => :json, :accept => :json) do |response, request, result|
          Debug.http response, url
          check response
        end
      rescue SocketError => e
        puts "\nConnection error.".color(:red)
        Errors.global_error("cnx.rb/put", url, e)
      rescue SystemCallError => e
        puts "\nConnection error.".color(:red)
        Errors.global_error("cnx.rb/put", url, e)
      rescue => e
        Errors.global_error("cnx.rb/put", [url, payload], e)
      end
    end

  end
end