import { InsufficientPermissions, formatDate, getDayOfWeekIndex, getFirstWorkDayOfWeek, getHumanDate, getIsWeekend, getWeekNumber } from "./utils";
import getStorage from "./storage";
import { getMenuSladovnicka, getMenuTechTower, getMenuUMotliku, getMenuZastavkaUmichala, getMenuSenkSerikova } from "./restaurants";
import { getTodayMock } from "./mock";
import { ClientData, DepartureTime, LunchChoice, Restaurant, RestaurantDayMenu, WeekMenu } from "../../types";

const storage = getStorage();
const MENU_PREFIX = 'menu';

/** Vrátí dnešní datum, případně fiktivní datum pro účely vývoje a testování. */
export function getToday(): Date {
    if (process.env.MOCK_DATA === 'true') {
        return getTodayMock();
    }
    return new Date();
}

/** Vrátí datum v aktuálním týdnu na základě předaného indexu (0 = pondělí). */
export const getDateForWeekIndex = (index: number) => {
    if (index < 0 || index > 4) {
        // Nechceme shodit server, vrátíme dnešek
        console.log('Neplatný index dne v týdnu: ' + index);
        return getToday();
    }
    const date = getToday();
    date.setDate(date.getDate() - getDayOfWeekIndex(date) + index);
    return date;
}

/** Vrátí "prázdná" (implicitní) data pro předaný den. */
function getEmptyData(date?: Date): ClientData {
    const usedDate = date || getToday();
    return {
        todayDayIndex: getDayOfWeekIndex(getToday()),
        date: getHumanDate(usedDate),
        isWeekend: getIsWeekend(usedDate),
        dayIndex: getDayOfWeekIndex(usedDate),
        choices: {},
    };
}

/**
 * Vrátí veškerá klientská data pro předaný den, nebo aktuální den, pokud není předán.
 */
export async function getData(date?: Date): Promise<ClientData> {
    const clientData = await getClientData(date);
    clientData.menus = {
        SLADOVNICKA: await getRestaurantMenu('SLADOVNICKA', date),
        // UMOTLIKU: await getRestaurantMenu('UMOTLIKU', date),
        TECHTOWER: await getRestaurantMenu('TECHTOWER', date),
        ZASTAVKAUMICHALA: await getRestaurantMenu('ZASTAVKAUMICHALA', date),
        SENKSERIKOVA: await getRestaurantMenu('SENKSERIKOVA', date),
    }
    return clientData;
}

/**
 * Vrátí klíč, pod kterým je uloženo menu pro týden příslušící předanému datu.
 *
 * @param date datum
 * @returns databázový klíč
 */
function getMenuKey(date: Date) {
    const weekNumber = getWeekNumber(date);
    return `${MENU_PREFIX}_${date.getFullYear()}_${weekNumber}`;
}

/**
 * Vrátí menu všech podniků pro celý týden do kterého spadá předané datum, pokud již existují.
 * 
 * @param date datum
 * @returns menu restaurací pro týden příslušící předanému datu
 */
async function getMenu(date: Date): Promise<WeekMenu | undefined> {
    return await storage.getData<WeekMenu | undefined>(getMenuKey(date));
}

// TODO přesun do restaurants.ts
/**
 * Vrátí menu dané restaurace pro předaný den.
 * Pokud neexistuje, provede stažení menu pro příslušný týden a uložení do DB.
 * 
 * @param restaurant restaurace
 * @param date datum, ke kterému získat menu
 * @param mock příznak, zda chceme pouze mock data
 */
