Testy datumových funkcí

client/src/App.tsx

@@ -14,7 +14,7 @@ import './App.css';
 import { SelectSearchOption } from 'react-select-search';
 import { faCircleCheck, faTrashCan } from '@fortawesome/free-regular-svg-icons';
 import { useBank } from './context/bank';
-import { ClientData, Restaurants, Food, Order, Locations, PizzaOrder, PizzaDayState, FoodChoices, Menu } from './types';
+import { ClientData, Restaurants, Food, Order, Locations, PizzaOrder, PizzaDayState, FoodChoices, DayMenu } from './types';
 import Footer from './components/Footer';
 import { faChainBroken, faChevronLeft, faChevronRight, faGear, faSatelliteDish, faSearch } from '@fortawesome/free-solid-svg-icons';
 import Loader from './components/Loader';
@@ -28,7 +28,7 @@ function App() {
   const bank = useBank();
   const [isConnected, setIsConnected] = useState<boolean>(false);
   const [data, setData] = useState<ClientData>();
-  const [food, setFood] = useState<{ [key in Restaurants]?: Menu }>();
+  const [food, setFood] = useState<{ [key in Restaurants]?: DayMenu }>();
   const [myOrder, setMyOrder] = useState<Order>();
   const [foodChoiceList, setFoodChoiceList] = useState<Food[]>();
   const [closed, setClosed] = useState<boolean>(false);
@@ -288,7 +288,7 @@ function App() {
     }
   }
 
