discourse/plugins/poll/assets/javascripts/initializers/extend-for-poll.js.es6

118 lines
2.7 KiB
JavaScript

import { withPluginApi } from "discourse/lib/plugin-api";
import { observes } from "ember-addons/ember-computed-decorators";
import { getRegister } from "discourse-common/lib/get-owner";
import WidgetGlue from "discourse/widgets/glue";
function initializePolls(api) {
const register = getRegister(api);
api.modifyClass("controller:topic", {
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();
}
});
let _glued = [];
let _interval = null;
function rerender() {
_glued.forEach(g => g.queueRerender());
}
api.modifyClass("model:post", {
_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 || {};
polls.forEach(p => {
const existing = this._polls[p.name];
if (existing) {
this._polls[p.name].setProperties(p);
} else {
this._polls[p.name] = Em.Object.create(p);
}
});
this.set("pollsObject", this._polls);
rerender();
}
}
});
function attachPolls($elem, helper) {
const $polls = $(".poll", $elem);
if (!$polls.length) {
return;
}
if (!helper) {
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;
}
_interval = _interval || setInterval(rerender, 30000);
$polls.each((idx, pollElem) => {
const $poll = $(pollElem);
const pollName = $poll.data("poll-name");
const poll = polls[pollName];
if (poll) {
const glue = new WidgetGlue("discourse-poll", register, {
id: `${pollName}-${post.id}`,
post,
poll,
vote: votes[pollName] || []
});
glue.appendTo(pollElem);
_glued.push(glue);
}
});
}
function cleanUpPolls() {
if (_interval) {
clearInterval(_interval);
_interval = null;
}
_glued.forEach(g => g.cleanUp());
_glued = [];
}
api.includePostAttributes("polls", "polls_votes");
api.decorateCooked(attachPolls, { onlyStream: true });
api.cleanupStream(cleanUpPolls);
}
export default {
name: "extend-for-poll",
initialize() {
withPluginApi("0.8.7", initializePolls);
}
};