weseek__growi/packages/app/src/client/services/PageContainer.js

import { Container } from 'unstated';


import * as entities from 'entities';
import * as toastr from 'toastr';
import { pagePathUtils } from '@growi/core';
import loggerFactory from '~/utils/logger';
import { toastError } from '../util/apiNotification';

import {
  DetachCodeBlockInterceptor,
  RestoreCodeBlockInterceptor,
} from '../util/interceptor/detach-code-blocks';

import {
  DrawioInterceptor,
} from '../util/interceptor/drawio-interceptor';

const { isTrashPage } = pagePathUtils;

const logger = loggerFactory('growi:services:PageContainer');

/**
 * Service container related to Page
 * @extends {Container} unstated Container
 */
export default class PageContainer extends Container {

  constructor(appContainer) {
    super();

    this.appContainer = appContainer;
    this.appContainer.registerContainer(this);

    this.state = {};

    const mainContent = document.querySelector('#content-main');
    if (mainContent == null) {
      logger.debug('#content-main element is not exists');
      return;
    }

    const revisionId = mainContent.getAttribute('data-page-revision-id');
    const path = decodeURI(mainContent.getAttribute('data-path'));

    this.state = {
      // local page data
      markdown: null, // will be initialized after initStateMarkdown()
      pageId: mainContent.getAttribute('data-page-id'),
      revisionId,
      revisionCreatedAt: +mainContent.getAttribute('data-page-revision-created'),
      path,
      tocHtml: '',

      isBookmarked: false,
      sumOfBookmarks: 0,

      seenUsers: [],
      seenUserIds: [],
      sumOfSeenUsers: [],

      isLiked: false,
      likers: [],
      likerIds: [],
      sumOfLikers: 0,

      createdAt: mainContent.getAttribute('data-page-created-at'),
      updatedAt: mainContent.getAttribute('data-page-updated-at'),
      deletedAt: mainContent.getAttribute('data-page-deleted-at') || null,

      isUserPage: JSON.parse(mainContent.getAttribute('data-page-user')) != null,
      isTrashPage: isTrashPage(path),
      isDeleted: JSON.parse(mainContent.getAttribute('data-page-is-deleted')),
      isDeletable: JSON.parse(mainContent.getAttribute('data-page-is-deletable')),
      isNotCreatable: JSON.parse(mainContent.getAttribute('data-page-is-not-creatable')),
      isAbleToDeleteCompletely: JSON.parse(mainContent.getAttribute('data-page-is-able-to-delete-completely')),
      isPageExist: mainContent.getAttribute('data-page-id') != null,

      pageUser: JSON.parse(mainContent.getAttribute('data-page-user')),
      tags: null,
      hasChildren: JSON.parse(mainContent.getAttribute('data-page-has-children')),
      templateTagData: mainContent.getAttribute('data-template-tags') || null,
      shareLinksNumber: mainContent.getAttribute('data-share-links-number'),
      shareLinkId: JSON.parse(mainContent.getAttribute('data-share-link-id') || null),

      // latest(on remote) information
      remoteRevisionId: revisionId,
      revisionIdHackmdSynced: mainContent.getAttribute('data-page-revision-id-hackmd-synced') || null,
      lastUpdateUsername: mainContent.getAttribute('data-page-last-update-username') || null,
      deleteUsername: mainContent.getAttribute('data-page-delete-username') || null,
      pageIdOnHackmd: mainContent.getAttribute('data-page-id-on-hackmd') || null,
      hasDraftOnHackmd: !!mainContent.getAttribute('data-page-has-draft-on-hackmd'),
      isHackmdDraftUpdatingInRealtime: false,
      isConflictingOnSave: false,
      isConflictDiffModalOpen: false,

      revisionsOnConflict: {},
    };

    // parse creator, lastUpdateUser and revisionAuthor
    try {
      this.state.creator = JSON.parse(mainContent.getAttribute('data-page-creator'));
    }
    catch (e) {
      logger.warn('The data of \'data-page-creator\' is invalid', e);
    }
    try {
      this.state.revisionAuthor = JSON.parse(mainContent.getAttribute('data-page-revision-author'));
    }
    catch (e) {
      logger.warn('The data of \'data-page-revision-author\' is invalid', e);
    }

    const { interceptorManager } = this.appContainer;
    interceptorManager.addInterceptor(new DetachCodeBlockInterceptor(appContainer), 10); // process as soon as possible
    interceptorManager.addInterceptor(new DrawioInterceptor(appContainer), 20);
    interceptorManager.addInterceptor(new RestoreCodeBlockInterceptor(appContainer), 900); // process as late as possible

    this.initStateMarkdown();
    this.checkAndUpdateImageUrlCached(this.state.likers);

    const { isSharedUser } = this.appContainer;

    // see https://dev.growi.org/5fabddf8bbeb1a0048bcb9e9
    const isAbleToGetAttachedInformationAboutPages = this.state.isPageExist && !isSharedUser;

    if (isAbleToGetAttachedInformationAboutPages) {
      // We don't retrieve bookmarks in the initial page load
      // as it is stored in a separate collection to like and seen user
      // data so it has a separate api endpoint.
      this.initialPageLoad();
      this.retrieveBookmarkInfo();
    }

    this.setTocHtml = this.setTocHtml.bind(this);
    this.save = this.save.bind(this);
    this.checkAndUpdateImageUrlCached = this.checkAndUpdateImageUrlCached.bind(this);

    this.emitJoinPageRoomRequest = this.emitJoinPageRoomRequest.bind(this);
    this.emitJoinPageRoomRequest();

    this.addWebSocketEventHandlers = this.addWebSocketEventHandlers.bind(this);
    this.addWebSocketEventHandlers();

    const unlinkPageButton = document.getElementById('unlink-page-button');
    if (unlinkPageButton != null) {
      unlinkPageButton.addEventListener('click', async() => {
        try {
          const res = await this.appContainer.apiPost('/pages.unlink', { path });
          window.location.href = encodeURI(`${res.path}?unlinked=true`);
        }
        catch (err) {
          toastError(err);
        }
      });
    }

  }

