feat: Initial commit

internal/services/env_service.go

@@ -0,0 +1,68 @@
+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
+}