discourse-table-builder/javascripts/discourse/components/spreadsheet-editor.js

import { action } from "@ember/object";
import loadScript from "discourse/lib/load-script";
import { arrayToTable, findTableRegex, tableToObj } from "../lib/utilities";
import Component from "@ember/component";
import { ajax } from "discourse/lib/ajax";
import { popupAjaxError } from "discourse/lib/ajax-error";
import I18n from "I18n";
import { schedule } from "@ember/runloop";

export default Component.extend({
  tagName: "",
  showEditReason: false,
  spreadsheet: null,

  // Lifecycle Methods:
  didInsertElement() {
    this._super(...arguments);

    schedule("afterRender", () => {
      this.loadLibraries().then(() => {
        if (this.isEditingTable) {
          this.buildPopulatedTable(this.tableHtml);
        } else {
          this.buildNewTable();
        }
      });
    });
  },

  willDestroyElement() {
    this._super(...arguments);
    this.spreadsheet?.destroy();
  },

  // Getters:
  get isEditingTable() {
    if (this.tableHtml) {
      return true;
    }

    return false;
  },

  get modalAttributes() {
    if (this.isEditingTable) {
      return {
        title: themePrefix("discourse_table_builder.edit.modal.title"),
        insertTable: {
          title: themePrefix("discourse_table_builder.edit.modal.create"),
          icon: "pencil-alt",
        },
      };
    } else {
      return {
        title: themePrefix("discourse_table_builder.modal.title"),
        insertTable: {
          title: themePrefix("discourse_table_builder.modal.create"),
          icon: "plus",
        },
      };
    }
  },

  // Actions:
  @action
  showEditReasonField() {
    if (this.showEditReason) {
      return this.set("showEditReason", false);
    } else {
      return this.set("showEditReason", true);
    }
  },

  @action
  cancelTableInsertion() {
    this.triggerModalClose();
  },

  @action
  insertTable() {
    const updatedHeaders = this.spreadsheet.getHeaders().split(","); // keys
    const updatedData = this.spreadsheet.getData(); // values
    const markdownTable = this.buildTableMarkdown(updatedHeaders, updatedData);

    if (!this.isEditingTable) {
      this.toolbarEvent.addText(markdownTable);
      return this.triggerModalClose();
    } else {
      return this.updateTable(markdownTable);
    }
  },

  // Helper Methods:
  loadLibraries() {
    return loadScript(settings.theme_uploads.jsuites).then(() => {
      return loadScript(settings.theme_uploads.jspreadsheet);
    });
  },

  buildNewTable() {
    const data = [
      ["", "", ""],
      ["", "", ""],
      ["", "", ""],
    ];

    const columns = [
      { title: "Column 1", width: 150 },
      { title: "Column 2", width: 150 },
      { title: "Column 3", width: 150 },
    ];

    return this.buildSpreadsheet(data, columns);
  },

  buildPopulatedTable(table) {
    const tableObject = tableToObj(table);
    const headings = [];
    const tableData = [];

    tableObject.forEach((object) => {
      // Build Headings
      if (!headings.includes(...Object.keys(object))) {
        headings.push(...Object.keys(object));
      }

      // Build Table Data
      tableData.push([...Object.values(object)]);
    });

    const columns = headings.map((heading) => {
      return {
        title: heading,
        width: heading.length * 15,
      };
    });

    return this.buildSpreadsheet(tableData, columns);
  },

  buildSpreadsheet(data, columns, opts = {}) {
    const spreadsheetContainer = document.querySelector("#spreadsheet");

    // eslint-disable-next-line no-undef
    this.spreadsheet = jspreadsheet(spreadsheetContainer, {
      data,
      columns,
      ...opts,
    });
  },

  buildUpdatedPost(tableId, raw, newRaw) {
    const tableToEdit = raw.match(findTableRegex());
    let editedTable;

    if (tableToEdit.length > 1) {
      editedTable = raw.replace(tableToEdit[tableId], newRaw);
    } else {
      editedTable = raw.replace(findTableRegex(), newRaw);
    }

    return editedTable;
  },

  updateTable(markdownTable) {
    const tableId = this.get("tableId");
    const postId = this.model.id;
    const newRaw = markdownTable;
    const editReason =
      this.get("editReason") ||
      I18n.t(themePrefix("discourse_table_builder.edit.default_edit_reason"));
    const raw = this.model.raw;
    const newPostRaw = this.buildUpdatedPost(tableId, raw, newRaw);

    return this.sendTableUpdate(postId, newPostRaw, editReason);
  },

  sendTableUpdate(postId, raw, edit_reason) {
    return ajax(`/posts/${postId}.json`, {
      type: "PUT",
      data: {
        post: {
          raw,
          edit_reason,
        },
      },
    })
      .catch(popupAjaxError)
      .finally(() => {
        this.triggerModalClose();
      });
  },

  buildTableMarkdown(headers, data) {
    const table = [];
    data.forEach((row) => {
      const result = {};

      headers.forEach((key, index) => (result[key] = row[index]));
      table.push(result);
    });

    return arrayToTable(table);
  },
});