import type { Ref } from 'vue'

export type DateShortcutType = 'all' | 'today' | 'tomorrow' | 'yesterday' | 'week'

function formatYmd(date: Date): string {
  const year = date.getFullYear()
  const month = String(date.getMonth() + 1).padStart(2, '0')
  const day = String(date.getDate()).padStart(2, '0')
  return `${year}-${month}-${day}`
}

/**
 * 列表工具栏计划日期快捷筛选（全部 / 昨日 / 今日 / 明日 / 本周）
 */
export function useListDateShortcut(dateRange: Ref<string[]>, onSearch: () => void) {
  const activeDateShortcut = ref<DateShortcutType>('all')

  function handleDateShortcut(type: DateShortcutType) {
    activeDateShortcut.value = type
    const today = new Date()

    if (type === 'all') {
      dateRange.value = []
    } else if (type === 'today') {
      const d = formatYmd(today)
      dateRange.value = [d, d]
    } else if (type === 'tomorrow') {
      const t = new Date(today)
      t.setDate(t.getDate() + 1)
      const d = formatYmd(t)
      dateRange.value = [d, d]
    } else if (type === 'yesterday') {
      const y = new Date(today)
      y.setDate(y.getDate() - 1)
      const d = formatYmd(y)
      dateRange.value = [d, d]
    } else if (type === 'week') {
      const start = new Date(today)
      const day = start.getDay() || 7
      start.setDate(start.getDate() - day + 1)
      const end = new Date(start)
      end.setDate(start.getDate() + 6)
      dateRange.value = [formatYmd(start), formatYmd(end)]
    }

    onSearch()
  }

  function resetDateShortcut() {
    activeDateShortcut.value = 'all'
    dateRange.value = []
  }

  return { activeDateShortcut, handleDateShortcut, resetDateShortcut }
}