From 9a5cee89fb0cd14b5d7583ae7b0ce60308fdb9c0 Mon Sep 17 00:00:00 2001
From: Iki <farazrizki13@gmail.com>
Date: Fri, 26 Jun 2026 11:23:54 +0900
Subject: [PATCH] add main disease

---
 src/controllers/main_disease.ts | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/controllers/main_disease.ts b/src/controllers/main_disease.ts
index aea4727..c023c8a 100644
--- a/src/controllers/main_disease.ts
+++ b/src/controllers/main_disease.ts
@@ -1,4 +1,4 @@
-import { MainDisease, Paging } from "entity";
+import { DiseaseCategory, MainDisease, Paging } from "entity";
 import { NextFunction, Response } from "express";
 import { Request } from "express-jwt";
 import Joi from "joi";
@@ -15,6 +15,8 @@ const log: Logger<ILogObj> = new Logger({
     type: "pretty",
 });
 
+const diseaseCategoryValues = Object.values(DiseaseCategory);
+
 export class MainDiseaseController {
     static async list(req: Request, res: Response, next: NextFunction): Promise<void | Response> {
         /*
@@ -82,6 +84,10 @@ export class MainDiseaseController {
             const schema = Joi.object().keys({
                 name: Joi.string().max(128).required().label("Name"),
                 status: Joi.string().required().label("Status"),
+                category: Joi.string()
+                    .valid(...diseaseCategoryValues)
+                    .required()
+                    .label("Category"),
             });
 
             let param: any = await schema.validateAsync(req.body);
@@ -93,6 +99,7 @@ export class MainDiseaseController {
             let entity = new MainDisease();
             entity.name = param.name;
             entity.status = param.status;
+            entity.category = param.category;
             entity.created_by = req.auth.data.name;
             entity.updated_by = req.auth.data.name;
             await queryRunner.manager.save(entity);
@@ -152,6 +159,10 @@ export class MainDiseaseController {
                 id: Joi.string().uuid().required().label("Main Disease ID"),
                 name: Joi.string().max(128).required().label("Name"),
                 status: Joi.string().required().label("Status"),
+                category: Joi.string()
+                    .valid(...diseaseCategoryValues)
+                    .required()
+                    .label("Category"),
             });
 
             let param: any = await schema.validateAsync(req.body);
@@ -166,6 +177,7 @@ export class MainDiseaseController {
 
             entity.name = param.name;
             entity.status = param.status;
+            entity.category = param.category;
             entity.updated_by = req.auth.data.name;
             await queryRunner.manager.save(entity);