FIX: Poll requires logged in user to log in again

FIX: Open/Closing the poll wasn't updating the UI
REFACTOR: ES6 FTW

plugins/poll/assets/javascripts/controllers/poll.js.es6

@@ -4,42 +4,44 @@ export default DiscourseController.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');
-  }.property('poll.post.topic.details.can_edit'),
+  showToggleClosePoll: Em.computed.alias('poll.post.topic.details.can_edit'),
 
   actions: {
-    selectOption: function(option) {
+    selectOption(option) {
       if (this.get('disableRadio')) {
         return;
       }
 
-      if (!this.get('currentUser.id')) {
+      if (!this.get('postController.currentUser.id')) {
         this.get('postController').send('showLogin');
         return;
       }
 
       this.set('loading', true);
+
+      const self = this;
       this.get('poll').saveVote(option).then(function() {
-        this.set('loading', false);
-        this.set('showResults', true);
-      }.bind(this));
+        self.setProperties({ loading: false, showResults: true});
+      });
     },
 
-    toggleShowResults: function() {
-      this.set('showResults', !this.get('showResults'));
+    toggleShowResults() {
+      this.toggleProperty('showResults');
     },
 
-    toggleClosePoll: function() {
+    toggleClosePoll() {
+      const self = this;
+
       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));
+
+      return Discourse.ajax('/poll/toggle_close', {
+        type: 'PUT',
+        data: { post_id: this.get('poll.post.id') }
+      }).then(function(result) {
+        self.set('poll.post.topic.title', result.basic_topic.title);
+        self.set('poll.post.topic.fancy_title', result.basic_topic.title);
+        self.set('loading', false);
+      });
     }
   }
 });

plugins/poll/assets/javascripts/discourse/templates/poll.hbs

