Add poll plugin.

plugins/poll/plugin.rb

@@ -0,0 +1,159 @@
+# 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