Zpracování chyb z API

2023-09-18 22:38:04 +02:00
parent 8a67325c85
commit bc181defa8
5 changed files with 96 additions and 45 deletions
--- a/server/src/index.ts
+++ b/server/src/index.ts
@@ -7,7 +7,7 @@ import dotenv from 'dotenv';
 import path from 'path';
 import { getQr } from "./qr";
 import { generateToken, getLogin, getTrusted, verify } from "./auth";
-import { getDayOfWeekIndex } from "./utils";
+import { InsufficientPermissions, getDayOfWeekIndex } from "./utils";

 const ENVIRONMENT = process.env.NODE_ENV || 'production';
 dotenv.config({ path: path.resolve(__dirname, `./.env.${ENVIRONMENT}`) });
@@ -219,7 +219,7 @@ app.post("/api/finishDelivery", async (req, res) => {
    res.status(200).json({});
 });

-app.post("/api/addChoice", async (req, res) => {
+app.post("/api/addChoice", async (req, res, next) => {
    const login = getLogin(parseToken(req));
    const trusted = getTrusted(parseToken(req));
    if (req.body.locationIndex > -1) {
@@ -233,15 +233,18 @@ app.post("/api/addChoice", async (req, res) => {
            }
            date = getDateForWeekIndex(dayIndex);
        }
-        const data = await addChoice(login, trusted, req.body.locationIndex, req.body.foodIndex, date);
-        io.emit("message", data);
-        return res.status(200).json(data);
+        try {
+            const data = await addChoice(login, trusted, req.body.locationIndex, req.body.foodIndex, date);
+            io.emit("message", data);
+            return res.status(200).json(data);
+        } catch (e: any) { next(e) }
    }
    return res.status(400); // TODO přidat popis chyby
 });

-app.post("/api/removeChoices", async (req, res) => {
+app.post("/api/removeChoices", async (req, res, next) => {
    const login = getLogin(parseToken(req));
+    const trusted = getTrusted(parseToken(req));
    let date = undefined;
    if (req.body.dayIndex != null) {
        let dayIndex;
@@ -252,13 +255,16 @@ app.post("/api/removeChoices", async (req, res) => {
        }
        date = getDateForWeekIndex(dayIndex);
    }
-    const data = await removeChoices(login, req.body.locationIndex, date);
-    io.emit("message", data);
-    res.status(200).json(data);
+    try {
+        const data = await removeChoices(login, trusted, req.body.locationIndex, date);
+        io.emit("message", data);
+        res.status(200).json(data);
+    } catch (e: any) { next(e) }
 });

-app.post("/api/removeChoice", async (req, res) => {
+app.post("/api/removeChoice", async (req, res, next) => {
    const login = getLogin(parseToken(req));
+    const trusted = getTrusted(parseToken(req));
    let date = undefined;
    if (req.body.dayIndex != null) {
        let dayIndex;
@@ -269,9 +275,11 @@ app.post("/api/removeChoice", async (req, res) => {
        }
        date = getDateForWeekIndex(dayIndex);
    }
-    const data = await removeChoice(login, req.body.locationIndex, req.body.foodIndex, date);
-    io.emit("message", data);
-    res.status(200).json(data);
+    try {
+        const data = await removeChoice(login, trusted, req.body.locationIndex, req.body.foodIndex, date);
+        io.emit("message", data);
+        res.status(200).json(data);
+    } catch (e: any) { next(e) }
 });

 // TODO přejmenovat, ať je jasné, že to patří k Pizza day
@@ -302,6 +310,16 @@ app.post("/api/changeDepartureTime", async (req, res) => {
    res.status(200).json(data);
 });

+// Middleware pro zpracování chyb
+app.use((err: any, req: any, res: any, next: any) => {
+    if (err instanceof InsufficientPermissions) {
+        res.status(403).send({ error: err.message })
+    } else {
+        res.status(500).send({ error: err.message })
+    }
+    next();
+});
+
 io.on("connection", (socket) => {
    console.log(`New client connected: ${socket.id}`);

--- a/server/src/service.ts
+++ b/server/src/service.ts
@@ -1,7 +1,7 @@
-import { formatDate, getDayOfWeekIndex, getHumanDate, getHumanTime, getIsWeekend } from "./utils";
+import { InsufficientPermissions, formatDate, getDayOfWeekIndex, getHumanDate, getHumanTime, getIsWeekend } from "./utils";
 import { callNotifikace } from "./notifikace";
 import { generateQr } from "./qr";
-import { ClientData, PizzaDayState, UdalostEnum, Pizza, PizzaSize, Order, PizzaOrder, Locations, Restaurants, Food, Menu } from "../../types";
+import { ClientData, PizzaDayState, UdalostEnum, Pizza, PizzaSize, Order, PizzaOrder, Locations, Restaurants, Menu } from "../../types";
 import getStorage from "./storage";
 import { getMenuSladovnicka, getMenuTechTower, getMenuUMotliku } from "./restaurants";
 import { downloadPizzy } from "./chefie";
@@ -345,14 +345,15 @@ export async function initIfNeeded(date?: Date) {
 * 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 location vybrané "umístění"
 * @param date datum, ke kterému se volba vztahuje
 * @returns 
 */
-export async function removeChoices(login: string, location: Locations, 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);
-    // TODO zajistit, že neověřený uživatel se stejným loginem nemůže mazat volby ověřeného
+    validateTrusted(data, login, trusted);
    if (location in data.choices) {
        if (login in data.choices[location]) {
            delete data.choices[location][login]
@@ -370,15 +371,16 @@ export async function removeChoices(login: string, location: Locations, date?: D
 * 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 location 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, location: Locations, foodIndex: number, date?: Date) {
+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);
-    // TODO řešit ověření uživatele
+    validateTrusted(data, login, trusted);
    if (location in data.choices) {
        if (login in data.choices[location]) {
            const index = data.choices[location][login].options.indexOf(foodIndex);
@@ -411,20 +413,13 @@ async function removeChoiceIfPresent(login: string, date: string) {
 }

 /**
- * Přidá volbu uživatele.
+ * Ověří, zda se neověřený uživatel nepokouší přepsat údaje ověřeného a případně vyhodí chybu.
 * 
- * @param login login uživatele
- * @param location vybrané "umístění"
- * @param foodIndex volitelný index jídla v daném umístění
+ * @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
- * @param date datum, ke kterému se volba vztahuje
- * @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);
-    // Ověření, že se neověřený užívatel nepokouší přepsat údaje ověřeného
+function validateTrusted(data: ClientData, login: string, trusted: boolean) {
    const locations = Object.values(data?.choices);
    let found = false;
    if (!trusted) {
@@ -435,8 +430,26 @@ export async function addChoice(login: string, trusted: boolean, location: Locat
        }
    }
    if (!trusted && found) {
-        throw Error("Nelze změnit volbu ověřeného uživatele");
+        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 location 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, location: Locations, foodIndex?: number, date?: Date) {
+    await initIfNeeded();
+    const selectedDate = formatDate(date ?? getToday());
+    let data: ClientData = await storage.getData(selectedDate);
+    validateTrusted(data, login, trusted);
    // Pokud měníme pouze lokaci, mažeme případné předchozí
    if (foodIndex == null) {
        data = await removeChoiceIfPresent(login, selectedDate);
--- a/server/src/utils.ts
+++ b/server/src/utils.ts
@@ -37,4 +37,7 @@ export const getDayOfWeekIndex = (date: Date) => {
 export function getIsWeekend(date: Date) {
    const index = getDayOfWeekIndex(date);
    return index == 5 || index == 6;
-}
+}
+
+// TODO umístit do samostatného souboru
+export class InsufficientPermissions extends Error { }