<template>
  <el-row :gutter="20">
    <el-col :span="6" :xs="24">
      <ContentWrap class="h-1/1">
        <el-input
          v-model="keyword"
          clearable
          :placeholder="t('DataCollection.DeviceParamAnalysis.treeSearchPlaceholder')"
          class="!w-1/1"
          @input="handleKeywordChange"
        />
        <div class="mt-12px">
          <el-tree
            ref="treeRef"
            v-loading="treeLoading"
            :data="treeData"
            :props="treeProps"
            node-key="id"
            show-checkbox
            check-strictly
            @check="handleTreeCheck"
          />
        </div>
      </ContentWrap>
    </el-col>

    <el-col :span="18" :xs="24">
      <div class="analysis-groups">
        <ContentWrap v-if="!selectedGroups.length">
          <el-empty :description="t('DataCollection.DeviceParamAnalysis.emptySelectNodeDescription')" />
        </ContentWrap>

        <ContentWrap v-for="group in selectedGroups" :key="group.id">
          <el-form class="-mb-15px device-param-analysis-form" :inline="true" label-width="auto">
            <el-form-item :label="t('DataCollection.DeviceParamAnalysis.formTimeLabel')">
              <el-date-picker
                v-model="group.dateRange"
                type="daterange"
                :start-placeholder="t('DataCollection.DeviceParamAnalysis.formTimeStartPlaceholder')"
                :end-placeholder="t('DataCollection.DeviceParamAnalysis.formTimeEndPlaceholder')"
                value-format="YYYY-MM-DD"
                :shortcuts="dateShortcuts"
                :default-time="[new Date('1 00:00:00'), new Date('1 23:59:59')]"
                class="!w-360px"
              />
            </el-form-item>
            <el-form-item>
              <el-button :disabled="group.chartLoading" @click="handleQuery(group)">
                <Icon icon="ep:search" class="mr-5px" />
                {{ t('DataCollection.DeviceParamAnalysis.searchButtonText') }}
              </el-button>
              <el-button :disabled="group.chartLoading" @click="resetQuery(group)">
                <Icon icon="ep:refresh" class="mr-5px" />
                {{ t('DataCollection.DeviceParamAnalysis.resetButtonText') }}
              </el-button>
            </el-form-item>
          </el-form>

          <el-alert :title="buildParamTitle(group.param)" type="info" :closable="false" class="mb-12px" />
          <div v-loading="group.chartLoading">
            <el-empty
              v-if="group.chartState === 'empty'"
              :description="t('DataCollection.DeviceParamAnalysis.emptyDescription')"
            />
            <EChart
              v-else-if="group.chartState === 'ready'"
              :key="group.chartRenderKey"
              :options="buildChartOption(group)"
              height="360px"
            />
          </div>
        </ContentWrap>
      </div>
    </el-col>
  </el-row>
</template>

<script setup lang="ts">
import type { EChartsOption } from 'echarts'
import dayjs from 'dayjs'
import { Echart as EChart } from '@/components/Echart'
import { OrganizationApi } from '@/api/mes/organization'
import { DeviceModelAttributeApi } from '@/api/iot/devicemodelattribute'
import { handleTree } from '@/utils/tree'

defineOptions({ name: 'DeviceParamAnalysis' })

const { t } = useI18n()

type TreeNodeType = 'device' | 'param'

type DeviceTreeNode = {
  id: string
  label: string
  type: TreeNodeType
  children?: DeviceTreeNode[]
  deviceId?: number
  modelId?: number
  paramKey?: string
  unit?: string
  paramCount?: number
  disabled?: boolean
}

type ApiTreeParameter = {
  id: number
  name: string
  code?: string
  unit?: string | null
}

type ApiTreeEquipment = {
  id: number
  name: string
  parameters?: ApiTreeParameter[]
}

type ApiTreeOrg = {
  id: number
  name: string
  orgClass?: string
  parentId?: number | null
  equipments?: ApiTreeEquipment[]
  children?: ApiTreeOrg[]
}

const message = useMessage()

const treeRef = ref()
const treeLoading = ref(false)
const keyword = ref('')
const treeProps = { children: 'children', label: 'label', disabled: 'disabled' }
const treeData = ref<DeviceTreeNode[]>([])

const buildDefaultDateRange = (): [string, string] => {
  const end = dayjs().endOf('day')
  const start = end.subtract(6, 'day').startOf('day')
  return [start.format('YYYY-MM-DD'), end.format('YYYY-MM-DD')]
}

