Parsování jídel na celý týden

This commit is contained in:
2023-10-15 19:05:19 +02:00
parent 74893c38eb
commit ca9a7c5c23
7 changed files with 338 additions and 260 deletions

View File

@@ -1,10 +1,11 @@
import { InsufficientPermissions, formatDate, getDayOfWeekIndex, getHumanDate, getHumanTime, getIsWeekend } from "./utils";
import { ClientData, Locations, Restaurants, Menu, DepartureTime } from "../../types";
import { InsufficientPermissions, formatDate, getDayOfWeekIndex, getFirstWorkDayOfWeek, getHumanDate, getHumanTime, getIsWeekend, getLastWorkDayOfWeek, getWeekNumber } from "./utils";
import { ClientData, Locations, Restaurants, DayMenu, DepartureTime, DayData, WeekMenu } from "../../types";
import getStorage from "./storage";
import { getMenuSladovnicka, getMenuTechTower, getMenuUMotliku } from "./restaurants";
import { getTodayMock } from "./mock";
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 {
@@ -34,7 +35,7 @@ function getEmptyData(date?: Date): ClientData {
isWeekend: getIsWeekend(usedDate),
weekIndex: getDayOfWeekIndex(usedDate),
choices: {},
departureTimes: Object.values(DepartureTime),
departureTimes: Object.values(DepartureTime), // TODO tohle zmizí, bude se přidávat do dat dynamicky
};
}
@@ -42,73 +43,112 @@ function getEmptyData(date?: Date): ClientData {
* 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 dateString = formatDate(date ?? getToday());
const data: ClientData = await storage.getData(dateString) || getEmptyData(date);
data.todayWeekIndex = getDayOfWeekIndex(getToday());
// Dotažení jídel, pokud je ještě nemáme
if (!data.menus) {
data.menus = {
[Restaurants.SLADOVNICKA]: await getRestaurantMenu(Restaurants.SLADOVNICKA, date ?? getToday()),
[Restaurants.UMOTLIKU]: await getRestaurantMenu(Restaurants.UMOTLIKU, date ?? getToday()),
[Restaurants.TECHTOWER]: await getRestaurantMenu(Restaurants.TECHTOWER, date ?? getToday()),
}
await storage.setData(dateString, data);
const targetDate = date ?? getToday();
const dateString = formatDate(targetDate);
const data: DayData = await storage.getData(dateString) || getEmptyData(date);
const clientData: ClientData = { ...data };
clientData.todayWeekIndex = getDayOfWeekIndex(getToday());
clientData.menus = {
[Restaurants.SLADOVNICKA]: await getRestaurantMenu(Restaurants.SLADOVNICKA, targetDate),
[Restaurants.UMOTLIKU]: await getRestaurantMenu(Restaurants.UMOTLIKU, targetDate),
[Restaurants.TECHTOWER]: await getRestaurantMenu(Restaurants.TECHTOWER, targetDate),
}
return data;
return clientData;
}
/**
* Vrátí klíč, pod kterým je uloženo menu pro předané datum.
*
* @param date datum
* @returns databázový klíč
*/
function getMenuKey(date: Date) {
const weekNumber = getWeekNumber(date);
return `${MENU_PREFIX}_${date.getFullYear()}_${weekNumber}`;
}
/**
* Vrátí menu restaurací pro předané datum, pokud již existují.
*
* @param date datum
* @returns menu restaurací pro předané datum
*/
async function getMenu(date: Date): Promise<WeekMenu | undefined> {
return await storage.getData(getMenuKey(date));
}
// TODO přesun do restaurants.ts
/**
* Vrátí menu dané restaurace pro předaný den. Pokud neexistuje, provede jeho stažení a uložení do DB.
* 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
* @param date datum, ke kterému získat menu
* @param mock příznak, zda chceme pouze mock data
*/
export async function getRestaurantMenu(restaurant: Restaurants, date?: Date): Promise<Menu> {
await initIfNeeded(date);
const selectedDay = formatDate(date ?? getToday());
const clientData: ClientData = await storage.getData(selectedDay);
if (!clientData.menus) {
clientData.menus = {};
storage.setData(selectedDay, clientData);
export async function getRestaurantMenu(restaurant: Restaurants, date?: Date): Promise<DayMenu> {
const usedDate = date ?? getToday();
const dayOfWeekIndex = getDayOfWeekIndex(usedDate);
let menus = await getMenu(usedDate);
if (menus == null) {
menus = [];
}
if (!clientData.menus[restaurant]) {
clientData.menus[restaurant] = {
lastUpdate: getHumanTime(new Date()),
closed: false,
food: [],
};
for (let i = 0; i < 5; i++) {
if (menus[i] == null) {
menus[i] = {};
}
if (menus[i][restaurant] == null) {
menus[i][restaurant] = {
lastUpdate: getHumanTime(new Date()),
closed: false,
food: [],
};
}
}
if (!menus[dayOfWeekIndex][restaurant]?.food?.length) {
const firstDay = getFirstWorkDayOfWeek(usedDate);
const mock = process.env.MOCK_DATA === 'true';
switch (restaurant) {
case Restaurants.SLADOVNICKA:
const sladovnickaFood = await getMenuSladovnicka(date, mock);
clientData.menus[restaurant]!.food = sladovnickaFood;
// Velice chatrný a nespolehlivý způsob detekce uzavření...
if (sladovnickaFood.length === 1 && sladovnickaFood[0].name.toLowerCase() === 'pro daný den nebyla nalezena denní nabídka') {
clientData.menus[restaurant]!.closed = true;
const sladovnickaFood = await getMenuSladovnicka(firstDay, mock);
for (let i = 0; i < sladovnickaFood.length; i++) {
menus[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') {
menus[i][restaurant]!.closed = true;
}
}
break;
case Restaurants.UMOTLIKU:
const uMotlikuFood = await getMenuUMotliku(date, mock);
clientData.menus[restaurant]!.food = uMotlikuFood;
if (uMotlikuFood.length === 1 && uMotlikuFood[0].name.toLowerCase() === 'zavřeno') {
clientData.menus[restaurant]!.closed = true;
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;
}
}
break;
case Restaurants.TECHTOWER:
const techTowerFood = await getMenuTechTower(date, mock);
clientData.menus[restaurant]!.food = techTowerFood;
if (techTowerFood.length === 1 && techTowerFood[0].name.toLowerCase() === 'svátek') {
clientData.menus[restaurant]!.closed = true;
const techTowerFood = await getMenuTechTower(firstDay, mock);
for (let i = 0; i < techTowerFood.length; i++) {
menus[i][restaurant]!.food = techTowerFood[i];
if (techTowerFood[i].length === 1 && techTowerFood[i][0].name.toLowerCase() === 'svátek') {
menus[i][restaurant]!.closed = true;
}
}
break;
}
storage.setData(selectedDay, clientData);
await storage.setData(getMenuKey(usedDate), menus);
}
return clientData.menus[restaurant]!;
return menus[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);
@@ -128,7 +168,7 @@ export async function initIfNeeded(date?: Date) {
*/
export async function removeChoices(login: string, trusted: boolean, location: Locations, date?: Date) {
const selectedDay = formatDate(date ?? getToday());
let data: ClientData = await storage.getData(selectedDay);
let data: DayData = await storage.getData(selectedDay);
validateTrusted(data, login, trusted);
if (location in data.choices) {
if (login in data.choices[location]) {
@@ -155,7 +195,7 @@ export async function removeChoices(login: string, trusted: boolean, location: L
*/
export async function removeChoice(login: string, trusted: boolean, location: Locations, foodIndex: number, date?: Date) {
const selectedDay = formatDate(date ?? getToday());
let data: ClientData = await storage.getData(selectedDay);
let data: DayData = await storage.getData(selectedDay);
validateTrusted(data, login, trusted);
if (location in data.choices) {
if (login in data.choices[location]) {
@@ -175,7 +215,7 @@ export async function removeChoice(login: string, trusted: boolean, location: Lo
* @param login login uživatele
*/
async function removeChoiceIfPresent(login: string, date: string) {
let data: ClientData = await storage.getData(date);
let data: DayData = await storage.getData(date);
for (const key of Object.keys(data.choices)) {
if (login in data.choices[key]) {
delete data.choices[key][login];
@@ -222,9 +262,10 @@ function validateTrusted(data: ClientData, login: string, trusted: boolean) {
* @returns aktuální data
*/
export async function addChoice(login: string, trusted: boolean, location: Locations, foodIndex?: number, date?: Date) {
await initIfNeeded();
const selectedDate = formatDate(date ?? getToday());
let data: ClientData = await storage.getData(selectedDate);
const usedDate = date ?? getToday();
await initIfNeeded(usedDate);
const selectedDate = formatDate(usedDate);
let data: DayData = await storage.getData(selectedDate);
validateTrusted(data, login, trusted);
// Pokud měníme pouze lokaci, mažeme případné předchozí
if (foodIndex == null) {
@@ -255,7 +296,7 @@ export async function addChoice(login: string, trusted: boolean, location: Locat
*/
export async function updateDepartureTime(login: string, time?: string, date?: Date) {
const selectedDate = formatDate(date ?? getToday());
let clientData: ClientData = await storage.getData(selectedDate);
let clientData: DayData = await storage.getData(selectedDate);
const found = Object.values(clientData.choices).find(location => login in location);
// TODO validace, že se jedná o restauraci
if (found) {