59 lines
2.1 KiB
Go
59 lines
2.1 KiB
Go
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
|
|
}
|