notesnook/packages/core/models/note.js

import MarkdownBuilder from "../utils/templates/markdown/builder";
import HTMLBuilder from "../utils/templates/html/builder";
import TextBuilder from "../utils/templates/text/builder";
import { getContentFromData } from "../content-types";
import { CHECK_IDS, sendCheckUserStatusEvent } from "../common";
import { addItem, deleteItem } from "../utils/array";

export default class Note {
  /**
   *
   * @param {import('../api').default} db
   * @param {Object} note
   */
  constructor(note, db) {
    this._note = note;
    this._db = db;
  }

  get data() {
    return this._note;
  }

  get headline() {
    return this._note.headline;
  }

  get title() {
    return this._note.title;
  }

  get tags() {
    return this._note.tags;
  }

  get colors() {
    return this._note.colors;
  }

  get id() {
    return this._note.id;
  }

  get notebooks() {
    return this._note.notebooks;
  }

  get deleted() {
    return this._note.deleted;
  }

  get dateEdited() {
    return this._note.dateEdited;
  }

  /**
   *
   * @param {"html"|"md"|"txt"} format - Format to export into
   * @param {string?} rawContent - Use this raw content instead of generating itself
   */
  async export(to = "html", rawContent) {
    const templateData = {
      metadata: this.data,
      title: this.title,
      editedOn: this.dateEdited,
      headline: this.headline,
      createdOn: this.data.dateCreated,
    };
    const { data, type } = await this._db.content.raw(this._note.contentId);
    let content = getContentFromData(type, data);

    if (to !== "txt" && !(await sendCheckUserStatusEvent(CHECK_IDS.noteExport)))
      return;

    switch (to) {
      case "html":
        templateData.content = rawContent || content.toHTML();
        return HTMLBuilder.buildHTML(templateData);
      case "txt":
        templateData.content = rawContent || content.toTXT();
        return TextBuilder.buildText(templateData);
      case "md":
        templateData.content = rawContent || content.toMD();
        return MarkdownBuilder.buildMarkdown(templateData);
      default:
        throw new Error("Export format not supported.");
    }
  }

  async content(withAttachments = false) {
    const content = await this._db.content.raw(
      this._note.contentId,
      withAttachments
    );
    return content.data;
  }

  async color(color) {
    if (!(await sendCheckUserStatusEvent(CHECK_IDS.noteColor))) return;
    await this.uncolor();
    let tag = await this._db.colors.add(color, this._note.id);
    await this._db.notes._collection.addItem({
      ...this._note,
      color: tag.title,
    });
  }

  async uncolor() {
    if (!this._note.color) return;
    await this._db.colors.untag(this._note.color, this._note.id);
    await this._db.notes._collection.addItem({
      ...this._note,
      color: undefined,
    });
  }

  async tag(tag) {
    if (
      this._db.tags.all.length >= 5 &&
      !(await sendCheckUserStatusEvent(CHECK_IDS.noteTag))
    )
      return;

    let tagItem = await this._db.tags.add(tag, this._note.id);
    if (addItem(this._note.tags, tagItem.title))
      await this._db.notes._collection.addItem(this._note);
  }

  async untag(tag) {
    if (deleteItem(this._note.tags, tag)) {
      await this._db.notes._collection.addItem(this._note);
    } else console.error("This note is not tagged by the specified tag.", tag);
    await this._db.tags.untag(tag, this._note.id);
  }

  _toggle(prop) {
    return this._db.notes.add({ id: this._note.id, [prop]: !this._note[prop] });
  }

  favorite() {
    return this._toggle("favorite");
  }

  pin() {
    return this._toggle("pinned");
  }
}