-  const renderFoodTable = (name: string, menu: Menu) => {
+  const renderFoodTable = (name: string, menu: DayMenu) => {
     let content;
     if (menu?.closed) {
       content = <h3>Zavřeno</h3>
@@ -352,13 +352,7 @@ function App() {
           <Alert variant={'primary'}>
             Poslední změny:
             <ul>
-              <li>Oprava generování QR kódů pro Pizza day</li>
-              <li>Serverová validace času odchodu</li>
-              <li>Loader při zakládání Pizza day</li>
-              <li>Možnost ručního zadání příplatku k Pizza day objednávkám</li>
-              <li>Vylepšená detekce uzavření pro podniky Sladovnická a TechTower</li>
-              <li>Úprava zvýraznění aktuálního dne</li>
-              <li>Možnost hlasování o nových funkcích</li>
+              <li>Parsování jídelních lístků na celý týden</li>
             </ul>
           </Alert>
           {dayIndex != null &&

server/src/restaurants.ts

@@ -2,7 +2,6 @@ import axios from "axios";
 import { load } from 'cheerio';
 import { Food } from "../../types";
 import { getMenuSladovnickaMock, getMenuTechTowerMock, getMenuUMotlikuMock } from "./mock";
-import { getDayOfWeekIndex } from "./utils";
 
 // Fráze v názvech jídel, které naznačují že se jedná o polévku
 const SOUP_NAMES = ['polévka', 'česnečka', 'česnekový krém', 'cibulačka', 'vývar']

server/src/service.ts

@@ -1,9 +1,8 @@
 import { InsufficientPermissions, formatDate, getDayOfWeekIndex, getFirstWorkDayOfWeek, getHumanDate, getHumanTime, getIsWeekend, getLastWorkDayOfWeek, getWeekNumber } from "./utils";
-import { ClientData, Locations, Restaurants, Menu, DepartureTime, DayData, WeekMenu } from "../../types";
+import { ClientData, Locations, Restaurants, DayMenu, DepartureTime, DayData, WeekMenu } from "../../types";
 import getStorage from "./storage";
 import { getMenuSladovnicka, getMenuTechTower, getMenuUMotliku } from "./restaurants";
 import { getTodayMock } from "./mock";
-import { first } from "cheerio/lib/api/traversing";
 
 const storage = getStorage();
 const MENU_PREFIX = 'menu';
@@ -49,92 +48,107 @@ export async function getData(date?: Date): Promise<ClientData> {
     const data: DayData = await storage.getData(dateString) || getEmptyData(date);
     const clientData: ClientData = { ...data };
     clientData.todayWeekIndex = getDayOfWeekIndex(getToday());
-    // Dotažení jídel, pokud je ještě nemáme
-    if (!data.menus) {
-        data.menus = {
-            [Restaurants.SLADOVNICKA]: await getRestaurantMenu(Restaurants.SLADOVNICKA, targetDate),
-            [Restaurants.UMOTLIKU]: await getRestaurantMenu(Restaurants.UMOTLIKU, targetDate),
-            [Restaurants.TECHTOWER]: await getRestaurantMenu(Restaurants.TECHTOWER, targetDate),
-        }
-        await storage.setData(dateString, data);
-        clientData.menus = data.menus;
+    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í.
- * Jinak založí a vrátí prázdný objekt.
  * 
  * @param date datum
  * @returns menu restaurací pro předané datum
  */
-async function getMenu(date: Date): Promise<WeekMenu> {
-    const weekNumber = getWeekNumber(date);
-    const menuKey = `${MENU_PREFIX}_${date.getFullYear()}_${weekNumber}`;
-    const menus: WeekMenu = await storage.getData(menuKey);
-    if (!menus) {
-        storage.setData(menuKey, {});
-        return {};
-    }
-    return menus;
+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> {
+export async function getRestaurantMenu(restaurant: Restaurants, date?: Date): Promise<DayMenu> {
     const usedDate = date ?? getToday();
-    await initIfNeeded(usedDate);
-    const selectedDay = formatDate(usedDate);
-    const clientData: DayData = await storage.getData(selectedDay);
-    const weekNumber = getWeekNumber(usedDate);
-    const menus = await getMenu(usedDate);
-    if (!menus[weekNumber]) {
-        menus[weekNumber] = {};
+    const dayOfWeekIndex = getDayOfWeekIndex(usedDate);
+
+    let menus = await getMenu(usedDate);
+    if (menus == null) {
+        menus = [];
     }
-    if (!menus[weekNumber][restaurant]) {
-        menus[weekNumber][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(firstDay, mock);
-                menus[weekNumber][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;
+                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 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);
@@ -248,8 +262,9 @@ 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());
+    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í

types/Types.ts

@@ -70,7 +70,7 @@ interface PizzaDay {
 /** Týdenní menu jednotlivých restaurací. */
 export interface WeekMenu {
     [dayIndex: number]: {
-        [restaurant in Restaurants]?: Menu
+        [restaurant in Restaurants]?: DayMenu
     }
 }
 
@@ -82,7 +82,7 @@ export interface DayData {
     choices: Choices, // seznam voleb uživatelů
     // TODO smazat
     departureTimes: DepartureTime[], // seznam možných časů odchodu
-    menus?: { [restaurant in Restaurants]?: Menu }, // menu jednotlivých restaurací
+    menus?: { [restaurant in Restaurants]?: DayMenu }, // menu jednotlivých restaurací
     pizzaDay?: PizzaDay, // pizza day pro dnešní den, pokud existuje
     pizzaList?: Pizza[], // seznam dostupných pizz pro dnešní den
     pizzaListLastUpdate?: Date, // datum a čas poslední aktualizace pizz
@@ -93,10 +93,10 @@ export interface ClientData extends DayData {
     todayWeekIndex?: number, // index dnešního dne v týdnu (0-6)
 }
 
-/** Nabídka jídel jednoho podniku. */
-export interface Menu {
+/** Nabídka jídel jednoho podniku pro jeden konkrétní den. */
+export interface DayMenu {
     lastUpdate: string, // human-readable čas poslední aktualizace menu
-    closed: boolean, // příznak, zda je daný podnik aktuálně zavřený
+    closed: boolean, // příznak, zda je daný podnik v tento den zavřený
     food: Food[], // seznam jídel v menu
 }