import {
  fetchMineProductionWorkOrderTabStepCountsDeduped,
  fetchProductionWorkOrderTabStepCountsDeduped,
  sumProductionWorkOrderTabStepCounts
} from '@/api/production-work-order'
import {
  PRODUCTION_MENU_CODE_MY_TASK,
  PRODUCTION_MENU_CODE_WORK_ORDER,
  useProductionMenuBadgeStore
} from '@/store/modules/production-menu-badge'
import { useMenuStore } from '@/store/modules/menu'
import type { AppRouteRecord } from '@/types/router'
import { PRODUCTION_WORK_ORDER_VISIBLE_TABS } from '@/utils/production-work-order-tabs'

const VISIBLE_TAB_KEYS = PRODUCTION_WORK_ORDER_VISIBLE_TABS
  .filter((tab) => tab.key !== 'all')
  .map((tab) => tab.key)

/** 解析菜单项对应的生产角标 menuCode（兼容后端菜单未带 menuCode） */
export function resolveProductionMenuBadgeCode(item: AppRouteRecord): string | undefined {
  const code = (item.meta.menuCode || '').trim()
  if (code === PRODUCTION_MENU_CODE_WORK_ORDER || code === PRODUCTION_MENU_CODE_MY_TASK) {
    return code
  }
  const name = String(item.name || '')
  if (name === 'ProductionWorkOrder') return PRODUCTION_MENU_CODE_WORK_ORDER
  if (name === 'ProductionMyProductionTask') return PRODUCTION_MENU_CODE_MY_TASK
  const path = String(item.path || '')
  if (path.endsWith('/production-work-order') || path === 'production-work-order') {
    return PRODUCTION_MENU_CODE_WORK_ORDER
  }
  if (path.endsWith('/my-production-task') || path === 'my-production-task') {
    return PRODUCTION_MENU_CODE_MY_TASK
  }
  return undefined
}

/** 从当前侧栏菜单树收集需展示角标的生产菜单 code */
export function collectVisibleProductionMenuBadgeCodes(menuList: AppRouteRecord[]): string[] {
  const codes = new Set<string>()
  const walk = (items: AppRouteRecord[]) => {
    for (const item of items) {
      const code = resolveProductionMenuBadgeCode(item)
      if (code) codes.add(code)
      if (item.children?.length) walk(item.children)
    }
  }
  walk(menuList || [])
  return [...codes]
}

/** 刷新侧栏「生产工单 / 我的生产任务」Tab 角标汇总 */
export async function refreshProductionMenuBadges() {
  const menuStore = useMenuStore()
  const badgeStore = useProductionMenuBadgeStore()
  const visibleCodes = collectVisibleProductionMenuBadgeCodes(menuStore.menuList)

  const tasks: Promise<void>[] = []

  if (visibleCodes.includes(PRODUCTION_MENU_CODE_WORK_ORDER)) {
    tasks.push(
      fetchProductionWorkOrderTabStepCountsDeduped()
        .then((data) =>
          badgeStore.setBadge(
            PRODUCTION_MENU_CODE_WORK_ORDER,
            sumProductionWorkOrderTabStepCounts(data, VISIBLE_TAB_KEYS)
          )
        )
        .catch(() =>
          badgeStore.setBadge(PRODUCTION_MENU_CODE_WORK_ORDER, {
            overdueIncomplete: 0,
            futureIncomplete: 0
          })
        )
    )
  }

  if (visibleCodes.includes(PRODUCTION_MENU_CODE_MY_TASK)) {
    tasks.push(
      fetchMineProductionWorkOrderTabStepCountsDeduped()
        .then((data) =>
          badgeStore.setBadge(
            PRODUCTION_MENU_CODE_MY_TASK,
            sumProductionWorkOrderTabStepCounts(data, VISIBLE_TAB_KEYS)
          )
        )
        .catch(() =>
          badgeStore.setBadge(PRODUCTION_MENU_CODE_MY_TASK, {
            overdueIncomplete: 0,
            futureIncomplete: 0
          })
        )
    )
  }

  if (!tasks.length) return

  await Promise.all(tasks)
}