rpush/rpush

module Rpush
  module Daemon
    module Wns
      class ToastRequest
        def self.create(notification, access_token)
          body = ToastRequestPayload.new(notification).to_xml
          uri  = URI.parse(notification.uri)
          headers = {
            "Content-Length" => body.length.to_s,
            "Content-Type" => "text/xml",
            "X-WNS-Type" => "wns/toast",
            "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 ToastRequestPayload
        def initialize(notification)
          @title = notification.data['title'] || ''
          @body = notification.data['body'] || ''
          @launch = notification.data['launch']
          @sound = notification.sound unless notification.sound.eql?("default".freeze)
        end

        def to_xml
          launch_string = "" unless @launch
          launch_string = " launch='#{CleanParamString.clean(@launch)}'" if @launch
          audio_string = "" unless @sound
          audio_string = "<audio src='#{CleanParamString.clean(@sound)}'/>" if @sound
          "<toast#{launch_string}>
            <visual version='1' lang='en-US'>
              <binding template='ToastText02'>
                <text id='1'>#{CleanParamString.clean(@title)}</text>
                <text id='2'>#{CleanParamString.clean(@body)}</text>
              </binding>
            </visual>
            #{audio_string}
          </toast>"
        end
      end

      class CleanParamString
        def self.clean(string)
          string.gsub(/&/, "&amp;").gsub(/</, "&lt;") \
            .gsub(/>/, "&gt;").gsub(/'/, "&apos;").gsub(/"/, "&quot;")
        end
      end
    end
  end
end