refactor: eiblog

pkg/core/blog/file/file.go

@@ -0,0 +1,49 @@
+// Package file provides ...
+package file
+
+import (
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+)
+
+// RegisterRoutes register routes
+func RegisterRoutes(e *gin.Engine) {
+	e.GET("/rss.html", handleFeed)
+	e.GET("/feed", handleFeed)
+	e.GET("/opensearch.xml", handleOpensearch)
+	e.GET("/sitemap.xml", handleSitemap)
+	e.GET("/robots.txt", handleRobots)
+	e.GET("/crossdomain.xml", handleCrossDomain)
+	e.GET("/favicon.ico", handleFavicon)
+}
+
+// handleFeed feed.xml
+func handleFeed(c *gin.Context) {
+	http.ServeFile(c.Writer, c.Request, "assets/feed.xml")
+}
+
+// handleOpensearch opensearch.xml
+func handleOpensearch(c *gin.Context) {
+	http.ServeFile(c.Writer, c.Request, "assets/opensearch.xml")
+}
+
+// handleRobots robotx.txt
+func handleRobots(c *gin.Context) {
+	http.ServeFile(c.Writer, c.Request, "assets/robots.txt")
+}
+
+// handleSitemap sitemap.xml
+func handleSitemap(c *gin.Context) {
+	http.ServeFile(c.Writer, c.Request, "assets/sitemap.xml")
+}
+
+// handleCrossDomain crossdomain.xml
+func handleCrossDomain(c *gin.Context) {
+	http.ServeFile(c.Writer, c.Request, "assets/crossdomain.xml")
+}
+
+// handleFavicon favicon.ico
+func handleFavicon(c *gin.Context) {
+	http.ServeFile(c.Writer, c.Request, "assets/favicon.ico")
+}

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
+	}
+}