a5f1245b7e
Now that we're rendering `shared/flag_actions` everywhere, we can use the same code in all cases.
28 lines
652 B
Ruby
28 lines
652 B
Ruby
module FlagActions
|
|
extend ActiveSupport::Concern
|
|
|
|
def flag
|
|
Flag.flag(current_user, flaggable)
|
|
|
|
render "shared/_refresh_flag_actions", locals: { flaggable: flaggable }
|
|
end
|
|
|
|
def unflag
|
|
Flag.unflag(current_user, flaggable)
|
|
|
|
render "shared/_refresh_flag_actions", locals: { flaggable: flaggable }
|
|
end
|
|
|
|
private
|
|
|
|
def flaggable
|
|
if resource_model.to_s == "Budget::Investment"
|
|
instance_variable_get("@investment")
|
|
elsif resource_model.to_s == "Legislation::Proposal"
|
|
instance_variable_get("@proposal")
|
|
else
|
|
instance_variable_get("@#{resource_model.to_s.downcase}")
|
|
end
|
|
end
|
|
end
|