package branches

import (
	"context"
	"furtuna-be/internal/common/logger"
	"furtuna-be/internal/entity"

	"go.uber.org/zap"
	"gorm.io/gorm"
)

type BranchRepository struct {
	db *gorm.DB
}

func NewBranchRepository(db *gorm.DB) *BranchRepository {
	return &BranchRepository{
		db: db,
	}
}

func (b *BranchRepository) CreateBranch(ctx context.Context, branch *entity.BranchDB) (*entity.BranchDB, error) {
	err := b.db.Create(branch).Error
	if err != nil {
		logger.ContextLogger(ctx).Error("error when create branch", zap.Error(err))
		return nil, err
	}
	return branch, nil
}

func (b *BranchRepository) UpdateBranch(ctx context.Context, branch *entity.BranchDB) (*entity.BranchDB, error) {
	if err := b.db.Save(branch).Error; err != nil {
		logger.ContextLogger(ctx).Error("error when update branch", zap.Error(err))
		return nil, err
	}
	return branch, nil
}

func (b *BranchRepository) GetBranchByID(ctx context.Context, id int64) (*entity.BranchDB, error) {
	branch := new(entity.BranchDB)
	if err := b.db.First(branch, id).Error; err != nil {
		logger.ContextLogger(ctx).Error("error when get by id branch", zap.Error(err))
		return nil, err
	}
	return branch, nil
}

func (b *BranchRepository) GetAllBranches(ctx context.Context, req entity.BranchSearch) (entity.BranchList, int, error) {
	var branches []*entity.BranchDB
	var total int64

	query := b.db
	query = query.Where("deleted_at  is null")

	if req.Search != "" {
		query = query.Where("name ILIKE ?", "%"+req.Search+"%")
	}

	if req.Name != "" {
		query = query.Where("name ILIKE ?", "%"+req.Name+"%")
	}

	if req.Limit > 0 {
		query = query.Limit(req.Limit)
	}

	if req.Offset > 0 {
		query = query.Offset(req.Offset)
	}

	if err := query.Find(&branches).Error; err != nil {
		logger.ContextLogger(ctx).Error("error when get all branches", zap.Error(err))
		return nil, 0, err
	}

	if err := b.db.Model(&entity.BranchDB{}).Where(query).Count(&total).Error; err != nil {
		logger.ContextLogger(ctx).Error("error when count branches", zap.Error(err))
		return nil, 0, err
	}

	return branches, int(total), nil
}

func (b *BranchRepository) DeleteBranch(ctx context.Context, id int64) error {
	branch := new(entity.BranchDB)
	branch.ID = id
	if err := b.db.Delete(branch).Error; err != nil {
		return err
	}
	return nil
}