CDJ11/CDJ

module FlagsHelper

  def show_flag_action?(flaggable)
    current_user && !own_flaggable?(flaggable) && !flagged?(flaggable)
  end

  def show_unflag_action?(flaggable)
    current_user && !own_flaggable?(flaggable) && flagged?(flaggable)
  end

  private

  def flagged?(flaggable)
    if flaggable.is_a? Comment
      @comment_flags[flaggable.id] unless flaggable.commentable_type == "Poll"
    else
      Flag.flagged?(current_user, flaggable)
    end
  end

  def own_flaggable?(flaggable)
    if flaggable.is_a? Comment
      flaggable.user_id == current_user.id
    else
      flaggable.author_id == current_user.id
    end
  end

end