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

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(...arguments);
      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(...arguments);
    }
  });

  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.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] = Ember.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, id: "discourse-poll" });
  api.cleanupStream(cleanUpPolls);
}

export default {
  name: "extend-for-poll",

  initialize() {
    withPluginApi("0.8.7", initializePolls);
  }
};