campaign

internal/repository/customer_points_repository.go

@@ -1,117 +1,97 @@
 package repository
 
 import (
-	"apskel-pos-be/internal/entities"
 	"context"
 	"fmt"
 
-	"github.com/google/uuid"
+	"apskel-pos-be/internal/entities"
+
 	"gorm.io/gorm"
 )
 
-type CustomerPointsRepository struct {
+type CustomerPointsRepository interface {
+	GetCustomerTotalPoints(ctx context.Context, customerID string) (int64, error)
+	GetCustomerTotalTokens(ctx context.Context, customerID string) (int64, error)
+	GetCustomerPointsHistory(ctx context.Context, customerID string, limit int) ([]entities.CustomerPoints, error)
+	GetCustomerTokensHistory(ctx context.Context, customerID string, limit int) ([]entities.CustomerTokens, error)
+}
+
+type customerPointsRepository struct {
 	db *gorm.DB
 }
 
-func NewCustomerPointsRepository(db *gorm.DB) *CustomerPointsRepository {
-	return &CustomerPointsRepository{db: db}
-}
-
-func (r *CustomerPointsRepository) Create(ctx context.Context, customerPoints *entities.CustomerPoints) error {
-	return r.db.WithContext(ctx).Create(customerPoints).Error
-}
-
-func (r *CustomerPointsRepository) GetByID(ctx context.Context, id uuid.UUID) (*entities.CustomerPoints, error) {
-	var customerPoints entities.CustomerPoints
-	err := r.db.WithContext(ctx).Preload("Customer").Where("id = ?", id).First(&customerPoints).Error
-	if err != nil {
-		return nil, err
+func NewCustomerPointsRepository(db *gorm.DB) CustomerPointsRepository {
+	return &customerPointsRepository{
+		db: db,
 	}
-	return &customerPoints, nil
 }
 
-func (r *CustomerPointsRepository) GetByCustomerID(ctx context.Context, customerID uuid.UUID) (*entities.CustomerPoints, error) {
-	var customerPoints entities.CustomerPoints
-	err := r.db.WithContext(ctx).Preload("Customer").Where("customer_id = ?", customerID).First(&customerPoints).Error
-	if err != nil {
-		return nil, err
-	}
-	return &customerPoints, nil
-}
+func (r *customerPointsRepository) GetCustomerTotalPoints(ctx context.Context, customerID string) (int64, error) {
+	var totalPoints int64
 
-func (r *CustomerPointsRepository) List(ctx context.Context, offset, limit int, search string, sortBy, sortOrder string) ([]entities.CustomerPoints, int64, error) {
-	var customerPoints []entities.CustomerPoints
-	var total int64
-
-	query := r.db.WithContext(ctx).Preload("Customer")
-
-	if search != "" {
-		searchTerm := "%" + search + "%"
-		query = query.Joins("JOIN customers ON customer_points.customer_id = customers.id").
-			Where("customers.name ILIKE ? OR customers.email ILIKE ?", searchTerm, searchTerm)
-	}
-
-	if err := query.Model(&entities.CustomerPoints{}).Count(&total).Error; err != nil {
-		return nil, 0, err
-	}
-
-	if sortBy != "" {
-		if sortOrder == "" {
-			sortOrder = "asc"
-		}
-		query = query.Order(fmt.Sprintf("customer_points.%s %s", sortBy, sortOrder))
-	} else {
-		query = query.Order("customer_points.created_at DESC")
-	}
-
-	err := query.Offset(offset).Limit(limit).Find(&customerPoints).Error
-	if err != nil {
-		return nil, 0, err
-	}
-
-	return customerPoints, total, nil
-}
-
-func (r *CustomerPointsRepository) Update(ctx context.Context, customerPoints *entities.CustomerPoints) error {
-	return r.db.WithContext(ctx).Save(customerPoints).Error
-}
-
-func (r *CustomerPointsRepository) Delete(ctx context.Context, id uuid.UUID) error {
-	return r.db.WithContext(ctx).Delete(&entities.CustomerPoints{}, id).Error
-}
-
-func (r *CustomerPointsRepository) AddPoints(ctx context.Context, customerID uuid.UUID, points int64) error {
-	return r.db.WithContext(ctx).Model(&entities.CustomerPoints{}).
+	err := r.db.WithContext(ctx).
+		Model(&entities.CustomerPoints{}).
 		Where("customer_id = ?", customerID).
-		Update("balance", gorm.Expr("balance + ?", points)).Error
-}
+		Select("COALESCE(SUM(balance), 0)").
+		Scan(&totalPoints).Error
 
-func (r *CustomerPointsRepository) DeductPoints(ctx context.Context, customerID uuid.UUID, points int64) error {
-	return r.db.WithContext(ctx).Model(&entities.CustomerPoints{}).
-		Where("customer_id = ? AND balance >= ?", customerID, points).
-		Update("balance", gorm.Expr("balance - ?", points)).Error
-}
-
-func (r *CustomerPointsRepository) EnsureCustomerPoints(ctx context.Context, customerID uuid.UUID) (*entities.CustomerPoints, error) {
-	customerPoints, err := r.GetByCustomerID(ctx, customerID)
-	if err == nil {
-		return customerPoints, nil
-	}
-
-	if err != gorm.ErrRecordNotFound {
-		return nil, err
-	}
-
-	// Create new customer points record
-	newCustomerPoints := &entities.CustomerPoints{
-		CustomerID: customerID,
-		Balance:    0,
-	}
-
-	err = r.Create(ctx, newCustomerPoints)
 	if err != nil {
-		return nil, err
+		return 0, fmt.Errorf("failed to get customer total points: %w", err)
 	}
 
-	return newCustomerPoints, nil
+	return totalPoints, nil
+}
+
+func (r *customerPointsRepository) GetCustomerTotalTokens(ctx context.Context, customerID string) (int64, error) {
+	var totalTokens int64
+
+	err := r.db.WithContext(ctx).
+		Model(&entities.CustomerTokens{}).
+		Where("customer_id = ?", customerID).
+		Select("COALESCE(SUM(tokens), 0)").
+		Scan(&totalTokens).Error
+
+	if err != nil {
+		return 0, fmt.Errorf("failed to get customer total tokens: %w", err)
+	}
+
+	return totalTokens, nil
+}
+
+func (r *customerPointsRepository) GetCustomerPointsHistory(ctx context.Context, customerID string, limit int) ([]entities.CustomerPoints, error) {
+	var pointsHistory []entities.CustomerPoints
+
+	query := r.db.WithContext(ctx).
+		Where("customer_id = ?", customerID).
+		Order("created_at DESC")
+
+	if limit > 0 {
+		query = query.Limit(limit)
+	}
+
+	err := query.Find(&pointsHistory).Error
+	if err != nil {
+		return nil, fmt.Errorf("failed to get customer points history: %w", err)
+	}
+
+	return pointsHistory, nil
+}
+
+func (r *customerPointsRepository) GetCustomerTokensHistory(ctx context.Context, customerID string, limit int) ([]entities.CustomerTokens, error) {
+	var tokensHistory []entities.CustomerTokens
+
+	query := r.db.WithContext(ctx).
+		Where("customer_id = ?", customerID).
+		Order("created_at DESC")
+
+	if limit > 0 {
+		query = query.Limit(limit)
+	}
+
+	err := query.Find(&tokensHistory).Error
+	if err != nil {
+		return nil, fmt.Errorf("failed to get customer tokens history: %w", err)
+	}
+
+	return tokensHistory, nil
 }