  /**
   * Workaround for the mangling in production build to break constructor.name
   */
  static getClassName() {
    return 'PageContainer';
  }


  get isAbleToOpenPageEditor() {
    const { isNotCreatable, isTrashPage } = this.state;
    const { isGuestUser } = this.appContainer;

    return (!isNotCreatable && !isTrashPage && !isGuestUser);
  }

  /**
   * whether to display reaction buttons
   * ex.) like, bookmark
   */
  get isAbleToShowPageReactionButtons() {
    const { isTrashPage, isPageExist } = this.state;
    const { isSharedUser } = this.appContainer;

    return (!isTrashPage && isPageExist && !isSharedUser);
  }

  /**
   * whether to display tag labels
   */
  get isAbleToShowTagLabel() {
    const { isUserPage } = this.state;
    const { isSharedUser } = this.appContainer;

    return (!isUserPage && !isSharedUser);
  }

  /**
   * whether to display page management
   * ex.) duplicate, rename
   */
  get isAbleToShowPageManagement() {
    const { isPageExist, isTrashPage } = this.state;
    const { isSharedUser } = this.appContainer;

    return (isPageExist && !isTrashPage && !isSharedUser);
  }

  /**
   * whether to display pageEditorModeManager
   * ex.) view, edit, hackmd
   */
  get isAbleToShowPageEditorModeManager() {
    const { isNotCreatable, isTrashPage } = this.state;
    const { isSharedUser } = this.appContainer;

    return (!isNotCreatable && !isTrashPage && !isSharedUser);
  }

  /**
   * whether to display pageAuthors
   * ex.) creator, lastUpdateUser
   */
  get isAbleToShowPageAuthors() {
    const { isPageExist, isUserPage } = this.state;

    return (isPageExist && !isUserPage);
  }

  /**
   * whether to like button
   * not displayed on user page
   */
  get isAbleToShowLikeButtons() {
    const { isUserPage } = this.state;
    const { isSharedUser } = this.appContainer;

    return (!isUserPage && !isSharedUser);
  }

  /**
   * whether to Empty Trash Page
   * not displayed when guest user and not on trash page
   */
  get isAbleToShowEmptyTrashButton() {
    const { currentUser } = this.appContainer;
    const { path, hasChildren } = this.state;

    return (currentUser != null && currentUser.admin && path === '/trash' && hasChildren);
  }

  /**
   * whether to display trash management buttons
   * ex.) undo, delete completly
   * not displayed when guest user
   */
  get isAbleToShowTrashPageManagementButtons() {
    const { currentUser } = this.appContainer;
    const { isDeleted } = this.state;

    return (isDeleted && currentUser != null);
  }

