discourse/plugins/poll/assets/javascripts/initializers/add-poll-ui-builder.js.es6

import { withPluginApi } from "discourse/lib/plugin-api";
import computed from "ember-addons/ember-computed-decorators";
import showModal from "discourse/lib/show-modal";

function initializePollUIBuilder(api) {
  api.modifyClass("controller:composer", {
    @computed(
      "siteSettings.poll_enabled",
      "siteSettings.poll_minimum_trust_level_to_create",
      "model.topic.pm_with_non_human_user"
    )
    canBuildPoll(pollEnabled, minimumTrustLevel, pmWithNonHumanUser) {
      return (
        pollEnabled &&
        (pmWithNonHumanUser ||
          (this.currentUser &&
            (this.currentUser.staff ||
              this.currentUser.trust_level >= minimumTrustLevel)))
      );
    },

    actions: {
      showPollBuilder() {
        showModal("poll-ui-builder").set(
          "toolbarEvent",
          this.get("toolbarEvent")
        );
      }
    }
  });

  api.addToolbarPopupMenuOptionsCallback(function() {
    return {
      action: "showPollBuilder",
      icon: "bar-chart-o",
      label: "poll.ui_builder.title",
      condition: "canBuildPoll"
    };
  });
}

export default {
  name: "add-poll-ui-builder",

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