ChineseSubFinder/cmd/chinesesubfinder/main.go

package main

import (
	"flag"
	"fmt"
	"net/http"
	_ "net/http/pprof"
	"path/filepath"
	"strings"
	"time"

	"github.com/allanpk716/ChineseSubFinder/pkg"

	"github.com/allanpk716/ChineseSubFinder/internal/dao"

	"github.com/allanpk716/ChineseSubFinder/pkg/logic/cron_helper"
	"github.com/allanpk716/ChineseSubFinder/pkg/logic/file_downloader"
	"github.com/allanpk716/ChineseSubFinder/pkg/logic/pre_job"

	"github.com/allanpk716/ChineseSubFinder/pkg/random_auth_key"

	"github.com/allanpk716/ChineseSubFinder/internal/backend"
	"github.com/allanpk716/ChineseSubFinder/pkg/cache_center"
	"github.com/allanpk716/ChineseSubFinder/pkg/common"
	"github.com/allanpk716/ChineseSubFinder/pkg/log_helper"
	"github.com/allanpk716/ChineseSubFinder/pkg/my_util"
	"github.com/allanpk716/ChineseSubFinder/pkg/settings"
	"github.com/sirupsen/logrus"
)

func newLog() *logrus.Logger {
	var level logrus.Level
	// --------------------------------------------------
	// 之前是读取配置文件，现在改为，读取当前目录下，是否有一个特殊的文件，有则启动 Debug 日志级别
	// 那么怎么写入这个文件，就靠额外的逻辑控制了
	if my_util.IsFile(filepath.Join(pkg.ConfigRootDirFPath(), log_helper.DebugFileName)) == true {
		level = logrus.DebugLevel
	} else {
		level = logrus.InfoLevel
	}
	logger := log_helper.NewLogHelper(
		log_helper.LogNameChineseSubFinder,
		pkg.ConfigRootDirFPath(),
		level, time.Duration(7*24)*time.Hour, time.Duration(24)*time.Hour,
		settings.GetSettings().ExperimentalFunction.ExtendLog)
	logger.AddHook(log_helper.NewLoggerHub())

	return logger
}

func init() {
	// 要先进行 flag 的读取，并且写入全局变量中，否则后续的逻辑由于顺序问题故障
	flag.Parse()
	pkg.SetLinuxConfigPathInSelfPath(*SetLinuxConfigPathInSelfPathFlag)

	loggerBase = newLog()
	// --------------------------------------------------
	if strings.ToLower(LiteMode) == "true" {
		loggerBase.Info("LiteMode is true")
		AppVersion += " Lite"
		pkg.SetLiteMode(true)
	} else {
		pkg.SetLiteMode(false)
	}

	loggerBase.Infoln("ChineseSubFinder Version:", AppVersion)
	pkg.SetAppVersion(AppVersion)
	pkg.SetExtEnCode(ExtEnCode)
	if my_util.ReadCustomAuthFile(loggerBase) == false {
		pkg.SetBaseKey(BaseKey)
		pkg.SetAESKey16(AESKey16)
		pkg.SetAESIv16(AESIv16)
	}
	// --------------------------------------------------
	if my_util.OSCheck() == false {
		loggerBase.Panicln(`You should search runtime.GOOS in the project, Implement unimplemented function`)
	}
	// --------------------------------------------------
	// 初始化设备的信息
	dao.UpdateInfo(AppVersion, settings.GetSettings())
}

