1 files changed, 19 insertions, 5 deletions
diff --git a/src/api/routes.js b/src/api/routes.js
index 0da8be9..09606f1 100644
--- a/src/api/routes.js
+++ b/src/api/routes.js
@@ -12,23 +12,37 @@ export function registerRoutes(app) {
             );
 
         if (route.onGet) {
-            app.get(route.route, route.onGet);
+            if (route.onGetValidation)
+                app.get(route.route, route.onGetValidation, route.onGet);
+            else app.get(route.route, route.onGet);
             routeCount++;
         }
         if (route.onPost) {
-            app.post(route.route, route.onPost);
+            if (route.onPostValidation)
+                app.post(route.route, route.onPostValidation, route.onPost);
+            else app.post(route.route, route.onPost);
             routeCount++;
         }
         if (route.onPut) {
-            app.put(route.route, route.onPut);
+            if (route.onPutValidation)
+                app.put(route.route, route.onPutValidation, route.onPut);
+            else app.put(route.route, route.onPut);
             routeCount++;
         }
         if (route.onDelete) {
-            app.put(route.route, route.onDelete);
+            if (route.onDeleteValidation)
+                app.delete(
+                    route.route,
+                    route.onDeleteValidation,
+                    route.onDelete
+                );
+            else app.delete(route.route, route.onDelete);
             routeCount++;
         }
         if (route.onPatch) {
-            app.patch(route.route, route.onPatch);
+            if (route.onPatchValidation)
+                app.patch(route.route, route.onPatchValidation, route.onPatch);
+            else app.patch(route.route, route.onPatch);
             routeCount++;
         }
     });