vac-optimizer/src/app/services/calendar-service.ts

import { Injectable } from '@angular/core';
import { DateRange } from '@angular/material/datepicker';
import Holidays, { HolidaysTypes } from 'date-holidays';
import dayjs from 'dayjs';
import weekOfYear from 'dayjs/plugin/weekOfYear';
import isoWeek from 'dayjs/plugin/isoWeek';
import { CalendarSettingsService } from './calendar-settings-service';

dayjs.extend(weekOfYear);
dayjs.extend(isoWeek);

const lang = 'en';

const allTypesOptions: HolidaysTypes.Options = {
	languages: [lang],
};

@Injectable({
	providedIn: 'root',
})
export class CalendarService {
	// Cache for holiday timestamps
	private holidayTimestampsCache: Set<number> | null = null;
	private holidayCacheYear: number | null = null;

	// Get selected dates from current date to the next year
	getSelectedDatesFromNow(): SelectedDates {
		const holidays = this.getHolidays();
		let selectedDates: SelectedDates = new SelectedDates();
		// Logic to populate selectedDates from now until the next year
		return selectedDates;
	}

	getSelectedDatesForYear(): SelectedDates {
		const holidays = this.getHolidays();
		const monthInAYear = CalendarService.monthsInAYearFromNow();
		let selectedDates: SelectedDates = new SelectedDates();

		// Cache today's date components to avoid repeated new Date() calls
		const today = new Date();
		const todayDate = today.getDate();
		const todayMonth = today.getMonth();
		const todayYear = today.getFullYear();

		// Create a Set of holiday timestamps for O(1) lookup instead of O(n)
		const holidayTimestamps = new Set(holidays.map((h) => h.getTime()));

		for (const weeksInAMonth of monthInAYear) {
			for (const daysInAWeek of weeksInAMonth) {
				for (const day of daysInAWeek) {
					if (day) {
						this.setWeekendSelection(day, selectedDates);
						if (holidayTimestamps.has(day.getTime())) {
							selectedDates._datesSelected.push(
								new SelectedDate(DayType.CLOSED_DAY, new DateRange<Date>(day, day)),
							);
						}
						if (
							day.getDate() === todayDate &&
							day.getMonth() === todayMonth &&
							day.getFullYear() === todayYear
						) {
							selectedDates._datesSelected.push(
								new SelectedDate(DayType.TODAY, new DateRange<Date>(day, day)),
							);
						}
					}
				}
			}
		}
		selectedDates.update();
		return selectedDates;
	}

	setWeekendSelection(date: Date, selectedDates: SelectedDates) {
		if (date.getDay() === 6) {
			selectedDates._datesSelected.push(
				new SelectedDate(
					DayType.WEEKEND,
					new DateRange<Date>(
						date,
						new Date(date.getFullYear(), date.getMonth(), date.getDate() + 1),
					),
				),
			);
		}
		if (date.getDate() === 1 && date.getDay() === 0) {
			selectedDates._datesSelected.push(
				new SelectedDate(
					DayType.WEEKEND,
					new DateRange<Date>(
						new Date(date.getFullYear(), date.getMonth(), date.getDate() - 1),
						date,
					),
				),
			);
		}
	}

	getHolidays(): Date[] {
		// Cache holidays to avoid recalculation
		const currentYear = new Date().getFullYear();
		if (this.holidayCacheYear === currentYear && this.holidayTimestampsCache) {
			return Array.from(this.holidayTimestampsCache).map((ts) => new Date(ts));
		}

		// Logic to return holidays for the given year
		const hd = new Holidays('FR', allTypesOptions);
		const holidays = hd.getHolidays(currentYear, lang);
		const holidaysPlusOne = hd.getHolidays(currentYear + 1, lang);
		holidays.push(...holidaysPlusOne);
		let holidaysDate: Date[] = holidays.map((holiday) => new Date(holiday.date));

		// Update cache
		this.holidayTimestampsCache = new Set(holidaysDate.map((d) => d.getTime()));
		this.holidayCacheYear = currentYear;

		return holidaysDate;
	}