export async function getRestaurantMenu(restaurant: keyof typeof Restaurant, date?: Date): Promise<RestaurantDayMenu> {
    const usedDate = date ?? getToday();
    const dayOfWeekIndex = getDayOfWeekIndex(usedDate);
    const now = new Date().getTime();
    if (getIsWeekend(usedDate)) {
        return {
            lastUpdate: now,
            closed: true,
            food: [],
        };
    }

    let weekMenu = await getMenu(usedDate);
    if (weekMenu == null) {
        weekMenu = [{}, {}, {}, {}, {}];
    }
    for (let i = 0; i < 5; i++) {
        if (weekMenu[i] == null) {
            weekMenu[i] = {};
        }
        if (weekMenu[i][restaurant] == null) {
            weekMenu[i][restaurant] = {
                lastUpdate: now,
                closed: false,
                food: [],
            };
        }
    }
    if (!weekMenu[dayOfWeekIndex][restaurant]?.food?.length) {
        const firstDay = getFirstWorkDayOfWeek(usedDate);
        const mock = process.env.MOCK_DATA === 'true';
        switch (restaurant) {
            case 'SLADOVNICKA':
                try {
                    const sladovnickaFood = await getMenuSladovnicka(firstDay, mock);
                    for (let i = 0; i < sladovnickaFood.length; i++) {
                        weekMenu[i][restaurant]!.food = sladovnickaFood[i];
                        // Velice chatrný a nespolehlivý způsob detekce uzavření...
                        if (sladovnickaFood[i].length === 1 && sladovnickaFood[i][0].name.toLowerCase() === 'pro daný den nebyla nalezena denní nabídka') {
                            weekMenu[i][restaurant]!.closed = true;
                        }
                    }
                } catch (e: any) {
                    console.error("Selhalo načtení jídel pro podnik Sladovnická", e);
                }
                break;
            // case 'UMOTLIKU':
            //     try {
            //         const uMotlikuFood = await getMenuUMotliku(firstDay, mock);
            //         for (let i = 0; i < uMotlikuFood.length; i++) {
            //             menus[i][restaurant]!.food = uMotlikuFood[i];
            //             if (uMotlikuFood[i].length === 1 && uMotlikuFood[i][0].name.toLowerCase() === 'zavřeno') {
            //                 menus[i][restaurant]!.closed = true;
            //             }
            //         }
            //     } catch (e: any) {
            //         console.error("Selhalo načtení jídel pro podnik U Motlíků", e);
            //     }
            //     break;
            case 'TECHTOWER':
                try {
                    const techTowerFood = await getMenuTechTower(firstDay, mock);
                    for (let i = 0; i < techTowerFood.length; i++) {
                        weekMenu[i][restaurant]!.food = techTowerFood[i];
                        if (techTowerFood[i]?.length === 1 && techTowerFood[i][0].name.toLowerCase() === 'svátek') {
                            weekMenu[i][restaurant]!.closed = true;
                        }
                    }
                    break;
                } catch (e: any) {
                    console.error("Selhalo načtení jídel pro podnik TechTower", e);
                }
            case 'ZASTAVKAUMICHALA':
                try {
                    const zastavkaUmichalaFood = await getMenuZastavkaUmichala(firstDay, mock);
                    for (let i = 0; i < zastavkaUmichalaFood.length; i++) {
                        weekMenu[i][restaurant]!.food = zastavkaUmichalaFood[i];
                        if (zastavkaUmichalaFood[i]?.length === 1 && zastavkaUmichalaFood[i][0].name === 'Pro tento den není uveřejněna nabídka jídel.') {
                            weekMenu[i][restaurant]!.closed = true;
                        }
                    }
                    break;
                } catch (e: any) {
                    console.error("Selhalo načtení jídel pro podnik Zastávka u Michala", e);
                }
            case 'SENKSERIKOVA':
                try {
                    const senkSerikovaFood = await getMenuSenkSerikova(firstDay, mock);
                    for (let i = 0; i < senkSerikovaFood.length; i++) {
                        weekMenu[i][restaurant]!.food = senkSerikovaFood[i];
                        if (senkSerikovaFood[i]?.length === 1 && senkSerikovaFood[i][0].name === 'Pro tento den nebylo zadáno menu.') {
                            weekMenu[i][restaurant]!.closed = true;
                        }
                    }
                    break;
                } catch (e: any) {
                    console.error("Selhalo načtení jídel pro podnik Pivovarský šenk Šeříková", e);
                }
        }
        await storage.setData(getMenuKey(usedDate), weekMenu);
    }
    return weekMenu[dayOfWeekIndex][restaurant]!;
}

/**
 * Inicializuje výchozí data pro předané datum, nebo dnešek, pokud není datum předáno.
 * 
 * @param date datum
 */
export async function initIfNeeded(date?: Date) {
    const usedDate = formatDate(date ?? getToday());
    const hasData = await storage.hasData(usedDate);
    if (!hasData) {
        await storage.setData(usedDate, getEmptyData(date || getToday()));
    }
}

/**
 * Odstraní kompletně volbu uživatele (včetně případných podřízených jídel).
 * 
 * @param login login uživatele
 * @param trusted příznak, zda se jedná o ověřeného uživatele
 * @param locationKey vybrané "umístění"
 * @param date datum, ke kterému se volba vztahuje
 * @returns 
 */
