添加 feedback 逻辑

Signed-off-by: allan716 <525223688@qq.com>
ChineseSubFinder · Jul 14, 2022 · ec84c3b · ec84c3b
1 parent 91b599c
commit ec84c3b
Show file tree

Hide file tree

Showing 5 changed files with 71 additions and 0 deletions.
diff --git a/cmd/chinesesubfinder/main.go b/cmd/chinesesubfinder/main.go
@@ -8,6 +8,8 @@ import (
 	"strings"
 	"time"
 
+	"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"
@@ -66,6 +68,9 @@ func init() {
 	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() {

diff --git a/internal/dao/info.go b/internal/dao/info.go
@@ -0,0 +1,40 @@
+package dao
+
+import (
+	"github.com/allanpk716/ChineseSubFinder/internal/models"
+	"github.com/allanpk716/ChineseSubFinder/pkg/my_util"
+	"github.com/allanpk716/ChineseSubFinder/pkg/settings"
+)
+
+func UpdateInfo(version string, settings *settings.Settings) *models.Info {
+	var infos []models.Info
+	GetDb().Find(&infos)
+
+	mediaServerName := ""
+	if settings.EmbySettings.Enable == true {
+		mediaServerName = "Emby"
+	} else {
+		mediaServerName = "None"
+	}
+	if len(infos) == 0 {
+		// 不存在则新增
+		saveInfo := &models.Info{
+			Id:           my_util.RandStringBytesMaskImprSrcSB(64),
+			MediaServer:  mediaServerName,
+			Version:      version,
+			EnableShare:  settings.ExperimentalFunction.ShareSubSettings.ShareSubEnabled,
+			EnableApiKey: settings.ExperimentalFunction.ApiKeySettings.Enabled,
+		}
+		GetDb().Save(saveInfo)
+
+		return saveInfo
+	} else {
+		// 存在则更新
+		infos[0].Version = version
+		infos[0].MediaServer = mediaServerName
+		infos[0].EnableShare = settings.ExperimentalFunction.ShareSubSettings.ShareSubEnabled
+		infos[0].EnableApiKey = settings.ExperimentalFunction.ApiKeySettings.Enabled
+		GetDb().Save(&infos[0])
+		return &infos[0]
+	}
+}
diff --git a/internal/dao/init.go b/internal/dao/init.go
@@ -71,6 +71,7 @@ func InitDb() error {
 		&models.ThirdPartSetVideoPlayedInfo{},
 		&models.MediaInfo{},
 		&models.LowVideoSubInfo{},
+		&models.Info{},
 	)
 	if err != nil {
 		return errors.New(fmt.Sprintf("db AutoMigrate error, %s", err.Error()))

diff --git a/internal/models/info.go b/internal/models/info.go
@@ -0,0 +1,12 @@
+package models
+
+import "gorm.io/gorm"
+
+type Info struct {
+	gorm.Model
+	Id           string // 当前设备的随机ID
+	Version      string // 当前设备的版本
+	MediaServer  string // 媒体服务的名称，没有使用则是 None
+	EnableShare  bool   // 是否开启了共享功能
+	EnableApiKey bool   // 是否开启本地 http api 功能
+}
diff --git a/pkg/logic/cron_helper/cron_helper.go b/pkg/logic/cron_helper/cron_helper.go
@@ -1,6 +1,8 @@
 package cron_helper
 
 import (
+	"github.com/allanpk716/ChineseSubFinder/internal/dao"
+	"github.com/allanpk716/ChineseSubFinder/pkg/global_value"
 	"github.com/allanpk716/ChineseSubFinder/pkg/logic/file_downloader"
 	"github.com/allanpk716/ChineseSubFinder/pkg/logic/scan_played_video_subinfo"
 
@@ -246,6 +248,17 @@ func (ch *CronHelper) Stop() {
 
 func (ch *CronHelper) scanPlayedVideoSub() {
 
+	ch.log.Infoln("Update Info...")
+	nowInfo := dao.UpdateInfo(global_value.AppVersion(), ch.Settings)
+	_, err := ch.FileDownloader.SubtitleBestApi.FeedBack(
+		nowInfo.Id,
+		nowInfo.Version, nowInfo.MediaServer,
+		nowInfo.EnableShare, nowInfo.EnableApiKey)
+	if err != nil {
+		ch.log.Errorln("FeedBack Error:", err)
+		return
+	}
+
 	ch.log.Infoln("------------------------------------------------------")
 	ch.log.Infoln("scanPlayedVideoSub Start")
 	startT := time.Now()