	static daysInWeek(weekIndex: number, monthIndex: number, year: number): Week {
		const firstDayOfMonth = new Date(year, monthIndex, 1);
		const firstDayOfWeek = new Date(firstDayOfMonth);
		let offset = firstDayOfMonth.getDay() - 1; // Adjust to make Monday the first day of the week
		if (offset < 0) {
			offset = 6; // Sunday case
		}
		firstDayOfWeek.setDate(firstDayOfMonth.getDate() + weekIndex * 7 - offset);

		const days: Week = [];
		for (let i = 0; i < 7; i++) {
			const currentDay = new Date(firstDayOfWeek);
			currentDay.setDate(firstDayOfWeek.getDate() + i);
			if (currentDay.getMonth() === monthIndex) {
				days.push(currentDay);
			} else {
				days.push(null);
			}
		}
		return days;
	}

	static weeksInMonth(monthIndex: number, year: number): Month {
		const weeks: Month = [];
		let weekIndex = 0;
		while (true) {
			const week = this.daysInWeek(weekIndex, monthIndex, year);
			if (week.every((day) => day === null)) {
				break;
			}
			weeks.push(week);
			weekIndex++;
		}
		return weeks;
	}

	static monthsInYear(year: number): Year {
		const months: Year = [];
		for (let monthIndex = 0; monthIndex < 12; monthIndex++) {
			months.push(this.weeksInMonth(monthIndex, year));
		}
		return months;
	}

	static monthsInAYearFromNow(): Year {
		const months: Year = [];
		const currentMonthIndex = new Date().getMonth();
		const currentYear = new Date().getFullYear();
		for (let i = 0; i < 12; i++) {
			const monthIndex = (currentMonthIndex + i) % 12;
			const year = currentYear + Math.floor((currentMonthIndex + i) / 12);
			months.push(this.weeksInMonth(monthIndex, year));
		}
		return months;
	}

	static getWeekNumber(date: Date): number {
		return dayjs(date).isoWeek();
	}
}

export type VacationsNumber = {
	rtt: number;
	cp: number;
	other: number;
};

export type Week = (Date | null)[];
export type Month = Week[];
export type Year = Month[];

export enum DayType {
	WEEKEND = 'weekend',
	CLOSED_DAY = 'closed-day',
	RTT = 'rtt',
	CP = 'cp',
	OTHER = 'other',
	TODAY = 'today',
}

export const tooltipTypeMapping: { [key in DayType]: string } = {
	[DayType.WEEKEND]: 'Weekend',
	[DayType.CLOSED_DAY]: 'Jour Férié',
	[DayType.RTT]: 'RTT',
	[DayType.CP]: 'Congé Payé',
	[DayType.OTHER]: 'Autre',
	[DayType.TODAY]: "Aujourd'hui",
};

export const monthNames: string[] = [
	'Janvier',
	'Février',
	'Mars',
	'Avril',
	'Mai',
	'Juin',
	'Juillet',
	'Août',
	'Septembre',
	'Octobre',
	'Novembre',
	'Décembre',
];

export interface SelectedDateInterface {
	isSelected(date: Date): boolean;

	isStart(date: Date): boolean;
	isEnd(date: Date): boolean;

	getClasses(date: Date): string;
}

export class SelectedDate implements SelectedDateInterface {
	type: DayType | null;
	range: DateRange<Date>;

	constructor(type: DayType | null, range: DateRange<Date>) {
		this.type = type;
		this.range = range;
	}

	isSelected(date: Date): boolean {
		return (
			this.type !== DayType.TODAY &&
			this.range.start !== null &&
			this.range.end !== null &&
			date >= this.range.start &&
			date <= this.range.end
		);
	}

	isStart(date: Date): boolean {
		return this.range.start !== null && date.getTime() === this.range.start.getTime();
	}

	isEnd(date: Date): boolean {
		return this.range.end !== null && date.getTime() === this.range.end.getTime();
	}

	getClasses(date: Date): string {
		let returnType: DayType | string = '';
		if (this.type && date >= this.range.start! && date <= this.range.end!) {
			returnType = returnType += this.type + ' ';
		}
		return returnType;
	}

	getTooltip(date: Date): string | null | undefined {
		if (this.isSelected(date)) {
			if (this.type) {
				return tooltipTypeMapping[this.type];
			}
		}
		return null;
	}