  /**
   * initialize state for markdown data
   */
  initStateMarkdown() {
    let pageContent = '';

    const rawText = document.getElementById('raw-text-original');
    if (rawText) {
      pageContent = rawText.innerHTML;
    }
    const markdown = entities.decodeHTML(pageContent);

    this.state.markdown = markdown;
  }


  async initialPageLoad() {
    {
      const {
        data: {
          likerIds, sumOfLikers, isLiked, seenUserIds, sumOfSeenUsers, isSeen,
        },
      } = await this.appContainer.apiv3Get('/page/info', { _id: this.state.pageId });

      await this.setState({
        sumOfLikers,
        isLiked,
        likerIds,
        seenUserIds,
        sumOfSeenUsers,
        isSeen,
      });
    }

    await this.retrieveLikersAndSeenUsers();
  }

  async toggleLike() {
    {
      const toggledIsLiked = !this.state.isLiked;
      await this.appContainer.apiv3Put('/page/likes', { pageId: this.state.pageId, bool: toggledIsLiked });

      await this.setState(state => ({
        isLiked: toggledIsLiked,
        sumOfLikers: toggledIsLiked ? state.sumOfLikers + 1 : state.sumOfLikers - 1,
        likerIds: toggledIsLiked
          ? [...this.state.likerIds, this.appContainer.currentUserId]
          : state.likerIds.filter(id => id !== this.appContainer.currentUserId),
      }));
    }

    await this.retrieveLikersAndSeenUsers();
  }

  async retrieveLikersAndSeenUsers() {
    const { users } = await this.appContainer.apiGet('/users.list', { user_ids: [...this.state.likerIds, ...this.state.seenUserIds].join(',') });

    await this.setState({
      likers: users.filter(({ id }) => this.state.likerIds.includes(id)).slice(0, 15),
      seenUsers: users.filter(({ id }) => this.state.seenUserIds.includes(id)).slice(0, 15),
    });

    this.checkAndUpdateImageUrlCached(users);
  }

  async retrieveBookmarkInfo() {
    const response = await this.appContainer.apiv3Get('/bookmarks/info', { pageId: this.state.pageId });
    this.setState({
      sumOfBookmarks: response.data.sumOfBookmarks,
      isBookmarked: response.data.isBookmarked,
    });
  }

  async toggleBookmark() {
    const bool = !this.state.isBookmarked;
    await this.appContainer.apiv3Put('/bookmarks', { pageId: this.state.pageId, bool });
    return this.retrieveBookmarkInfo();
  }

  async checkAndUpdateImageUrlCached(users) {
    const noImageCacheUsers = users.filter((user) => { return user.imageUrlCached == null });
    if (noImageCacheUsers.length === 0) {
      return;
    }

    const noImageCacheUserIds = noImageCacheUsers.map((user) => { return user.id });
    try {
      await this.appContainer.apiv3Put('/users/update.imageUrlCache', { userIds: noImageCacheUserIds });
    }
    catch (err) {
      // Error alert doesn't apear, because user don't need to notice this error.
      logger.error(err);
    }
  }

  get navigationContainer() {
    return this.appContainer.getContainer('NavigationContainer');
  }

  setLatestRemotePageData(s2cMessagePageUpdated) {
    const newState = {
      remoteRevisionId: s2cMessagePageUpdated.revisionId,
      revisionIdHackmdSynced: s2cMessagePageUpdated.revisionIdHackmdSynced,
      lastUpdateUsername: s2cMessagePageUpdated.lastUpdateUsername,
    };

    if (s2cMessagePageUpdated.hasDraftOnHackmd != null) {
      newState.hasDraftOnHackmd = s2cMessagePageUpdated.hasDraftOnHackmd;
    }

    this.setState(newState);
  }

  setTocHtml(tocHtml) {
    if (this.state.tocHtml !== tocHtml) {
      this.setState({ tocHtml });
    }
  }

