'use client'
import type { DocType } from '@/models/datasets'
import type { I18nKeysByPrefix } from '@/types/i18n'
import { useTranslation } from 'react-i18next'
import useTimestamp from '@/hooks/use-timestamp'
import { ChunkingMode } from '@/models/datasets'
import { formatFileSize, formatNumber, formatTime } from '@/utils/format'

export type inputType = 'input' | 'select' | 'textarea'
type MetadataType = DocType | 'originInfo' | 'technicalParameters'

type MetadataMap = Record<
  MetadataType,
  {
    text: string
    allowEdit?: boolean
    icon?: React.ReactNode
    iconName?: string
    subFieldsMap: Record<
      string,
      {
        label: string
        inputType?: inputType
        field?: string
        render?: (value: any, total?: number) => React.ReactNode | string
      }
    >
  }
>

const fieldPrefix = 'metadata.field'

export const useMetadataMap = (): MetadataMap => {
  const { t } = useTranslation()
  const { formatTime: formatTimestamp } = useTimestamp()

  return {
    book: {
      text: t(($) => $['metadata.type.book'], { ns: 'datasetDocuments' }),
      iconName: 'bookOpen',
      subFieldsMap: {
        title: { label: t(($) => $[`${fieldPrefix}.book.title`], { ns: 'datasetDocuments' }) },
        language: {
          label: t(($) => $[`${fieldPrefix}.book.language`], { ns: 'datasetDocuments' }),
          inputType: 'select',
        },
        author: { label: t(($) => $[`${fieldPrefix}.book.author`], { ns: 'datasetDocuments' }) },
        publisher: {
          label: t(($) => $[`${fieldPrefix}.book.publisher`], { ns: 'datasetDocuments' }),
        },
        publication_date: {
          label: t(($) => $[`${fieldPrefix}.book.publicationDate`], { ns: 'datasetDocuments' }),
        },
        isbn: { label: t(($) => $[`${fieldPrefix}.book.ISBN`], { ns: 'datasetDocuments' }) },
        category: {
          label: t(($) => $[`${fieldPrefix}.book.category`], { ns: 'datasetDocuments' }),
          inputType: 'select',
        },
      },
    },
    web_page: {
      text: t(($) => $['metadata.type.webPage'], { ns: 'datasetDocuments' }),
      iconName: 'globe',
      subFieldsMap: {
        title: { label: t(($) => $[`${fieldPrefix}.webPage.title`], { ns: 'datasetDocuments' }) },
        url: { label: t(($) => $[`${fieldPrefix}.webPage.url`], { ns: 'datasetDocuments' }) },
        language: {
          label: t(($) => $[`${fieldPrefix}.webPage.language`], { ns: 'datasetDocuments' }),
          inputType: 'select',
        },
        'author/publisher': {
          label: t(($) => $[`${fieldPrefix}.webPage.authorPublisher`], { ns: 'datasetDocuments' }),
        },
        publish_date: {
          label: t(($) => $[`${fieldPrefix}.webPage.publishDate`], { ns: 'datasetDocuments' }),
        },
        'topic/keywords': {
          label: t(($) => $[`${fieldPrefix}.webPage.topicKeywords`], { ns: 'datasetDocuments' }),
        },
        description: {
          label: t(($) => $[`${fieldPrefix}.webPage.description`], { ns: 'datasetDocuments' }),
        },
      },
    },
    paper: {
      text: t(($) => $['metadata.type.paper'], { ns: 'datasetDocuments' }),
      iconName: 'graduationHat',
      subFieldsMap: {
        title: { label: t(($) => $[`${fieldPrefix}.paper.title`], { ns: 'datasetDocuments' }) },
        language: {
          label: t(($) => $[`${fieldPrefix}.paper.language`], { ns: 'datasetDocuments' }),
          inputType: 'select',
        },
        author: { label: t(($) => $[`${fieldPrefix}.paper.author`], { ns: 'datasetDocuments' }) },
        publish_date: {
          label: t(($) => $[`${fieldPrefix}.paper.publishDate`], { ns: 'datasetDocuments' }),
        },
        'journal/conference_name': {
          label: t(($) => $[`${fieldPrefix}.paper.journalConferenceName`], {
            ns: 'datasetDocuments',
          }),
        },
        'volume/issue/page_numbers': {
          label: t(($) => $[`${fieldPrefix}.paper.volumeIssuePage`], { ns: 'datasetDocuments' }),
        },
        doi: { label: t(($) => $[`${fieldPrefix}.paper.DOI`], { ns: 'datasetDocuments' }) },
        'topic/keywords': {
          label: t(($) => $[`${fieldPrefix}.paper.topicsKeywords`], { ns: 'datasetDocuments' }),
        },
        abstract: {
          label: t(($) => $[`${fieldPrefix}.paper.abstract`], { ns: 'datasetDocuments' }),
          inputType: 'textarea',
        },
      },
    },
    social_media_post: {
      text: t(($) => $['metadata.type.socialMediaPost'], { ns: 'datasetDocuments' }),
      iconName: 'atSign',
      subFieldsMap: {
        platform: {
          label: t(($) => $[`${fieldPrefix}.socialMediaPost.platform`], { ns: 'datasetDocuments' }),
        },
        'author/username': {
          label: t(($) => $[`${fieldPrefix}.socialMediaPost.authorUsername`], {
            ns: 'datasetDocuments',
          }),
        },
        publish_date: {
          label: t(($) => $[`${fieldPrefix}.socialMediaPost.publishDate`], {
            ns: 'datasetDocuments',
          }),
        },
        post_url: {
          label: t(($) => $[`${fieldPrefix}.socialMediaPost.postURL`], { ns: 'datasetDocuments' }),
        },
        'topics/tags': {
          label: t(($) => $[`${fieldPrefix}.socialMediaPost.topicsTags`], {
            ns: 'datasetDocuments',
          }),
        },
      },
    },
    personal_document: {
      text: t(($) => $['metadata.type.personalDocument'], { ns: 'datasetDocuments' }),
      iconName: 'file',
      subFieldsMap: {
        title: {
          label: t(($) => $[`${fieldPrefix}.personalDocument.title`], { ns: 'datasetDocuments' }),
        },
        author: {
          label: t(($) => $[`${fieldPrefix}.personalDocument.author`], { ns: 'datasetDocuments' }),
        },
        creation_date: {
          label: t(($) => $[`${fieldPrefix}.personalDocument.creationDate`], {
            ns: 'datasetDocuments',
          }),
        },
        last_modified_date: {
          label: t(($) => $[`${fieldPrefix}.personalDocument.lastModifiedDate`], {
            ns: 'datasetDocuments',
          }),
        },
        document_type: {
          label: t(($) => $[`${fieldPrefix}.personalDocument.documentType`], {
            ns: 'datasetDocuments',
          }),
          inputType: 'select',
        },
        'tags/category': {
          label: t(($) => $[`${fieldPrefix}.personalDocument.tagsCategory`], {
            ns: 'datasetDocuments',
          }),
        },
      },
    },
    business_document: {
      text: t(($) => $['metadata.type.businessDocument'], { ns: 'datasetDocuments' }),
      iconName: 'briefcase',
      subFieldsMap: {
        title: {
          label: t(($) => $[`${fieldPrefix}.businessDocument.title`], { ns: 'datasetDocuments' }),
        },
        author: {
          label: t(($) => $[`${fieldPrefix}.businessDocument.author`], { ns: 'datasetDocuments' }),
        },
        creation_date: {
          label: t(($) => $[`${fieldPrefix}.businessDocument.creationDate`], {
            ns: 'datasetDocuments',
          }),
        },
        last_modified_date: {
          label: t(($) => $[`${fieldPrefix}.businessDocument.lastModifiedDate`], {
            ns: 'datasetDocuments',
          }),
        },
        document_type: {
          label: t(($) => $[`${fieldPrefix}.businessDocument.documentType`], {
            ns: 'datasetDocuments',
          }),
          inputType: 'select',
        },
        'department/team': {
          label: t(($) => $[`${fieldPrefix}.businessDocument.departmentTeam`], {
            ns: 'datasetDocuments',
          }),
        },
      },
    },
    im_chat_log: {
      text: t(($) => $['metadata.type.IMChat'], { ns: 'datasetDocuments' }),
      iconName: 'messageTextCircle',
      subFieldsMap: {
        chat_platform: {
          label: t(($) => $[`${fieldPrefix}.IMChat.chatPlatform`], { ns: 'datasetDocuments' }),
        },
        'chat_participants/group_name': {
          label: t(($) => $[`${fieldPrefix}.IMChat.chatPartiesGroupName`], {
            ns: 'datasetDocuments',
          }),
        },
        start_date: {
          label: t(($) => $[`${fieldPrefix}.IMChat.startDate`], { ns: 'datasetDocuments' }),
        },
        end_date: {
          label: t(($) => $[`${fieldPrefix}.IMChat.endDate`], { ns: 'datasetDocuments' }),
        },
        participants: {
          label: t(($) => $[`${fieldPrefix}.IMChat.participants`], { ns: 'datasetDocuments' }),
        },
        topicKeywords: {
          label: t(($) => $[`${fieldPrefix}.IMChat.topicsKeywords`], { ns: 'datasetDocuments' }),
          inputType: 'textarea',
        },
        fileType: {
          label: t(($) => $[`${fieldPrefix}.IMChat.fileType`], { ns: 'datasetDocuments' }),
        },
      },
    },
    wikipedia_entry: {
      text: t(($) => $['metadata.type.wikipediaEntry'], { ns: 'datasetDocuments' }),
      allowEdit: false,
      subFieldsMap: {
        title: {
          label: t(($) => $[`${fieldPrefix}.wikipediaEntry.title`], { ns: 'datasetDocuments' }),
        },
        language: {
          label: t(($) => $[`${fieldPrefix}.wikipediaEntry.language`], { ns: 'datasetDocuments' }),
          inputType: 'select',
        },
        web_page_url: {
          label: t(($) => $[`${fieldPrefix}.wikipediaEntry.webpageURL`], {
            ns: 'datasetDocuments',
          }),
        },
        'editor/contributor': {
          label: t(($) => $[`${fieldPrefix}.wikipediaEntry.editorContributor`], {
            ns: 'datasetDocuments',
          }),
        },
        last_edit_date: {
          label: t(($) => $[`${fieldPrefix}.wikipediaEntry.lastEditDate`], {
            ns: 'datasetDocuments',
          }),
        },
        'summary/introduction': {
          label: t(($) => $[`${fieldPrefix}.wikipediaEntry.summaryIntroduction`], {
            ns: 'datasetDocuments',
          }),
          inputType: 'textarea',
        },
      },
    },
    synced_from_notion: {
      text: t(($) => $['metadata.type.notion'], { ns: 'datasetDocuments' }),
      allowEdit: false,
      subFieldsMap: {
        title: { label: t(($) => $[`${fieldPrefix}.notion.title`], { ns: 'datasetDocuments' }) },
        language: {
          label: t(($) => $[`${fieldPrefix}.notion.language`], { ns: 'datasetDocuments' }),
          inputType: 'select',
        },
        'author/creator': {
          label: t(($) => $[`${fieldPrefix}.notion.author`], { ns: 'datasetDocuments' }),
        },
        creation_date: {
          label: t(($) => $[`${fieldPrefix}.notion.createdTime`], { ns: 'datasetDocuments' }),
        },
        last_modified_date: {
          label: t(($) => $[`${fieldPrefix}.notion.lastModifiedTime`], { ns: 'datasetDocuments' }),
        },
        notion_page_link: {
          label: t(($) => $[`${fieldPrefix}.notion.url`], { ns: 'datasetDocuments' }),
        },
        'category/tags': {
          label: t(($) => $[`${fieldPrefix}.notion.tag`], { ns: 'datasetDocuments' }),
        },
        description: {
          label: t(($) => $[`${fieldPrefix}.notion.description`], { ns: 'datasetDocuments' }),
        },
      },
    },
    synced_from_github: {
      text: t(($) => $['metadata.type.github'], { ns: 'datasetDocuments' }),
      allowEdit: false,
      subFieldsMap: {
        repository_name: {
          label: t(($) => $[`${fieldPrefix}.github.repoName`], { ns: 'datasetDocuments' }),
        },
        repository_description: {
          label: t(($) => $[`${fieldPrefix}.github.repoDesc`], { ns: 'datasetDocuments' }),
        },
        'repository_owner/organization': {
          label: t(($) => $[`${fieldPrefix}.github.repoOwner`], { ns: 'datasetDocuments' }),
        },
        code_filename: {
          label: t(($) => $[`${fieldPrefix}.github.fileName`], { ns: 'datasetDocuments' }),
        },
        code_file_path: {
          label: t(($) => $[`${fieldPrefix}.github.filePath`], { ns: 'datasetDocuments' }),
        },
        programming_language: {
          label: t(($) => $[`${fieldPrefix}.github.programmingLang`], { ns: 'datasetDocuments' }),
        },
        github_link: {
          label: t(($) => $[`${fieldPrefix}.github.url`], { ns: 'datasetDocuments' }),
        },
        open_source_license: {
          label: t(($) => $[`${fieldPrefix}.github.license`], { ns: 'datasetDocuments' }),
        },
        commit_date: {
          label: t(($) => $[`${fieldPrefix}.github.lastCommitTime`], { ns: 'datasetDocuments' }),
        },
        commit_author: {
          label: t(($) => $[`${fieldPrefix}.github.lastCommitAuthor`], { ns: 'datasetDocuments' }),
        },
      },
    },
    originInfo: {
      text: '',
      allowEdit: false,
      subFieldsMap: {
        name: {
          label: t(($) => $[`${fieldPrefix}.originInfo.originalFilename`], {
            ns: 'datasetDocuments',
          }),
        },
        'data_source_info.upload_file.size': {
          label: t(($) => $[`${fieldPrefix}.originInfo.originalFileSize`], {
            ns: 'datasetDocuments',
          }),
          render: (value) => formatFileSize(value),
        },
        created_at: {
          label: t(($) => $[`${fieldPrefix}.originInfo.uploadDate`], { ns: 'datasetDocuments' }),
          render: (value) =>
            formatTimestamp(
              value,
              t(($) => $['metadata.dateTimeFormat'], { ns: 'datasetDocuments' }) as string,
            ),
        },
        completed_at: {
          label: t(($) => $[`${fieldPrefix}.originInfo.lastUpdateDate`], {
            ns: 'datasetDocuments',
          }),
          render: (value) =>
            formatTimestamp(
              value,
              t(($) => $['metadata.dateTimeFormat'], { ns: 'datasetDocuments' }) as string,
            ),
        },
        data_source_type: {
          label: t(($) => $[`${fieldPrefix}.originInfo.source`], { ns: 'datasetDocuments' }),
          render: (
            value: I18nKeysByPrefix<'datasetDocuments', 'metadata.source.'> | 'notion_import',
          ) =>
            t(($) => $[`metadata.source.${value === 'notion_import' ? 'notion' : value}`], {
              ns: 'datasetDocuments',
            }),
        },
      },
    },
    technicalParameters: {
      text: t(($) => $['metadata.type.technicalParameters'], { ns: 'datasetDocuments' }),
      allowEdit: false,
      subFieldsMap: {
        doc_form: {
          label: t(($) => $[`${fieldPrefix}.technicalParameters.segmentSpecification`], {
            ns: 'datasetDocuments',
          }),
          render: (value) => {
            if (value === ChunkingMode.text)
              return t(($) => $['chunkingMode.general'], { ns: 'dataset' })
            if (value === ChunkingMode.qa) return t(($) => $['chunkingMode.qa'], { ns: 'dataset' })
            if (value === ChunkingMode.parentChild)
              return t(($) => $['chunkingMode.parentChild'], { ns: 'dataset' })
            return '--'
          },
        },
        'dataset_process_rule.rules.segmentation.max_tokens': {
          label: t(($) => $[`${fieldPrefix}.technicalParameters.segmentLength`], {
            ns: 'datasetDocuments',
          }),
          render: (value) => formatNumber(value),
        },
        average_segment_length: {
          label: t(($) => $[`${fieldPrefix}.technicalParameters.avgParagraphLength`], {
            ns: 'datasetDocuments',
          }),
          render: (value) => `${formatNumber(value)} characters`,
        },
        segment_count: {
          label: t(($) => $[`${fieldPrefix}.technicalParameters.paragraphs`], {
            ns: 'datasetDocuments',
          }),
          render: (value) => `${formatNumber(value)} paragraphs`,
        },
        hit_count: {
          label: t(($) => $[`${fieldPrefix}.technicalParameters.hitCount`], {
            ns: 'datasetDocuments',
          }),
          render: (value, total) => {
            const v = value || 0
            return `${!total ? 0 : ((v / total) * 100).toFixed(2)}% (${v}/${total})`
          },
        },
        indexing_latency: {
          label: t(($) => $[`${fieldPrefix}.technicalParameters.embeddingTime`], {
            ns: 'datasetDocuments',
          }),
          render: (value) => formatTime(value),
        },
        tokens: {
          label: t(($) => $[`${fieldPrefix}.technicalParameters.embeddedSpend`], {
            ns: 'datasetDocuments',
          }),
          render: (value) => `${formatNumber(value)} tokens`,
        },
      },
    },
  }
}

