require 'net/http'
require 'net/https'
require 'uri'
require 'digest/sha1'
require 'wagon/ward'

module Wagon
  
  class AuthenticationFailure < StandardError; end
  
  class Connection
    HOST        = 'secure.lds.org'
    LOGIN_PATH  = '/units/a/login/1,21568,779-1,00.html?URL='
    CACHE_PATH  = File.join(File.expand_path('~'), '.wagon_cache')
    
    def initialize(username, password)
      response    = post(LOGIN_PATH, 'username' => username, 'password' => password)
      @cookies    = response['set-cookie']
      @home_path  = URI.parse(response['location']).path

      raise AuthenticationFailure.new("Invalid username and/or password") unless @cookies
    end
    
    def home_path
      @home_path
    end
    
    def ward
      @ward ||= Ward.new(self, home_path)
    end
    
    def get(path)
      Connection.perform_caching? ? get_with_caching(path) : get_without_caching(path)
    end
    
    def get_without_caching(path)
      _http.request(Net::HTTP::Get.new(path, {'Cookie' => @cookies || ''})).body
    end
    
    def get_with_caching(path)
      FileUtils::mkdir(CACHE_PATH) unless File.directory?(CACHE_PATH)
      cache_path = File.join(CACHE_PATH, Digest::SHA1.hexdigest(path) + ".cache")
      return open(cache_path).read if File.exists?(cache_path)
      open(cache_path, "w").write(data = get_without_caching(path))
      data
    end
    
    def self.perform_caching?
      @@perform_caching ||= true
    end
    
    def self.perform_caching(true_or_false)
      @@perform_caching = true_or_false
    end

    def post(path, data)
      request = Net::HTTP::Post.new(path, {'Cookie' => @cookies || ''})
      request.set_form_data(data)
      _http.request(request)
    end
    
    def _dump(depth)
      Marshal.dump([@cookies, @home_path])
    end
    
    def self._load(string)
      attributes = Marshal.restore(string)
      connection = Connection.allocate()
      connection.instance_variable_set(:@cookies, attributes.shift)
      connection.instance_variable_set(:@home_path, attributes.shift)
      connection
    end
    
    private
    def _http
      return @http unless @http.nil?
      @http              = Net::HTTP.new(HOST, 443)
      @http.use_ssl      = true
      @http.verify_mode  = OpenSSL::SSL::VERIFY_NONE
      @http
    end
  end
end