	getLengthInDays(): number {
		if (this.range.start && this.range.end) {
			const timeDiff = this.range.end.getTime() - this.range.start.getTime();
			return Math.floor(timeDiff / (1000 * 3600 * 24)) + 1;
		}
		return 0;
	}
}

export class SelectedDates implements SelectedDateInterface {
	_datesSelected: SelectedDate[] = [];
	datesSelected: SelectedDate[] = [];

	private minDate: Date | null = null;
	private maxDate: Date | null = null;

	update(): void {
		this.datesSelected = this._datesSelected.slice();
		this.grouping();
	}

	push(selectedDate: SelectedDate) {
		this._datesSelected.push(selectedDate);
		this.datesSelected = this._datesSelected.slice();
		this.updateMinMaxDates();
		this.grouping();
	}

	private updateMinMaxDates(): void {
		if (this._datesSelected.length === 0) {
			this.minDate = null;
			this.maxDate = null;
			return;
		}

		this.minDate = this._datesSelected.reduce((min, date) => {
			if (!date.range.start) return min;
			if (!min) return date.range.start;
			return date.range.start < min ? date.range.start : min;
		}, this._datesSelected[0].range.start);

		this.maxDate = this._datesSelected.reduce((max, date) => {
			if (!date.range.end) return max;
			if (!max) return date.range.end;
			return date.range.end > max ? date.range.end : max;
		}, this._datesSelected[0].range.end);
	}

	// make a "mother group" with type = null when two or more groups are adjacent or overlapping; keep the children groups
	grouping(): void {
		if (this._datesSelected.length <= 1) {
			return;
		}

		const oneDayInMs = 24 * 60 * 60 * 1000;

		// Sort dates by start date
		const sorted = [...this._datesSelected].sort((a, b) => {
			const aStart = a.range.start?.getTime() ?? 0;
			const bStart = b.range.start?.getTime() ?? 0;
			return aStart - bStart;
		});

		// Group adjacent or overlapping dates
		const groups: SelectedDate[][] = [];
		let currentGroup: SelectedDate[] = [sorted[0]];
		let currentGroupMaxEnd = sorted[0].range.end?.getTime() ?? 0;

		for (let i = 1; i < sorted.length; i++) {
			const current = sorted[i];
			const currStart = current.range.start?.getTime() ?? 0;
			const currEnd = current.range.end?.getTime() ?? 0;

			// Check if current date is adjacent or overlapping with the maximum end of the current group
			if (currStart - currentGroupMaxEnd <= oneDayInMs) {
				currentGroup.push(current);
				currentGroupMaxEnd = Math.max(currentGroupMaxEnd, currEnd);
			} else {
				if (currentGroup.length > 1) {
					groups.push(currentGroup);
				}
				currentGroup = [current];
				currentGroupMaxEnd = Math.max(currentGroupMaxEnd, currEnd);
			}
		}

		// Add the last group if it has more than one element
		if (currentGroup.length > 1) {
			groups.push(currentGroup);
		}

		// Create mother groups and add to result
		const result: SelectedDate[] = [...this._datesSelected];

		for (const group of groups) {
			const groupStart = Math.min(...group.map((d) => d.range.start?.getTime() ?? Infinity));
			const groupEnd = Math.max(...group.map((d) => d.range.end?.getTime() ?? 0));

			result.push(
				new SelectedDate(
					null,
					new DateRange<Date>(new Date(groupStart), new Date(groupEnd)),
				),
			);
		}

		this.datesSelected = result;
	}

	isSelected(date: Date): boolean {
		return this.datesSelected.some((selectedDate) => selectedDate.isSelected(date));
	}

	private getRangeBoundaryCounts(date: Date): { starts: number; ends: number } {
		let starts = 0;
		let ends = 0;
		for (const selected of this.datesSelected) {
			if (selected.range.start && date.getTime() === selected.range.start.getTime()) {
				starts++;
			}
			if (selected.range.end && date.getTime() === selected.range.end.getTime()) {
				ends++;
			}
		}
		return { starts, ends };
	}

	isStart(date: Date): boolean {
		if (this.existInARangeButNotABoundary(date)) return false;
		const { starts, ends } = this.getRangeBoundaryCounts(date);
		return starts >= ends;
	}

