package leads

import (
	"errors"
	"time"

	"github.com/google/uuid"
	"gorm.io/gorm"
)

// GormStore : Store adossé à PostgreSQL via GORM.
type GormStore struct {
	db *gorm.DB
}

func NewGormStore(db *gorm.DB) *GormStore {
	return &GormStore{db: db}
}

func (s *GormStore) Create(l Lead) (Lead, error) {
	l.ID = uuid.NewString()
	l.Status = StatusNew
	l.CreatedAt = time.Now().UTC()
	if err := s.db.Create(&l).Error; err != nil {
		return Lead{}, err
	}
	return l, nil
}

func (s *GormStore) List() ([]Lead, error) {
	var out []Lead
	if err := s.db.Order("created_at DESC").Find(&out).Error; err != nil {
		return nil, err
	}
	return out, nil
}

func (s *GormStore) Get(id string) (Lead, bool) {
	var l Lead
	err := s.db.Where("id = ?", id).First(&l).Error
	if errors.Is(err, gorm.ErrRecordNotFound) || err != nil {
		return Lead{}, false
	}
	return l, true
}

func (s *GormStore) SetStatus(id string, status Status) (Lead, bool) {
	var l Lead
	if err := s.db.Where("id = ?", id).First(&l).Error; err != nil {
		return Lead{}, false
	}
	l.Status = status
	if err := s.db.Save(&l).Error; err != nil {
		return Lead{}, false
	}
	return l, true
}