package main

import (
	"Blog/bootstrap"
	"Blog/internal/client"
	"Blog/internal/consts"
	"Blog/internal/controller"
	"Blog/internal/model"
	"Blog/internal/repository"
	"Blog/internal/service"
	"context"
	"flag"
	"fmt"
	"io"
	"os"
	"time"

	"github.com/redis/go-redis/v9"
	"github.com/sirupsen/logrus"
	"gopkg.in/yaml.v2"
)

func main() {
	profile := flag.String("profile", "dev", "启动的配置文件")
	flag.Parse()
	inits(*profile)

	runServer()

	defer func() {
		logrus.Info("程序结束，关闭资源")
		// db.Close()
		client.RedisClient().Close()
	}()
}

func inits(profile string) {
	//日志文件位置
	f, err := os.OpenFile("./Iris.log", os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0644)
	if err != nil {
		panic("日志文件错误")
	}
	//控制台和日志文件都输出
	iw := io.MultiWriter(f, os.Stdout)
	//日志设置
	// logrus.SetFormatter(log.Ldate | log.Ltime | log.Lmicroseconds | log.Ltime | log.Llongfile)
	logrus.SetOutput(iw)
	logrus.SetFormatter(&logrus.TextFormatter{
		ForceColors:     true,
		FullTimestamp:   true,
		TimestampFormat: "2006-01-02 15:04:05.000",
	})
	logrus.SetReportCaller(true)
	logrus.Info("init bootstrap")
	filePath := fmt.Sprintf("conf/bootstrap-%v.yaml", profile)
	logrus.Info("当前Profile文件位置:", filePath)
	b, err := os.ReadFile(filePath)
	if err != nil {
		panic(err)
	}
	err3 := yaml.Unmarshal(b, &bootstrap.Config)
	if err3 != nil {
		panic(err3)
	}

	// profileName := bootstrap.Config.Profile
	logrus.Info("当前Profile:", profile)
	if profile == "dev" {
		logrus.SetLevel(logrus.DebugLevel)
	}

	//初始化相关数据的配置
	InitData()
}

func runServer() {
	port := bootstrap.Config.Iris.Port
	controller.Router(port)
}

func InitData() {
	initArticleData()
	service.DiaryService.RefreshLatest()
	initContentData()
	initViewRecordData()
}

func initArticleData() {
	var slices []model.BlogArticle = repository.ArticleRepository().FindList(nil, "state = ?", consts.ARTICLE_STATE_PUBLISH)
	logrus.Info("文章初始化数据加载量:", len(slices))
	ctx := context.Background()
	for _, article := range slices {
		// logrus.Info(article)
		publishTime := article.PublishTime
		err := client.RedisClient().ZAdd(ctx, consts.REDIS_BLOG_ARTICLE_LATEST, redis.Z{Score: float64(publishTime), Member: &article}).Err()
		if err != nil {
			logrus.Info(err)
		}
	}
}

func initContentData() {
	var slices []model.BlogTextContent = repository.TextContentRepository().FindList(nil, "state = ?", consts.CONTENT_STATE_PUBLISH)
	logrus.Info("大文本初始化数据加载量:", len(slices))
	for _, content := range slices {
		client.RedisClient().Set(context.Background(), consts.REDIS_BLOG_CONTENT+content.RelId, &content, time.Duration(0))
	}
}

func initViewRecordData() {
	type viewRecord struct {
		RelId  string
		Counts int
	}
	var slices []viewRecord
	repository.ViewRecordRep().Table(consts.TABLE_BLOG_VIEW_RECORD).Select("rel_id ,count(1) counts").Group("rel_id").Find(&slices)
	logrus.Info("点击量初始化数据加载量:", len(slices))
	for _, val := range slices {
		client.RedisClient().HSet(context.Background(), consts.REDIS_BLOG_VIEW_RECORD, map[string]any{val.RelId: val.Counts})
	}
}