import { withPluginApi } from 'discourse/lib/plugin-api'; function createPollView(container, post, poll, vote) { const controller = container.lookup("controller:poll", { singleton: false }); const view = container.lookup("view:poll"); controller.set("vote", vote); controller.setProperties({ model: poll, post }); view.set("controller", controller); return view; } let _pollViews; function initializePolls(api) { const TopicController = api.container.lookupFactory('controller:topic'); TopicController.reopen({ subscribe(){ this._super(); this.messageBus.subscribe("/polls/" + this.get("model.id"), msg => { const post = this.get('model.postStream').findLoadedPost(msg.post_id); if (post) { post.set('polls', msg.polls); } }); }, unsubscribe(){ this.messageBus.unsubscribe('/polls/*'); this._super(); } }); const Post = api.container.lookupFactory('model:post'); Post.reopen({ _polls: null, pollsObject: null, // we need a proper ember object so it is bindable pollsChanged: function(){ const polls = this.get("polls"); if (polls) { this._polls = this._polls || {}; _.map(polls, (v,k) => { const existing = this._polls[k]; if (existing) { this._polls[k].setProperties(v); } else { this._polls[k] = Em.Object.create(v); } }); this.set("pollsObject", this._polls); } }.observes("polls") }); function cleanUpPollViews() { if (_pollViews) { Object.keys(_pollViews).forEach(pollName => _pollViews[pollName].destroy()); } _pollViews = null; } function createPollViews($elem, helper) { const $polls = $('.poll', $elem); if (!$polls.length) { return; } const post = helper.getModel(); api.preventCloak(post.id); const votes = post.get('polls_votes') || {}; post.pollsChanged(); const polls = post.get("pollsObject"); if (!polls) { return; } cleanUpPollViews(); const postPollViews = {}; $polls.each((idx, pollElem) => { const $div = $("