ChineseSubFinder/cmd/chinesesubfinder/main.go

package main

import (
	"github.com/allanpk716/ChineseSubFinder/internal"
	"github.com/allanpk716/ChineseSubFinder/internal/logic/sub_timeline_fixer"
	"github.com/allanpk716/ChineseSubFinder/internal/pkg"
	"github.com/allanpk716/ChineseSubFinder/internal/pkg/hot_fix"
	"github.com/allanpk716/ChineseSubFinder/internal/pkg/log_helper"
	"github.com/allanpk716/ChineseSubFinder/internal/pkg/notify_center"
	"github.com/allanpk716/ChineseSubFinder/internal/pkg/proxy_helper"
	"github.com/allanpk716/ChineseSubFinder/internal/pkg/rod_helper"
	"github.com/allanpk716/ChineseSubFinder/internal/pkg/sub_formatter"
	"github.com/allanpk716/ChineseSubFinder/internal/pkg/sub_formatter/common"
	"github.com/allanpk716/ChineseSubFinder/internal/types"
	"github.com/robfig/cron/v3"
	"github.com/sirupsen/logrus"
)

func init() {

	log = log_helper.GetLogger()

	log.Infoln("ChineseSubFinder Version:", appVersion)

	if pkg.OSCheck() == false {
		panic(`only support Linux and Windows, if you want support MacOS, 
you need implement getDbName() in file: internal/dao/init.go 
and 
implement getSpeFileName() in internal/logic/forced_scan_and_down_sub/forced_scan_and_down_sub.go`)
	}

	config = pkg.GetConfig()
}

func main() {
	if log == nil {
		panic("log init error")
	}
	if config == nil {
		panic("read config error")
	}
	httpProxy := config.HttpProxy
	if config.UseProxy == false {
		httpProxy = ""
	}
	if config.UseProxy == false {
		log.Infoln("UseProxy = false")
	} else {
		log.Infoln("UseProxy:", httpProxy)
		proxySpeed, proxyStatus, err := proxy_helper.ProxyTest(httpProxy)
		if err != nil {
			log.Errorln("ProxyTest Target Site http://google.com", err)
			return
		} else {
			log.Infoln("ProxyTest Target Site http://google.com", "Speed:", proxySpeed, "Status:", proxyStatus)
		}
	}

	// 判断文件夹是否存在
	if pkg.IsDir(config.MovieFolder) == false {
		log.Errorln("MovieFolder not found")
		return
	}
	if pkg.IsDir(config.SeriesFolder) == false {
		log.Errorln("SeriesFolder not found")
		return
	}
	// 读取到的文件夹信息展示
	log.Infoln("MovieFolder:", config.MovieFolder)
	log.Infoln("SeriesFolder:", config.SeriesFolder)

	// ------ Hot Fix Start ------
	// 开始修复
	log.Infoln("HotFix Start...")
	err := hot_fix.HotFixProcess(types.HotFixParam{
		MovieRootDir:  config.MovieFolder,
		SeriesRootDir: config.SeriesFolder,
	})
	if err != nil {
		log.Errorln("HotFixProcess()", err)
		log.Infoln("HotFix End")
		return
	}
	log.Infoln("HotFix End")
	// ------ Hot Fix End ------

	// ------ Change SubName Format Start ------
	/*
		字幕命名格式转换，需要数据库支持
		如果数据库没有记录经过转换，那么默认从 Emby 的格式作为检测的起点，转换到目标的格式
		然后需要在数据库中记录本次的转换结果
	*/
	log.Infoln("Change Sub Name Format Start...")

	renameResults, err := sub_formatter.SubFormatChangerProcess(config.MovieFolder, config.SeriesFolder, common.FormatterName(config.SubNameFormatter))
	// 出错的文件有哪一些
	for s, i := range renameResults.ErrFiles {
		log_helper.GetLogger().Errorln("reformat ErrFile:"+s, i)
	}
	if err != nil {
		log.Errorln("SubFormatChangerProcess()", err)
		return
	}

	log.Infoln("Change Sub Name Format End")
	// ------ Change SubName Format End ------

	// 初始化通知缓存模块
	notify_center.Notify = notify_center.NewNotifyCenter(config.WhenSubSupplierInvalidWebHook)

	log.Infoln("ReloadBrowser Start...")
	// ReloadBrowser 提前把浏览器下载好
	rod_helper.ReloadBrowser()
	log.Infoln("ReloadBrowser End")

	// 任务还没执行完，下一次执行时间到来，下一次执行就跳过不执行
	c := cron.New(cron.WithChain(cron.SkipIfStillRunning(cron.DefaultLogger)))
	// 定时器
	entryID, err := c.AddFunc("@every "+config.EveryTime, func() {

		DownLoadStart(httpProxy)
	})
	if err != nil {
		log.Errorln("cron entryID:", entryID, "Error:", err)
		return
	}

	if config.RunAtStartup == true {
		log.Infoln("First Time Download Start")

		DownLoadStart(httpProxy)

		log.Infoln("First Time Download End")
	} else {
		log.Infoln("config.yaml set RunAtStartup: false, so will not Run At Startup, wait", config.EveryTime, "to Download")
	}

	c.Start()
	// 阻塞
	select {}
}

