From c58e7b9504c1142a5561184de286e235cbed5188 Mon Sep 17 00:00:00 2001
From: Natthapol SERMSARAN <natthapolwork@gmail.com>
Date: Thu, 13 Feb 2025 23:31:15 +0700
Subject: [PATCH] feat: add farm repository

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

diff --git a/backend/internal/repository/postgres_farm.go b/backend/internal/repository/postgres_farm.go
new file mode 100644
index 0000000..e0fa6a4
--- /dev/null
+++ b/backend/internal/repository/postgres_farm.go
@@ -0,0 +1,102 @@
+package repository
+
+import (
+	"context"
+	"strings"
+
+	"github.com/google/uuid"
+
+	"github.com/forfarm/backend/internal/domain"
+)
+
+type postgresFarmRepository struct {
+	conn Connection
+}
+
+func NewPostgresFarm(conn Connection) domain.FarmRepository {
+	return &postgresFarmRepository{conn: conn}
+}
+
+func (p *postgresFarmRepository) fetch(ctx context.Context, query string, args ...interface{}) ([]domain.Farm, error) {
+	rows, err := p.conn.Query(ctx, query, args...)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var farms []domain.Farm
+	for rows.Next() {
+		var f domain.Farm
+		if err := rows.Scan(
+			&f.UUID,
+			&f.Name,
+			&f.Lat,
+			&f.Lon,
+			&f.CreatedAt,
+			&f.UpdatedAt,
+			&f.OwnerID,
+		); err != nil {
+			return nil, err
+		}
+		farms = append(farms, f)
+	}
+	return farms, nil
+}
+
+func (p *postgresFarmRepository) GetByID(ctx context.Context, uuid string) (domain.Farm, error) {
+	query := `
+		SELECT uuid, name, lat, lon, created_at, updated_at, owner_id
+		FROM farms
+		WHERE uuid = $1`
+
+	farms, err := p.fetch(ctx, query, uuid)
+	if err != nil {
+		return domain.Farm{}, err
+	}
+	if len(farms) == 0 {
+		return domain.Farm{}, domain.ErrNotFound
+	}
+	return farms[0], nil
+}
+
+func (p *postgresFarmRepository) GetByOwnerID(ctx context.Context, ownerID string) ([]domain.Farm, error) {
+	query := `
+		SELECT uuid, name, lat, lon, created_at, updated_at, owner_id
+		FROM farms
+		WHERE owner_id = $1`
+
+	return p.fetch(ctx, query, ownerID)
+}
+
+func (p *postgresFarmRepository) CreateOrUpdate(ctx context.Context, f *domain.Farm) error {
+	if strings.TrimSpace(f.UUID) == "" {
+		f.UUID = uuid.New().String()
+	}
+
+	query := `  
+		INSERT INTO farms (uuid, name, lat, lon, created_at, updated_at, owner_id)  
+		VALUES ($1, $2, $3, $4, NOW(), NOW(), $5)
+		ON CONFLICT (uuid) DO UPDATE
+		SET name = EXCLUDED.name,
+		    lat = EXCLUDED.lat,
+		    lon = EXCLUDED.lon,
+		    updated_at = NOW(),
+		    owner_id = EXCLUDED.owner_id
+		RETURNING uuid, created_at, updated_at`
+
+	return p.conn.QueryRow(
+		ctx,
+		query,
+		f.UUID,
+		f.Name,
+		f.Lat,
+		f.Lon,
+		f.OwnerID,
+	).Scan(&f.CreatedAt, &f.UpdatedAt)
+}
+
+func (p *postgresFarmRepository) Delete(ctx context.Context, uuid string) error {
+	query := `DELETE FROM farms WHERE uuid = $1`
+	_, err := p.conn.Exec(ctx, query, uuid)
+	return err
+}