From 4225d5114091763506040e057e7f1e50d55306ae Mon Sep 17 00:00:00 2001
From: Douglas Barone <douglasvbarone@gmail.com>
Date: Tue, 20 Jun 2023 15:48:25 -0400
Subject: [PATCH] Use middleware

---
 src/controllers/PrinterController.ts       | 4 ++--
 src/controllers/PrinterStatusController.ts | 3 +++
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/controllers/PrinterController.ts b/src/controllers/PrinterController.ts
index 87dafff..aff04f3 100644
--- a/src/controllers/PrinterController.ts
+++ b/src/controllers/PrinterController.ts
@@ -67,7 +67,7 @@ class PrinterController {
 
   static async edit(req: Request, res: Response) {
     const { id } = req.params
-    const { friendlyName, ip } = req.body
+    const { friendlyName } = req.body
 
     // Verify if printer exists
     const printerExists = await prisma.printer.findUnique({
@@ -77,7 +77,7 @@ class PrinterController {
     if (printerExists) {
       const printer = await prisma.printer.update({
         where: { id: Number(id) },
-        data: { friendlyName, ip }
+        data: { friendlyName }
       })
 
       res.json(printer)
diff --git a/src/controllers/PrinterStatusController.ts b/src/controllers/PrinterStatusController.ts
index 2058851..b659014 100644
--- a/src/controllers/PrinterStatusController.ts
+++ b/src/controllers/PrinterStatusController.ts
@@ -1,6 +1,7 @@
 import { Router, Request, Response } from 'express'
 import { prisma } from '../prisma.js'
 import { PrinterStatusService } from '../services/PrinterStatusService.js'
+import { hasRolesMiddleware } from '../middlewares/hasRolesMiddleware.js'
 
 const router = Router()
 
@@ -16,6 +17,8 @@ class PrinterStatusController {
   }
 }
 
+router.use(hasRolesMiddleware(['ADMIN', 'INSPECTOR']))
+
 router.post('/update', PrinterStatusController.update)
 
 export default router