  /**
   * save success handler
   * @param {object} page Page instance
   * @param {Array[Tag]} tags Array of Tag
   * @param {object} revision Revision instance
   */
  updateStateAfterSave(page, tags, revision) {
    const { editorMode } = this.navigationContainer.state;

    // update state of PageContainer
    const newState = {
      pageId: page._id,
      revisionId: revision._id,
      revisionCreatedAt: new Date(revision.createdAt).getTime() / 1000,
      remoteRevisionId: revision._id,
      revisionIdHackmdSynced: page.revisionHackmdSynced,
      hasDraftOnHackmd: page.hasDraftOnHackmd,
      markdown: revision.body,
      createdAt: page.createdAt,
      updatedAt: page.updatedAt,
    };
    if (tags != null) {
      newState.tags = tags;
    }
    this.setState(newState);

    // PageEditor component
    const pageEditor = this.appContainer.getComponentInstance('PageEditor');
    if (pageEditor != null) {
      if (editorMode !== 'edit') {
        pageEditor.updateEditorValue(newState.markdown);
      }
    }
    // PageEditorByHackmd component
    const pageEditorByHackmd = this.appContainer.getComponentInstance('PageEditorByHackmd');
    if (pageEditorByHackmd != null) {
      // reset
      if (editorMode !== 'hackmd') {
        pageEditorByHackmd.reset();
      }
    }

    // hidden input
    $('input[name="revision_id"]').val(newState.revisionId);
  }

  /**
   * Save page
   * @param {string} markdown
   * @param {object} optionsToSave
   * @return {object} { page: Page, tags: Tag[] }
   */
  async save(markdown, optionsToSave = {}) {
    const { editorMode } = this.navigationContainer.state;

    const { pageId, path } = this.state;
    let { revisionId } = this.state;
    const options = Object.assign({}, optionsToSave);

    if (editorMode === 'hackmd') {
      // set option to sync
      options.isSyncRevisionToHackmd = true;
      revisionId = this.state.revisionIdHackmdSynced;
    }

    let res;
    if (pageId == null) {
      res = await this.createPage(path, markdown, options);
    }
    else {
      res = await this.updatePage(pageId, revisionId, markdown, options);
    }

    this.updateStateAfterSave(res.page, res.tags, res.revision);
    return res;
  }

  async saveAndReload(optionsToSave) {
    if (optionsToSave == null) {
      const msg = '\'saveAndReload\' requires the \'optionsToSave\' param';
      throw new Error(msg);
    }

    const { editorMode } = this.navigationContainer.state;
    if (editorMode == null) {
      logger.warn('\'saveAndReload\' requires the \'errorMode\' param');
      return;
    }

    const { pageId, path } = this.state;
    let { revisionId } = this.state;

    const options = Object.assign({}, optionsToSave);

    let markdown;
    if (editorMode === 'hackmd') {
      const pageEditorByHackmd = this.appContainer.getComponentInstance('PageEditorByHackmd');
      markdown = await pageEditorByHackmd.getMarkdown();
      // set option to sync
      options.isSyncRevisionToHackmd = true;
      revisionId = this.state.revisionIdHackmdSynced;
    }
    else {
      const pageEditor = this.appContainer.getComponentInstance('PageEditor');
      markdown = pageEditor.getMarkdown();
    }

    let res;
    if (pageId == null) {
      res = await this.createPage(path, markdown, options);
    }
    else {
      res = await this.updatePage(pageId, revisionId, markdown, options);
    }

    const editorContainer = this.appContainer.getContainer('EditorContainer');
    editorContainer.clearDraft(path);
    window.location.href = path;

    return res;
  }

  async createPage(pagePath, markdown, tmpParams) {
    const socketIoContainer = this.appContainer.getContainer('SocketIoContainer');

    // clone
    const params = Object.assign(tmpParams, {
      path: pagePath,
      body: markdown,
    });

    const res = await this.appContainer.apiv3Post('/pages/', params);
    const { page, tags, revision } = res.data;

    return { page, tags, revision };
  }

  async updatePage(pageId, revisionId, markdown, tmpParams) {
    const socketIoContainer = this.appContainer.getContainer('SocketIoContainer');

    // clone
    const params = Object.assign(tmpParams, {
      page_id: pageId,
      revision_id: revisionId,
      body: markdown,
    });

    const res = await this.appContainer.apiPost('/pages.update', params);
    if (!res.ok) {
      throw new Error(res.error);
    }
    return res;
  }

  deletePage(isRecursively, isCompletely) {
    const socketIoContainer = this.appContainer.getContainer('SocketIoContainer');

    // control flag
    const completely = isCompletely ? true : null;
    const recursively = isRecursively ? true : null;

    return this.appContainer.apiPost('/pages.remove', {
      recursively,
      completely,
      page_id: this.state.pageId,
      revision_id: this.state.revisionId,
    });

  }