	isEnd(date: Date): boolean {
		if (this.existInARangeButNotABoundary(date)) return false;
		const { starts, ends } = this.getRangeBoundaryCounts(date);
		return ends >= starts;
	}

	/**
	 * Check whether a given date is selected and lies strictly inside a selected range
	 * (i.e., it is part of at least one selected range but is not the start or end of that range).
	 *
	 * This method returns true only if:
	 *  - `this.isSelected(dateTest)` is true (the date is considered selected), and
	 *  - there exists at least one element in `this.datesSelected` for which
	 *    `dateSelected.isStart(dateTest)` and `dateSelected.isEnd(dateTest)` are both false.
	 *
	 * Notes:
	 *  - The exact equality/comparison semantics are delegated to `isSelected`, `isStart`
	 *    and `isEnd` implementations on the respective objects.
	 *  - `dateTest` is expected to be a valid Date. Behavior for null/undefined/invalid Date
	 *    values is not defined here.
	 *
	 * @private
	 * @param dateTest - The date to test.
	 * @returns True if the date is an interior day of at least one selected range; otherwise false.
	 */
	private existInARangeButNotABoundary(dateTest: Date): boolean {
		return this.datesSelected.some(
			(dateSelected) =>
				dateSelected.isSelected(dateTest) &&
				!dateSelected.isStart(dateTest) &&
				!dateSelected.isEnd(dateTest),
		);
	}

	getClasses(date: Date): string {
		return this.datesSelected.map((dateSelected) => dateSelected.getClasses(date)).join(' ');
	}

	getTooltip(date: Date): string | null | undefined {
		return this.datesSelected
			.map((dateSelected) => dateSelected.getTooltip(date))
			.filter((tooltip) => tooltip)
			.join(' + ');
	}

	computeHeuristics(): number {
		if (this.datesSelected.length === 0) {
			return 0;
		}

		let totalScore = 0;

		// Sort all dates by start date to find next closest dates
		const sorted = [...this.datesSelected]
			.filter((date) => date.type !== null)
			.sort((a, b) => {
				const aStart = a.range.start?.getTime() ?? 0;
				const bStart = b.range.start?.getTime() ?? 0;
				return aStart - bStart;
			});

		const oneDayInMs = 24 * 60 * 60 * 1000;

		// PRIMARY: Optimize adjacency - reduce "little holes" between vacation periods
		for (let i = 0; i < sorted.length - 1; i++) {
			const current = sorted[i];
			const next = sorted[i + 1];

			// Use timestamps directly - dates should already be normalized
			const currentEndTime = current.range.end?.getTime() ?? 0;
			const nextStartTime = next.range.start?.getTime() ?? 0;

			// Check if adjacent (next day) - this is the best case!
			if (nextStartTime - currentEndTime === oneDayInMs) {
				totalScore += 10; // Big bonus for adjacency
			} else if (nextStartTime > currentEndTime) {
				// Penalty based on days between the two dates
				const daysBetween = Math.floor((nextStartTime - currentEndTime) / oneDayInMs) - 1;
				// Ensure daysBetween is non-negative before taking sqrt
				totalScore -= Math.sqrt(Math.max(0, daysBetween));
			}
		}

		// SECONDARY: Penalize uneven monthly distribution to encourage spread
		const monthCounts = new Map<number, number>(); // Use year*12+month as key to avoid strings
		for (const selectedDate of this.datesSelected) {
			if (!selectedDate.range.start || !selectedDate.range.end) continue;

			// Count days in this selected date range
			const daysInRange = selectedDate.getLengthInDays();

			// For each day in the range, add it to the appropriate month
			let currentDate = new Date(selectedDate.range.start);
			for (let i = 0; i < daysInRange; i++) {
				const monthKey = currentDate.getFullYear() * 12 + currentDate.getMonth();
				monthCounts.set(monthKey, (monthCounts.get(monthKey) ?? 0) + 1);
				currentDate.setDate(currentDate.getDate() + 1);
			}
		}

		// Apply small penalty for uneven distribution (secondary priority)
		for (const count of monthCounts.values()) {
			totalScore -= count; // Smaller penalty than gap filling
		}

		return totalScore;
	}

