'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 = Readonly<{
  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
