import {ImageIcon} from '@sanity/icons'
import {Box, Spinner} from '@sanity/ui'
import {useContext, useMemo} from 'react'
import {DocumentFieldAction, DocumentFieldActionGroup, DocumentFieldActionItem} from 'sanity'

import {useAssistDocumentContext} from '../assistDocument/AssistDocumentContext'
import {useAiAssistanceConfig} from '../assistLayout/AiAssistanceConfigContext'
import {ImageContext} from '../components/ImageContext'
import {usePathKey} from '../helpers/misc'
import {useApiClient, useGenerateImage} from '../useApiClient'

function node(node: DocumentFieldActionItem | DocumentFieldActionGroup) {
  return node
}

export const generateImagActions: DocumentFieldAction = {
  name: 'sanity-assist-generate-image',
  useAction(props) {
    const pathKey = usePathKey(props.path)

    const {config} = useAiAssistanceConfig()
    const apiClient = useApiClient(config?.__customApiClient)
    const {generateImage, loading} = useGenerateImage(apiClient)

    const imageContext = useContext(ImageContext)

    if (imageContext && pathKey === imageContext?.imageInstructionPath) {
      //if this is true, it is stable, and not breaking rules of hooks
      // eslint-disable-next-line react-hooks/rules-of-hooks
      const {assistableDocumentId} = useAssistDocumentContext()
      // eslint-disable-next-line react-hooks/rules-of-hooks
      return useMemo(() => {
        return node({
          type: 'action',
          icon: loading
            ? () => (
                <Box style={{height: 17}}>
                  <Spinner style={{transform: 'translateY(6px)'}} />
                </Box>
              )
            : ImageIcon,
          title: 'Generate image from prompt',
          onAction: () => {
            if (loading) {
              return
            }
            generateImage({path: pathKey, documentId: assistableDocumentId})
          },
          renderAsButton: true,
          disabled: loading,
        })
      }, [generateImage, pathKey, assistableDocumentId, loading])
    }

    // works but not supported by types
    return undefined as unknown as DocumentFieldActionItem
  },
}