From 54afcdb8b759c1a5833908e5d0cd343a6301d614 Mon Sep 17 00:00:00 2001
From: Buravit Yenjit <buravit.yj@gmail.com>
Date: Thu, 13 Feb 2025 23:49:05 +0700
Subject: [PATCH] feat:add repository/postgres_plant.go

---
 backend/internal/repository/postgres_plant.go | 85 +++++++++++++++++++
 1 file changed, 85 insertions(+)
 create mode 100644 backend/internal/repository/postgres_plant.go

diff --git a/backend/internal/repository/postgres_plant.go b/backend/internal/repository/postgres_plant.go
new file mode 100644
index 0000000..797f498
--- /dev/null
+++ b/backend/internal/repository/postgres_plant.go
@@ -0,0 +1,85 @@
+package repository
+
+import (
+	"context"
+	"strings"
+
+	"github.com/forfarm/backend/internal/domain"
+	"github.com/google/uuid"
+)
+
+type postgresPlantRepository struct {
+	conn Connection
+}
+
+func NewPostgresPlant(conn Connection) domain.PlantRepository {
+	return &postgresPlantRepository{conn: conn}
+}
+
+func (p *postgresPlantRepository) fetch(ctx context.Context, query string, args ...interface{}) ([]domain.Plant, error) {
+	rows, err := p.conn.Query(ctx, query, args...)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var plants []domain.Plant
+	for rows.Next() {
+		var plant domain.Plant
+		if err := rows.Scan(
+			&plant.UUID, &plant.Name, &plant.Variety,
+			&plant.RowSpacing, &plant.OptimalTemp, &plant.PlantingDepth,
+			&plant.AverageHeight, &plant.LightProfileID, &plant.SoilConditionID,
+			&plant.PlantingDetail, &plant.IsPerennial, &plant.DaysToEmerge,
+			&plant.DaysToFlower, &plant.DaysToMaturity, &plant.HarvestWindow,
+			&plant.PHValue, &plant.EstimateLossRate, &plant.EstimateRevenuePerHU,
+			&plant.HarvestUnitID, &plant.WaterNeeds, &plant.CreatedAt, &plant.UpdatedAt,
+		); err != nil {
+			return nil, err
+		}
+		plants = append(plants, plant)
+	}
+	return plants, nil
+}
+
+func (p *postgresPlantRepository) GetByUUID(ctx context.Context, uuid string) (domain.Plant, error) {
+	query := `SELECT * FROM plants WHERE uuid = $1`
+	plants, err := p.fetch(ctx, query, uuid)
+	if err != nil || len(plants) == 0 {
+		return domain.Plant{}, domain.ErrNotFound
+	}
+	return plants[0], nil
+}
+
+func (p *postgresPlantRepository) GetAll(ctx context.Context) ([]domain.Plant, error) {
+	query := `SELECT * FROM plants`
+	return p.fetch(ctx, query)
+}
+
+func (p *postgresPlantRepository) Create(ctx context.Context, plant *domain.Plant) error {
+	if strings.TrimSpace(plant.UUID) == "" {
+		plant.UUID = uuid.New().String()
+	}
+	if err := plant.Validate(); err != nil {
+		return err
+	}
+	query := `INSERT INTO plants (uuid, name, light_profile_id, soil_condition_id, harvest_unit_id, created_at, updated_at)
+	VALUES ($1, $2, $3, $4, $5, NOW(), NOW()) RETURNING created_at, updated_at`
+	return p.conn.QueryRow(ctx, query, plant.UUID, plant.Name, plant.LightProfileID, plant.SoilConditionID, plant.HarvestUnitID).Scan(&plant.CreatedAt, &plant.UpdatedAt)
+}
+
+func (p *postgresPlantRepository) Update(ctx context.Context, plant *domain.Plant) error {
+	if err := plant.Validate(); err != nil {
+		return err
+	}
+	query := `UPDATE plants SET name = $2, light_profile_id = $3, soil_condition_id = $4,
+		harvest_unit_id = $5, updated_at = NOW() WHERE uuid = $1`
+	_, err := p.conn.Exec(ctx, query, plant.UUID, plant.Name, plant.LightProfileID, plant.SoilConditionID, plant.HarvestUnitID)
+	return err
+}
+
+func (p *postgresPlantRepository) Delete(ctx context.Context, uuid string) error {
+	query := `DELETE FROM plants WHERE uuid = $1`
+	_, err := p.conn.Exec(ctx, query, uuid)
+	return err
+}