export async function removeChoices(login: string, trusted: boolean, locationKey: keyof typeof LunchChoice, date?: Date) {
    const selectedDay = formatDate(date ?? getToday());
    let data = await getClientData(date);
    validateTrusted(data, login, trusted);
    if (locationKey in data.choices) {
        if (data.choices[locationKey] && login in data.choices[locationKey]) {
            delete data.choices[locationKey][login]
            if (Object.keys(data.choices[locationKey]).length === 0) {
                delete data.choices[locationKey]
            }
            await storage.setData(selectedDay, data);
        }
    }
    return data;
}

/**
 * Odstraní konkrétní volbu jídla uživatele.
 * Neodstraňuje volbu samotnou, k tomu slouží {@link removeChoices}.
 * 
 * @param login login uživatele
 * @param trusted příznak, zda se jedná o ověřeného uživatele
 * @param locationKey vybrané "umístění"
 * @param foodIndex index jídla v jídelním lístku daného umístění, pokud existuje
 * @param date datum, ke kterému se volba vztahuje
 * @returns 
 */
export async function removeChoice(login: string, trusted: boolean, locationKey: keyof typeof LunchChoice, foodIndex: number, date?: Date) {
    const selectedDay = formatDate(date ?? getToday());
    let data = await getClientData(date);
    validateTrusted(data, login, trusted);
    if (locationKey in data.choices) {
        if (data.choices[locationKey] && login in data.choices[locationKey]) {
            const index = data.choices[locationKey][login].selectedFoods?.indexOf(foodIndex);
            if (index && index > -1) {
                data.choices[locationKey][login].selectedFoods?.splice(index, 1);
                await storage.setData(selectedDay, data);
            }
        }
    }
    return data;
}

/**
 * Odstraní kompletně volbu uživatele, vyjma ignoredLocationKey (pokud byla předána a existuje).
 * 
 * @param login login uživatele
 * @param date datum, ke kterému se volby vztahují
 * @param ignoredLocationKey volba, která nebude odstraněna, pokud existuje
 */
async function removeChoiceIfPresent(login: string, date?: Date, ignoredLocationKey?: keyof typeof LunchChoice) {
    const usedDate = date ?? getToday();
    let data = await getClientData(usedDate);
    for (const key of Object.keys(data.choices)) {
        const locationKey = key as keyof typeof LunchChoice;
        if (ignoredLocationKey != null && ignoredLocationKey == locationKey) {
            continue;
        }
        if (data.choices[locationKey] && login in data.choices[locationKey]) {
            delete data.choices[locationKey][login];
            if (Object.keys(data.choices[locationKey]).length === 0) {
                delete data.choices[locationKey];
            }
            await storage.setData(formatDate(usedDate), data);
        }
    }
    return data;
}

/**
 * Ověří, zda se neověřený uživatel nepokouší přepsat údaje ověřeného a případně vyhodí chybu.
 * 
 * @param data aktuální klientská data
 * @param login přihlašovací jméno uživatele
 * @param trusted příznak, zda se jedná o ověřeného uživatele
 */
function validateTrusted(data: ClientData, login: string, trusted: boolean) {
    const locations = Object.values(data?.choices);
    let found = false;
    if (!trusted) {
        for (const location of locations) {
            if (Object.keys(location).includes(login) && location[login].trusted) {
                found = true;
            }
        }
    }
    if (!trusted && found) {
        throw new InsufficientPermissions("Nelze změnit volbu ověřeného uživatele");
    }
}

/**
 * Přidá volbu uživatele.
 * 
 * @param login login uživatele
 * @param trusted příznak, zda se jedná o ověřeného uživatele
 * @param locationKey vybrané "umístění"
 * @param foodIndex volitelný index jídla v daném umístění
 * @param trusted příznak, zda se jedná o ověřeného uživatele
 * @param date datum, ke kterému se volba vztahuje
 * @returns aktuální data
 */
