lloydmeta/push_to_devices

# encoding: utf-8
class Notification
  include Mongoid::Document
  include Mongoid::Timestamps # adds created_at and updated_at fields

  # field <name>, :type => <type>, :default => <value>
  field :ios_specific_fields, :type => String
  field :android_specific_fields, :type => String

  validate :fields_all_json

  embedded_in :user

  after_create :increment_user_notifications_count
  after_destroy :decrement_user_notifications_count

  DEFAULT_NOTIFICATION_IOS = {alert: "Hi", badge: 1, sound: "default"}

  DEFAULT_NOTIFICATION_OPTIONS_ANDROID = {options: {time_to_live: 1.week.to_i}}
  DEFAULT_NOTIFICATION_DATA_ANDROID = {data: {title: "My App", text: "Hello"}}

  # You can define indexes on documents using the index macro:
  # index :field <, :unique => true>

  # You can create a composite key in mongoid to replace the default id using the key macro:
  # key :field <, :another_field, :one_more ....>

  def ios_version
    if ios_specific_fields && ! ios_specific_fields.empty?
      DEFAULT_NOTIFICATION_IOS.merge(JSON.parse(ios_specific_fields).symbolize_keys)
    else
      {}
    end
  end

  def android_version
    if android_specific_fields && ! android_specific_fields.empty?
      android_notification = JSON.parse(android_specific_fields).symbolize_keys
      DEFAULT_NOTIFICATION_DATA_ANDROID.merge(DEFAULT_NOTIFICATION_OPTIONS_ANDROID).merge(android_notification)
    else
      {}
    end
  end

  # Returns a sendable APNS or GCM Notification, ready to be consumed by
  # a connection provided by PushMeUp
  def sendable(type)
    if type == :ios
      user.apn_device_tokens.map{|apn_device_token|
        APNS::Notification.new(apn_device_token.device_id, ios_version)
      }
    elsif type == :android
      sendable_android
    end
  end

  def sendable_android
    android_noti = android_version
    unless android_noti.empty? || user.gcm_device_tokens.empty?
      GCM::Notification.new(user.gcm_device_tokens.map(&:device_id),  android_noti.delete(:data), android_noti.delete(:options))
    end
  end

  private
    def fields_all_json
      [ios_specific_fields, android_specific_fields].each do |field|
        begin
          JSON.parse(field) unless field.nil? || field.empty?
        rescue
          errors.add :base, "Invalid JSON: #{field}"
        end
      end
    end

    def increment_user_notifications_count
      parent = self._parent
      parent.inc(:notifications_count, 1)
    end

    def decrement_user_notifications_count
      parent = self._parent
      parent.inc(:notifications_count, -1)
    end
end