require 'fog'

module Fog
  module Radosgw
    module MultipartUtils
      require 'net/http'

      class Headers
        include Net::HTTPHeader

        def initialize
          initialize_http_header({})
        end

        # Parse a single header line into its key and value
        # @param [String] chunk a single header line
        def self.parse(chunk)
          line = chunk.strip
          # thanks Net::HTTPResponse
          return [nil,nil] if chunk =~ /\AHTTP(?:\/(\d+\.\d+))?\s+(\d\d\d)\s*(.*)\z/in
          m = /\A([^:]+):\s*/.match(line)
          [m[1], m.post_match] rescue [nil, nil]
        end

        # Parses a header line and adds it to the header collection
        # @param [String] chunk a single header line
        def parse(chunk)
          key, value = self.class.parse(chunk)
          add_field(key, value) if key && value
        end
      end

      def parse(data, boundary)
        contents = data.match(end_boundary_regex(boundary)).pre_match rescue ""
        contents.split(inner_boundary_regex(boundary)).reject(&:empty?).map do |part|
          parse_multipart_section(part)
        end.compact
      end

      def extract_boundary(header_string)
        $1 if header_string =~ /boundary=([A-Za-z0-9\'()+_,-.\/:=?]+)/
      end

      private
      def end_boundary_regex(boundary)
        /\r?\n--#{Regexp.escape(boundary)}--\r?\n?/
      end

      def inner_boundary_regex(boundary)
        /\r?\n--#{Regexp.escape(boundary)}\r?\n/
      end

      def parse_multipart_section(part)
        headers = Headers.new
        if md = part.match(/\r?\n\r?\n/)
          body = md.post_match
          md.pre_match.split(/\r?\n/).each do |line|
            headers.parse(line)
          end

          if headers["content-type"] =~ /multipart\/mixed/
            boundary = extract_boundary(headers.to_hash["content-type"].first)
            parse(body, boundary)
          else
            {:headers => headers.to_hash, :body => body}
          end
        end
      end
    end

    module UserUtils

      def update_radosgw_user(user_id, user)
        path         = "admin/user"
        user_id      = Fog::AWS.escape(user_id)
        params       = {
          :method => 'POST',
          :path => path,
        }
        query        = "?uid=#{user_id}&format=json&suspended=#{user[:suspended]}"
        begin
          response = Excon.post("#{@scheme}://#{@host}/#{path}#{query}",
                                :headers => signed_headers(params))
          if !response.body.empty?
            case response.headers['Content-Type']
            when 'application/json'
              response.body = Fog::JSON.decode(response.body)
            end
          end
          response
        rescue Excon::Errors::NotFound => e
          raise Fog::Radosgw::Provisioning::NoSuchUser.new
        rescue Excon::Errors::BadRequest => e
          raise Fog::Radosgw::Provisioning::ServiceUnavailable.new
        end
      end

      def update_mock_user(user_id, user)
        if data[user_id]
          if suspended = user[:suspended]
            data[user_id][:suspended] = suspended
          end

          Excon::Response.new.tap do |response|
            response.status = 200
            response.body   = data[user_id]
          end
        else
          Excon::Response.new.tap do |response|
            response.status = 403
          end
        end
      end
    end

    module Utils
      def configure_uri_options(options = {})
        @host       = options[:host]       || 'localhost'
        @persistent = options[:persistent] || true
        @port       = options[:port]       || 8080
        @scheme     = options[:scheme]     || 'http'
      end

      def radosgw_uri
        "#{@scheme}://#{@host}:#{@port}"
      end

      def signed_headers(params)
        expires = Fog::Time.now.to_date_header
        auth   =  @s3_connection.signature(params,expires)
        awskey =  @radosgw_access_key_id
        headers = {
          'Date'          => expires,
          'Authorization' => "AWS #{awskey}:#{auth}"
        }
      end
    end

    extend Fog::Provider

    service(:provisioning, 'Provisioning')
    service(:usage,        'Usage')
  end
end