import { withPluginApi } from 'discourse/lib/plugin-api'; import { observes } from "ember-addons/ember-computed-decorators"; function createPollView(container, post, poll, vote, publicPoll) { const controller = container.lookup("controller:poll", { singleton: false }); const view = container.lookup("view:poll"); controller.setProperties({ model: poll, vote: vote, public: publicPoll, 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 @observes("polls") pollsChanged() { 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); } } }); 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; } const postPollViews = {}; $polls.each((idx, pollElem) => { const $div = $("
"); const $poll = $(pollElem); const pollName = $poll.data("poll-name"); const publicPoll = $poll.data("poll-public"); const pollId = `${pollName}-${post.id}`; const pollView = createPollView( helper.container, post, polls[pollName], votes[pollName], publicPoll ); $poll.replaceWith($div); Em.run.schedule('afterRender', () => pollView.renderer.replaceIn(pollView, $div[0])); postPollViews[pollId] = pollView; }); _pollViews = postPollViews; } api.includePostAttributes("polls", "polls_votes"); api.decorateCooked(createPollViews, { onlyStream: true }); api.cleanupStream(cleanUpPollViews); } export default { name: "extend-for-poll", initialize() { withPluginApi('0.1', initializePolls); } };