/**
 * A股交易日历：基于 chinese-days 节假日数据，严格判断某日期是否为交易日
 * 交易日 = 周一至周五 且 不在法定节假日
 * 调休补班日（周末变工作日）A股仍休市，故不视为交易日
 */

const CDN_BASE = 'https://cdn.jsdelivr.net/npm/chinese-days@1/dist/years';
const yearCache = new Map(); // year -> Set<dateStr> (holidays)

/**
 * 加载某年的节假日数据
 * @param {number} year
 * @returns {Promise<Set<string>>} 节假日日期集合，格式 YYYY-MM-DD
 */
export async function loadHolidaysForYear(year) {
  if (yearCache.has(year)) {
    return yearCache.get(year);
  }
  try {
    const res = await fetch(`${CDN_BASE}/${year}.json`);
    if (!res.ok) throw new Error(`HTTP ${res.status}`);
    const data = await res.json();
    const holidays = new Set(Object.keys(data?.holidays ?? {}));
    yearCache.set(year, holidays);
    return holidays;
  } catch (e) {
    console.warn(`[tradingCalendar] 加载 ${year} 年节假日失败:`, e);
    yearCache.set(year, new Set());
    return yearCache.get(year);
  }
}

/**
 * 加载多个年份的节假日数据
 * @param {number[]} years
 */
export async function loadHolidaysForYears(years) {
  await Promise.all([...new Set(years)].map(loadHolidaysForYear));
}

/**
 * 判断某日期是否为 A股交易日
 * @param {dayjs.Dayjs} date - dayjs 对象
 * @param {Map<number, Set<string>>} [cache] - 可选，已加载的年份缓存，默认使用内部 yearCache
 * @returns {boolean}
 */
export function isTradingDay(date, cache = yearCache) {
  const dayOfWeek = date.day(); // 0=周日, 6=周六
  if (dayOfWeek === 0 || dayOfWeek === 6) return false;

  const dateStr = date.format('YYYY-MM-DD');
  const year = date.year();
  const holidays = cache.get(year);
  if (!holidays) return true; // 未加载该年数据时，仅排除周末
  return !holidays.has(dateStr);
}