'use client'
import type {
  ConversationVariable,
} from '@/app/components/workflow/types'
import { cn } from '@langgenius/dify-ui/cn'
import { Dialog, DialogContent } from '@langgenius/dify-ui/dialog'
import { RiCloseLine } from '@remixicon/react'
import { useMount } from 'ahooks'
import copy from 'copy-to-clipboard'
import { capitalize } from 'es-toolkit/string'
import * as React from 'react'
import { useCallback } from 'react'
import { useTranslation } from 'react-i18next'
import {
  Copy,
  CopyCheck,
} from '@/app/components/base/icons/src/vender/line/files'
import { BubbleX } from '@/app/components/base/icons/src/vender/line/others'
import CodeEditor from '@/app/components/workflow/nodes/_base/components/editor/code-editor'
import { CodeLanguage } from '@/app/components/workflow/nodes/code/types'
import { ChatVarType } from '@/app/components/workflow/panel/chat-variable-panel/type'
import { useStore } from '@/app/components/workflow/store'
import useTimestamp from '@/hooks/use-timestamp'
import { fetchCurrentValueOfConversationVariable } from '@/service/workflow'

type Props = {
  conversationID: string
  onHide: () => void
}

const ConversationVariableModal = ({
  conversationID,
  onHide,
}: Props) => {
  const { t } = useTranslation()
  const { formatTime } = useTimestamp()
  const varList = useStore(s => s.conversationVariables) as ConversationVariable[]
  const appID = useStore(s => s.appId)
  const [currentVar, setCurrentVar] = React.useState<ConversationVariable>(varList[0]!)
  const [latestValueMap, setLatestValueMap] = React.useState<Record<string, string>>({})
  const [latestValueTimestampMap, setLatestValueTimestampMap] = React.useState<Record<string, number>>({})

  const getChatVarLatestValues = useCallback(async () => {
    if (conversationID && varList.length > 0) {
      const res = await fetchCurrentValueOfConversationVariable({
        url: `/apps/${appID}/conversation-variables`,
        params: { conversation_id: conversationID },
      })
      if (res.data.length > 0) {
        const valueMap = res.data.reduce((acc: any, cur) => {
          acc[cur.id] = cur.value
          return acc
        }, {})
        setLatestValueMap(valueMap)
        const timestampMap = res.data.reduce((acc: any, cur) => {
          acc[cur.id] = cur.updated_at
          return acc
        }, {})
        setLatestValueTimestampMap(timestampMap)
      }
    }
  }, [appID, conversationID, varList.length])

  const [isCopied, setIsCopied] = React.useState(false)
  const handleCopy = useCallback(() => {
    copy(currentVar.value)
    setIsCopied(true)
    setTimeout(() => {
      setIsCopied(false)
    }, 2000)
  }, [currentVar.value])

  useMount(() => {
    getChatVarLatestValues()
  })

  return (
    <Dialog open>
      <DialogContent className={cn('w-full overflow-hidden! border-none text-left align-middle', cn('h-[min(640px,calc(100dvh-2rem))] max-h-none! w-[920px] max-w-[calc(100vw-2rem)] p-0'))}>

        <button
          type="button"
          aria-label={t('operation.close', { ns: 'common' })}
          className="absolute top-4 right-4 cursor-pointer border-none bg-transparent p-2 focus-visible:ring-1 focus-visible:ring-components-input-border-active focus-visible:outline-hidden"
          onClick={onHide}
        >
          <RiCloseLine className="size-4 text-text-tertiary" aria-hidden="true" />
        </button>
        <div className="flex size-full">
          {/* LEFT */}
          <div className="flex h-full w-[224px] shrink-0 flex-col border-r border-divider-burn bg-background-sidenav-bg">
            <div className="shrink-0 pt-5 pr-4 pb-3 pl-5 system-xl-semibold text-text-primary">{t('chatVariable.panelTitle', { ns: 'workflow' })}</div>
            <div className="grow overflow-y-auto px-3 py-2">
              {varList.map(chatVar => (
                <button
                  key={chatVar.id}
                  type="button"
                  className={cn('group mb-0.5 flex w-full cursor-pointer items-center rounded-lg border-none bg-transparent p-2 text-left hover:bg-state-base-hover focus-visible:ring-1 focus-visible:ring-components-input-border-active focus-visible:outline-hidden', currentVar.id === chatVar.id && 'bg-state-base-hover')}
                  onClick={() => setCurrentVar(chatVar)}
                >
                  <BubbleX className={cn('mr-1 size-4 shrink-0 text-text-tertiary group-hover:text-util-colors-teal-teal-700', currentVar.id === chatVar.id && 'text-util-colors-teal-teal-700')} aria-hidden="true" />
                  <div title={chatVar.name} className={cn('truncate system-sm-medium text-text-tertiary group-hover:text-util-colors-teal-teal-700', currentVar.id === chatVar.id && 'text-util-colors-teal-teal-700')}>{chatVar.name}</div>
                </button>
              ))}
            </div>
          </div>
          {/* RIGHT */}
          <div className="flex h-full w-0 grow flex-col bg-components-panel-bg">
            <div className="shrink-0 p-4 pb-2">
              <div className="flex items-center gap-1 py-1">
                <div className="system-xl-semibold text-text-primary">{currentVar.name}</div>
                <div className="system-xs-medium text-text-tertiary">{capitalize(currentVar.value_type)}</div>
              </div>
            </div>
            <div className="flex h-0 grow flex-col p-4 pt-2">
              <div className="mb-2 flex shrink-0 items-center gap-2">
                <div className="shrink-0 system-xs-medium-uppercase text-text-tertiary">{t('chatVariable.storedContent', { ns: 'workflow' }).toLocaleUpperCase()}</div>
                <div
                  className="h-px grow"
                  style={{
                    background: 'linear-gradient(to right, rgba(16, 24, 40, 0.08) 0%, rgba(255, 255, 255) 100%)',
                  }}
                >
                </div>
                {!!latestValueTimestampMap[currentVar.id] && (
                  <div className="shrink-0 system-xs-regular text-text-tertiary">
                    {t('chatVariable.updatedAt', { ns: 'workflow' })}
                    {formatTime(latestValueTimestampMap[currentVar.id]!, t('dateTimeFormat', { ns: 'appLog' }) as string)}
                  </div>
                )}
              </div>
              <div className="grow overflow-y-auto">
                {currentVar.value_type !== ChatVarType.Number && currentVar.value_type !== ChatVarType.String && (
                  <div className="flex h-full flex-col rounded-lg bg-components-input-bg-normal px-2 pb-2">
                    <div className="flex h-7 shrink-0 items-center justify-between pt-1 pr-2 pl-3">
                      <div className="system-xs-semibold text-text-secondary">JSON</div>
                      <div className="flex items-center p-1">
                        {!isCopied
                          ? (
                              <Copy className="size-4 cursor-pointer text-text-tertiary" onClick={handleCopy} />
                            )
                          : (
                              <CopyCheck className="size-4 text-text-tertiary" />
                            )}
                      </div>
                    </div>
                    <div className="grow pl-4">
                      <CodeEditor
                        readOnly
                        noWrapper
                        isExpand
                        language={CodeLanguage.json}
                        value={latestValueMap[currentVar.id] || ''}
                        isJSONStringifyBeauty
                      />
                    </div>
                  </div>
                )}
                {(currentVar.value_type === ChatVarType.Number || currentVar.value_type === ChatVarType.String) && (
                  <div className="h-full overflow-x-hidden overflow-y-auto rounded-lg bg-components-input-bg-normal px-4 py-3 system-md-regular text-components-input-text-filled">{latestValueMap[currentVar.id] || ''}</div>
                )}
              </div>
            </div>
          </div>
        </div>
      </DialogContent>
    </Dialog>
  )
}

export default ConversationVariableModal