const dateShortcuts = [
  {
    text: t('DataCollection.DeviceParamAnalysis.shortcutLast7Days'),
    value: () => {
      const end = dayjs().endOf('day').toDate()
      const start = dayjs().subtract(6, 'day').startOf('day').toDate()
      return [start, end]
    }
  },
  {
    text: t('DataCollection.DeviceParamAnalysis.shortcutLastWeek'),
    value: () => {
      const start = dayjs().subtract(1, 'week').startOf('week').toDate()
      const end = dayjs().subtract(1, 'week').endOf('week').toDate()
      return [start, end]
    }
  },
  {
    text: t('DataCollection.DeviceParamAnalysis.shortcutLastMonth'),
    value: () => {
      const start = dayjs().subtract(1, 'month').startOf('month').toDate()
      const end = dayjs().subtract(1, 'month').endOf('month').toDate()
      return [start, end]
    }
  },
  {
    text: t('DataCollection.DeviceParamAnalysis.shortcutLast3Months'),
    value: () => {
      const end = dayjs().endOf('day').toDate()
      const start = dayjs().subtract(3, 'month').startOf('day').toDate()
      return [start, end]
    }
  }
]

type ChartState = 'idle' | 'loading' | 'empty' | 'ready'

type SelectedGroup = {
  id: string
  param: DeviceTreeNode
  deviceId?: number
  modelId?: number
  dateRange: [string, string]
  requestSeq: number
  chartLoading: boolean
  chartState: ChartState
  chartXAxis: string[]
  chartSeries: Array<{ name: string; data: Array<number | null> }>
  chartRenderKey: number
}

const selectedGroups = ref<SelectedGroup[]>([])

let keywordTimer: number | undefined
const handleKeywordChange = () => {
  if (keywordTimer) window.clearTimeout(keywordTimer)
  keywordTimer = window.setTimeout(() => {
    keywordTimer = undefined
    loadTree()
  }, 300)
}

const toFiniteId = (value: any): number | undefined => {
  if (typeof value === 'number') return Number.isFinite(value) ? value : undefined
  if (typeof value === 'string') {
    if (/^\d+$/.test(value.trim())) return Number(value)
    const n = Number(value)
    return Number.isFinite(n) ? n : undefined
  }
  return undefined
}

const toNodeIds = (node: DeviceTreeNode): { deviceId?: number; modelId?: number } => {
  const modelId = typeof node.modelId === 'number' && Number.isFinite(node.modelId) ? node.modelId : undefined
  const deviceId = typeof node.deviceId === 'number' && Number.isFinite(node.deviceId) ? node.deviceId : undefined
  if (typeof deviceId === 'number') return { deviceId, modelId }

  const parts = String(node.id ?? '').split('-').filter(Boolean)
  const last = parts.length ? parts[parts.length - 1] : undefined
  const parsed = toFiniteId(last)
  if (typeof parsed !== 'number' || !Number.isFinite(parsed) || parsed <= 0) {
    return { modelId }
  }
  return { deviceId: parsed, modelId }
}

const buildParamTitle = (param: DeviceTreeNode) => {
  const unitText = param.unit ? ` (${param.unit})` : ''
  return t('DataCollection.DeviceParamAnalysis.selectedParamTitle', {
    label: param.label,
    unit: unitText
  })
}

const buildChartOption = (group: SelectedGroup): EChartsOption => {
  const unit = group.param?.unit
  return {
    tooltip: { trigger: 'axis' },
    legend: { top: 10, left: 'center', type: 'scroll' },
    grid: { left: 50, right: 60, top: 80, bottom: 40, containLabel: true },
    xAxis: {
      type: 'category',
      boundaryGap: false,
      data: group.chartXAxis
    },
    yAxis: {
      type: 'value',
      name: unit || ''
    },
    series: group.chartSeries.map((s) => ({
      type: 'line',
      name: s.name,
      smooth: false,
      showSymbol: false,
      data: s.data
    }))
  }
}