	optimizeVacations(
		vacationsNumber: VacationsNumber,
		calendarSettingsService: CalendarSettingsService,
	): void {
		if (calendarSettingsService.samediMalin()) this.samediMalin(vacationsNumber);
		while (vacationsNumber.cp > 0) {
			const result = this.lookForVacation(DayType.CP);
			if (result.heuristic === -1) {
				break;
			}
			vacationsNumber.cp -= 1;
		}
		while (vacationsNumber.rtt > 0) {
			const result = this.lookForVacation(DayType.RTT);
			if (result.heuristic === -1) {
				break;
			}
			vacationsNumber.rtt -= 1;
		}
		while (vacationsNumber.other > 0) {
			const result = this.lookForVacation(DayType.OTHER);
			if (result.heuristic === -1) {
				break;
			}
			vacationsNumber.other -= 1;
		}
	}

	samediMalin(vacationsNumber: VacationsNumber): void {
		const now = new Date();
		now.setHours(0, 0, 0, 0);
		const nowTime = now.getTime();

		const samedi_ferie = this._datesSelected.filter(
			(date) => date.type === DayType.CLOSED_DAY && date.range.start?.getDay() === 6,
		);

		for (const holiday of samedi_ferie) {
			if (!holiday.range.start) continue;

			// Skip if no CP days available
			if (vacationsNumber.cp < 2) {
				continue;
			}

			// Get the Saturday date (holiday) and the following Sunday
			const saturdayDate = new Date(holiday.range.start);
			const sundayDate = new Date(saturdayDate);
			sundayDate.setDate(saturdayDate.getDate() + 1);

			// Calculate dates for all three strategies (2 CP days total each)
			// Strategy 1: 2 days before (Thursday + Friday)
			const twoDaysBefore1 = new Date(saturdayDate);
			twoDaysBefore1.setDate(saturdayDate.getDate() - 2);
			const twoDaysBefore2 = new Date(saturdayDate);
			twoDaysBefore2.setDate(saturdayDate.getDate() - 1);

			// Strategy 2: 2 days after (Monday + Tuesday)
			const twoDaysAfter1 = new Date(sundayDate);
			twoDaysAfter1.setDate(sundayDate.getDate() + 1);
			const twoDaysAfter2 = new Date(sundayDate);
			twoDaysAfter2.setDate(sundayDate.getDate() + 2);

			// Strategy 3: 1 day before + 1 day after (Friday + Monday)
			const oneDayBefore = new Date(saturdayDate);
			oneDayBefore.setDate(saturdayDate.getDate() - 1);
			const oneDayAfter = new Date(sundayDate);
			oneDayAfter.setDate(sundayDate.getDate() + 1);

			// Check if we can apply each strategy (and not in the past)
			const canApplyTwoBeforeStrategy =
				twoDaysBefore1.getTime() >= nowTime &&
				!this.isSelected(twoDaysBefore1) &&
				!this.isSelected(twoDaysBefore2);

			const canApplyTwoAfterStrategy =
				twoDaysAfter1.getTime() >= nowTime &&
				!this.isSelected(twoDaysAfter1) &&
				!this.isSelected(twoDaysAfter2);

			const canApplyMixedStrategy =
				oneDayBefore.getTime() >= nowTime &&
				!this.isSelected(oneDayBefore) &&
				!this.isSelected(oneDayAfter);

			// Test all applicable strategies and choose the one with the best heuristic score
			const strategies: { apply: () => void; heuristic: number }[] = [];
			const tempDatesBackup = [...this._datesSelected];

			if (canApplyTwoBeforeStrategy) {
				this.push(
					new SelectedDate(
						DayType.CP,
						new DateRange<Date>(twoDaysBefore1, twoDaysBefore2),
					),
				);
				const heuristic = this.computeHeuristics();
				strategies.push({
					apply: () => {
						this._datesSelected = [...tempDatesBackup];
						this.push(
							new SelectedDate(
								DayType.CP,
								new DateRange<Date>(twoDaysBefore1, twoDaysBefore2),
							),
						);
						vacationsNumber.cp -= 2;
						vacationsNumber.other += 1; // Reward: gain 1 other day
					},
					heuristic: heuristic,
				});
				this._datesSelected = [...tempDatesBackup];
				this.datesSelected = this._datesSelected.slice();
				this.grouping();
			}

			if (canApplyTwoAfterStrategy) {
				this.push(
					new SelectedDate(DayType.CP, new DateRange<Date>(twoDaysAfter1, twoDaysAfter2)),
				);
				const heuristic = this.computeHeuristics();
				strategies.push({
					apply: () => {
						this._datesSelected = [...tempDatesBackup];
						this.push(
							new SelectedDate(
								DayType.CP,
								new DateRange<Date>(twoDaysAfter1, twoDaysAfter2),
							),
						);
						vacationsNumber.cp -= 2;
						vacationsNumber.other += 1; // Reward: gain 1 other day
					},
					heuristic: heuristic,
				});
				this._datesSelected = [...tempDatesBackup];
				this.datesSelected = this._datesSelected.slice();
				this.grouping();
			}

			if (canApplyMixedStrategy) {
				this.push(
					new SelectedDate(DayType.CP, new DateRange<Date>(oneDayBefore, oneDayBefore)),
				);
				this.push(
					new SelectedDate(DayType.CP, new DateRange<Date>(oneDayAfter, oneDayAfter)),
				);
				const heuristic = this.computeHeuristics();
				strategies.push({
					apply: () => {
						this._datesSelected = [...tempDatesBackup];
						this.push(
							new SelectedDate(
								DayType.CP,
								new DateRange<Date>(oneDayBefore, oneDayBefore),
							),
						);
						this.push(
							new SelectedDate(
								DayType.CP,
								new DateRange<Date>(oneDayAfter, oneDayAfter),
							),
						);
						vacationsNumber.cp -= 2;
						vacationsNumber.other += 1; // Reward: gain 1 other day
					},
					heuristic: heuristic,
				});
				this._datesSelected = [...tempDatesBackup];
				this.datesSelected = this._datesSelected.slice();
				this.grouping();
			}

			// Apply the strategy with the best heuristic score
			if (strategies.length > 0) {
				const bestStrategy = strategies.reduce((best, current) =>
					current.heuristic > best.heuristic ? current : best,
				);
				bestStrategy.apply();
			}
		}
	}

