discourse/plugins/poll/plugin.rb

# 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,
# 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.
        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

        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/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;
}

.poll-ui button i.fa {
  margin-right: 2px;
}

CSS