add article poster text feature

mooncreeks · Jul 5, 2018 · ee7073d · ee7073d
1 parent 45174b7
commit ee7073d
Show file tree

Hide file tree

Showing 6 changed files with 81 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -2,4 +2,5 @@
 .DS_Store
 
 runtime/*
-!runtime/qrcode/bg.jpg
+!runtime/qrcode/bg.jpg
+!runtime/fonts
diff --git a/conf/app.ini b/conf/app.ini
@@ -12,6 +12,7 @@ ImageAllowExts = .jpg,.jpeg,.png
 
 ExportSavePath = export/
 QrCodeSavePath = qrcode/
+FontSavePath = fonts/
 
 LogSavePath = logs/
 LogSaveName = log

diff --git a/pkg/setting/setting.go b/pkg/setting/setting.go
@@ -20,6 +20,7 @@ type App struct {
 
 	ExportSavePath string
 	QrCodeSavePath string
+	FontSavePath string
 
 	LogSavePath string
 	LogSaveName string

diff --git a/routers/api/v1/article.go b/routers/api/v1/article.go
@@ -331,4 +331,4 @@ func GenerateArticlePoster(c *gin.Context) {
 		"poster_url":      qrcode.GetQrCodeFullUrl(posterName),
 		"poster_save_url": filePath + posterName,
 	})
-}
+}
diff --git a/runtime/fonts/msyhbd.ttc b/runtime/fonts/msyhbd.ttc
diff --git a/service/article_service/article_poster.go b/service/article_service/article_poster.go
@@ -8,6 +8,9 @@ import (
 
 	"github.com/EDDYCJY/go-gin-example/pkg/file"
 	"github.com/EDDYCJY/go-gin-example/pkg/qrcode"
+	"github.com/EDDYCJY/go-gin-example/pkg/setting"
+	"github.com/golang/freetype"
+	"io/ioutil"
 )
 
 type ArticlePoster struct {
@@ -74,6 +77,61 @@ func NewArticlePosterBg(name string, ap *ArticlePoster, rect *Rect, pt *Pt) *Art
 	}
 }
 
+type DrawText struct {
+	JPG    draw.Image
+	Merged *os.File
+
+	Title string
+	X0    int
+	Y0    int
+	Size0 float64
+
+	SubTitle string
+	X1       int
+	Y1       int
+	Size1    float64
+}
+
+func (a *ArticlePosterBg) DrawPoster(d *DrawText, fontName string) error {
+	fontSource := setting.AppSetting.RuntimeRootPath + setting.AppSetting.FontSavePath + fontName
+	fontSourceBytes, err := ioutil.ReadFile(fontSource)
+	if err != nil {
+		return err
+	}
+
+	trueTypeFont, err := freetype.ParseFont(fontSourceBytes)
+	if err != nil {
+		return err
+	}
+
+	fc := freetype.NewContext()
+	fc.SetDPI(72)
+	fc.SetFont(trueTypeFont)
+	fc.SetFontSize(d.Size0)
+	fc.SetClip(d.JPG.Bounds())
+	fc.SetDst(d.JPG)
+	fc.SetSrc(image.Black)
+
+	pt := freetype.Pt(d.X0, d.Y0)
+	_, err = fc.DrawString(d.Title, pt)
+	if err != nil {
+		return err
+	}
+
+	fc.SetFontSize(d.Size1)
+	_, err = fc.DrawString(d.SubTitle, freetype.Pt(d.X1, d.Y1))
+	if err != nil {
+		return err
+	}
+
+	err = jpeg.Encode(d.Merged, d.JPG, nil)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
 func (a *ArticlePosterBg) Generate() (string, string, error) {
 	fullPath := qrcode.GetQrCodeFullPath()
 	fileName, path, err := a.Qr.Encode(fullPath)
@@ -114,7 +172,24 @@ func (a *ArticlePosterBg) Generate() (string, string, error) {
 		draw.Draw(jpg, jpg.Bounds(), bgImage, bgImage.Bounds().Min, draw.Over)
 		draw.Draw(jpg, jpg.Bounds(), qrImage, qrImage.Bounds().Min.Sub(image.Pt(a.Pt.X, a.Pt.Y)), draw.Over)
 
-		jpeg.Encode(mergedF, jpg, nil)
+		err = a.DrawPoster(&DrawText{
+			JPG:    jpg,
+			Merged: mergedF,
+
+			Title: "Golang Gin 系列文章",
+			X0:    80,
+			Y0:    160,
+			Size0: 42,
+
+			SubTitle: "---煎鱼",
+			X1:       320,
+			Y1:       220,
+			Size1:    36,
+		}, "msyhbd.ttc")
+
+		if err != nil {
+			return "", "", err
+		}
 	}
 
 	return fileName, path, nil
-Original file line number
+Diff line change
@@ Expand Up / @@ -331,4 +331,4 @@ func GenerateArticlePoster(c *gin.Context) { @@
     		"poster_url":      qrcode.GetQrCodeFullUrl(posterName),
     		"poster_save_url": filePath + posterName,
     	})
-    }
+    }