discourse-solved/assets/javascripts/discourse/initializers/extend-for-solved-button.js.es6

import Topic from "discourse/models/topic";
import User from "discourse/models/user";
import TopicStatus from "discourse/raw-views/topic-status";
import TopicStatusIcons from "discourse/helpers/topic-status-icons";
import { popupAjaxError } from "discourse/lib/ajax-error";
import { withPluginApi } from "discourse/lib/plugin-api";
import { ajax } from "discourse/lib/ajax";
import PostCooked from "discourse/widgets/post-cooked";
import { formatUsername } from "discourse/lib/utilities";
import { iconHTML } from "discourse-common/lib/icon-library";

function clearAccepted(topic) {
  const posts = topic.get("postStream.posts");
  posts.forEach(post => {
    if (post.get("post_number") > 1) {
      post.set("accepted_answer", false);
      post.set("can_accept_answer", true);
      post.set("can_unaccept_answer", false);
    }
  });
}

function unacceptPost(post) {
  if (!post.get("can_unaccept_answer")) {
    return;
  }
  const topic = post.topic;

  post.setProperties({
    can_accept_answer: true,
    can_unaccept_answer: false,
    accepted_answer: false
  });
  topic.set("accepted_answer", undefined);

  ajax("/solution/unaccept", {
    type: "POST",
    data: { id: post.get("id") }
  }).catch(popupAjaxError);
}

function acceptPost(post) {
  const topic = post.topic;

  clearAccepted(topic);

  post.setProperties({
    can_unaccept_answer: true,
    can_accept_answer: false,
    accepted_answer: true
  });

  topic.set("accepted_answer", {
    username: post.get("username"),
    post_number: post.get("post_number"),
    excerpt: post.get("cooked")
  });

  ajax("/solution/accept", {
    type: "POST",
    data: { id: post.get("id") }
  }).catch(popupAjaxError);
}

