1 files changed, 15 insertions(+), 2 deletions(-)

M src/index.js

@@ 1,5 1,6 @@
 import Koa from 'koa';
 import Router from 'koa-router';
+import { HTTPError } from 'got';
 import { getDailyMenu, getFullWeekMenu } from './scrape.js';
 
 const app = new Koa();


@@ 7,12 8,24 @@ const router = new Router();
 
 router.get('/:servery', async (ctx) => {
 	const { servery } = ctx.params;
-	ctx.body = await getDailyMenu(servery);
+
+	try {
+		ctx.body = await getDailyMenu(servery);
+	} catch (err) {
+		ctx.body = [];
+		ctx.status = (err instanceof HTTPError) ? 404 : 500;
+	}
 });
 
 router.get('/:servery/full-week-menu', async (ctx) => {
 	const { servery } = ctx.params;
-	ctx.body = await getFullWeekMenu(servery);
+
+	try {
+		ctx.body = await getFullWeekMenu(servery);
+	} catch (err) {
+		ctx.body = [];
+		ctx.status = (err instanceof HTTPError) ? 404 : 500;
+	}
 });
 
 app.use(router.routes());