feat: Initial commit

internal/services/script_service.go

@@ -0,0 +1,52 @@
+package services
+
+import (
+	"baihu/internal/database"
+	"baihu/internal/models"
+)
+
+type ScriptService struct{}
+
+func NewScriptService() *ScriptService {
+	return &ScriptService{}
+}
+
+func (ss *ScriptService) CreateScript(name, content string, userID int) *models.Script {
+	script := &models.Script{
+		Name:    name,
+		Content: content,
+		UserID:  uint(userID),
+	}
+	database.DB.Create(script)
+	return script
+}
+
+func (ss *ScriptService) GetScriptsByUserID(userID int) []models.Script {
+	var scripts []models.Script
+	database.DB.Where("user_id = ?", userID).Find(&scripts)
+	return scripts
+}
+
+func (ss *ScriptService) GetScriptByID(id int) *models.Script {
+	var script models.Script
+	if err := database.DB.First(&script, id).Error; err != nil {
+		return nil
+	}
+	return &script
+}
+
+func (ss *ScriptService) UpdateScript(id int, name, content string) *models.Script {
+	var script models.Script
+	if err := database.DB.First(&script, id).Error; err != nil {
+		return nil
+	}
+	script.Name = name
+	script.Content = content
+	database.DB.Save(&script)
+	return &script
+}
+
+func (ss *ScriptService) DeleteScript(id int) bool {
+	result := database.DB.Delete(&models.Script{}, id)
+	return result.RowsAffected > 0
+}