discourse/plugins/poll/plugin.rb

160 lines
3.6 KiB
Ruby

# name: poll
# about: adds poll support to Discourse
# version: 0.1
# authors: Vikhyat Korrapati
load File.expand_path("../poll.rb", __FILE__)
# Without this line we can't lookup the constant inside the after_initialize blocks,
# probably 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.is_poll?
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])
render json: poll.serialize(current_user)
end
end
end
PollPlugin::Engine.routes.draw do
put '/' => 'poll#vote'
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.
# Need to add an error when:
# * there is no list of options.
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
if self.created_at and self.created_at < 5.minutes.ago and poll.options.sort != poll.details.keys.sort
self.errors.add(:raw, I18n.t('poll.cannot_have_modified_options'))
end
end
end
# Save the list of options to PluginStore after the post is saved.
Post.class_eval do
after_save :save_poll_options_to_topic_metadata
def save_poll_options_to_topic_metadata
poll = PollPlugin::Poll.new(self)
if poll.is_poll?
details = poll.details || {}
new_options = poll.options
details.each do |key, value|
unless new_options.include? key
details.delete(key)
end
end
new_options.each do |key|
details[key] ||= 0
end
poll.set_details! details
end
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).is_poll?
end
end
end
# Poll UI.
register_asset "javascripts/discourse/templates/poll.js.handlebars"
register_asset "javascripts/poll_ui.js"
register_asset "javascripts/poll_bbcode.js", :server_side
register_css <<CSS
.poll-ui table {
margin-bottom: 5px;
}
.poll-ui tr {
cursor: pointer;
}
.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 {
border: none;
}
CSS