  revertRemove(isRecursively) {
    const socketIoContainer = this.appContainer.getContainer('SocketIoContainer');

    // control flag
    const recursively = isRecursively ? true : null;

    return this.appContainer.apiPost('/pages.revertRemove', {
      recursively,
      page_id: this.state.pageId,
    });
  }

  rename(newPagePath, isRecursively, isRenameRedirect, isRemainMetadata) {
    const socketIoContainer = this.appContainer.getContainer('SocketIoContainer');
    const { pageId, revisionId, path } = this.state;

    return this.appContainer.apiv3Put('/pages/rename', {
      revisionId,
      pageId,
      isRecursively,
      isRenameRedirect,
      isRemainMetadata,
      newPagePath,
      path,
    });
  }

  showSuccessToastr() {
    toastr.success(undefined, 'Saved successfully', {
      closeButton: true,
      progressBar: true,
      newestOnTop: false,
      showDuration: '100',
      hideDuration: '100',
      timeOut: '1200',
      extendedTimeOut: '150',
    });
  }

  showErrorToastr(error) {
    toastr.error(error.message, 'Error occured', {
      closeButton: true,
      progressBar: true,
      newestOnTop: false,
      showDuration: '100',
      hideDuration: '100',
      timeOut: '3000',
    });
  }

  // request to server so the client to join a room for each page
  emitJoinPageRoomRequest() {
    const socketIoContainer = this.appContainer.getContainer('SocketIoContainer');
    const socket = socketIoContainer.getSocket();
    socket.emit('join:page', { socketId: socket.id, pageId: this.state.pageId });
  }

  addWebSocketEventHandlers() {
    // eslint-disable-next-line @typescript-eslint/no-this-alias
    const pageContainer = this;
    const socketIoContainer = this.appContainer.getContainer('SocketIoContainer');
    const socket = socketIoContainer.getSocket();

    socket.on('page:create', (data) => {
      logger.debug({ obj: data }, `websocket on 'page:create'`); // eslint-disable-line quotes

      // update remote page data
      const { s2cMessagePageUpdated } = data;
      if (s2cMessagePageUpdated.pageId === pageContainer.state.pageId) {
        pageContainer.setLatestRemotePageData(s2cMessagePageUpdated);
      }
    });

    socket.on('page:update', (data) => {
      logger.debug({ obj: data }, `websocket on 'page:update'`); // eslint-disable-line quotes

      // update remote page data
      const { s2cMessagePageUpdated } = data;
      if (s2cMessagePageUpdated.pageId === pageContainer.state.pageId) {
        pageContainer.setLatestRemotePageData(s2cMessagePageUpdated);
      }
    });

    socket.on('page:delete', (data) => {
      logger.debug({ obj: data }, `websocket on 'page:delete'`); // eslint-disable-line quotes

      // update remote page data
      const { s2cMessagePageUpdated } = data;
      if (s2cMessagePageUpdated.pageId === pageContainer.state.pageId) {
        pageContainer.setLatestRemotePageData(s2cMessagePageUpdated);
      }
    });

    socket.on('page:editingWithHackmd', (data) => {
      logger.debug({ obj: data }, `websocket on 'page:editingWithHackmd'`); // eslint-disable-line quotes

      // update isHackmdDraftUpdatingInRealtime
      const { s2cMessagePageUpdated } = data;
      if (s2cMessagePageUpdated.pageId === pageContainer.state.pageId) {
        pageContainer.setState({ isHackmdDraftUpdatingInRealtime: true });
      }
    });

  }

  /* TODO GW-325 */
  retrieveMyBookmarkList() {
  }

  async resolveConflict(pageId, revisionId, markdown, optionsToSave) {

    if (optionsToSave == null) {
      const msg = '\'saveAndReload\' requires the \'optionsToSave\' param';
      throw new Error(msg);
    }

    const { path } = this.state;

    const params = Object.assign(optionsToSave, {
      page_id: pageId,
      revision_id: revisionId,
      body: markdown,
    });

    const res = await this.appContainer.apiPost('/pages.update', params);
    if (!res.ok) {
      throw new Error(res.error);
    }

    const editorContainer = this.appContainer.getContainer('EditorContainer');
    editorContainer.clearDraft(path);

    return res;
  }

  async resolveConflictAndReload(pageId, revisionId, markdown, optionsToSave) {
    await this.resolveConflict(pageId, revisionId, markdown, optionsToSave);
    window.location.href = this.state.path;
  }

}