110 lines
2.3 KiB
Go
110 lines
2.3 KiB
Go
package db_bridge
|
|
|
|
import (
|
|
"github.com/goccy/go-json"
|
|
"seanime/internal/database/db"
|
|
"seanime/internal/database/models"
|
|
"seanime/internal/library/anime"
|
|
)
|
|
|
|
var CurrAutoDownloaderRules []*anime.AutoDownloaderRule
|
|
|
|
func GetAutoDownloaderRules(db *db.Database) ([]*anime.AutoDownloaderRule, error) {
|
|
|
|
//if CurrAutoDownloaderRules != nil {
|
|
// return CurrAutoDownloaderRules, nil
|
|
//}
|
|
|
|
var res []*models.AutoDownloaderRule
|
|
err := db.Gorm().Find(&res).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Unmarshal the data
|
|
var rules []*anime.AutoDownloaderRule
|
|
for _, r := range res {
|
|
smBytes := r.Value
|
|
var sm anime.AutoDownloaderRule
|
|
if err := json.Unmarshal(smBytes, &sm); err != nil {
|
|
return nil, err
|
|
}
|
|
sm.DbID = r.ID
|
|
rules = append(rules, &sm)
|
|
}
|
|
|
|
//CurrAutoDownloaderRules = rules
|
|
|
|
return rules, nil
|
|
}
|
|
|
|
func GetAutoDownloaderRule(db *db.Database, id uint) (*anime.AutoDownloaderRule, error) {
|
|
var res models.AutoDownloaderRule
|
|
err := db.Gorm().First(&res, id).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Unmarshal the data
|
|
smBytes := res.Value
|
|
var sm anime.AutoDownloaderRule
|
|
if err := json.Unmarshal(smBytes, &sm); err != nil {
|
|
return nil, err
|
|
}
|
|
sm.DbID = res.ID
|
|
|
|
return &sm, nil
|
|
}
|
|
|
|
func GetAutoDownloaderRulesByMediaId(db *db.Database, mediaId int) (ret []*anime.AutoDownloaderRule) {
|
|
rules, err := GetAutoDownloaderRules(db)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
for _, rule := range rules {
|
|
if rule.MediaId == mediaId {
|
|
ret = append(ret, rule)
|
|
}
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func InsertAutoDownloaderRule(db *db.Database, sm *anime.AutoDownloaderRule) error {
|
|
|
|
CurrAutoDownloaderRules = nil
|
|
|
|
// Marshal the data
|
|
bytes, err := json.Marshal(sm)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Save the data
|
|
return db.Gorm().Create(&models.AutoDownloaderRule{
|
|
Value: bytes,
|
|
}).Error
|
|
}
|
|
|
|
func DeleteAutoDownloaderRule(db *db.Database, id uint) error {
|
|
|
|
CurrAutoDownloaderRules = nil
|
|
|
|
return db.Gorm().Delete(&models.AutoDownloaderRule{}, id).Error
|
|
}
|
|
|
|
func UpdateAutoDownloaderRule(db *db.Database, id uint, sm *anime.AutoDownloaderRule) error {
|
|
|
|
CurrAutoDownloaderRules = nil
|
|
|
|
// Marshal the data
|
|
bytes, err := json.Marshal(sm)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Save the data
|
|
return db.Gorm().Model(&models.AutoDownloaderRule{}).Where("id = ?", id).Update("value", bytes).Error
|
|
}
|