notesnook/packages/core/models/topic.js

import { qclone } from "qclone";
import sort from "fast-sort";

export default class Topic {
  /**
   * @param {Object} topic
   * @param {string} notebookId
   * @param {import('../api').default} db
   */
  constructor(topic, notebookId, db) {
    this._topic = topic;
    this._db = db;
    this._notebookId = notebookId;
  }

  get totalNotes() {
    return this._topic.notes.length;
  }

  has(noteId) {
    return this._topic.notes.findIndex((n) => n === noteId) > -1;
  }

  async add(...noteIds) {
    const topic = qclone(this._topic);
    for (let noteId of noteIds) {
      let note = this._db.notes.note(noteId);
      if (this.has(noteId) || !note || note.data.deleted) continue;

      let array = note.notebooks || [];
      const notebookIndex = array.findIndex((nb) => nb.id === this._notebookId);
      if (notebookIndex === -1) {
        let notebook = {};
        notebook.id = this._notebookId;
        notebook.topics = [topic.id];
        array.push(notebook);
      } else {
        const topicIndex = array[notebookIndex].topics.indexOf(topic.id);
        if (topicIndex > -1) return;
        array[notebookIndex].topics.push(topic.id);
      }

      await this._db.notes.add({
        id: noteId,
        notebooks: array,
      });

      topic.notes.push(noteId);
    }
    return await this._save(topic);
  }

  async delete(...noteIds) {
    const topic = qclone(this._topic);
    for (let noteId of noteIds) {
      let note = this._db.notes.note(noteId);

      if (!this.has(noteId) || !note || note.data.deleted) return this;

      let index = topic.notes.indexOf(noteId);
      topic.notes.splice(index, 1);

      let array = note.notebooks || [];
      const notebookIndex = array.findIndex((nb) => nb.id === this._notebookId);
      if (notebookIndex === -1) break;

      const topicIndex = array[notebookIndex].topics.indexOf(topic.id);
      if (topicIndex === -1) break;

      array[notebookIndex].topics.splice(topicIndex, 1);
      if (array[notebookIndex].topics.length <= 0)
        array.splice(notebookIndex, 1);

      await this._db.notes.add({
        id: noteId,
        notebooks: array,
      });
    }
    return await this._save(topic);
  }

  async _save(topic) {
    await this._db.notebooks.notebook(this._notebookId).topics.add(topic);
    return this;
  }

  get all() {
    const notes = this._topic.notes.reduce((arr, noteId) => {
      let note = this._db.notes.note(noteId);
      if (note) arr.push(note.data);
      return arr;
    }, []);
    return sort(notes).desc((note) => note.dateCreated);
  }
}