package system

import (
	"context"

	"management/internal/db/model/dto"
	"management/internal/erpserver/model/system"

	"gorm.io/gorm"
)

// UserStore 定义了 user 模块在 store 层所实现的方法.
type UserStore interface {
	Create(ctx context.Context, obj *system.User) error
	Update(ctx context.Context, obj *system.User) error
	Get(ctx context.Context, id int32) (*system.User, error)
	GetByEmail(ctx context.Context, email string) (*system.User, error)
	All(ctx context.Context) ([]*system.User, error)
	List(ctx context.Context, q dto.SearchDto) ([]*system.User, int64, error)

	UserExpansion
}

// UserExpansion 定义了用户操作的附加方法.
type UserExpansion interface{}

// userStore 是 UserStore 接口的实现.
type userStore struct {
	db *gorm.DB
}

// 确保 userStore 实现了 UserStore 接口.
var _ UserStore = (*userStore)(nil)

// NewUserStore 创建 userStore 的实例.
func NewUserStore(db *gorm.DB) *userStore {
	return &userStore{
		db: db,
	}
}

func (s *userStore) Create(ctx context.Context, obj *system.User) error {
	return s.db.WithContext(ctx).Create(obj).Error
}

func (s *userStore) Update(ctx context.Context, obj *system.User) error {
	return s.db.WithContext(ctx).Save(obj).Error
}

func (s *userStore) Get(ctx context.Context, id int32) (*system.User, error) {
	var user system.User
	err := s.db.WithContext(ctx).Where("id = ?", id).First(&user).Error
	if err != nil {
		return nil, err
	}
	return &user, nil
}

func (s *userStore) GetByEmail(ctx context.Context, email string) (*system.User, error) {
	var user system.User
	err := s.db.WithContext(ctx).Where("email = ?", email).First(&user).Error
	if err != nil {
		return nil, err
	}
	return &user, nil
}

func (s *userStore) All(ctx context.Context) ([]*system.User, error) {
	var users []*system.User
	err := s.db.WithContext(ctx).Find(&users).Error
	if err != nil {
		return nil, err
	}
	return users, nil
}

func (s *userStore) List(ctx context.Context, q dto.SearchDto) ([]*system.User, int64, error) {
	query := s.db.WithContext(ctx).
		Model(&system.User{}).
		Where("created_at BETWEEN? AND?", q.SearchTimeBegin, q.SearchTimeEnd)
	if q.SearchID != 0 {
		query = query.Where("id = ?", q.SearchID)
	}
	if q.SearchName != "" {
		query = query.Where("username LIKE ?", "%"+q.SearchName+"%")
	}
	if q.SearchEmail != "" {
		query = query.Where("email LIKE ?", "%"+q.SearchEmail+"%")
	}
	if q.SearchStatus != 9999 {
		query = query.Where("status = ?", q.SearchStatus)
	}

	var count int64
	err := query.Count(&count).Error
	if err != nil {
		return nil, 0, err
	}

	var users []*system.User
	err = query.
		Order("id DESC").
		Offset((q.Page - 1) * q.Rows).
		Limit(q.Rows).
		Find(&users).Error
	if err != nil {
		return nil, 0, err
	}
	return users, count, nil
}