import type {RegisterFn} from '@roots/wordpress-hmr'

import * as api from '@roots/wordpress-hmr/block'
import * as editor from '@roots/wordpress-hmr/editor'
import {dispatch, select} from '@wordpress/data'

let selected = null

const before = () => {
  selected = select(`core/block-editor`).getSelectedBlockClientId()
  dispatch(`core/block-editor`).clearSelectedBlock()
}

const after = (changed?: Array<{name: string}>) => {
  if (!changed?.length) return

  select(`core/block-editor`)
    .getBlocks()
    .forEach(({clientId, name}) => {
      changed
        ?.filter(module => module?.name)
        .map(module => module.name)
        .includes(name) &&
        dispatch(`core/block-editor`).selectBlock(clientId)
    })

  selected
    ? dispatch(`core/block-editor`).selectBlock(selected)
    : dispatch(`core/block-editor`).clearSelectedBlock()

  selected = null
}

export const register: RegisterFn = (getContext, accept) =>
  editor.load({accept, after, api, before, getContext})