refactor: eiblog

pkg/core/blog/api.go

@@ -0,0 +1,80 @@
+// Package eiblog provides ...
+package eiblog
+
+import (
+	"net/http"
+
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
+)
+
+// @title APP Demo API
+// @version 1.0
+// @description This is a sample server celler server.
+
+// @BasePath /api
+
+// LogStatus log status
+type LogStatus int
+
+// user log status
+var (
+	LogStatusOut LogStatus = 0
+	LogStatusTFA LogStatus = 1
+	LogStatusIn  LogStatus = 2
+)
+
+// AuthFilter auth filter
+func AuthFilter(c *gin.Context) {
+	if !IsLogined(c) {
+		c.Abort()
+		c.Status(http.StatusUnauthorized)
+		return
+	}
+
+	c.Next()
+}
+
+// SetLogStatus login user
+func SetLogStatus(c *gin.Context, uid string, status LogStatus) {
+	session := sessions.Default(c)
+	session.Set("uid", uid)
+	session.Set("status", int(status))
+	session.Save()
+}
+
+// SetLogout logout user
+func SetLogout(c *gin.Context) {
+	session := sessions.Default(c)
+	session.Set("status", int(LogStatusOut))
+	session.Save()
+}
+
+// IsLogined account logined
+func IsLogined(c *gin.Context) bool {
+	status := GetLogStatus(c)
+	if status < 0 {
+		return false
+	}
+	return status == LogStatusIn
+}
+
+// GetUserID get logined account uuid
+func GetUserID(c *gin.Context) string {
+	session := sessions.Default(c)
+	uid := session.Get("uid")
+	if uid == nil {
+		return ""
+	}
+	return uid.(string)
+}
+
+// GetLogStatus get account log status
+func GetLogStatus(c *gin.Context) LogStatus {
+	session := sessions.Default(c)
+	status := session.Get("status")
+	if status == nil {
+		return -1
+	}
+	return LogStatus(status.(int))
+}

pkg/core/blog/docs/docs.go

@@ -0,0 +1,65 @@
+// GENERATED BY THE COMMAND ABOVE; DO NOT EDIT
+// This file was generated by swaggo/swag at
+// 2021-04-26 15:31:15.52194 +0800 CST m=+0.022347488
+
+package docs
+
+import (
+	"bytes"
+	"encoding/json"
+
+	"github.com/alecthomas/template"
+	"github.com/swaggo/swag"
+)
+
+var doc = `{
+    "schemes": {{ marshal .Schemes }},
+    "swagger": "2.0",
+    "info": {
+        "description": "This is a sample server celler server.",
+        "title": "APP Demo API",
+        "contact": {},
+        "license": {},
+        "version": "1.0"
+    },
+    "host": "{{.Host}}",
+    "basePath": "/api",
+    "paths": {}
+}`
+
+type swaggerInfo struct {
+	Version     string
+	Host        string
+	BasePath    string
+	Schemes     []string
+	Title       string
+	Description string
+}
+
+// SwaggerInfo holds exported Swagger Info so clients can modify it
+var SwaggerInfo = swaggerInfo{ Schemes: []string{}}
+
+type s struct{}
+
+func (s *s) ReadDoc() string {
+	t, err := template.New("swagger_info").Funcs(template.FuncMap{
+		"marshal": func(v interface {}) string {
+			a, _ := json.Marshal(v)
+			return string(a)
+		},
+	}).Parse(doc)
+	if err != nil {
+		return doc
+	}
+
+	var tpl bytes.Buffer
+	if err := t.Execute(&tpl, SwaggerInfo); err != nil {
+		return doc
+	}
+
+	return tpl.String()
+}
+
+func init() {
+	swag.Register(swag.Name, &s{})
+}

pkg/core/blog/docs/swagger.json

@@ -0,0 +1,13 @@
+{
+    "swagger": "2.0",
+    "info": {
+        "description": "This is a sample server celler server.",
+        "title": "APP Demo API",
+        "contact": {},
+        "license": {},
+        "version": "1.0"
+    },
+    "host": "{{.Host}}",
+    "basePath": "/api",
+    "paths": {}
+}

