discourse-ai/assets/javascripts/discourse/components/ai-suggestion-dropdown.gjs

import Component from "@glimmer/component";
import { tracked } from "@glimmer/tracking";
import { fn } from "@ember/helper";
import { action } from "@ember/object";
import didInsert from "@ember/render-modifiers/modifiers/did-insert";
import { inject as service } from "@ember/service";
import DButton from "discourse/components/d-button";
import { ajax } from "discourse/lib/ajax";
import { popupAjaxError } from "discourse/lib/ajax-error";
import { bind } from "discourse-common/utils/decorators";
import I18n from "I18n";

export default class AISuggestionDropdown extends Component {
  @service dialog;
  @service siteSettings;
  @service composer;

  @tracked loading = false;
  @tracked showMenu = false;
  @tracked generatedSuggestions = [];
  @tracked suggestIcon = "discourse-sparkles";
  @tracked showErrors = false;
  @tracked error = "";

  SUGGESTION_TYPES = {
    title: "suggest_title",
    category: "suggest_category",
    tag: "suggest_tags",
  };

  willDestroy() {
    super.willDestroy(...arguments);
    document.removeEventListener("click", this.onClickOutside);
  }

  get showAIButton() {
    const minCharacterCount = 40;
    const isShowAIButton = this.composer.model.replyLength > minCharacterCount;
    const composerFields = document.querySelector(".composer-fields");

    if (composerFields) {
      if (isShowAIButton) {
        composerFields.classList.add("showing-ai-suggestions");
      } else {
        composerFields.classList.remove("showing-ai-suggestions");
      }
    }

    return isShowAIButton;
  }

  get disableSuggestionButton() {
    return this.loading;
  }

  @action
  applyClasses() {
    if (this.showAIButton) {
      document
        .querySelector(".composer-fields")
        ?.classList.add("showing-ai-suggestions");
    } else {
      document
        .querySelector(".composer-fields")
        ?.classList.remove("showing-ai-suggestions");
    }
  }

  @bind
  onClickOutside(event) {
    const menu = document.querySelector(".ai-title-suggestions-menu");

    if (event.target === menu) {
      return;
    }

    return this.#closeMenu();
  }

  @action
  handleClickOutside() {
    document.addEventListener("click", this.onClickOutside);
  }

  @action
  applySuggestion(suggestion) {
    if (!this.args.mode) {
      return;
    }

    const composer = this.args?.composer;
    if (!composer) {
      return;
    }

    if (this.args.mode === this.SUGGESTION_TYPES.title) {
      composer.set("title", suggestion);
      return this.#closeMenu();
    }

    if (this.args.mode === this.SUGGESTION_TYPES.category) {
      const selectedCategoryId = this.composer.categories.find(
        (c) => c.slug === suggestion
      ).id;
      composer.set("categoryId", selectedCategoryId);
      return this.#closeMenu();
    }

    if (this.args.mode === this.SUGGESTION_TYPES.tag) {
      this.#updateTags(suggestion, composer);
    }
  }

  @action
  async performSuggestion() {
    if (!this.args.mode) {
      return;
    }

    if (this.composer.model.replyLength === 0) {
      return this.dialog.alert(
        I18n.t("discourse_ai.ai_helper.missing_content")
      );
    }

    this.loading = true;
    this.suggestIcon = "spinner";

    return ajax(`/discourse-ai/ai-helper/${this.args.mode}`, {
      method: "POST",
      data: { text: this.composer.model.reply },
    })
      .then((data) => {
        this.#assignGeneratedSuggestions(data, this.args.mode);
      })
      .catch(popupAjaxError)
      .finally(() => {
        this.loading = false;
        this.suggestIcon = "sync-alt";
        this.showMenu = true;

        if (this.args.mode === "suggest_category") {
          document
            .querySelector(".category-input")
            ?.classList.add("showing-ai-suggestion-menu");
        }
      });
  }

  #closeMenu() {
    if (this.showMenu && this.args.mode === "suggest_category") {
      document
        .querySelector(".category-input")
        ?.classList.remove("showing-ai-suggestion-menu");
    }

    this.suggestIcon = "discourse-sparkles";
    this.showMenu = false;
    this.showErrors = false;
    this.errors = "";
  }

  #updateTags(suggestion, composer) {
    const maxTags = this.siteSettings.max_tags_per_topic;

    if (!composer.tags) {
      composer.set("tags", [suggestion]);
      // remove tag from the list of suggestions once added
      this.generatedSuggestions = this.generatedSuggestions.filter(
        (s) => s !== suggestion
      );
      return;
    }
    const tags = composer.tags;

    if (tags?.length >= maxTags) {
      // Show error if trying to add more tags than allowed
      this.showErrors = true;
      this.error = I18n.t("select_kit.max_content_reached", { count: maxTags });
      return;
    }

    tags.push(suggestion);
    composer.set("tags", [...tags]);
    // remove tag from the list of suggestions once added
    return (this.generatedSuggestions = this.generatedSuggestions.filter(
      (s) => s !== suggestion
    ));
  }

  #tagSelectorHasValues() {
    return this.args.composer?.tags && this.args.composer?.tags.length > 0;
  }

  #assignGeneratedSuggestions(data, mode) {
    if (mode === this.SUGGESTION_TYPES.title) {
      return (this.generatedSuggestions = data.suggestions);
    }

    const suggestions = data.assistant.map((s) => s.name);

    if (mode === this.SUGGESTION_TYPES.tag) {
      if (this.#tagSelectorHasValues()) {
        // Filter out tags if they are already selected in the tag input
        return (this.generatedSuggestions = suggestions.filter(
          (t) => !this.args.composer.tags.includes(t)
        ));
      } else {
        return (this.generatedSuggestions = suggestions);
      }
    }

    return (this.generatedSuggestions = suggestions);
  }

  <template>
    {{#if this.showAIButton}}
      <DButton
        @icon={{this.suggestIcon}}
        @title="discourse_ai.ai_helper.suggest"
        @action={{this.performSuggestion}}
        @disabled={{this.disableSuggestionButton}}
        class="suggestion-button {{if this.loading 'is-loading'}}"
        ...attributes
      />
    {{/if}}

    {{#if this.showMenu}}
      {{! template-lint-disable modifier-name-case }}
      <ul
        class="popup-menu ai-suggestions-menu"
        {{didInsert this.handleClickOutside}}
      >
        {{#if this.showErrors}}
          <li class="ai-suggestions-menu__errors">{{this.error}}</li>
        {{/if}}
        {{#each this.generatedSuggestions as |suggestion index|}}
          <li data-name={{suggestion}} data-value={{index}}>
            <DButton
              @translatedLabel={{suggestion}}
              @action={{fn this.applySuggestion suggestion}}
              class="popup-menu-btn"
            />
          </li>
        {{/each}}
      </ul>
    {{/if}}
  </template>
}