	lookForVacation(vacType: DayType): { apply: () => void; heuristic: number } {
		const daysOfYear = CalendarService.monthsInAYearFromNow();
		const now = new Date();
		now.setHours(0, 0, 0, 0);
		const nowTime = now.getTime();

		// Preprocess: filter out past days and already selected days in one pass
		const candidateDays: Date[] = [];
		for (const month of daysOfYear) {
			for (const week of month) {
				for (const day of week) {
					if (day) {
						day.setHours(0, 0, 0, 0);
						if (day.getTime() >= nowTime && !this.isSelected(day)) {
							candidateDays.push(day);
						}
					}
				}
			}
		}

		// Early exit if no candidates
		if (candidateDays.length === 0) {
			return { apply: () => {}, heuristic: -1 };
		}

		const tempDatesBackup = [...this._datesSelected];
		let bestStrategy: { apply: () => void; heuristic: number; value: Date } | null = null;
		let bestHeuristic = -Infinity;

		// Evaluate each candidate day
		for (const day of candidateDays) {
			// Reset to backup state
			this._datesSelected = [...tempDatesBackup];
			this.datesSelected = this._datesSelected.slice();
			this.grouping();

			// Mark this day as vacation
			const startDay = new Date(day);
			const endDay = new Date(day);
			this.datesSelected.push(
				new SelectedDate(vacType, new DateRange<Date>(startDay, endDay)),
			);

			const heuristic = this.computeHeuristics();

			// Track only if better than current best
			if (heuristic > bestHeuristic) {
				bestHeuristic = heuristic;
				const capturedDay = startDay;
				bestStrategy = {
					apply: () => {
						this._datesSelected = [...tempDatesBackup];
						this.push(
							new SelectedDate(
								vacType,
								new DateRange<Date>(capturedDay, capturedDay),
							),
						);
					},
					heuristic: heuristic,
					value: capturedDay,
				};
			}
		}

		// Restore backup before returning
		this._datesSelected = [...tempDatesBackup];
		this.datesSelected = this._datesSelected.slice();
		this.grouping();

		// Apply best strategy if found
		if (bestStrategy) {
			bestStrategy.apply();
			return bestStrategy;
		}

		return { apply: () => {}, heuristic: -1 };
	}
}