import { yDocToProsemirrorJSON } from "@getoutline/y-prosemirror";
import { uniq } from "lodash";
import { Node } from "prosemirror-model";
import * as Y from "yjs";
import { sequelize } from "@server/database/sequelize";
import { schema, serializer } from "@server/editor";
import { Document, Event } from "@server/models";

export default async function documentCollaborativeUpdater({
  documentId,
  ydoc,
  userId,
}: {
  documentId: string;
  ydoc: Y.Doc;
  userId?: string;
}) {
  return sequelize.transaction(async (transaction) => {
    const document = await Document.unscoped()
      .scope("withState")
      .findByPk(documentId, {
        transaction,
        lock: {
          of: Document,
          level: transaction.LOCK.UPDATE,
        },
        rejectOnEmpty: true,
        paranoid: false,
      });

    const state = Y.encodeStateAsUpdate(ydoc);
    const node = Node.fromJSON(schema, yDocToProsemirrorJSON(ydoc, "default"));
    const text = serializer.serialize(node, undefined);
    const isUnchanged = document.text === text;
    const hasMultiplayerState = !!document.state;

    if (isUnchanged && hasMultiplayerState) {
      return;
    }

    // extract collaborators from doc user data
    const pud = new Y.PermanentUserData(ydoc);
    const pudIds = Array.from(pud.clients.values());
    const existingIds = document.collaboratorIds;
    const collaboratorIds = uniq([...pudIds, ...existingIds]);

    await document.update(
      {
        text,
        state: Buffer.from(state),
        lastModifiedById:
          isUnchanged || !userId ? document.lastModifiedById : userId,
        collaboratorIds,
      },
      {
        transaction,
        silent: isUnchanged,
        hooks: false,
      }
    );

    if (isUnchanged) {
      return;
    }

    await Event.schedule({
      name: "documents.update",
      documentId: document.id,
      collectionId: document.collectionId,
      teamId: document.teamId,
      actorId: userId,
      data: {
        multiplayer: true,
        title: document.title,
      },
    });
  });
}