const buildTreeFromApi = (orgs: ApiTreeOrg[]): DeviceTreeNode[] => {
  const normalizeOrg = (org: any): ApiTreeOrg => {
    const children = Array.isArray(org?.children) ? org.children.map(normalizeOrg) : undefined
    const equipments = Array.isArray(org?.equipments) ? org.equipments : undefined
    const parentId = typeof org?.parentId === 'number' ? org.parentId : Number(org?.parentId ?? 0) || 0
    return {
      id: Number(org?.id ?? 0) || 0,
      name: String(org?.name ?? ''),
      orgClass: org?.orgClass ? String(org.orgClass) : undefined,
      parentId,
      equipments,
      children
    }
  }

  const normalizedOrgs: ApiTreeOrg[] = Array.isArray(orgs) ? orgs.map(normalizeOrg) : []
  const isNestedTree = normalizedOrgs.some((o) => typeof (o as any)?.children !== 'undefined')
  const orgTree = isNestedTree ? normalizedOrgs : (handleTree(normalizedOrgs, 'id', 'parentId', 'children') as ApiTreeOrg[])

  const toOrgNode = (org: ApiTreeOrg): DeviceTreeNode => {
    const orgChildren = Array.isArray(org?.children) ? org.children.map(toOrgNode) : []

    const equipments = Array.isArray(org?.equipments) ? org.equipments : []
    const equipmentNodes: DeviceTreeNode[] = equipments.map((eq) => {
      const params = Array.isArray(eq?.parameters) ? eq.parameters : []
      const paramNodes: DeviceTreeNode[] = params.map((p) => ({
        id: `param-${eq.id}-${p.id}`,
        label: p?.name ?? p?.code ?? String(p?.id ?? ''),
        type: 'param',
        deviceId: toFiniteId(eq.id),
        modelId: toFiniteId(p.id),
        paramKey: p?.code ? String(p.code) : undefined,
        unit: p?.unit ? String(p.unit) : undefined,
        disabled: false
      }))

      return {
        id: `equipment-${org.id}-${eq.id}`,
        label: eq?.name ?? String(eq?.id ?? ''),
        type: 'device',
        deviceId: toFiniteId(eq.id),
        paramCount: params.length,
        children: paramNodes.length ? paramNodes : undefined,
        disabled: true
      }
    })

    const children = [...orgChildren, ...equipmentNodes]
    return {
      id: `org-${org.id}`,
      label: org?.name ?? String(org?.id ?? ''),
      type: 'device',
      children: children.length ? children : undefined,
      disabled: true
    }
  }

  return Array.isArray(orgTree) ? orgTree.map(toOrgNode) : []
}

const extractApiOrgs = (res: any): ApiTreeOrg[] => {
  if (Array.isArray(res)) return res as ApiTreeOrg[]
  if (Array.isArray(res?.data)) return res.data as ApiTreeOrg[]
  if (Array.isArray(res?.result)) return res.result as ApiTreeOrg[]
  if (Array.isArray(res?.list)) return res.list as ApiTreeOrg[]
  return []
}

const loadTree = async () => {
  treeLoading.value = true
  try {
    const res = await OrganizationApi.deviceParameterAnalysis({ keyword: keyword.value || undefined, showDevices: 1 })
    treeData.value = buildTreeFromApi(extractApiOrgs(res))
    if (keyword.value) {
      treeRef.value?.setCurrentKey?.(undefined)
      treeRef.value?.setCheckedKeys?.([])
      selectedGroups.value = []
    }
  } catch {
    message.error(t('DataCollection.DeviceParamAnalysis.messageLoadTreeFailed'))
  } finally {
    treeLoading.value = false
  }
}

const ensureDateRange = (group: SelectedGroup) => {
  if (!group.dateRange || group.dateRange.length !== 2) {
    group.dateRange = buildDefaultDateRange()
  }
}

const resetChartData = (group: SelectedGroup) => {
  group.chartXAxis = []
  group.chartSeries = []
  group.chartRenderKey++
}

const toNumber = (value: any) => {
  if (typeof value === 'number') return value
  if (typeof value === 'string') {
    const n = Number(value)
    return Number.isFinite(n) ? n : 0
  }
  if (Array.isArray(value)) return 0
  if (value && typeof value === 'object') {
    const maybe = value.value ?? value.val ?? value.count ?? value.amount
    const n = Number(maybe)
    return Number.isFinite(n) ? n : 0
  }
  return 0
}

const extractChartRows = (res: any): Record<string, any>[] => {
  if (Array.isArray(res)) return res as Record<string, any>[]
  if (Array.isArray(res?.data)) return res.data as Record<string, any>[]
  if (Array.isArray(res?.result)) return res.result as Record<string, any>[]
  if (Array.isArray(res?.list)) return res.list as Record<string, any>[]
  return []
}