function initializeWithApi(api, container) {
  const currentUser = api.getCurrentUser();
  const { mobileView } = container.lookup("site:main");

  TopicStatusIcons.addObject([
    "has_accepted_answer",
    "far-check-square",
    "solved"
  ]);

  api.includePostAttributes(
    "can_accept_answer",
    "can_unaccept_answer",
    "accepted_answer"
  );

  if (api.addDiscoveryQueryParam) {
    api.addDiscoveryQueryParam("solved", { replace: true, refreshModel: true });
  }

  api.addPostMenuButton("solved", attrs => {
    const isOp = currentUser && currentUser.id === attrs.topicCreatedById;
    const position =
      !attrs.accepted_answer && attrs.can_accept_answer && !isOp
        ? "second-last-hidden"
        : "first";

    if (!mobileView && !attrs.accepted_answer && attrs.can_accept_answer) {
      return {
        action: "acceptAnswer",
        icon: "far-check-square",
        title: "solved.accept_answer",
        className: "unaccepted",
        position,
        afterButton(h) {
          return h("span.unaccepted", I18n.t("solved.mark_as_solution"));
        }
      };
    } else if (attrs.accepted_answer) {
      return {
        action: attrs.can_unaccept_answer ? "unacceptAnswer" : "",
        icon: "check-square",
        title: "solved.accept_answer",
        className: "accepted",
        position: "first",
        afterButton(h) {
          return h("span.accepted", I18n.t("solved.solution"));
        }
      };
    }
  });

  api.decorateWidget("post-contents:after-cooked", dec => {
    if (dec.attrs.post_number === 1) {
      const postModel = dec.getModel();
      if (postModel) {
        const topic = postModel.get("topic");
        if (topic.get("accepted_answer")) {
          const hasExcerpt = !!topic.get("accepted_answer").excerpt;

          const withExcerpt = `
            <aside class='quote accepted-answer' data-post="${
              topic.get("accepted_answer").post_number
            }" data-topic="${topic.get("id")}">
              <div class='title'>
                ${topic.get(
                  "acceptedAnswerHtml"
                )} <div class="quote-controls"><\/div>
              </div>
              <blockquote>
                ${topic.get("accepted_answer").excerpt}
              </blockquote>
            </aside>`;

          const withoutExcerpt = `
            <aside class='quote accepted-answer'>
              <div class='title title-only'>
                ${topic.get("acceptedAnswerHtml")}
              </div>
            </aside>`;

          var cooked = new PostCooked({
            cooked: hasExcerpt ? withExcerpt : withoutExcerpt
          });

          var html = cooked.init();

          return dec.rawHtml(html);
        }
      }
    }
  });

  if (mobileView) {
    api.decorateWidget("post-contents:after-cooked", dec => {
      const model = dec.getModel();
      const result = [];
      if (!model.accepted_answer && model.can_accept_answer) {
        result.push(
          dec.attach("button", {
            label: "solved.mark_as_solution",
            title: "solved.mark_as_solution",
            icon: "far-check-square",
            action: "acceptAnswer",
            className: "solve"
          })
        );
      }
      return dec.h("div.solved-container", result);
    });
  }

  api.attachWidgetAction("post", "acceptAnswer", function() {
    const post = this.model;
    const current = post.get("topic.postStream.posts").filter(p => {
      return p.get("post_number") === 1 || p.get("accepted_answer");
    });
    acceptPost(post);

    current.forEach(p =>
      this.appEvents.trigger("post-stream:refresh", { id: p.id })
    );
  });

  api.attachWidgetAction("post", "unacceptAnswer", function() {
    const post = this.model;
    const op = post
      .get("topic.postStream.posts")
      .find(p => p.get("post_number") === 1);
    unacceptPost(post);
    this.appEvents.trigger("post-stream:refresh", { id: op.get("id") });
  });

  if (api.registerConnectorClass) {
    api.registerConnectorClass("user-activity-bottom", "solved-list", {
      shouldRender(args, component) {
        return component.siteSettings.solved_enabled;
      }
    });
    api.registerConnectorClass("user-summary-stat", "solved-count", {
      shouldRender(args, component) {
        return (
          component.siteSettings.solved_enabled && args.model.solved_count > 0
        );
      },
      setupComponent() {
        this.set("classNames", ["linked-stat"]);
      }
    });
  }
}

export default {
  name: "extend-for-solved-button",
  initialize(container) {
    Topic.reopen({
      // keeping this here cause there is complex localization
      acceptedAnswerHtml: function() {
        const username = this.get("accepted_answer.username");
        const postNumber = this.get("accepted_answer.post_number");

        if (!username || !postNumber) {
          return "";
        }

        return I18n.t("solved.accepted_html", {
          icon: iconHTML("check-square", { class: "accepted" }),
          username_lower: username.toLowerCase(),
          username: formatUsername(username),
          post_path: this.get("url") + "/" + postNumber,
          post_number: postNumber,
          user_path: User.create({ username }).get("path")
        });
      }.property("accepted_answer", "id")
    });

    TopicStatus.reopen({
      statuses: function() {
        const results = this._super();
        if (this.topic.has_accepted_answer) {
          results.push({
            openTag: "span",
            closeTag: "span",
            title: I18n.t("topic_statuses.solved.help"),
            icon: "far-check-square"
          });
        } else if (
          this.topic.can_have_answer &&
          this.siteSettings.solved_enabled &&
          this.siteSettings.empty_box_on_unsolved
        ) {
          results.push({
            openTag: "span",
            closeTag: "span",
            title: I18n.t("solved.has_no_accepted_answer"),
            icon: "square-o"
          });
        }
        return results;
      }.property()
    });

    withPluginApi("0.1", api => initializeWithApi(api, container));

    withPluginApi("0.8.10", api => {
      api.replaceIcon(
        "notification.solved.accepted_notification",
        "far-check-square"
      );
    });
  }
};