crispgm/holy-images

class NotificationController < ApplicationController
  include TimeHelper

  before_action :require_login

  def list
    @user = current_user
    @notifications = Notification.where(user_id: @user.id).last(20).reverse

    render "notification/list"
  end

  def read
    id = params[:id]
    user = current_user

    result = nil
    if id == "all"
      result = Notification.where(status: 0, user_id: user.id).update_all(status: 1)
      redirect_to notification_url
    else
      id = id.to_i
      result = Notification.where(status: 0, user_id: user.id, id: id).update(status: 1)
      notif = Notification.find(id)
      target_id = case notif.event_type
      when 1
        target_id = notif.like.image_id
      when 2
        target_id = notif.comment.image_id
      else
        nil
      end
      redirect_url = "/image/#{target_id}"
      redirect_url << "#comments" if notif.event_type == 2

      unless target_id.nil?
        redirect_to redirect_url
      else
        redirect_to root_url
      end
    end
  end

  private
  def require_login
    unless logged_in?
      flash[:warning] = I18n.t(:login_required)
      redirect_to root_url
    end
  end
end