feat: seed default dataset sources

backend/internal/initdb/dataset.go

@@ -0,0 +1,58 @@
+package initdb
+
+import (
+	"errors"
+	"fmt"
+
+	"gorm.io/gorm"
+	"senlinai-agent/backend/internal/models"
+)
+
+var defaultDatasetSources = []struct {
+	Name    string
+	URL     string
+	IconURL string
+}{
+	{Name: "彭博社最新报道", URL: "https://bbg.buzzing.cc/feed.json", IconURL: "https://bbg.buzzing.cc/apple-touch-icon.png"},
+	{Name: "国外科技头条", URL: "https://tech.buzzing.cc/feed.json", IconURL: "https://tech.buzzing.cc/apple-touch-icon.png"},
+	{Name: "国外财经新闻", URL: "https://finance.buzzing.cc/feed.json", IconURL: "https://finance.buzzing.cc/apple-touch-icon.png"},
+	{Name: "华尔街日报热门", URL: "https://wsj.buzzing.cc/feed.json", IconURL: "https://wsj.buzzing.cc/apple-touch-icon.png"},
+	{Name: "Product Hunt", URL: "https://ph.buzzing.cc/feed.json", IconURL: "https://ph.buzzing.cc/apple-touch-icon.png"},
+	{Name: "Dev.to", URL: "https://dev.buzzing.cc/feed.json", IconURL: "https://dev.buzzing.cc/apple-touch-icon.png"},
+}
+
+// InitDataset creates the default Buzzing RSS sources for the default account when the source table is empty.
+func InitDataset(database *gorm.DB) error {
+	var count int64
+	if err := database.Model(&models.SaDatasetSource{}).Count(&count).Error; err != nil {
+		return err
+	}
+	if count > 0 {
+		return nil
+	}
+
+	var owner models.SaUser
+	err := database.Where("email = ?", rootUsername).First(&owner).Error
+	if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+		return fmt.Errorf("find root user for default dataset sources: %w", err)
+	}
+	if errors.Is(err, gorm.ErrRecordNotFound) {
+		if err := database.Order("id asc").First(&owner).Error; err != nil {
+			return fmt.Errorf("find owner for default dataset sources: %w", err)
+		}
+	}
+
+	sources := make([]models.SaDatasetSource, 0, len(defaultDatasetSources))
+	for _, source := range defaultDatasetSources {
+		sources = append(sources, models.SaDatasetSource{
+			CreatedBy:   owner.ID,
+			Name:        source.Name,
+			Kind:        "rss",
+			URL:         source.URL,
+			IconURL:     source.IconURL,
+			Description: "Buzzing 中文聚合订阅",
+			Enabled:     true,
+		})
+	}
+	return database.Create(&sources).Error
+}