package services

import (
	"baihu/internal/database"
	"baihu/internal/models"
)

type EnvService struct{}

func NewEnvService() *EnvService {
	return &EnvService{}
}

func (es *EnvService) CreateEnvVar(name, value, remark string, userID int) *models.EnvironmentVariable {
	env := &models.EnvironmentVariable{
		Name:   name,
		Value:  value,
		Remark: remark,
		UserID: uint(userID),
	}
	database.DB.Create(env)
	return env
}

func (es *EnvService) GetEnvVarsByUserID(userID int) []models.EnvironmentVariable {
	var envs []models.EnvironmentVariable
	database.DB.Where("user_id = ?", userID).Find(&envs)
	return envs
}

func (es *EnvService) GetEnvVarsWithPagination(userID int, name string, page, pageSize int) ([]models.EnvironmentVariable, int64) {
	var envs []models.EnvironmentVariable
	var total int64

	query := database.DB.Model(&models.EnvironmentVariable{}).Where("user_id = ?", userID)
	if name != "" {
		query = query.Where("name LIKE ?", "%"+name+"%")
	}

	query.Count(&total)
	query.Order("id DESC").Offset((page - 1) * pageSize).Limit(pageSize).Find(&envs)
	return envs, total
}

func (es *EnvService) GetEnvVarByID(id int) *models.EnvironmentVariable {
	var env models.EnvironmentVariable
	if err := database.DB.First(&env, id).Error; err != nil {
		return nil
	}
	return &env
}

func (es *EnvService) UpdateEnvVar(id int, name, value, remark string) *models.EnvironmentVariable {
	var env models.EnvironmentVariable
	if err := database.DB.First(&env, id).Error; err != nil {
		return nil
	}
	env.Name = name
	env.Value = value
	env.Remark = remark
	database.DB.Save(&env)
	return &env
}

func (es *EnvService) DeleteEnvVar(id int) bool {
	result := database.DB.Delete(&models.EnvironmentVariable{}, id)
	return result.RowsAffected > 0
}