patch-db/client/lib/json-patch-lib.ts

import { DBCache, PatchOp } from './types'

export interface BaseOperation {
  path: string
}

export interface AddOperation<T> extends BaseOperation {
  op: PatchOp.ADD
  value: T
}

export interface RemoveOperation extends BaseOperation {
  op: PatchOp.REMOVE
}

export interface ReplaceOperation<T> extends BaseOperation {
  op: PatchOp.REPLACE
  value: T
}

export type Operation<T> = AddOperation<T> | RemoveOperation | ReplaceOperation<T>

export function getValueByPointer<T extends Record<string, T>> (data: T, pointer: string): any {
  if (pointer === '/') return data

  try {
    return jsonPathToKeyArray(pointer).reduce((acc, next) => acc[next], data)
  } catch (e) {
    return undefined
  }
}

export function applyOperation<T> (
  doc: DBCache<Record<string, any>>,
  { path, op, value }: Operation<T> & { value?: T },
): Operation<T> | null {
  const current = getValueByPointer(doc.data, path)
  const remove = { op: PatchOp.REMOVE, path} as const
  const add = { op: PatchOp.ADD, path, value: current} as const
  const replace = { op: PatchOp.REPLACE, path, value: current } as const

  doc.data = recursiveApply(doc.data, jsonPathToKeyArray(path), value)

  switch (op) {
    case PatchOp.REMOVE:
      return current === undefined
        ? null
        : add
    case PatchOp.REPLACE:
    case PatchOp.ADD:
      return current === undefined
        ? remove
        : replace
  }
}

function recursiveApply<T extends Record<string, T>> (data: T, path: readonly string[], value?: any): T {
  if (!path.length) return value

  if (!isObject(data)) {
    throw Error('Patch cannot be applied. Path contains non object')
  }

  const updated = recursiveApply(data[path[0]], path.slice(1), value)
  const result = {
    ...data,
    [path[0]]: updated,
  }

  if (updated === undefined) {
    delete result[path[0]]
  }

  return result
}

function isObject (val: any): val is Record<string, unknown> {
  return typeof val === 'object' && !Array.isArray(val) && val !== null
}

function jsonPathToKeyArray (path: string): string[] {
  return path.split('/').slice(1)
}