discourse/plugins/poll/plugin.rb

204 lines
4.9 KiB
Ruby

# name: poll
# about: adds poll support to Discourse
# version: 0.1
# authors: Vikhyat Korrapati
# url: https://github.com/discourse/discourse/tree/master/plugins/poll
load File.expand_path("../poll.rb", __FILE__)
# Without this line we can't lookup the constant inside the after_initialize blocks,
# because all of this is instance_eval'd inside an instance of Plugin::Instance.
PollPlugin = PollPlugin
after_initialize do
# Rails Engine for accepting votes.
module PollPlugin
class Engine < ::Rails::Engine
engine_name "poll_plugin"
isolate_namespace PollPlugin
end
class PollController < ActionController::Base
include CurrentUser
def vote
if current_user.nil?
render status: :forbidden, json: false
return
end
if params[:post_id].nil? or params[:option].nil?
render status: 400, json: false
return
end
post = Post.find(params[:post_id])
poll = PollPlugin::Poll.new(post)
unless poll.has_poll_details?
render status: 400, json: false
return
end
options = poll.details
unless options.keys.include? params[:option]
render status: 400, json: false
return
end
poll.set_vote!(current_user, params[:option])
MessageBus.publish("/topic/#{post.topic_id}", {
id: post.id,
post_number: post.post_number,
updated_at: Time.now,
type: "revised"
},
group_ids: post.topic.secure_group_ids
)
render json: poll.serialize(current_user)
end
def toggle_close
post = Post.find(params[:post_id])
topic = post.topic
poll = PollPlugin::Poll.new(post)
# Make sure the user is allowed to close the poll.
Guardian.new(current_user).ensure_can_edit!(topic)
# Make sure this is actually a poll.
unless poll.has_poll_details?
render status: 400, json: false
return
end
# Make sure the topic is not closed.
if topic.closed?
render status: 400, json: false
return
end
# Modify topic title.
I18n.with_locale(topic.user.effective_locale) do
if topic.title =~ /^(#{I18n.t('poll.prefix').strip})\s?:/i
topic.title = topic.title.gsub(/^(#{I18n.t('poll.prefix').strip})\s?:/i, I18n.t('poll.closed_prefix') + ':')
elsif topic.title =~ /^(#{I18n.t('poll.closed_prefix').strip})\s?:/i
topic.title = topic.title.gsub(/^(#{I18n.t('poll.closed_prefix').strip})\s?:/i, I18n.t('poll.prefix') + ':')
end
end
topic.acting_user = current_user
topic.save!
render json: topic, serializer: BasicTopicSerializer
end
end
end
PollPlugin::Engine.routes.draw do
put '/' => 'poll#vote'
put '/toggle_close' => 'poll#toggle_close'
end
Discourse::Application.routes.append do
mount ::PollPlugin::Engine, at: '/poll'
end
# Starting a topic title with "Poll:" will create a poll topic. If the title
# starts with "poll:" but the first post doesn't contain a list of options in
# it we need to raise an error.
Post.class_eval do
validate :poll_options
def poll_options
poll = PollPlugin::Poll.new(self)
return unless poll.is_poll?
if poll.options.length == 0
self.errors.add(:raw, I18n.t('poll.must_contain_poll_options'))
end
poll.ensure_can_be_edited!
end
end
# Save the list of options to PluginStore after the post is saved.
Post.class_eval do
after_save :save_poll_options_to_plugin_store
def save_poll_options_to_plugin_store
PollPlugin::Poll.new(self).update_options!
end
end
# Add poll details into the post serializer.
PostSerializer.class_eval do
attributes :poll_details
def poll_details
PollPlugin::Poll.new(object).serialize(scope.user)
end
def include_poll_details?
PollPlugin::Poll.new(object).has_poll_details?
end
end
end
# Poll UI.
register_asset "javascripts/controllers/poll.js.es6"
register_asset "javascripts/discourse/templates/poll.hbs"
register_asset "javascripts/initializers/poll.js.es6"
register_asset "javascripts/poll_bbcode.js", :server_side
register_css <<CSS
.poll-ui table {
margin-bottom: 5px;
margin-left: 20px;
}
.poll-ui tr {
cursor: pointer;
}
.poll-ui .row {
padding-left: 15px;
padding-top: 10px;
}
.poll-ui td.radio input {
margin-left: -10px !important;
}
.poll-ui td {
padding: 4px 8px;
}
.poll-ui td.option .option {
float: left;
}
.poll-ui td.option .result {
float: right;
margin-left: 50px;
}
.poll-ui tr.active {
background-color: #FFFFB3;
}
.poll-ui button i.fa {
margin-right: 2px;
}
.poll-ui .radio {
margin-right: 0px;
}
.poll-ui .toggle-poll {
float: right;
}
CSS