@@ -1,12 +1,12 @@
 <table>
   {{#each po in poll.options}}
-    <tr {{bind-attr class=po.checked:active}} {{action "selectOption" po.option}}>
+    <tr {{bind-attr class="po.checked:active"}} {{action "selectOption" po.option}}>
       <td class="radio">
-        <input type="radio" name="poll" {{bind-attr checked=po.checked disabled=controller.disableRadio}}>
+        <input type="radio" name="poll" {{bind-attr checked="po.checked" disabled="disableRadio"}}>
       </td>
       <td class="option">
         <div class="option">{{{po.option}}}</div>
-        {{#if controller.showResults}}
+        {{#if showResults}}
           <div class="result">{{i18n 'poll.voteCount' count=po.votes}}</div>
         {{/if}}
       </td>
@@ -15,27 +15,23 @@
 </table>
 
 <div class='row'>
-<button {{action "toggleShowResults"}} class="btn btn-small show-results">
-  {{#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 showToggleClosePoll}}
-  <button {{action "toggleClosePoll"}} class="btn btn-small toggle-poll">
-    {{#if poll.closed}}
-      <i class="fa fa-unlock-alt"></i>
-      {{i18n 'poll.open_poll'}}
+  <button {{action "toggleShowResults"}} class="btn btn-small show-results">
+    {{#if showResults}}
+      {{fa-icon "eye-slash"}} {{i18n 'poll.results.hide'}}
     {{else}}
-      <i class="fa fa-lock"></i>
-      {{i18n 'poll.close_poll'}}
+      {{fa-icon "eye"}} {{i18n 'poll.results.show'}}
     {{/if}}
   </button>
-{{/if}}
+
+  {{#if showToggleClosePoll}}
+    <button {{action "toggleClosePoll"}} class="btn btn-small toggle-poll">
+      {{#if poll.closed}}
+        {{fa-icon "unlock-alt"}} {{i18n 'poll.open_poll'}}
+      {{else}}
+        {{fa-icon "lock"}} {{i18n 'poll.close_poll'}}
+      {{/if}}
+    </button>
+  {{/if}}
 </div>
 
 {{loading-spinner condition=loading}}

plugins/poll/assets/javascripts/initializers/poll.js.es6

@@ -1,65 +1,17 @@
+import Poll from "discourse/plugins/poll/models/poll";
+import PollView from "discourse/plugins/poll/views/poll";
 import PollController from "discourse/plugins/poll/controllers/poll";
+
 import PostView from "discourse/views/post";
 
-var Poll = Discourse.Model.extend({
-  post: null,
-  options: [],
-  closed: false,
-
-  postObserver: function() {
-    this.updateFromJson(this.get('post.poll_details'));
-  }.observes('post.poll_details'),
-
-  fetchNewPostDetails: Discourse.debounce(function() {
-    var self = this;
-    Discourse.debounce(function() {
-      self.get('post.topic.postStream').triggerChangedPost(self.get('post.id'), self.get('post.topic.updated_at'));
-    }, 500);
-  }).observes('post.topic.title'),
-
-  updateFromJson: function(json) {
-    var selectedOption = json["selected"];
-
-    var options = [];
-    Object.keys(json["options"]).forEach(function(option) {
-      options.push(Ember.Object.create({
-        option: option,
-        votes: json["options"][option],
-        checked: (option === selectedOption)
-      }));
-    });
-
-    this.set('options', options);
-    this.set('closed', json.closed);
-  },
-
-  saveVote: function(option) {
-    this.get('options').forEach(function(opt) {
-      opt.set('checked', opt.get('option') === option);
-    });
-
-    return Discourse.ajax("/poll", {
-      type: "PUT",
-      data: {post_id: this.get('post.id'), option: option}
-    }).then(function(newJSON) {
-      this.updateFromJson(newJSON);
-    }.bind(this));
-  }
-});
-
-var PollView = Ember.View.extend({
-  templateName: "poll",
-  classNames: ['poll-ui'],
-});
-
 function initializePollView(self) {
-  var post = self.get('post');
-  var pollDetails = post.get('poll_details');
+  const post = self.get('post'),
+        pollDetails = post.get('poll_details');
 
-  var poll = Poll.create({post: post});
+  let poll = Poll.create({ post: post });
   poll.updateFromJson(pollDetails);
 
-  var pollController = PollController.create({
+  const pollController = PollController.create({
     poll: poll,
     showResults: pollDetails["selected"],
     postController: self.get('controller')
@@ -74,29 +26,25 @@ export default {
   initialize: function() {
     PostView.reopen({
       createPollUI: function($post) {
-        var post = this.get('post');
-
-        if (!post.get('poll_details')) {
+        if (!this.get('post').get('poll_details')) {
           return;
         }
 
-        var view = initializePollView(this);
-        var pollContainer = $post.find(".poll-ui:first");
+        let view = initializePollView(this),
+            pollContainer = $post.find(".poll-ui:first");
+
         if (pollContainer.length === 0) {
           pollContainer = $post.find("ul:first");
         }
 
-        var $div = $('<div>');
+        let $div = $('<div>');
         pollContainer.replaceWith($div);
         view.constructor.renderer.appendTo(view, $div[0]);
         this.set('pollView', view);
-
       }.on('postViewInserted'),
 
       clearPollView: function() {
-        if (this.get('pollView')) {
-          this.get('pollView').destroy();
-        }
+        if (this.get('pollView')) { this.get('pollView').destroy(); }
       }.on('willClearRender')
     });
   }

plugins/poll/assets/javascripts/models/poll.js.es6

@@ -0,0 +1,42 @@
+export default Discourse.Model.extend({
+  post: null,
+  options: [],
+  closed: false,
+
+  postObserver: function() {
+    this.updateFromJson(this.get('post.poll_details'));
+  }.observes('post.poll_details'),
+
+  fetchNewPostDetails: Discourse.debounce(function() {
+    this.get('post.topic.postStream').triggerChangedPost(this.get('post.id'), this.get('post.topic.updated_at'));
+  }, 250).observes('post.topic.title'),
+
+  updateFromJson(json) {
+    const selectedOption = json["selected"];
+    let options = [];
+
+    Object.keys(json["options"]).forEach(function(option) {
+      options.push(Ember.Object.create({
+        option: option,
+        votes: json["options"][option],
+        checked: option === selectedOption
+      }));
+    });
+
+    this.setProperties({ options: options, closed: json.closed });
+  },
+
+  saveVote(option) {
+    this.get('options').forEach(function(opt) {
+      opt.set('checked', opt.get('option') === option);
+    });
+
+    const self = this;
+    return Discourse.ajax("/poll", {
+      type: "PUT",
+      data: { post_id: this.get('post.id'), option: option }
+    }).then(function(newJSON) {
+      self.updateFromJson(newJSON);
+    });
+  }
+});

plugins/poll/assets/javascripts/poll_bbcode.js

@@ -4,6 +4,6 @@ Discourse.Dialect.inlineBetween({
   rawContents: true,
   emitter: function(contents) {
     var list = Discourse.Dialect.cook(contents, {});
-    return ['div', {class: 'poll-ui'}, list];
+    return ['div', { class: 'poll-ui' }, list];
   }
 });

plugins/poll/assets/javascripts/views/poll.js.es6

@@ -0,0 +1,4 @@
+export default Ember.View.extend({
+  templateName: "poll",
+  classNames: ['poll-ui'],
+});

plugins/poll/plugin.rb

@@ -145,7 +145,9 @@ after_initialize do
 end
 
 # Poll UI.
+register_asset "javascripts/models/poll.js.es6"
 register_asset "javascripts/controllers/poll.js.es6"
+register_asset "javascripts/views/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