const fetchGroupChart = async (group: SelectedGroup, notify: boolean) => {
  if (typeof group.deviceId !== 'number' || !Number.isFinite(group.deviceId)) return

  ensureDateRange(group)
  const [start, end] = group.dateRange
  if (!start || !end) return

  const buildDateTime = (date: string, suffix: string) => {
    return date.includes(':') ? date : `${date} ${suffix}`
  }

  const requestSeq = ++group.requestSeq
  group.chartState = 'loading'
  group.chartLoading = true
  resetChartData(group)
  try {
    const req: Record<string, any> = {
      deviceId: group.deviceId,
      collectionStartTime: buildDateTime(start, '00:00:00'),
      collectionEndTime: buildDateTime(end, '23:59:59')
    }
    if (typeof group.modelId === 'number' && Number.isFinite(group.modelId)) {
      req.modelId = group.modelId
    }
    const data = await DeviceModelAttributeApi.operationAnalysisDetails(req as any)
    if (requestSeq !== group.requestSeq) return
    const rows: Record<string, any>[] = extractChartRows(data)
    if (!rows.length) {
      group.chartXAxis = []
      group.chartSeries = []
      group.chartState = 'empty'
      if (notify) {
        message.warning(t('DataCollection.DeviceParamAnalysis.messageNodeNoParams'))
      }
      return
    }

    const firstRow = rows[0] || {}
    const keys = Object.keys(firstRow)
    const xKey = keys.find((k) => /time|date|时间|日期/i.test(k))
    if (xKey) {
      const sortedRows = [...rows].sort((a, b) =>
        String(a?.[xKey] ?? '').localeCompare(String(b?.[xKey] ?? ''))
      )
      if (requestSeq !== group.requestSeq) return
      group.chartXAxis = sortedRows.map((r) => String(r?.[xKey] ?? ''))

      const seriesKeys = keys.filter((k) => k !== xKey)
      const seriesMap = new Map<string, Array<number | null>>()
      const ensureSeries = (name: string) => {
        if (!seriesMap.has(name)) {
          seriesMap.set(name, Array(sortedRows.length).fill(null))
        }
        return seriesMap.get(name)!
      }

      sortedRows.forEach((row, rowIndex) => {
        seriesKeys.forEach((k) => {
          const v = row?.[k]
          if (Array.isArray(v)) {
            v.forEach((item) => {
              const name = String(item?.attributeName ?? k)
              const series = ensureSeries(name)
              series[rowIndex] = toNumber(item?.addressValue ?? item?.value ?? item?.val)
            })
            return
          }
          const series = ensureSeries(k)
          series[rowIndex] = toNumber(v)
        })
      })

      group.chartSeries = Array.from(seriesMap.entries()).map(([name, data]) => ({ name, data }))
    } else {
      group.chartXAxis = keys
      const values = keys.map((k) => toNumber(firstRow[k]))
      group.chartSeries = [
        { name: group.param?.label || t('DataCollection.DeviceParamAnalysis.defaultSeriesName'), data: values }
      ]
    }

    group.chartState = group.chartXAxis.length && group.chartSeries.length ? 'ready' : 'empty'
  } catch {
    if (notify) {
      message.error(t('DataCollection.DeviceParamAnalysis.messageFetchChartFailed'))
    }
    group.chartState = 'idle'
  } finally {
    if (requestSeq === group.requestSeq) {
      group.chartLoading = false
    }
  }
}

const syncGroupsByCheckedNodes = (checkedNodes: DeviceTreeNode[]) => {
  const ids = checkedNodes.map((n) => String(n.id))
  const existingIds = new Set(selectedGroups.value.map((g) => g.id))
  const nextIds = new Set(ids)

  selectedGroups.value = selectedGroups.value.filter((g) => nextIds.has(g.id))

  const toAdd = checkedNodes.filter((n) => !existingIds.has(String(n.id)))
  const addedGroups: SelectedGroup[] = []
  for (const node of toAdd) {
    const { deviceId, modelId } = toNodeIds(node)
    if (typeof deviceId !== 'number') {
      continue
    }
    const group: SelectedGroup = {
      id: String(node.id),
      param: node,
      deviceId,
      modelId,
      dateRange: buildDefaultDateRange(),
      requestSeq: 0,
      chartLoading: false,
      chartState: 'idle',
      chartXAxis: [],
      chartSeries: [],
      chartRenderKey: 0
    }
    selectedGroups.value.push(group)
    addedGroups.push(group)
  }
  return addedGroups
}

const handleTreeCheck = async () => {
  const checkedNodes = (treeRef.value?.getCheckedNodes?.(true) ?? []) as DeviceTreeNode[]
  const addedGroups = syncGroupsByCheckedNodes(checkedNodes)
  const needFetchGroups = selectedGroups.value.filter((g) => g.chartState === 'idle' && !g.chartLoading)
  const groups = [...new Set([...addedGroups, ...needFetchGroups])]
  await Promise.allSettled(groups.map((g) => fetchGroupChart(g, false)))
}

const handleQuery = async (group: SelectedGroup) => {
  await fetchGroupChart(group, true)
}

const resetQuery = async (group: SelectedGroup) => {
  group.dateRange = buildDefaultDateRange()
  await fetchGroupChart(group, true)
}

onMounted(async () => {
  await loadTree()
})
</script>

<style scoped lang="scss">
.device-param-analysis-form{
    margin-bottom: 0px
}
.device-param-analysis-form :deep(.el-form-item__label) {
  min-width: 100px;
}

.analysis-groups {
  display: flex;
  flex-direction: column;
  gap: 12px;
}

:deep(.el-tree) {
  max-height: calc(100vh - 280px);
  overflow: auto;
}
</style>
