rpush/rpush

module Rpush
  module Daemon
    module Wns
      class BadgeRequest
        def self.create(notification, access_token)
          body = BadgeRequestPayload.new(notification).to_xml
          uri = URI.parse(notification.uri)
          headers = {
            "Content-Length" => body.length.to_s,
            "Content-Type" => "text/xml",
            "X-WNS-Type" => "wns/badge",
            "X-WNS-RequestForStatus" => "true",
            "Authorization" => "Bearer #{access_token}"
          }
          headers['X-WNS-PRIORITY'] = notification.priority.to_s if notification.priority

          post = Net::HTTP::Post.new(
            uri.request_uri,
            headers
          )
          post.body = body
          post
        end
      end

      class BadgeRequestPayload
        def initialize(notification)
          @badge = notification.badge || 0
        end

        def to_xml
          "<badge value=\"#{@badge}\"/>"
        end
      end
    end
  end
end