Function createViewer

packages/plugins/image/src/create-viewer.tsx:37–86

(editor: Editable)

export const VIEWER_CACHE_WEAKMAP = new WeakMap<Editor, ReturnType<typeof createViewer>>()

const createViewer = (editor: Editable) => {
  const viewer = {
    images: (force = false) => findImages(editor, force),

    count: () => {
      return viewer.images().length
    },

    open: (image: Image) => {
      const store = getViewerStore(editor)
      const images = viewer.images(true)
      const key = Editable.findKey(editor, image).id
      return store.setState({
        visible: true,
        index: images.findIndex(image => image.key === key),
      })
    },

    close: () => {
      const store = getViewerStore(editor)
      store.setState({
        visible: false,
        index: -1,
      })
    },

    prev: () => {
      const store = getViewerStore(editor)
      const count = viewer.count()
      const index = store.getState().index
      let val = index - 1
      if (val < 0) val = count - 1
      store.setState({
        index: val,
      })
    },

    next: () => {
      const store = getViewerStore(editor)
      const count = viewer.count()
      const index = store.getState().index
      let val = index + 1
      if (val === count) val = 0
      store.setState({
        index: val,
      })
    },
  }
  return viewer
}

export const getViewer = (editor: Editable) => {
  let viewer = VIEWER_CACHE_WEAKMAP.get(editor)