func DownLoadStart(httpProxy string) {
	defer func() {
		log.Infoln("Download One End...")
		notify_center.Notify.Send()
		pkg.CloseChrome()
		rod_helper.Clear()
	}()
	notify_center.Notify.Clear()

	// 下载实例
	downloader := internal.NewDownloader(sub_formatter.GetSubFormatter(config.SubNameFormatter),
		types.ReqParam{
			HttpProxy:                     httpProxy,
			DebugMode:                     config.DebugMode,
			SaveMultiSub:                  config.SaveMultiSub,
			Threads:                       config.Threads,
			SubTypePriority:               config.SubTypePriority,
			WhenSubSupplierInvalidWebHook: config.WhenSubSupplierInvalidWebHook,
			EmbyConfig:                    config.EmbyConfig,
			SaveOneSeasonSub:              config.SaveOneSeasonSub,
		})

	log.Infoln("Download One Started...")

	// 优先级最高。读取特殊文件，启用一些特殊的功能，比如 forced_scan_and_down_sub
	err := downloader.ReadSpeFile()
	if err != nil {
		log.Errorln("ReadSpeFile", err)
	}
	// 刷新 Emby 的字幕，如果下载了字幕倒是没有刷新，则先刷新一次，便于后续的 Emby api 统计逻辑
	err = downloader.RefreshEmbySubList()
	if err != nil {
		log.Errorln("RefreshEmbySubList", err)
		return
	}
	err = downloader.GetUpdateVideoListFromEmby(config.MovieFolder, config.SeriesFolder)
	if err != nil {
		log.Errorln("GetUpdateVideoListFromEmby", err)
		return
	}
	// 开始下载，电影
	err = downloader.DownloadSub4Movie(config.MovieFolder)
	if err != nil {
		log.Errorln("DownloadSub4Movie", err)
		return
	}
	// 开始下载，连续剧
	err = downloader.DownloadSub4Series(config.SeriesFolder)
	if err != nil {
		log.Errorln("DownloadSub4Series", err)
		return
	}
	// 刷新 Emby 的字幕，下载完毕字幕了，就统一刷新一下
	err = downloader.RefreshEmbySubList()
	if err != nil {
		log.Errorln("RefreshEmbySubList", err)
		return
	}

	// 开始字幕的统一校正
	log.Infoln("Auto Fix Sub Timeline Start...")
	fixer := sub_timeline_fixer.NewSubTimelineFixerHelper(config.EmbyConfig)
	err = fixer.FixRecentlyItemsSubTimeline(config.MovieFolder, config.SeriesFolder)
	if err != nil {
		log.Errorln("FixRecentlyItemsSubTimeline", err)
		return
	}
	log.Infoln("Auto Fix Sub Timeline End")

	// 再次刷新
	// 刷新 Emby 的字幕，下载完毕字幕了，就统一刷新一下
	err = downloader.RefreshEmbySubList()
	if err != nil {
		log.Errorln("RefreshEmbySubList", err)
		return
	}
}

var (
	log    *logrus.Logger
	config *types.Config
)

/*
	没有很好的想法，因为喜欢使用 tag 进行版本的输出标记，但是 tag 的时候编译 docker 前确实可以修改源码替换关键词做到版本与 tag 同步变更
	但是， goreleaser 却不支持这样，会提示源码被改了，无法进行编译发布
	除非不发布、编译 Linux 和 Windows 程序，这样就能做到 tag 与 程序内部输出版本一致。
*/
const appVersion = "v0.18.0"