refactor: eiblog

pkg/core/blog/file/timer.go

@@ -0,0 +1,163 @@
+// Package file provides ...
+package file
+
+import (
+	"os"
+	"path/filepath"
+	"text/template"
+	"time"
+
+	"github.com/eiblog/eiblog/v2/pkg/cache"
+	"github.com/eiblog/eiblog/v2/pkg/config"
+
+	"github.com/eiblog/utils/tmpl"
+	"github.com/sirupsen/logrus"
+)
+
+var xmlTmpl *template.Template
+
+func init() {
+	root := filepath.Join(config.WorkDir, "conf", "tpl", "*.xml")
+
+	var err error
+	xmlTmpl, err = template.New("").Funcs(template.FuncMap{
+		"dateformat": tmpl.DateFormat,
+	}).ParseGlob(root)
+	if err != nil {
+		panic(err)
+	}
+	go timerFeed()
+	go timerSitemap()
+}
+
+// timerFeed 定时刷新feed
+func timerFeed() {
+	tpl := xmlTmpl.Lookup("feedTpl.xml")
+	if tpl == nil {
+		logrus.Info("file: not found: feedTpl.xml")
+		return
+	}
+
+	t := time.NewTicker(time.Hour * 4)
+	for now := range t.C {
+		_, _, articles := cache.Ei.PageArticles(1, 20)
+		params := map[string]interface{}{
+			"Titile":    cache.Ei.Blogger.BTitle,
+			"SubTitle":  cache.Ei.Blogger.SubTitle,
+			"Host":      config.Conf.BlogApp.Host,
+			"FeedrURL":  config.Conf.BlogApp.FeedRPC.FeedrURL,
+			"BuildDate": now.Format(time.RFC1123Z),
+			"Articles":  articles,
+		}
+		f, err := os.OpenFile("assets/feed.xml", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666)
+		if err != nil {
+			logrus.Error("file: timerFeed.OpenFile: ", err)
+			continue
+		}
+		defer f.Close()
+		err = tpl.Execute(f, params)
+		if err != nil {
+			logrus.Error("file: timerFeed.Execute: ", err)
+			continue
+		}
+	}
+}
+
+// timerSitemap 定时刷新sitemap
+func timerSitemap() {
+	tpl := xmlTmpl.Lookup("sitemapTpl.xml")
+	if tpl == nil {
+		logrus.Info("file: not found: sitemapTpl.xml")
+		return
+	}
+
+	t := time.NewTicker(time.Hour * 4)
+	for range t.C {
+		params := map[string]interface{}{
+			"Articles": cache.Ei.Articles,
+			"Host":     config.Conf.BlogApp.Host,
+		}
+		f, err := os.OpenFile("assets/sitemap.xml", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666)
+		if err != nil {
+			logrus.Error("file: timerSitemap.OpenFile: ", err)
+			continue
+		}
+		defer f.Close()
+		err = tpl.Execute(f, params)
+		if err != nil {
+			logrus.Error("file: timerSitemap.Execute: ", err)
+			continue
+		}
+	}
+}
+
+// generateOpensearch 生成opensearch.xml
+func generateOpensearch() {
+	tpl := xmlTmpl.Lookup("opensearchTpl.xml")
+	if tpl == nil {
+		logrus.Info("file: not found: opensearchTpl.xml")
+		return
+	}
+	params := map[string]string{
+		"BTitle":   cache.Ei.Blogger.BTitle,
+		"SubTitle": cache.Ei.Blogger.SubTitle,
+		"Host":     config.Conf.BlogApp.Host,
+	}
+	f, err := os.OpenFile("static/opensearch.xml", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666)
+	if err != nil {
+		logrus.Error("file: generateOpensearch.OpenFile: ", err)
+		return
+	}
+	defer f.Close()
+	err = tpl.Execute(f, params)
+	if err != nil {
+		logrus.Error("file: generateOpensearch.Execute: ", err)
+		return
+	}
+}
+
+// generateRobots 生成robots.txt
+func generateRobots() {
+	tpl := xmlTmpl.Lookup("robotsTpl.xml")
+	if tpl == nil {
+		logrus.Info("file: not found: robotsTpl.xml")
+		return
+	}
+	params := map[string]string{
+		"Host": config.Conf.BlogApp.Host,
+	}
+	f, err := os.OpenFile("static/robots.txt", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666)
+	if err != nil {
+		logrus.Error("file: generateRobots.OpenFile: ", err)
+		return
+	}
+	defer f.Close()
+	err = tpl.Execute(f, params)
+	if err != nil {
+		logrus.Error("file: generateRobots.Execute: ", err)
+		return
+	}
+}
+
+// generateCrossdomain 生成crossdomain.xml
+func generateCrossdomain() {
+	tpl := xmlTmpl.Lookup("crossdomainTpl.xml")
+	if tpl == nil {
+		logrus.Info("file: not found: crossdomainTpl.xml")
+		return
+	}
+	params := map[string]string{
+		"Host": config.Conf.BlogApp.Host,
+	}
+	f, err := os.OpenFile("static/crossdomain.xml", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666)
+	if err != nil {
+		logrus.Error("file: generateCrossdomain.OpenFile: ", err)
+		return
+	}
+	defer f.Close()
+	err = tpl.Execute(f, params)
+	if err != nil {
+		logrus.Error("file: generateCrossdomain.Execute: ", err)
+		return
+	}
+}