From f3b14cb8f25a10b1a111b98b3d3756a8bb252655 Mon Sep 17 00:00:00 2001
From: aguhh18 <teguh180902@gmail.com>
Date: Thu, 27 Nov 2025 14:28:48 +0700
Subject: [PATCH] Feat[BE]: create project budget repo, entity, and migration

---
 ...1127070744_create_project_budgets.down.sql |  2 ++
 ...251127070744_create_project_budgets.up.sql | 31 +++++++++++++++++++
 internal/entities/project_budget.go           | 15 +++++++++
 .../repositories/project_budget.repository.go | 23 ++++++++++++++
 4 files changed, 71 insertions(+)
 create mode 100644 internal/database/migrations/20251127070744_create_project_budgets.down.sql
 create mode 100644 internal/database/migrations/20251127070744_create_project_budgets.up.sql
 create mode 100644 internal/entities/project_budget.go
 create mode 100644 internal/modules/production/project_flocks/repositories/project_budget.repository.go

diff --git a/internal/database/migrations/20251127070744_create_project_budgets.down.sql b/internal/database/migrations/20251127070744_create_project_budgets.down.sql
new file mode 100644
index 00000000..55bfdb3d
--- /dev/null
+++ b/internal/database/migrations/20251127070744_create_project_budgets.down.sql
@@ -0,0 +1,2 @@
+DROP Table IF EXISTS project_budgets;
+
diff --git a/internal/database/migrations/20251127070744_create_project_budgets.up.sql b/internal/database/migrations/20251127070744_create_project_budgets.up.sql
new file mode 100644
index 00000000..db4a713b
--- /dev/null
+++ b/internal/database/migrations/20251127070744_create_project_budgets.up.sql
@@ -0,0 +1,31 @@
+CREATE TABLE project_budgets (
+    id BIGSERIAL PRIMARY KEY,
+    project_flock_id BIGINT NOT NULL,
+    nonstock_id BIGINT NOT NULL,
+    qty NUMERIC(15, 3) NOT NULL,
+    price NUMERIC(15, 3) NOT NULL,
+    created_at TIMESTAMPTZ DEFAULT now()
+);
+
+-- Tambahkan Foreign Key ke project_flocks
+DO $$
+BEGIN
+    IF EXISTS (SELECT 1 FROM pg_tables WHERE tablename = 'project_flocks') THEN
+        ALTER TABLE project_budgets
+            ADD CONSTRAINT fk_project_budgets_project_flock_id 
+            FOREIGN KEY (project_flock_id) REFERENCES project_flocks(id);
+    END IF;
+END $$;
+-- Tambahkan Foreign Key ke nonstocks
+DO $$
+BEGIN
+    IF EXISTS (SELECT 1 FROM pg_tables WHERE tablename = 'nonstocks') THEN
+        ALTER TABLE project_budgets
+            ADD CONSTRAINT fk_project_budgets_nonstock_id 
+            FOREIGN KEY (nonstock_id) REFERENCES nonstocks(id);
+    END IF;
+END $$;
+-- Index
+CREATE INDEX idx_project_budgets_project_flock_id ON project_budgets (project_flock_id);
+
+CREATE INDEX idx_project_budgets_nonstock_id ON project_budgets (nonstock_id);
\ No newline at end of file
diff --git a/internal/entities/project_budget.go b/internal/entities/project_budget.go
new file mode 100644
index 00000000..23c521ac
--- /dev/null
+++ b/internal/entities/project_budget.go
@@ -0,0 +1,15 @@
+package entities
+
+import (
+	"time"
+)
+
+type ProjectBudget struct {
+	Id        uint      `gorm:"primaryKey"`
+	Qty       float64   `gorm:"type:numeric(15,3);not null"`
+	Price     float64   `gorm:"type:numeric(15,3);not null"`
+	CreatedAt time.Time `gorm:"autoCreateTime"`
+
+	Nonstock     *Nonstock     `gorm:"foreignKey:Id;references:Id"`
+	ProjectFlock *ProjectFlock `gorm:"foreignKey:Id;references:Id"`
+}
diff --git a/internal/modules/production/project_flocks/repositories/project_budget.repository.go b/internal/modules/production/project_flocks/repositories/project_budget.repository.go
new file mode 100644
index 00000000..943a22b3
--- /dev/null
+++ b/internal/modules/production/project_flocks/repositories/project_budget.repository.go
@@ -0,0 +1,23 @@
+package repository
+
+import (
+	"gitlab.com/mbugroup/lti-api.git/internal/common/repository"
+	entity "gitlab.com/mbugroup/lti-api.git/internal/entities"
+	"gorm.io/gorm"
+)
+
+type ProjectBudgetRepository interface {
+	repository.BaseRepository[entity.ProjectBudget]
+}
+
+type ProjectBudgetRepositoryImpl struct {
+	*repository.BaseRepositoryImpl[entity.ProjectBudget]
+	db *gorm.DB
+}
+
+func NewProjectBudgetRepository(db *gorm.DB) ProjectBudgetRepository {
+	return &ProjectBudgetRepositoryImpl{
+		BaseRepositoryImpl: repository.NewBaseRepository[entity.ProjectBudget](db),
+		db:                 db,
+	}
+}