nocodb/packages/nc-gui/composables/useUndoRedo.ts

import type { Ref } from 'vue'
import rfdc from 'rfdc'
import type { ProjectType, TableType, ViewType } from 'nocodb-sdk'
import { createSharedComposable, ref, useRouter } from '#imports'
import type { UndoRedoAction } from '#imports'

export const useUndoRedo = createSharedComposable(() => {
  const clone = rfdc()

  const router = useRouter()

  const route = router.currentRoute

  // keys: projectType | projectId | type | title | viewTitle
  const scope = computed<{ key: string; param: string }[]>(() => {
    const tempScope: { key: string; param: string }[] = [{ key: 'root', param: 'root' }]
    for (const [key, param] of Object.entries(route.value.params)) {
      if (Array.isArray(param)) {
        tempScope.push({ key, param: param.join(',') })
      } else {
        tempScope.push({ key, param })
      }
    }

    return tempScope
  })

  const isSameScope = (sc: { key: string; param: string | string[] }[]) => {
    return sc.every((s) => {
      return scope.value.some(
        // viewTitle is optional for default view
        (s2) => {
          if (Array.isArray(s.param)) {
            return (
              (s.key === 'viewTitle' && s2.key === 'viewTitle' && s2.param === '') ||
              (s.key === s2.key && s.param.includes(s2.param))
            )
          } else {
            return (
              (s.key === 'viewTitle' && s2.key === 'viewTitle' && s2.param === '') || (s.key === s2.key && s.param === s2.param)
            )
          }
        },
      )
    })
  }

  const undoQueue: Ref<UndoRedoAction[]> = ref([])

  const redoQueue: Ref<UndoRedoAction[]> = ref([])

  const addUndo = (action: UndoRedoAction, fromRedo = false) => {
    // remove all redo actions that are in the same scope
    if (!fromRedo) redoQueue.value = redoQueue.value.filter((a) => !isSameScope(a.scope || []))
    undoQueue.value.push(action)
  }

  const addRedo = (action: UndoRedoAction) => {
    redoQueue.value.push(action)
  }

  const undo = async () => {
    let actionIndex = -1
    for (let i = undoQueue.value.length - 1; i >= 0; i--) {
      const elScope = undoQueue.value[i].scope || [{ key: 'root', param: 'root' }]
      if (isSameScope(elScope)) {
        actionIndex = i
        break
      }
    }

    if (actionIndex === -1) return

    const action = undoQueue.value.splice(actionIndex, 1)[0]
    if (action) {
      try {
        await action.undo.fn.apply(action, action.undo.args)
        addRedo(action)
      } catch (e) {
        message.warn('Error while undoing action, it is skipped.')
      }
    }
  }

  const redo = async () => {
    let actionIndex = -1
    for (let i = redoQueue.value.length - 1; i >= 0; i--) {
      const elScope = redoQueue.value[i].scope || [{ key: 'root', param: 'root' }]
      if (isSameScope(elScope)) {
        actionIndex = i
        break
      }
    }

    if (actionIndex === -1) return

    const action = redoQueue.value.splice(actionIndex, 1)[0]
    if (action) {
      try {
        await action.redo.fn.apply(action, action.redo.args)
        addUndo(action, true)
      } catch (e) {
        message.warn('Error while redoing action, it is skipped.')
      }
    }
  }

  const defineRootScope = () => {
    return [{ key: 'root', param: 'root' }]
  }

  const defineProjectScope = (param: { project?: ProjectType; model?: TableType; view?: ViewType; project_id?: string }) => {
    if (param.project) {
      return [{ key: 'projectId', param: param.project.id! }]
    } else if (param.model) {
      return [{ key: 'projectId', param: param.model.project_id! }]
    } else if (param.view) {
      return [{ key: 'projectId', param: param.view.project_id! }]
    } else {
      return [{ key: 'projectId', param: param.project_id! }]
    }
  }

  const defineModelScope = (param: { model?: TableType; view?: ViewType; project_id?: string; model_id?: string }) => {
    if (param.model) {
      return [
        { key: 'projectId', param: param.model.project_id! },
        { key: 'viewId', param: param.model.id! },
      ]
    } else if (param.view) {
      return [
        { key: 'projectId', param: param.view.project_id! },
        { key: 'viewId', param: param.view.fk_model_id! },
      ]
    } else {
      return [
        { key: 'projectId', param: param.project_id! },
        { key: 'viewId', param: param.model_id! },
      ]
    }
  }

  const defineViewScope = (param: { view?: ViewType; project_id?: string; model_id?: string; title?: string; id?: string }) => {
    if (param.view) {
      return [
        { key: 'projectId', param: param.view.project_id! },
        { key: 'viewId', param: param.view.fk_model_id! },
        { key: 'viewTitle', param: [param.view.title, param.view.id!] },
      ]
    } else {
      return [
        { key: 'projectId', param: param.project_id! },
        { key: 'viewId', param: param.model_id! },
        { key: 'viewTitle', param: [param.title!, param.id!] },
      ]
    }
  }

  useEventListener(document, 'keydown', async (e: KeyboardEvent) => {
    const cmdOrCtrl = isMac() ? e.metaKey : e.ctrlKey
    if (cmdOrCtrl && !e.altKey) {
      switch (e.keyCode) {
        case 90: {
          e.preventDefault()
          // CMD + z and CMD + shift + z
          if (!e.shiftKey) {
            if (undoQueue.value.length) {
              undo()
            }
          } else {
            if (redoQueue.value.length) {
              redo()
            }
          }
          break
        }
      }
    }
  })

  return {
    addUndo,
    undo,
    clone,
    defineRootScope,
    defineProjectScope,
    defineModelScope,
    defineViewScope,
  }
})