From 6efffc9056a64399e76d989b01800b48420e5d6c Mon Sep 17 00:00:00 2001
From: dpurbosakti <dwiatmokop@gmail.com>
Date: Tue, 7 Oct 2025 16:02:12 +0700
Subject: [PATCH] feat (infra): add rooms relation

---
 internal/domain/main-entities/infra/dto.go        | 3 +++
 internal/domain/main-entities/room/base/entity.go | 4 ++++
 2 files changed, 7 insertions(+)

diff --git a/internal/domain/main-entities/infra/dto.go b/internal/domain/main-entities/infra/dto.go
index 34a0774d..938ac42a 100644
--- a/internal/domain/main-entities/infra/dto.go
+++ b/internal/domain/main-entities/infra/dto.go
@@ -3,6 +3,7 @@ package infra
 import (
 	ecore "simrs-vx/internal/domain/base-entities/core"
 	ei "simrs-vx/internal/domain/main-entities/item"
+	erb "simrs-vx/internal/domain/main-entities/room/base"
 
 	ero "simrs-vx/internal/domain/references/organization"
 )
@@ -67,6 +68,7 @@ type ResponseDto struct {
 	Childrens       []Infra            `json:"childrens,omitempty"`
 	Item_Id         *uint              `json:"item_id"`
 	Item            *ei.Item           `json:"item,omitempty"`
+	Rooms           []erb.Basic        `json:"rooms,omitempty"`
 }
 
 func (d Infra) ToResponse() ResponseDto {
@@ -79,6 +81,7 @@ func (d Infra) ToResponse() ResponseDto {
 		Childrens:       d.Childrens,
 		Item_Id:         d.Item_Id,
 		Item:            d.Item,
+		Rooms:           d.Rooms,
 	}
 	resp.SmallMain = d.SmallMain
 	return resp
diff --git a/internal/domain/main-entities/room/base/entity.go b/internal/domain/main-entities/room/base/entity.go
index bbf87db8..e45a441d 100644
--- a/internal/domain/main-entities/room/base/entity.go
+++ b/internal/domain/main-entities/room/base/entity.go
@@ -17,3 +17,7 @@ type Basic struct {
 	Subspecialist_Id *uint16            `json:"subspecialist_id"`
 	Subspecialist    *ess.Subspecialist `json:"subspecialist,omitempty" gorm:"foreignKey:Subspecialist_Id"`
 }
+
+func (Basic) TableName() string {
+	return "Room"
+}