module ProMotion
  module DelegateNotifications

    attr_accessor :aps_notification

    def check_for_push_notification(options)
      if options && options[UIApplicationLaunchOptionsRemoteNotificationKey]
        received_push_notification options[UIApplicationLaunchOptionsRemoteNotificationKey], true
      end
    end

    def register_for_push_notifications(*notification_types)
      notification_types = Array.new(notification_types)
      notification_types = [ :badge, :sound, :alert, :newsstand ] if notification_types.include?(:all)

      types = UIRemoteNotificationTypeNone
      notification_types.each { |t| types = types | map_notification_symbol(t) }

      UIApplication.sharedApplication.registerForRemoteNotificationTypes types
    end

    def unregister_for_push_notifications
      UIApplication.sharedApplication.unregisterForRemoteNotifications
    end

    def registered_push_notifications
      mask = UIApplication.sharedApplication.enabledRemoteNotificationTypes
      types = []

      types << :badge     if mask & UIRemoteNotificationTypeBadge > 0
      types << :sound     if mask & UIRemoteNotificationTypeSound > 0
      types << :alert     if mask & UIRemoteNotificationTypeAlert > 0
      types << :newsstand if mask & UIRemoteNotificationTypeNewsstandContentAvailability > 0

      types
    end

    def received_push_notification(notification, was_launched)
      @aps_notification = PM::PushNotification.new(notification)
      on_push_notification(@aps_notification, was_launched) if respond_to?(:on_push_notification)
    end

    # CocoaTouch

    def application(application, didRegisterForRemoteNotificationsWithDeviceToken:device_token)
      on_push_registration(device_token, nil) if respond_to?(:on_push_registration)
    end

    def application(application, didFailToRegisterForRemoteNotificationsWithError:error)
      on_push_registration(nil, error) if respond_to?(:on_push_registration)
    end

    def application(application, didReceiveRemoteNotification:notification)
      received_push_notification(notification, application.applicationState != UIApplicationStateActive)
    end

    protected

    def map_notification_symbol(symbol)
      {
        none:       UIRemoteNotificationTypeNone,
        badge:      UIRemoteNotificationTypeBadge,
        sound:      UIRemoteNotificationTypeSound,
        alert:      UIRemoteNotificationTypeAlert,
        newsstand:  UIRemoteNotificationTypeNewsstandContentAvailability
      }[symbol] || UIRemoteNotificationTypeNone
    end

  end
end