func main() {

	// ------------------------------------------------------------------------
	// 如果是 Debug 模式，那么就需要写入特殊文件
	if settings.GetSettings().AdvancedSettings.DebugMode == true {
		err := log_helper.WriteDebugFile()
		if err != nil {
			loggerBase.Errorln("log_helper.WriteDebugFile " + err.Error())
		}
		loggerBase = newLog()
		loggerBase.Infoln("Reload Log Settings, level = Debug")
	} else {
		err := log_helper.DeleteDebugFile()
		if err != nil {
			loggerBase.Errorln("log_helper.DeleteDebugFile " + err.Error())
		}
		loggerBase = newLog()
		loggerBase.Infoln("Reload Log Settings, level = Info")
	}
	if pkg.LinuxConfigPathInSelfPath() != "" {

		loggerBase.Infoln("SetLinuxConfigPathInSelfPath:", pkg.LinuxConfigPathInSelfPath())

		if my_util.IsDir(pkg.LinuxConfigPathInSelfPath()) == false {
			// 如果设置了这个路径，但是不存在则会崩溃
			loggerBase.Panicln("LinuxConfigPathInSelfPath", pkg.LinuxConfigPathInSelfPath(), "is not dir")
		}
	}

	// ------------------------------------------------------------------------
	// 设置接口的 API TOKEN
	if settings.GetSettings().ExperimentalFunction.ApiKeySettings.Enabled == true {
		common.SetApiToken(settings.GetSettings().ExperimentalFunction.ApiKeySettings.Key)
	} else {
		common.SetApiToken("")
	}
	// 是否开启开发模式，跳过某些流程
	settings.GetSettings().SpeedDevMode = false
	if settings.GetSettings().SpeedDevMode == true {
		loggerBase.Infoln("Speed Dev Mode is On")
	} else {
		loggerBase.Infoln("Speed Dev Mode is Off")
	}
	if settings.GetSettings().AdvancedSettings.DebugMode == true {
		// 如果是 DebugMode 那么开启性能监控
		go func() {
			// 开启pprof，监听请求
			ip := "0.0.0.0:8080"
			if err := http.ListenAndServe(ip, nil); err != nil {
				fmt.Printf("start pprof failed on %s\n", ip)
			}
		}()
	}
	// ------------------------------------------------------------------------
	// 前置的任务，热修复、字幕修改文件名格式、提前下载好浏览器
	if settings.GetSettings().SpeedDevMode == false {
		pj := pre_job.NewPreJob(settings.GetSettings(), loggerBase)

		if pkg.LiteMode() == true {
			// 不启用 Chrome 相关操作
			err := pj.HotFix().ChangeSubNameFormat().Wait()
			if err != nil {
				loggerBase.Panicln("pre_job", err)
			}
		} else {
			// 完整版模式，启用 Chrome 相关操作
			err := pj.HotFix().ChangeSubNameFormat().ReloadBrowser().Wait()
			if err != nil {
				loggerBase.Panicln("pre_job", err)
			}
		}

	}
	// ----------------------------------------------
	fileDownloader := file_downloader.NewFileDownloader(
		cache_center.NewCacheCenter("local_task_queue", settings.GetSettings(), loggerBase),
		random_auth_key.AuthKey{
			BaseKey:  pkg.BaseKey(),
			AESKey16: pkg.AESKey16(),
			AESIv16:  pkg.AESIv16(),
		})
	// ----------------------------------------------
	cronHelper := cron_helper.NewCronHelper(fileDownloader)
	if settings.GetSettings().UserInfo.Username == "" || settings.GetSettings().UserInfo.Password == "" {
		// 如果没有完成，那么就不开启
		loggerBase.Infoln("Need do Setup")
	} else {
		// 是否完成了 Setup，如果完成了，那么就开启第一次的扫描
		go func() {
			loggerBase.Infoln("Setup is Done")
			cronHelper.Start(settings.GetSettings().CommonSettings.RunScanAtStartUp)
		}()
	}

	nowPort := my_util.ReadCustomPortFile(loggerBase)
	loggerBase.Infoln(fmt.Sprintf("WebUI will listen at 0.0.0.0:%d", nowPort))
	// 支持在外部配置特殊的端口号，以防止本地本占用了无法使用
	restartSignal := make(chan interface{}, 1)
	defer close(restartSignal)
	bend := backend.NewBackEnd(loggerBase, settings.GetSettings(), cronHelper, nowPort, restartSignal)
	go bend.Restart()
	restartSignal <- 1
	// 阻塞
	select {}
}

/*
	使用 git tag 来做版本描述，然后在编译的时候传入版本号信息到这个变量上
*/
var AppVersion = "unknow"

// go build -ldflags="-X main.AppVersion=aabb -X main.ExtEnCode=ccdd" .
var ExtEnCode = "abcdefg1234567890"

// 针对制作群晖的 SPK 应用，无法写入默认的 /config 目录而给出的新的编译条件，直接指向这个目录到当前程序的目录
var SetLinuxConfigPathInSelfPathFlag = flag.String("setconfigselfpath", "", "针对制作群晖的 SPK 应用，无法写入默认的 /config 目录而给出的新的编译条件，直接指向这个目录到当前程序的目录")

var (
	BaseKey  = "0123456789123456789" // 基础的密钥，密钥会基于这个基础的密钥生成
	AESKey16 = "1234567890123456"    // AES密钥
	AESIv16  = "1234567890123456"    // 初始化向量
)

var LiteMode = "false" // 是否轻量级运行模式（不支持Chrome相关操作，也就是无法支持 subhd 和 zimuku 等类似需要复杂爬虫的字幕源）

var loggerBase *logrus.Logger