Merge pull request #2262 from vikhyat/poll-plugin

Add open/close buttons to poll UI

plugins/poll/README.md

@@ -9,7 +9,7 @@ Allows you to add a poll to the first post of a topic.
 
 ## Closing the poll
 
-Change the start of the topic title from "Poll: " to "Closed Poll: "
+Change the start of the topic title from "Poll: " to "Closed Poll: ". This feature is disabled if the `allow_user_locale` site setting is enabled.
 
 _Note: closing a topic will also close the poll._
 
@@ -21,12 +21,12 @@ list to be used like this:
 
 ```
 Intro Text
-   
+
 - Item one
 - Item two
-   
+
 Here are your choices:
-   
+
 [poll]
 - Option 1
 - Option 2

plugins/poll/assets/javascripts/discourse/templates/poll.js.handlebars

@@ -14,14 +14,28 @@
   {{/each}}
 </table>
 
-<button {{action toggleShowResults}}>
+<button {{action toggleShowResults}} class="btn btn-small">
   {{#if showResults}}
+    <i class="fa fa-eye-slash"></i>
     {{i18n poll.results.hide}}
   {{else}}
+    <i class="fa fa-eye"></i>
     {{i18n poll.results.show}}
   {{/if}}
 </button>
 
+{{#if controller.showToggleClosePoll}}
+  <button {{action toggleClosePoll}} class="btn btn-small">
+    {{#if poll.closed}}
+      <i class="fa fa-unlock-alt"></i>
+      {{i18n poll.open_poll}}
+    {{else}}
+      <i class="fa fa-lock"></i>
+      {{i18n poll.close_poll}}
+    {{/if}}
+  </button>
+{{/if}}
+
 {{#if loading}}
   <i class="fa fa-spin fa-spinner"></i>
 {{/if}}

plugins/poll/assets/javascripts/poll_ui.js

@@ -44,8 +44,10 @@ var Poll = Discourse.Model.extend({
 var PollController = Discourse.Controller.extend({
   poll: null,
   showResults: Em.computed.oneWay('poll.closed'),
-
   disableRadio: Em.computed.any('poll.closed', 'loading'),
+  showToggleClosePoll: function() {
+    return this.get('poll.post.topic.details.can_edit') && !Discourse.SiteSettings.allow_user_locale;
+  }.property('poll.post.topic.details.can_edit'),
 
   actions: {
     selectOption: function(option) {
@@ -67,6 +69,18 @@ var PollController = Discourse.Controller.extend({
 
     toggleShowResults: function() {
       this.set('showResults', !this.get('showResults'));
+    },
+
+    toggleClosePoll: function() {
+      this.set('loading', true);
+      return Discourse.ajax("/poll/toggle_close", {
+        type: "PUT",
+        data: {post_id: this.get('poll.post.id')}
+      }).then(function(topicJson) {
+        this.set('poll.post.topic.title', topicJson.basic_topic.title);
+        this.set('poll.post.topic.fancy_title', topicJson.basic_topic.title);
+        this.set('loading', false);
+      }.bind(this));
     }
   }
 });

plugins/poll/config/locales/client.en.yml

@@ -15,3 +15,6 @@ en:
       results:
         show: Show Results
         hide: Hide Results
+
+      close_poll: "Close Poll"
+      open_poll: "Open Poll"

plugins/poll/plugin.rb

@@ -58,11 +58,45 @@ after_initialize do
 
         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
@@ -147,4 +181,8 @@ register_css <<CSS
   border: none;
 }
 
+.poll-ui button i.fa {
+  margin-right: 2px;
+}
+
 CSS

plugins/poll/poll.rb

@@ -60,7 +60,7 @@ module ::PollPlugin
     end
 
     def is_closed?
-      @post.topic.closed? || (@post.topic.title =~ /^#{I18n.t('poll.closed_prefix')}/i) === 0
+      @post.topic.closed? || @post.topic.archived? || (!SiteSetting.allow_user_locale? && (@post.topic.title =~ /^#{I18n.t('poll.closed_prefix')}/i) === 0)
     end
 
     def options

plugins/poll/spec/poll_plugin/poll_controller_spec.rb

@@ -2,52 +2,89 @@ require 'spec_helper'
 
 describe PollPlugin::PollController, type: :controller do
   let(:topic) { create_topic(title: "Poll: Chitoge vs Onodera") }
-  let(:post) { create_post(topic: topic, raw: "Pick one.\n\n[poll]\n* Chitoge\n* Onodera\n[/poll]") }
+  let!(:post) { create_post(topic: topic, raw: "Pick one.\n\n[poll]\n* Chitoge\n* Onodera\n[/poll]") }
   let(:user1) { Fabricate(:user) }
   let(:user2) { Fabricate(:user) }
+  let(:admin) { Fabricate(:admin) }
 
-  it "should return 403 if no user is logged in" do
-    xhr :put, :vote, post_id: post.id, option: "Chitoge", use_route: :poll
-    response.should be_forbidden
+  describe 'vote' do
+    it "returns 403 if no user is logged in" do
+      xhr :put, :vote, post_id: post.id, option: "Chitoge", use_route: :poll
+      response.should be_forbidden
+    end
+
+    it "returns 400 if post_id or invalid option is not specified" do
+      log_in_user user1
+      xhr :put, :vote, use_route: :poll
+      response.status.should eq(400)
+      xhr :put, :vote, post_id: post.id, use_route: :poll
+      response.status.should eq(400)
+      xhr :put, :vote, option: "Chitoge", use_route: :poll
+      response.status.should eq(400)
+      xhr :put, :vote, post_id: post.id, option: "Tsugumi", use_route: :poll
+      response.status.should eq(400)
+    end
+
+    it "returns 400 if post_id doesn't correspond to a poll post" do
+      log_in_user user1
+      post2 = create_post(topic: topic, raw: "Generic reply")
+      xhr :put, :vote, post_id: post2.id, option: "Chitoge", use_route: :poll
+    end
+
+    it "saves votes correctly" do
+      MessageBus.expects(:publish).times(3)
+
+      log_in_user user1
+      xhr :put, :vote, post_id: post.id, option: "Chitoge", use_route: :poll
+      PollPlugin::Poll.new(post).get_vote(user1).should eq("Chitoge")
+
+      log_in_user user2
+      xhr :put, :vote, post_id: post.id, option: "Onodera", use_route: :poll
+      PollPlugin::Poll.new(post).get_vote(user2).should eq("Onodera")
+
+      PollPlugin::Poll.new(post).details["Chitoge"].should eq(1)
+      PollPlugin::Poll.new(post).details["Onodera"].should eq(1)
+
+      xhr :put, :vote, post_id: post.id, option: "Chitoge", use_route: :poll
+      PollPlugin::Poll.new(post).get_vote(user2).should eq("Chitoge")
+
+      PollPlugin::Poll.new(post).details["Chitoge"].should eq(2)
+      PollPlugin::Poll.new(post).details["Onodera"].should eq(0)
+    end
   end
 
-  it "should return 400 if post_id or invalid option is not specified" do
-    log_in_user user1
-    xhr :put, :vote, use_route: :poll
-    response.status.should eq(400)
-    xhr :put, :vote, post_id: post.id, use_route: :poll
-    response.status.should eq(400)
-    xhr :put, :vote, option: "Chitoge", use_route: :poll
-    response.status.should eq(400)
-    xhr :put, :vote, post_id: post.id, option: "Tsugumi", use_route: :poll
-    response.status.should eq(400)
-  end
+  describe 'toggle_close' do
+    it "returns 400 if post_id doesn't correspond to a poll post" do
+      log_in_user admin
+      post2 = create_post(topic: topic, raw: "Generic reply")
+      xhr :put, :toggle_close, post_id: post2.id, use_route: :poll
+      response.status.should eq(400)
+    end
 
-  it "should return 400 if post_id doesn't correspond to a poll post" do
-    log_in_user user1
-    post2 = create_post(topic: topic, raw: "Generic reply")
-    xhr :put, :vote, post_id: post2.id, option: "Chitoge", use_route: :poll
-    response.status.should eq(400)
-  end
+    it "returns 400 if the topic is locked" do
+      log_in_user admin
+      topic.update_attributes closed: true
+      xhr :put, :toggle_close, post_id: post.id, use_route: :poll
+      response.status.should eq(400)
+    end
 
-  it "should save votes correctly" do
-    MessageBus.expects(:publish).times(4)
+    it "raises Discourse::InvalidAccess is the user is not authorized" do
+      log_in_user user1
+      expect do
+        xhr :put, :toggle_close, post_id: post.id, use_route: :poll
+      end.to raise_error(Discourse::InvalidAccess)
+    end
 
-    log_in_user user1
-    xhr :put, :vote, post_id: post.id, option: "Chitoge", use_route: :poll
-    PollPlugin::Poll.new(post).get_vote(user1).should eq("Chitoge")
-
-    log_in_user user2
-    xhr :put, :vote, post_id: post.id, option: "Onodera", use_route: :poll
-    PollPlugin::Poll.new(post).get_vote(user2).should eq("Onodera")
-
-    PollPlugin::Poll.new(post).details["Chitoge"].should eq(1)
-    PollPlugin::Poll.new(post).details["Onodera"].should eq(1)
-
-    xhr :put, :vote, post_id: post.id, option: "Chitoge", use_route: :poll
-    PollPlugin::Poll.new(post).get_vote(user2).should eq("Chitoge")
-
-    PollPlugin::Poll.new(post).details["Chitoge"].should eq(2)
-    PollPlugin::Poll.new(post).details["Onodera"].should eq(0)
+    it "renames the topic" do
+      I18n.stubs(:t).with('poll.prefix').returns("Poll ")
+      I18n.stubs(:t).with('poll.closed_prefix').returns("Closed Poll ")
+      log_in_user admin
+      xhr :put, :toggle_close, post_id: post.id, use_route: :poll
+      response.status.should eq(200)
+      topic.reload.title.should == "Closed Poll : Chitoge vs Onodera"
+      xhr :put, :toggle_close, post_id: post.id, use_route: :poll
+      response.status.should eq(200)
+      topic.reload.title.should == "Poll : Chitoge vs Onodera"
+    end
   end
 end