pkg/core/blog/docs/swagger.yaml

@@ -0,0 +1,10 @@
+basePath: /api
+host: '{{.Host}}'
+info:
+  contact: {}
+  description: This is a sample server celler server.
+  license: {}
+  title: APP Demo API
+  version: "1.0"
+paths: {}
+swagger: "2.0"

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

pkg/core/blog/page/page.go

@@ -0,0 +1,293 @@
+// Package page provides ...
+package page
+
+import (
+	"bytes"
+	"fmt"
+	htemplate "html/template"
+	"io/ioutil"
+	"net/http"
+	"path/filepath"
+	"strconv"
+	"strings"
+	"text/template"
+	"time"
+
+	"github.com/eiblog/eiblog/setting"
+	"github.com/eiblog/eiblog/v2/pkg/cache"
+	"github.com/eiblog/eiblog/v2/pkg/config"
+	"github.com/eiblog/eiblog/v2/pkg/internal"
+	"github.com/eiblog/eiblog/v2/tools"
+	"github.com/eiblog/utils/tmpl"
+
+	"github.com/gin-gonic/gin"
+	"github.com/sirupsen/logrus"
+)
+
+// htmlTmpl html template cache
+var htmlTmpl *template.Template
+
+func init() {
+	htmlTmpl = template.New("eiblog").Funcs(tmpl.TplFuncMap)
+	root := filepath.Join(config.WorkDir, "website")
+	files := tools.ReadDirFiles(root, func(name string) bool {
+		if name == ".DS_Store" {
+			return true
+		}
+		return false
+	})
+	_, err := htmlTmpl.ParseFiles(files...)
+	if err != nil {
+		panic(err)
+	}
+}
+
+// RegisterRoutes register routes
+func RegisterRoutes(e *gin.Engine) {
+	e.NoRoute(handleNotFound)
+
+	e.GET("/", handleHomePage)
+	e.GET("/post/:slug", handleArticlePage)
+	e.GET("/series.html", handleSeriesPage)
+	e.GET("/archives.html", handleArchivePage)
+	e.GET("/search.html", handleSearchPage)
+	e.GET("/disqus/form/post-:slug", handleDisqusPage)
+	e.GET("/beacon.html", handleBeaconPage)
+}
+
+// baseParams 基础参数
+func baseParams(c *gin.Context) gin.H {
+	version := 0
+
+	cookie, err := c.Request.Cookie("v")
+	if err != nil || cookie.Value !=
+		fmt.Sprint(config.Conf.BlogApp.StaticVersion) {
+		version = config.Conf.BlogApp.StaticVersion
+	}
+	return gin.H{
+		"BlogName": cache.Ei.Blogger.BlogName,
+		"SubTitle": cache.Ei.Blogger.SubTitle,
+		"BTitle":   cache.Ei.Blogger.BTitle,
+		"BeiAn":    cache.Ei.Blogger.BeiAn,
+		"Domain":   config.Conf.BlogApp.Host,
+		"CopyYear": time.Now().Year(),
+		"Twitter":  config.Conf.BlogApp.Twitter,
+		"Qiniu":    config.Conf.BlogApp.Qiniu,
+		"Disqus":   config.Conf.BlogApp.Disqus,
+		"Version":  version,
+	}
+}
+
+// handleNotFound not found page
+func handleNotFound(c *gin.Context) {
+	params := baseParams(c)
+	params["title"] = "Not Found"
+	params["Description"] = "404 Not Found"
+	params["Path"] = ""
+	c.Status(http.StatusNotFound)
+	renderHTMLHomeLayout(c, "notfound", params)
+}
+
+// handleHomePage 首页
+func handleHomePage(c *gin.Context) {
+	params := baseParams(c)
+	params["title"] = cache.Ei.Blogger.BTitle + " | " + cache.Ei.Blogger.SubTitle
+	params["Description"] = "博客首页，" + cache.Ei.Blogger.SubTitle
+	params["Path"] = c.Request.URL.Path
+	params["CurrentPage"] = "blog-home"
+	pn, err := strconv.Atoi(c.Query("pn"))
+	if err != nil || pn < 1 {
+		pn = 1
+	}
+	params["Prev"], params["Next"], params["List"] = cache.Ei.PageArticles(pn,
+		config.Conf.BlogApp.General.PageNum)
+
+	renderHTMLHomeLayout(c, "home", params)
+}
+
+// handleArticlePage 文章页
+func handleArticlePage(c *gin.Context) {
+	slug := c.Param("slug")
+	if !strings.HasSuffix(slug, ".html") || cache.Ei.ArticlesMap[slug[:len(slug)-5]] == nil {
+		handleNotFound(c)
+		return
+	}
+	article := cache.Ei.ArticlesMap[slug[:len(slug)-5]]
+	params := baseParams(c)
+	params["Title"] = article.Title + " | " + cache.Ei.Blogger.BTitle
+	params["Path"] = c.Request.URL.Path
+	params["CurrentPage"] = "post-" + article.Slug
+	params["Article"] = article
+
+	var name string
+	switch slug {
+	case "blogroll.html":
+		name = "blogroll"
+		params["Description"] = "友情连接，" + cache.Ei.Blogger.SubTitle
+	case "about.html":
+		name = "about"
+		params["Description"] = "关于作者，" + cache.Ei.Blogger.SubTitle
+	default:
+		params["Description"] = article.Desc + "，" + cache.Ei.Blogger.SubTitle
+		name = "article"
+		params["Copyright"] = cache.Ei.Blogger.Copyright
+		if !article.UpdateTime.IsZero() {
+			params["Days"] = int(time.Now().Sub(article.UpdateTime).Hours()) / 24
+		} else {
+			params["Days"] = int(time.Now().Sub(article.CreateTime).Hours()) / 24
+		}
+		if article.SerieID > 0 {
+			for _, series := range cache.Ei.Series {
+				if series.ID == article.SerieID {
+					params["Serie"] = series
+				}
+			}
+		}
+	}
+	renderHTMLHomeLayout(c, name, params)
+}
+
+// handleSeriesPage 专题页
+func handleSeriesPage(c *gin.Context) {
+	params := baseParams(c)
+	params["Title"] = "专题 | " + cache.Ei.Blogger.BTitle
+	params["Description"] = "专题列表，" + cache.Ei.Blogger.SubTitle
+	params["Path"] = c.Request.URL.Path
+	params["CurrentPage"] = "series"
+	params["Article"] = cache.Ei.PageSeries
+	renderHTMLHomeLayout(c, "series", params)
+}
+
+// handleArchivePage 归档页
+func handleArchivePage(c *gin.Context) {
+	params := baseParams(c)
+	params["Title"] = "归档 | " + cache.Ei.Blogger.BTitle
+	params["Description"] = "博客归档，" + cache.Ei.Blogger.SubTitle
+	params["Path"] = c.Request.URL.Path
+	params["CurrentPage"] = "archives"
+	params["Article"] = cache.Ei.PageArchives
+	renderHTMLHomeLayout(c, "archives", params)
+}
+
+// handleSearchPage 搜索页
+func handleSearchPage(c *gin.Context) {
+	params := baseParams(c)
+	params["Title"] = "站内搜索 | " + cache.Ei.Blogger.BTitle
+	params["Description"] = "站内搜索，" + cache.Ei.Blogger.SubTitle
+	params["Path"] = ""
+	params["CurrentPage"] = "search-post"
+
+	q := strings.TrimSpace(c.Query("q"))
+	if q != "" {
+		start, err := strconv.Atoi(c.Query("start"))
+		if start < 1 || err != nil {
+			start = 1
+		}
+		params["Word"] = q
+
+		vals := c.Request.URL.Query()
+		result, err := internal.ElasticSearch(q, config.Conf.BlogApp.General.PageNum, start-1)
+		if err != nil {
+			logrus.Error("HandleSearchPage.ElasticSearch: ", err)
+		} else {
+			result.Took /= 1000
+			for i, v := range result.Hits.Hits {
+				article := cache.Ei.ArticlesMap[v.Source.Slug]
+				if len(v.Highlight.Content) == 0 && article != nil {
+					result.Hits.Hits[i].Highlight.Content = []string{article.Excerpt}
+				}
+			}
+			params["SearchResult"] = result
+			if num := start - config.Conf.BlogApp.General.PageNum; num > 0 {
+				vals.Set("start", fmt.Sprint(num))
+				params["Prev"] = vals.Encode()
+			}
+			if num := start + config.Conf.BlogApp.General.PageNum; result.Hits.Total >= num {
+				vals.Set("start", fmt.Sprint(num))
+				params["Next"] = vals.Encode()
+			}
+		}
+	} else {
+		params["HotWords"] = config.Conf.BlogApp.HotWords
+	}
+	renderHTMLHomeLayout(c, "search", params)
+}
+
+// handleDisqusPage 评论页
+func handleDisqusPage(c *gin.Context) {
+	array := strings.Split(c.Param("slug"), "|")
+	if len(array) != 4 || array[1] == "" {
+		c.String(http.StatusOK, "出错啦。。。")
+		return
+	}
+	article := cache.Ei.ArticlesMap[array[0]]
+	params := gin.H{
+		"Titile": "发表评论 | " + config.Conf.BlogApp.Blogger.BTitle,
+		"ATitle": article.Title,
+		"Thread": array[1],
+		"Slug":   article.Slug,
+	}
+	err := htmlTmpl.ExecuteTemplate(c.Writer, "disqus.html", params)
+	if err != nil {
+		panic(err)
+	}
+	c.Header("Content-Type", "text/html; charset=utf-8")
+}
+
+// handleBeaconPage 服务端推送谷歌统计
+func handleBeaconPage(c *gin.Context) {
+	ua := c.Request.UserAgent()
+
+	vals := c.Request.URL.Query()
+	vals.Set("v", setting.Conf.Google.V)
+	vals.Set("tid", setting.Conf.Google.Tid)
+	vals.Set("t", setting.Conf.Google.T)
+	cookie, _ := c.Cookie("u")
+	vals.Set("cid", cookie)
+
+	vals.Set("dl", c.Request.Referer())
+	vals.Set("uip", c.ClientIP())
+	go func() {
+		req, err := http.NewRequest("POST", config.Conf.BlogApp.Google.URL,
+			strings.NewReader(vals.Encode()))
+		if err != nil {
+			logrus.Error("HandleBeaconPage.NewRequest: ", err)
+			return
+		}
+		req.Header.Set("User-Agent", ua)
+		req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
+		res, err := http.DefaultClient.Do(req)
+		if err != nil {
+			logrus.Error("HandleBeaconPage.Do: ", err)
+			return
+		}
+		defer res.Body.Close()
+		data, err := ioutil.ReadAll(res.Body)
+		if err != nil {
+			logrus.Error("HandleBeaconPage.ReadAll: ", err)
+			return
+		}
+		if res.StatusCode/100 != 2 {
+			logrus.Error(string(data))
+		}
+	}()
+	c.Status(http.StatusNoContent)
+}
+
+// renderHTMLHomeLayout homelayout html
+func renderHTMLHomeLayout(c *gin.Context, name string, data gin.H) {
+	buf := bytes.Buffer{}
+	err := htmlTmpl.ExecuteTemplate(&buf, name, data)
+	if err != nil {
+		panic(err)
+	}
+	data["LayoutContent"] = htemplate.HTML(buf.String())
+	err = htmlTmpl.ExecuteTemplate(c.Writer, "homelayout.html", data)
+	if err != nil {
+		panic(err)
+	}
+	if c.Writer.Status() == 0 {
+		c.Status(http.StatusOK)
+	}
+	c.Header("Content-Type", "text/html; charset=utf-8")
+}

pkg/core/blog/swag/swag.go

@@ -0,0 +1,15 @@
+// Package swag provides ...
+package swag
+
+import (
+	_ "github.com/eiblog/eiblog/v2/pkg/core/blog/docs" // docs
+
+	"github.com/gin-gonic/gin"
+	ginSwagger "github.com/swaggo/gin-swagger"
+	"github.com/swaggo/gin-swagger/swaggerFiles"
+)
+
+// RegisterRoutes register routes
+func RegisterRoutes(group gin.IRoutes) {
+	group.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))
+}