require "aptly_cli/version" 
require "aptly_load"
require "httmultiparty"
require "json"

module AptlyCli
  class AptlyRepo
  
    include HTTMultiParty
    
    # Load aptly-cli.conf and establish base_uri
    config = AptlyCli::AptlyLoad.new.configure_with("/etc/aptly-cli.conf")
    base_uri "http://#{config[:server]}:#{config[:port]}/api"

    if config[:username]
      if config[:password]
        basic_auth "#{config[:username]}", "#{config[:password]}"
      end
    end

    if config[:debug] == true
      debug_output $stdout
    end

    def repo_create(repo_options = {:name => nil, :comment => nil, :DefaultDistribution => nil, :DefaultComponent => nil})
      uri = "/repos"
      name = repo_options[:name]
      comment = repo_options[:comment]
      default_distribution = repo_options[:DefaultDistribution]
      default_component = repo_options[:DefaultComponent]
      
      self.class.post(uri, :query => { 'Name' => name, 'Comment' => comment, 'DefaultDistribution' => default_distribution, 'DefaultComponent' => default_component }.to_json, :headers => {'Content-Type'=>'application/json'}) 
    end

    def repo_delete(repo_options = {:name => nil, :force => nil})
      uri = "/repos/" + repo_options[:name]
      
      if repo_options[:force] == true 
        uri = uri + "?force=1"
      end

      self.class.delete(uri)
    end

    def repo_edit(name, repo_options = { k => v })
      repo_option = String.new
      repo_value = String.new 
      
      if name == nil
        raise ArgumentError.new('Must pass a repository name')
      else
        uri = "/repos/" + name
      end

      repo_options.each do |k, v|
        repo_option = k
        repo_value = v
      end

      self.class.put(uri, :query => { repo_option => repo_value }.to_json, :headers => {'Content-Type'=>'application/json'})
    end

    def repo_list()
      uri = "/repos"
      
      self.class.get(uri)
    end
    
    def repo_package_query(repo_options = {:name => nil, :query => nil, :withdeps => false, :format => nil})
      if repo_options[:name] == nil
        raise ArgumentError.new('Must pass a repository name')
      else
        uri = "/repos/" + repo_options[:name] + "/packages"
      end

      if repo_options[:query]
        uri = uri + "?q=" + repo_options[:query]
        if repo_options[:withdeps] or repo_options[:format]
          puts "When specifiying specific package query, other options are invalid."
        end 
      elsif repo_options[:format]
        uri = uri + "?format=#{repo_options[:format]}"
      elsif repo_options[:withdeps] == true
        uri = uri + "?withDeps=1"
      end

      self.class.get uri 

    end

    def repo_show(name)
      if name == nil
        uri = "/repos"
      else
        uri = "/repos/" + name 
      end
      
      self.class.get uri 
    end

    def repo_upload(repo_options = {:name => nil, :dir => nil, :file => nil, 
                                    :noremove => false, :forcereplace => false})

      name = repo_options[:name]
      dir  = repo_options[:dir]
      file = repo_options[:file]
      noremove = repo_options[:noremove]
      forcereplace = repo_options[:forcereplace]

      if file == nil 
        uri = "/repos/#{name}/file/#{dir}"
      else
        uri = "/repos/#{name}/file/#{dir}/#{file}"
      end

      if forcereplace == true 
        uri = uri + "?forceReplace=1"
      end
      
      if noremove == true 
        uri = uri + "?noRemove=1"
      end
      
      response = self.class.post(uri)
      
      case response.code
        when 404
          puts 'repository with such name does not exist'
      end

      json_response = JSON.parse(response.body)
      
      unless json_response["FailedFiles"].empty?
        begin
        rescue StandardError => e
          puts "Files that failed to upload... #{json_response["FailedFiles"]}"
          puts e
        end
      end

      unless json_response["Report"]["Warnings"].empty?
        begin
        rescue StandardError => e
          puts "File upload warning message[s]...#{json_response["Report"]["Warnings"]}"
          puts e
        end
      end
      
      return response

    end 

  end
end