const langPrefix = 'metadata.languageMap.'

export const useLanguages = () => {
  const { t } = useTranslation()
  return {
    zh: t(($) => $[`${langPrefix}zh`], { ns: 'datasetDocuments' }),
    en: t(($) => $[`${langPrefix}en`], { ns: 'datasetDocuments' }),
    es: t(($) => $[`${langPrefix}es`], { ns: 'datasetDocuments' }),
    fr: t(($) => $[`${langPrefix}fr`], { ns: 'datasetDocuments' }),
    de: t(($) => $[`${langPrefix}de`], { ns: 'datasetDocuments' }),
    ja: t(($) => $[`${langPrefix}ja`], { ns: 'datasetDocuments' }),
    ko: t(($) => $[`${langPrefix}ko`], { ns: 'datasetDocuments' }),
    ru: t(($) => $[`${langPrefix}ru`], { ns: 'datasetDocuments' }),
    ar: t(($) => $[`${langPrefix}ar`], { ns: 'datasetDocuments' }),
    pt: t(($) => $[`${langPrefix}pt`], { ns: 'datasetDocuments' }),
    it: t(($) => $[`${langPrefix}it`], { ns: 'datasetDocuments' }),
    nl: t(($) => $[`${langPrefix}nl`], { ns: 'datasetDocuments' }),
    pl: t(($) => $[`${langPrefix}pl`], { ns: 'datasetDocuments' }),
    sv: t(($) => $[`${langPrefix}sv`], { ns: 'datasetDocuments' }),
    tr: t(($) => $[`${langPrefix}tr`], { ns: 'datasetDocuments' }),
    he: t(($) => $[`${langPrefix}he`], { ns: 'datasetDocuments' }),
    hi: t(($) => $[`${langPrefix}hi`], { ns: 'datasetDocuments' }),
    da: t(($) => $[`${langPrefix}da`], { ns: 'datasetDocuments' }),
    fi: t(($) => $[`${langPrefix}fi`], { ns: 'datasetDocuments' }),
    no: t(($) => $[`${langPrefix}no`], { ns: 'datasetDocuments' }),
    hu: t(($) => $[`${langPrefix}hu`], { ns: 'datasetDocuments' }),
    el: t(($) => $[`${langPrefix}el`], { ns: 'datasetDocuments' }),
    cs: t(($) => $[`${langPrefix}cs`], { ns: 'datasetDocuments' }),
    th: t(($) => $[`${langPrefix}th`], { ns: 'datasetDocuments' }),
    id: t(($) => $[`${langPrefix}id`], { ns: 'datasetDocuments' }),
    ro: t(($) => $[`${langPrefix}ro`], { ns: 'datasetDocuments' }),
  }
}