notesnook/apps/mobile/app/hooks/use-actions.js

/*
This file is part of the Notesnook project (https://notesnook.com/)

Copyright (C) 2022 Streetwriters (Private) Limited

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

import Clipboard from "@react-native-clipboard/clipboard";
import React, { useEffect, useState } from "react";
import { Platform } from "react-native";
import Share from "react-native-share";
import { notesnook } from "../../e2e/test.ids";
import { db } from "../common/database";
import { presentDialog } from "../components/dialog/functions";
import NoteHistory from "../components/note-history";
import { MoveNotes } from "../components/sheets/move-notes/movenote";
import {
  eSendEvent,
  eSubscribeEvent,
  eUnSubscribeEvent,
  openVault,
  presentSheet,
  ToastEvent
} from "../services/event-manager";
import Navigation from "../services/navigation";
import Notifications from "../services/notifications";
import { useEditorStore } from "../stores/use-editor-store";
import { useMenuStore } from "../stores/use-menu-store";
import useNavigationStore from "../stores/use-navigation-store";
import { useSelectionStore } from "../stores/use-selection-store";
import { useTagStore } from "../stores/use-tag-store";
import { useThemeStore } from "../stores/use-theme-store";
import { useUserStore } from "../stores/use-user-store";
import { toTXT } from "../utils";
import { toggleDarkMode } from "../utils/color-scheme/utils";
import {
  eOpenAddNotebookDialog,
  eOpenAddTopicDialog,
  eOpenAttachmentsDialog,
  eOpenExportDialog,
  eOpenLoginDialog,
  eOpenMoveNoteDialog,
  eOpenPublishNoteDialog
} from "../utils/events";
import { deleteItems } from "../utils/functions";
import { sleep } from "../utils/time";
import { useCallback } from "react";

export const useActions = ({ close = () => null, item }) => {
  const colors = useThemeStore((state) => state.colors);
  const clearSelection = useSelectionStore((state) => state.clearSelection);
  const setSelectedItem = useSelectionStore((state) => state.setSelectedItem);
  const setMenuPins = useMenuStore((state) => state.setMenuPins);
  const [isPinnedToMenu, setIsPinnedToMenu] = useState(
    db.settings.isPinned(item.id)
  );
  const user = useUserStore((state) => state.user);
  const [notifPinned, setNotifPinned] = useState(null);
  const alias = item.alias || item.title;

  const isPublished =
    item.type === "note" && db.monographs.isPublished(item.id);

  useEffect(() => {
    if (item.id === null) return;
    checkNotifPinned();
    if (item.type !== "note") {
      setIsPinnedToMenu(db.settings.isPinned(item.id));
    }
  }, [checkNotifPinned, item]);

  const checkNotifPinned = useCallback(() => {
    let pinned = Notifications.getPinnedNotes();
    if (!pinned) {
      setNotifPinned(null);
      return;
    }

    let index = pinned.findIndex((notif) => notif.tag === item.id);
    if (index !== -1) {
      setNotifPinned(pinned[index]);
    } else {
      setNotifPinned(null);
    }
  }, [item.id]);

  const isNoteInTopic = () => {
    const currentScreen = useNavigationStore.getState().currentScreen;
    if (item.type !== "note" || currentScreen.name !== "TopicNotes") return;
    return db.notebooks
      .notebook(currentScreen.notebookId)
      .topics.topic(currentScreen.id)
      .has(item.id);
  };

  const onUpdate = useCallback(
    async (type) => {
      if (type === "unpin") {
        await sleep(1000);
        await Notifications.get();
        checkNotifPinned();
      }
    },
    [checkNotifPinned]
  );

  useEffect(() => {
    eSubscribeEvent("onUpdate", onUpdate);

    return () => {
      eUnSubscribeEvent("onUpdate", onUpdate);
    };
  }, [item, onUpdate]);

  function switchTheme() {
    toggleDarkMode();
  }

  function addTo() {
    close();
    clearSelection(true);
    setSelectedItem(item);
    setTimeout(() => {
      eSendEvent(eOpenMoveNoteDialog, item);
    }, 300);
  }

  async function addToFavorites() {
    if (!item.id) return;
    close();
    await db.notes.note(item.id).favorite();
    Navigation.queueRoutesForUpdate(
      "TaggedNotes",
      "ColoredNotes",
      "TopicNotes",
      "Favorites",
      "Notes"
    );
  }

  async function pinItem() {
    if (!item.id) return;
    close();
    let type = item.type;
    if (db[`${type}s`].pinned.length === 3 && !item.pinned) {
      ToastEvent.show({
        heading: `Cannot pin more than 3 ${type}s`,
        type: "error"
      });
      return;
    }
    await db[`${type}s`][type](item.id).pin();
    Navigation.queueRoutesForUpdate(
      "TaggedNotes",
      "ColoredNotes",
      "TopicNotes",
      "Favorites",
      "Notes",
      "Notebooks"
    );
  }

  async function pinToNotifications() {
    if (!checkNoteSynced()) return;
    if (Platform.OS === "ios") return;

    if (notifPinned !== null) {
      Notifications.remove(item.id, notifPinned.identifier);
      await sleep(1000);
      await Notifications.get();
      checkNotifPinned();
      return;
    }
    if (item.locked) return;
    let html = await db.notes.note(item.id).content();
    let text = await toTXT(item);
    Notifications.present({
      title: item.title,
      message: item.headline || text,
      subtitle: item.headline || text,
      bigText: html,
      ongoing: true,
      actions: ["UNPIN"],
      tag: item.id
    });
    await sleep(1000);
    await Notifications.get();
    checkNotifPinned();
  }

  async function restoreTrashItem() {
    if (!checkNoteSynced()) return;
    close();
    await db.trash.restore(item.id);
    Navigation.queueRoutesForUpdate(
      "TaggedNotes",
      "ColoredNotes",
      "TopicNotes",
      "Favorites",
      "Notes",
      "Notebooks",
      "Trash"
    );
    let type = item.type === "trash" ? item.itemType : item.type;
    ToastEvent.show({
      heading:
        type === "note"
          ? "Note restored from trash"
          : "Notebook restored from trash",
      type: "success"
    });
  }

  async function copyContent() {
    if (!checkNoteSynced()) return;
    if (item.locked) {
      close();
      await sleep(300);
      openVault({
        copyNote: true,
        novault: true,
        locked: true,
        item: item,
        title: "Copy note",
        description: "Unlock note to copy to clipboard."
      });
    } else {
      Clipboard.setString(await toTXT(item));
      ToastEvent.show({
        heading: "Note copied to clipboard",
        type: "success",
        context: "local"
      });
    }
  }

  async function publishNote() {
    if (!checkNoteSynced()) return;
    if (!user) {
      ToastEvent.show({
        heading: "Login required",
        message: "Login to publish note",
        context: "local",
        func: () => {
          eSendEvent(eOpenLoginDialog);
        },
        actionText: "Login"
      });
      return;
    }

    if (!user?.isEmailConfirmed) {
      ToastEvent.show({
        heading: "Email is not verified",
        message: "Please verify your email first.",
        context: "local"
      });
      return;
    }
    if (item.locked) {
      ToastEvent.show({
        heading: "Locked notes cannot be published",
        type: "error",
        context: "local"
      });
      return;
    }
    close();
    await sleep(300);
    eSendEvent(eOpenPublishNoteDialog, item);
  }

  const checkNoteSynced = () => {
    if (item.type !== "note" || item.itemType !== "note") return true;
    let isTrash = item.itemType === "note";
    if (!isTrash && !db.notes.note(item.id).synced()) {
      ToastEvent.show({
        context: "local",
        heading: "Note not synced",
        message: "Please run sync before making changes",
        type: "error"
      });
      return false;
    }

    if (isTrash && !db.trash.synced(item.id)) {
      ToastEvent.show({
        context: "local",
        heading: "Note not synced",
        message: "Please run sync before making changes",
        type: "error"
      });
      return false;
    }

    return true;
  };

  async function addToVault() {
    if (!item.id) return;
    if (!checkNoteSynced()) return;
    if (item.locked) {
      close();
      await sleep(300);
      openVault({
        item: item,
        novault: true,
        locked: true,
        permanant: true,
        title: "Unlock note",
        description: "Remove note from the vault."
      });
      return;
    }
    try {
      await db.vault.add(item.id);
      let note = db.notes.note(item.id).data;
      if (note.locked) {
        close();
        Navigation.queueRoutesForUpdate(
          "TaggedNotes",
          "ColoredNotes",
          "TopicNotes",
          "Favorites",
          "Notes"
        );
      }
    } catch (e) {
      close();
      await sleep(300);
      switch (e.message) {
        case db.vault.ERRORS.noVault:
          openVault({
            item: item,
            novault: false,
            title: "Create vault",
            description: "Set a password to create a vault and lock note."
          });
          break;
        case db.vault.ERRORS.vaultLocked:
          openVault({
            item: item,
            novault: true,
            locked: true,
            title: "Lock note",
            description: "Give access to vault to lock this note."
          });
          break;
      }
    }
  }

  async function createMenuShortcut() {
    close();
    try {
      if (isPinnedToMenu) {
        await db.settings.unpin(item.id);
      } else {
        if (item.type === "topic") {
          await db.settings.pin(item.type, {
            id: item.id,
            notebookId: item.notebookId
          });
        } else {
          await db.settings.pin(item.type, { id: item.id });
        }
      }
      setIsPinnedToMenu(db.settings.isPinned(item.id));
      setMenuPins();
    } catch (e) {
      console.error(e);
    }
  }

  async function renameTag() {
    close();
    await sleep(300);
    presentDialog({
      title: "Rename tag",
      paragraph: "Change the title of the tag " + alias,
      positivePress: async (value) => {
        if (!value || value === "" || value.trimStart().length == 0) return;
        await db.tags.rename(item.id, db.tags.sanitize(value));
        useTagStore.getState().setTags();
        useMenuStore.getState().setMenuPins();
        Navigation.queueRoutesForUpdate(
          "TaggedNotes",
          "ColoredNotes",
          "TopicNotes",
          "Favorites",
          "Notes",
          "Tags"
        );
      },
      input: true,
      defaultValue: alias,
      inputPlaceholder: "Enter title of tag",
      positiveText: "Save"
    });
  }

  async function shareNote() {
    if (!checkNoteSynced()) return;
    if (item.locked) {
      close();
      await sleep(300);
      openVault({
        item: item,
        novault: true,
        locked: true,
        share: true,
        title: "Share note",
        description: "Unlock note to share it."
      });
    } else {
      Share.open({
        title: "Share note to",
        failOnCancel: false,
        message: await toTXT(item)
      });
    }
  }

  async function deleteItem() {
    if (!checkNoteSynced()) return;
    close();
    if (item.type === "tag") {
      await sleep(300);
      presentDialog({
        title: "Delete tag",
        paragraph: "This tag will be removed from all notes.",
        positivePress: async () => {
          await db.tags.remove(item.id);
          useTagStore.getState().setTags();
          Navigation.queueRoutesForUpdate(
            "TaggedNotes",
            "ColoredNotes",
            "TopicNotes",
            "Favorites",
            "Notes",
            "Tags"
          );
        },
        positiveText: "Delete",
        positiveType: "errorShade"
      });
      return;
    }
    if (item.locked) {
      await sleep(300);
      openVault({
        deleteNote: true,
        novault: true,
        locked: true,
        item: item,
        title: "Delete note",
        description: "Unlock note to delete it."
      });
    } else {
      try {
        close();
        await deleteItems(item);
      } catch (e) {
        console.error(e);
      }
    }
  }
  async function removeNoteFromTopic() {
    const currentScreen = useNavigationStore.getState().currentScreen;
    if (currentScreen.name !== "TopicNotes") return;
    await db.notebooks
      .notebook(currentScreen.notebookId)
      .topics.topic(currentScreen.id)
      .delete(item.id);
    Navigation.queueRoutesForUpdate(
      "TaggedNotes",
      "ColoredNotes",
      "TopicNotes",
      "Favorites",
      "Notes",
      "Notebook",
      "Notebooks"
    );
    close();
  }

  async function deleteTrashItem() {
    if (!checkNoteSynced()) return;
    close();
    await sleep(300);
    presentDialog({
      title: "Permanent delete",
      paragraph: `Are you sure you want to delete this ${item.itemType} permanantly from trash?`,
      positiveText: "Delete",
      negativeText: "Cancel",
      positivePress: async () => {
        await db.trash.delete(item.id);
        Navigation.queueRoutesForUpdate("Trash");
        useSelectionStore.getState().setSelectionMode(false);
        ToastEvent.show({
          heading: "Permanantly deleted items",
          type: "success",
          context: "local"
        });
      },
      positiveType: "errorShade"
    });
  }

  async function openHistory() {
    close();
    await sleep(300);
    presentSheet({
      component: (ref) => <NoteHistory fwdRef={ref} note={item} />
    });
  }

  async function showAttachments() {
    close();
    await sleep(300);
    eSendEvent(eOpenAttachmentsDialog, item);
  }

  async function exportNote() {
    close();
    await sleep(300);
    eSendEvent(eOpenExportDialog, [item]);
  }

  async function toggleLocalOnly() {
    if (!checkNoteSynced() || !user) return;
    db.notes.note(item.id).localOnly();
    Navigation.queueRoutesForUpdate(
      "TaggedNotes",
      "ColoredNotes",
      "TopicNotes",
      "Favorites",
      "Notes"
    );
    close();
  }

  const toggleReadyOnlyMode = async () => {
    await db.notes.note(item.id).readonly();
    let current = db.notes.note(item.id).data.readonly;
    if (useEditorStore.getState().currentEditingNote === item.id) {
      useEditorStore.getState().setReadonly(current);
      //  tiny.call(EditorWebView, tiny.toogleReadMode(current ? 'readonly' : 'design'));
    }
    Navigation.queueRoutesForUpdate(
      "TaggedNotes",
      "ColoredNotes",
      "TopicNotes",
      "Favorites",
      "Notes"
    );
    close();
  };

  const duplicateNote = async () => {
    if (!checkNoteSynced()) return;
    await db.notes.note(item.id).duplicate();
    Navigation.queueRoutesForUpdate(
      "TaggedNotes",
      "ColoredNotes",
      "TopicNotes",
      "Favorites",
      "Notes"
    );
    close();
  };
  const actions = [
    {
      name: "Add to notebook",
      title: "Add to notebook",
      icon: "book-outline",
      func: addTo
    },
    {
      name: "Move notes",
      title: "Add notes",
      icon: "plus",
      func: async () => {
        close();
        await sleep(500);
        MoveNotes.present(db.notebooks.notebook(item.notebookId).data, item);
      }
    },

    {
      name: "Pin",
      title: item.pinned ? "Unpin" : "Pin to top",
      icon: item.pinned ? "pin-off-outline" : "pin-outline",
      func: pinItem,
      close: false,
      check: true,
      on: item.pinned,
      nopremium: true,
      id: notesnook.ids.dialogs.actionsheet.pin
    },
    {
      name: "Favorite",
      title: !item.favorite ? "Favorite" : "Unfavorite",
      icon: item.favorite ? "star-off" : "star-outline",
      func: addToFavorites,
      close: false,
      check: true,
      on: item.favorite,
      nopremium: true,
      id: notesnook.ids.dialogs.actionsheet.favorite,
      color: "orange"
    },
    {
      name: "PinToNotif",
      title:
        notifPinned !== null
          ? "Unpin from Notifications"
          : "Pin to Notifications",
      icon: "bell",
      on: notifPinned !== null,
      func: pinToNotifications
    },

    {
      name: "Edit Notebook",
      title: "Edit notebook",
      icon: "square-edit-outline",
      func: async () => {
        close();
        await sleep(300);
        eSendEvent(eOpenAddNotebookDialog, item);
      }
    },
    {
      name: "Edit Topic",
      title: "Edit topic",
      icon: "square-edit-outline",
      func: async () => {
        close();
        await sleep(300);
        eSendEvent(eOpenAddTopicDialog, {
          notebookId: item.notebookId,
          toEdit: item
        });
      }
    },
    {
      name: "Copy",
      title: "Copy",
      icon: "content-copy",
      func: copyContent
    },
    {
      name: "Restore",
      title: "Restore " + item.itemType,
      icon: "delete-restore",
      func: restoreTrashItem
    },

    {
      name: "Publish",
      title: isPublished ? "Published" : "Publish",
      icon: "cloud-upload-outline",
      on: isPublished,
      func: publishNote
    },
    {
      name: "Vault",
      title: item.locked ? "Remove from vault" : "Add to vault",
      icon: item.locked ? "shield-off-outline" : "shield-outline",
      func: addToVault,
      on: item.locked
    },

    {
      name: "Add Shortcut",
      title: isPinnedToMenu ? "Remove Shortcut" : "Add Shortcut",
      icon: isPinnedToMenu ? "link-variant-remove" : "link-variant",
      func: createMenuShortcut,
      close: false,
      check: true,
      on: isPinnedToMenu,
      nopremium: true,
      id: notesnook.ids.dialogs.actionsheet.pinMenu
    },
    {
      name: "Rename Tag",
      title: "Rename tag",
      icon: "square-edit-outline",
      func: renameTag
    },
    {
      name: "Share",
      title: "Share",
      icon: "share-variant",
      func: shareNote
    },
    {
      name: "Attachments",
      title: "Attachments",
      icon: "attachment",
      func: showAttachments
    },
    {
      name: "Export",
      title: "Export",
      icon: "export",
      func: exportNote
    },
    {
      name: "RemoveTopic",
      title: "Remove from topic",
      hidden: !isNoteInTopic(),
      icon: "minus-circle-outline",
      func: removeNoteFromTopic
    },
    {
      name: "Delete",
      title:
        item.type !== "notebook" && item.type !== "note"
          ? "Delete " + item.type
          : "Move to trash",
      icon: "delete-outline",
      type: "error",
      func: deleteItem
    },
    {
      name: "PermDelete",
      title: "Delete " + item.itemType,
      icon: "delete",
      func: deleteTrashItem
    },
    {
      name: "ReadOnly",
      title: "Read only",
      icon: "pencil-lock",
      func: toggleReadyOnlyMode,
      on: item.readonly
    },
    {
      name: "Local only",
      title: "Local only",
      icon: "sync-off",
      func: toggleLocalOnly,
      on: item.localOnly
    },

    {
      name: "History",
      title: "History",
      icon: "history",
      func: openHistory
    },
    {
      name: "Duplicate",
      title: "Duplicate",
      icon: "content-duplicate",
      func: duplicateNote
    },
    {
      name: "Dark Mode",
      title: "Dark mode",
      icon: "theme-light-dark",
      func: switchTheme,
      switch: true,
      on: colors.night ? true : false,
      close: false,
      nopremium: true,
      id: notesnook.ids.dialogs.actionsheet.night
    }
  ];

  return actions;
};