Insynia/user_notif

module UserNotif
  module ViewHelpers
    include ::ActionView::Context
    include ::ActionView::Helpers

    def list_user_notifs(notifs = nil, small = true)
      return unless notifs
      output = ''

      notifs.each { |n| output << user_notif(n, small) }

      wrap output
    end

    def user_notif(notif, small = true)
      @notif = notif
      @target = notif.target
      render("notifs/#{small ? 'small' : 'full'}/#{notif.template_name}")
    end

    def notif_badge(notifs = nil)
      nb = notifs.count
      nb == 0 ? '' : "<span class=\"notif-badge\">#{nb}</span>".html_safe
    end

    private

    def wrap(elem)
      "<div class=\"user-notif-wrapper\">#{elem}</div>".html_safe
    end
  end
end