export async function addChoice(login: string, trusted: boolean, locationKey: keyof typeof LunchChoice, foodIndex?: number, date?: Date) {
    const usedDate = date ?? getToday();
    await initIfNeeded(usedDate);
    let data = await getClientData(usedDate);
    validateTrusted(data, login, trusted);
    await validateFoodIndex(locationKey, foodIndex, date);
    // Pokud měníme pouze lokaci, mažeme případné předchozí
    if (foodIndex == null) {
        data = await removeChoiceIfPresent(login, usedDate);
    } else {
        // Mažeme případné ostatní volby (měla by být maximálně jedna)
        removeChoiceIfPresent(login, usedDate, locationKey);
    }
    // TODO vytáhnout inicializaci "prázdné struktury" do vlastní funkce
    if (!(data.choices[locationKey])) {
        data.choices[locationKey] = {}
    }
    if (!(login in data.choices[locationKey])) {
        if (!data.choices[locationKey]) {
            data.choices[locationKey] = {}
        }
        data.choices[locationKey][login] = {
            trusted,
            selectedFoods: []
        };
    }
    if (foodIndex != null && !data.choices[locationKey][login].selectedFoods?.includes(foodIndex)) {
        data.choices[locationKey][login].selectedFoods?.push(foodIndex);
    }
    const selectedDate = formatDate(usedDate);
    await storage.setData(selectedDate, data);
    return data;
}

/**
 * Zvaliduje platnost indexu jídla pro vybranou lokalitu a datum.
 * 
 * @param locationKey vybraná lokalita
 * @param foodIndex index jídla pro danou lokalitu
 * @param date datum, pro které je validace prováděna
 */
async function validateFoodIndex(locationKey: keyof typeof LunchChoice, foodIndex?: number, date?: Date) {
    if (foodIndex != null) {
        if (typeof foodIndex !== 'number') {
            throw Error(`Neplatný index ${foodIndex} typu ${typeof foodIndex}`);
        }
        if (foodIndex < 0) {
            throw Error(`Neplatný index ${foodIndex}`);
        }
        if (!Object.keys(Restaurant).includes(locationKey)) {
            throw Error(`Neplatný index ${foodIndex} pro lokalitu ${locationKey} nepodporující indexy`);
        }
        const usedDate = date ?? getToday();
        const menu = await getRestaurantMenu(locationKey as keyof typeof Restaurant, usedDate);
        if (menu.food?.length && foodIndex > (menu.food.length - 1)) {
            throw new Error(`Neplatný index ${foodIndex} pro lokalitu ${locationKey}`);
        }
    }
}

/**
 * Aktualizuje poznámku k aktuálně vybrané možnosti.
 * 
 * @param login login uživatele
 * @param trusted příznak, zda se jedná o ověřeného uživatele
 * @param note poznámka
 * @param date datum, ke kterému se volba vztahuje
 */
export async function updateNote(login: string, trusted: boolean, note?: string, date?: Date) {
    const usedDate = date ?? getToday();
    await initIfNeeded(usedDate);
    let data = await getClientData(usedDate);
    validateTrusted(data, login, trusted);
    const userEntry = data.choices != null && Object.entries(data.choices).find(entry => entry[1][login] != null);
    if (userEntry) {
        if (!note?.length) {
            delete userEntry[1][login].note;
        } else {
            userEntry[1][login].note = note;
        }
        const selectedDate = formatDate(usedDate);
        await storage.setData(selectedDate, data);
    }
    return data;
}

/**
 * Aktualizuje preferovaný čas odchodu strávníka.
 * 
 * @param login login uživatele
 * @param time preferovaný čas odchodu
 * @param date datum, ke kterému se čas vztahuje
 */
export async function updateDepartureTime(login: string, time?: string, date?: Date) {
    const usedDate = date ?? getToday();
    let clientData = await getClientData(usedDate);
    const found = Object.values(clientData.choices).find(location => login in location);
    // TODO validace, že se jedná o restauraci
    if (found) {
        if (!time?.length) {
            delete found[login].departureTime;
        } else {
            if (!Object.values<string>(DepartureTime).includes(time)) {
                throw Error(`Neplatný čas odchodu ${time}`);
            }
            found[login].departureTime = time;
        }
        await storage.setData(formatDate(usedDate), clientData);
    }
    return clientData;
}

/**
 * Vrátí data pro klienta pro předaný nebo aktuální den.
 * 
 * @param date datum pro který vrátit data, pokud není vyplněno, je použit dnešní den
 * @returns data pro klienta
 */
export async function getClientData(date?: Date): Promise<ClientData> {
    const targetDate = date ?? getToday();
    const dateString = formatDate(targetDate);
    const clientData = await storage.getData<ClientData>(dateString) || getEmptyData(date);
    return {
        ...clientData,
        todayDayIndex: getDayOfWeekIndex(getToday()),
    }
}