import { Redis } from '@upstash/redis'

export interface VisitorStats {
  total: number       // 总访问量
  today: number       // 今日访问量
  online: number      // 当前在线人数
  todayRecords: Record<string, { count: number; lastVisit: number }>  // 今日访客记录
  lastResetTime: number  // 上次重置时间（用于判断是否跨天）
}

const ONLINE_TIMEOUT = 300000  // 在线超时时间：5分钟

// 默认统计数据
const defaultStats: VisitorStats = {
  total: 0,
  today: 0,
  online: 0,
  todayRecords: {},
  lastResetTime: Date.now(),
}

// 创建Redis客户端（自动从环境变量读取配置）
const redis = Redis.fromEnv()

// 判断是否跨天（用于重置今日统计）
function isNewDay(lastReset: number): boolean {
  const lastDate = new Date(lastReset)
  const now = new Date()
  return (
    lastDate.getDate() !== now.getDate() ||
    lastDate.getMonth() !== now.getMonth() ||
    lastDate.getFullYear() !== now.getFullYear()
  )
}

// 计算当前在线人数（5分钟内有活动的访客）
function calculateOnline(stats: VisitorStats): number {
  const now = Date.now()
  let onlineCount = 0
  for (const record of Object.values(stats.todayRecords)) {
    if (now - record.lastVisit <= ONLINE_TIMEOUT) {
      onlineCount++
    }
  }
  return onlineCount
}

// 记录访客访问
export async function recordVisit(ip: string, visitorId?: string): Promise<VisitorStats> {
  // 从Redis获取当前统计数据
  let stats = (await redis.get<VisitorStats>('visitor-stats')) || defaultStats
  
  // 如果跨天了，重置今日统计
  if (isNewDay(stats.lastResetTime)) {
    stats = {
      total: stats.total,  // 总访问量保留
      today: 0,
      online: 0,
      todayRecords: {},
      lastResetTime: Date.now(),
    }
  }
  
  const now = Date.now()
  // 使用访客ID（优先）或IP地址作为唯一标识
  const recordKey = visitorId ? `vid:${visitorId}` : `ip:${ip}`
  
  // 判断是否是新访客
  if (!stats.todayRecords[recordKey]) {
    stats.total++  // 总访问量+1
    stats.todayRecords[recordKey] = { count: 1, lastVisit: now }
  } else {
    // 老访客只更新时间戳
    stats.todayRecords[recordKey].lastVisit = now
  }
  
  // 重新计算今日访问量和在线人数
  stats.today = Object.keys(stats.todayRecords).length
  stats.online = calculateOnline(stats)
  
  // 保存到Redis
  await redis.set('visitor-stats', stats)
  return stats
}

// 获取统计数据
export async function getStats(): Promise<VisitorStats> {
  let stats = (await redis.get<VisitorStats>('visitor-stats')) || defaultStats
  
  // 跨天重置
  if (isNewDay(stats.lastResetTime)) {
    stats = {
      total: stats.total,
      today: 0,
      online: 0,
      todayRecords: {},
      lastResetTime: Date.now(),
    }
    await redis.set('visitor-stats', stats)
  }
  
  stats.today = Object.keys(stats.todayRecords).length
  stats.online = calculateOnline(stats)
  
  return stats
}