Skip to content

Commit

Permalink
v0.11.1
Browse files Browse the repository at this point in the history
- 使用统一的models包
- 常量移动到constant包
- 写文件默认0664权限
- 修正import顺序
  • Loading branch information
wetor committed Jan 30, 2024
1 parent 856f43f commit f2f5123
Show file tree
Hide file tree
Showing 72 changed files with 521 additions and 518 deletions.
24 changes: 9 additions & 15 deletions cmd/animego/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -25,13 +25,7 @@ import (
"github.com/wetor/AnimeGo/internal/animego/anisource/bangumi"
"github.com/wetor/AnimeGo/internal/animego/anisource/mikan"
"github.com/wetor/AnimeGo/internal/animego/anisource/themoviedb"
"github.com/wetor/AnimeGo/internal/animego/clientnotifier"
"github.com/wetor/AnimeGo/internal/animego/database"
"github.com/wetor/AnimeGo/internal/animego/downloader"
"github.com/wetor/AnimeGo/internal/animego/feed"
"github.com/wetor/AnimeGo/internal/animego/filter"
"github.com/wetor/AnimeGo/internal/animego/parser"
"github.com/wetor/AnimeGo/internal/animego/renamer"
"github.com/wetor/AnimeGo/internal/constant"
"github.com/wetor/AnimeGo/internal/logger"
"github.com/wetor/AnimeGo/internal/models"
Expand Down Expand Up @@ -210,7 +204,7 @@ func Main() {
}
}
// 初始化rename
renameSrv := wire.GetRenamer(&renamer.Options{
renameSrv := wire.GetRenamer(&models.RenamerOptions{
WG: &wg,
RefreshSecond: config.RefreshSecond,
}, renamePlugin)
Expand All @@ -219,7 +213,7 @@ func Main() {

// ===============================================================================================================
// 初始化database配置
databaseInst, err := wire.GetDatabase(&database.Options{
databaseInst, err := wire.GetDatabase(&models.DatabaseOptions{
SavePath: xpath.P(config.SavePath),
}, bolt)
if err != nil {
Expand All @@ -229,21 +223,21 @@ func Main() {
// ===============================================================================================================
// 初始化downloader

downloadCallback := &clientnotifier.Callback{}
downloaderSrv := wire.GetDownloader(&downloader.Options{
downloadCallback := &models.Callback{}
downloaderSrv := wire.GetDownloader(&models.DownloaderOptions{
RefreshSecond: config.RefreshSecond,
Category: config.Category,
Tag: config.Tag,
AllowDuplicateDownload: config.Download.AllowDuplicateDownload,
WG: &wg,
}, clientSrv, &clientnotifier.Options{
}, clientSrv, &models.NotifierOptions{
DownloadPath: xpath.P(config.DownloadPath),
SavePath: xpath.P(config.SavePath),
Rename: config.Advanced.Download.Rename,
Callback: downloadCallback,
}, databaseInst, renameSrv)

downloadCallback.Renamed = func(data any) error {
downloadCallback.Func = func(data any) error {
return downloaderSrv.Delete(data.(string))
}
// 启动downloader
Expand All @@ -260,9 +254,9 @@ func Main() {
}
// 初始化filter

filterSrv := wire.GetFilter(&filter.Options{
filterSrv := wire.GetFilter(&models.FilterOptions{
DelaySecond: config.Advanced.Feed.DelaySecond,
}, downloaderSrv, &parser.Options{
}, downloaderSrv, &models.ParserOptions{
TMDBFailSkip: config.Default.TMDBFailSkip,
TMDBFailUseTitleSeason: config.Default.TMDBFailUseTitleSeason,
TMDBFailUseFirstSeason: config.Default.TMDBFailUseFirstSeason,
Expand All @@ -272,7 +266,7 @@ func Main() {
}
// ===============================================================================================================
// 初始化定时任务
scheduleSrv := schedule.NewSchedule(&schedule.Options{
scheduleSrv := schedule.NewSchedule(&models.ScheduleOptions{
WG: &wg,
})
// 添加定时任务
Expand Down
14 changes: 7 additions & 7 deletions cmd/plugin/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -9,9 +9,9 @@ import (
"time"

"github.com/wetor/AnimeGo/cmd/common"
feedPlugin "github.com/wetor/AnimeGo/internal/animego/feed"
filterPlugin "github.com/wetor/AnimeGo/internal/animego/filter"
renamerPlugin "github.com/wetor/AnimeGo/internal/animego/renamer"
"github.com/wetor/AnimeGo/internal/animego/feed"
"github.com/wetor/AnimeGo/internal/animego/filter"
"github.com/wetor/AnimeGo/internal/animego/renamer"
"github.com/wetor/AnimeGo/internal/constant"
"github.com/wetor/AnimeGo/internal/models"
"github.com/wetor/AnimeGo/internal/plugin"
Expand Down Expand Up @@ -78,14 +78,14 @@ func pluginPython(entryFunc string, info *models.Plugin) map[string]any {
}

func pluginFilter(items []*models.FeedItem, info *models.Plugin) []*models.FeedItem {
f := filterPlugin.NewFilterPlugin(info)
f := filter.NewFilterPlugin(info)
res, _ := f.FilterAll(items)
return res
}

func pluginRename(anime *models.AnimeEntity, info *models.Plugin) []*models.RenameResult {
result := make([]*models.RenameResult, len(anime.Ep))
r := renamerPlugin.NewRenamePlugin(info)
r := renamer.NewRenamePlugin(info)
var err error
for i, ep := range anime.Ep {
result[i], err = r.Rename(anime, i, ep.Src)
Expand Down Expand Up @@ -134,7 +134,7 @@ func Main() {
plugin.Init(&plugin.Options{
Path: dir,
Debug: pDebug,
Feed: feedPlugin.NewRss(),
Feed: feed.NewRss(),
})

pluginInfo := &models.Plugin{
Expand Down Expand Up @@ -193,7 +193,7 @@ func Main() {
log.Infof("[%d] filter结果: %v", i, string(jsonData))
}
case constant.PluginTemplateSchedule, constant.PluginTemplateFeed:
s := schedule.NewSchedule(&schedule.Options{
s := schedule.NewSchedule(&models.ScheduleOptions{
WG: &wg,
})
if pPlugin == constant.PluginTemplateSchedule {
Expand Down
23 changes: 12 additions & 11 deletions configs/config_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ import (

"github.com/stretchr/testify/assert"
"github.com/wetor/AnimeGo/configs"
"github.com/wetor/AnimeGo/internal/constant"
"github.com/wetor/AnimeGo/pkg/log"
"github.com/wetor/AnimeGo/pkg/utils"
"github.com/wetor/AnimeGo/test"
Expand Down Expand Up @@ -40,7 +41,7 @@ func TestUpdateConfig_120(t *testing.T) {
configs.ConfigVersion = "1.2.0"

file, _ := test.GetData(testdata, "animego_110.yaml")
_ = os.WriteFile("data/animego.yaml", file, 0666)
_ = os.WriteFile("data/animego.yaml", file, constant.WriteFilePerm)
configs.UpdateConfig("data/animego.yaml", false)

EqualFile(t, "data/animego.yaml", test.GetDataPath(testdata, "animego_120.yaml"))
Expand All @@ -49,7 +50,7 @@ func TestUpdateConfig_120(t *testing.T) {
func TestUpdateConfig_130(t *testing.T) {
configs.ConfigVersion = "1.3.0"
file, _ := test.GetData(testdata, "animego_120.yaml")
_ = os.WriteFile("data/animego.yaml", file, 0666)
_ = os.WriteFile("data/animego.yaml", file, constant.WriteFilePerm)
configs.UpdateConfig("data/animego.yaml", false)

EqualFile(t, "data/animego.yaml", test.GetDataPath(testdata, "animego_130.yaml"))
Expand All @@ -58,7 +59,7 @@ func TestUpdateConfig_130(t *testing.T) {
func TestUpdateConfig_140(t *testing.T) {
configs.ConfigVersion = "1.4.0"
file, _ := test.GetData(testdata, "animego_130.yaml")
_ = os.WriteFile("data/animego.yaml", file, 0666)
_ = os.WriteFile("data/animego.yaml", file, constant.WriteFilePerm)
configs.UpdateConfig("data/animego.yaml", false)

EqualFile(t, "data/animego.yaml", test.GetDataPath(testdata, "animego_140.yaml"))
Expand All @@ -67,7 +68,7 @@ func TestUpdateConfig_140(t *testing.T) {
func TestUpdateConfig_141(t *testing.T) {
configs.ConfigVersion = "1.4.1"
file, _ := test.GetData(testdata, "animego_140.yaml")
_ = os.WriteFile("data/animego.yaml", file, 0666)
_ = os.WriteFile("data/animego.yaml", file, constant.WriteFilePerm)
configs.UpdateConfig("data/animego.yaml", false)

EqualFile(t, "data/animego.yaml", test.GetDataPath(testdata, "animego_141.yaml"))
Expand All @@ -76,7 +77,7 @@ func TestUpdateConfig_141(t *testing.T) {
func TestUpdateConfig_150(t *testing.T) {
configs.ConfigVersion = "1.5.0"
file, _ := test.GetData(testdata, "animego_141.yaml")
_ = os.WriteFile("data/animego.yaml", file, 0666)
_ = os.WriteFile("data/animego.yaml", file, constant.WriteFilePerm)
configs.UpdateConfig("data/animego.yaml", false)

EqualFile(t, "data/animego.yaml", test.GetDataPath(testdata, "animego_150.yaml"))
Expand All @@ -85,7 +86,7 @@ func TestUpdateConfig_150(t *testing.T) {
func TestUpdateConfig_151(t *testing.T) {
configs.ConfigVersion = "1.5.1"
file, _ := test.GetData(testdata, "animego_150.yaml")
_ = os.WriteFile("data/animego.yaml", file, 0666)
_ = os.WriteFile("data/animego.yaml", file, constant.WriteFilePerm)
configs.UpdateConfig("data/animego.yaml", false)

EqualFile(t, "data/animego.yaml", test.GetDataPath(testdata, "animego_151.yaml"))
Expand All @@ -94,7 +95,7 @@ func TestUpdateConfig_151(t *testing.T) {
func TestUpdateConfig_152(t *testing.T) {
configs.ConfigVersion = "1.5.2"
file, _ := test.GetData(testdata, "animego_151.yaml")
_ = os.WriteFile("data/animego.yaml", file, 0666)
_ = os.WriteFile("data/animego.yaml", file, constant.WriteFilePerm)
configs.UpdateConfig("data/animego.yaml", false)

EqualFile(t, "data/animego.yaml", test.GetDataPath(testdata, "animego_152.yaml"))
Expand All @@ -103,7 +104,7 @@ func TestUpdateConfig_152(t *testing.T) {
func TestUpdateConfig_160(t *testing.T) {
configs.ConfigVersion = "1.6.0"
file, _ := test.GetData(testdata, "animego_152.yaml")
_ = os.WriteFile("data/animego.yaml", file, 0666)
_ = os.WriteFile("data/animego.yaml", file, constant.WriteFilePerm)
configs.UpdateConfig("data/animego.yaml", false)

EqualFile(t, "data/animego.yaml", test.GetDataPath(testdata, "animego_160.yaml"))
Expand All @@ -112,7 +113,7 @@ func TestUpdateConfig_160(t *testing.T) {
func TestUpdateConfig_161(t *testing.T) {
configs.ConfigVersion = "1.6.1"
file, _ := test.GetData(testdata, "animego_160.yaml")
_ = os.WriteFile("data/animego.yaml", file, 0666)
_ = os.WriteFile("data/animego.yaml", file, constant.WriteFilePerm)
configs.UpdateConfig("data/animego.yaml", false)

EqualFile(t, "data/animego.yaml", test.GetDataPath(testdata, "animego_161.yaml"))
Expand All @@ -121,7 +122,7 @@ func TestUpdateConfig_161(t *testing.T) {
func TestUpdateConfig_162(t *testing.T) {
configs.ConfigVersion = "1.6.2"
file, _ := test.GetData(testdata, "animego_161.yaml")
_ = os.WriteFile("data/animego.yaml", file, 0666)
_ = os.WriteFile("data/animego.yaml", file, constant.WriteFilePerm)
configs.UpdateConfig("data/animego.yaml", false)

EqualFile(t, "data/animego.yaml", test.GetDataPath(testdata, "animego_162.yaml"))
Expand All @@ -130,7 +131,7 @@ func TestUpdateConfig_162(t *testing.T) {
func TestUpdateConfig_170(t *testing.T) {
configs.ConfigVersion = "1.7.0"
file, _ := test.GetData(testdata, "animego_162.yaml")
_ = os.WriteFile("data/animego.yaml", file, 0666)
_ = os.WriteFile("data/animego.yaml", file, constant.WriteFilePerm)
configs.UpdateConfig("data/animego.yaml", false)

EqualFile(t, "data/animego.yaml", test.GetDataPath(testdata, "animego_170.yaml"))
Expand Down
3 changes: 2 additions & 1 deletion configs/default.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@ package configs
import (
"os"

"github.com/wetor/AnimeGo/internal/constant"
encoder "github.com/wetor/AnimeGo/third_party/yaml-encoder"
)

Expand Down Expand Up @@ -190,7 +191,7 @@ func DefaultFile(filename string) error {
return err
}
// 所有者可读可写,其他用户只读
err = os.WriteFile(filename, data, 0644)
err = os.WriteFile(filename, data, constant.WriteFilePerm)
if err != nil {
return err
}
Expand Down
3 changes: 2 additions & 1 deletion configs/init.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@ import (
"path"

"github.com/caarlos0/env/v9"
"github.com/wetor/AnimeGo/internal/constant"
"github.com/wetor/AnimeGo/pkg/utils"
)

Expand Down Expand Up @@ -63,7 +64,7 @@ func InitEnvConfig(configFile, saveFile string) {
panic(err)
}

err = os.WriteFile(saveFile, data, 0644)
err = os.WriteFile(saveFile, data, constant.WriteFilePerm)
if err != nil {
panic(err)
}
Expand Down
Loading

0 comments on commit f2f5123

Please sign in to comment.