From 0181ef270d91d5d1787c6a0f0258bf35fbd16487 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=A1=9C=E8=8F=AF?= <huan0016@gmail.com>
Date: Wed, 7 Jun 2023 15:30:33 +0800
Subject: [PATCH] img

---
 README.md      | 12 +++++++++
 routers/img.go | 66 ++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 78 insertions(+)
 create mode 100644 routers/img.go

diff --git a/README.md b/README.md
index e67e394..d9c3453 100644
--- a/README.md
+++ b/README.md
@@ -158,6 +158,18 @@ response=$(curl -X POST -H "Content-Type: multipart/form-data" -F "file=@./data/
 message "$response" "上傳圖片" true
 ```
 
+
+提供webp生成服务
+
+```javascript
+// GET /img/{id}-{version}-{width}-{height}-{fit}.{format}
+// @id: 取图片ID
+// @version: 取 hash 前 6 位
+// @width: 所需宽度 1x 2x 3x 倍图直接输入尺寸
+// @height: 所需高度 1x 2x 3x 倍图直接输入尺寸
+// @fit: 裁切方式 cover contain fill auto
+```
+
 任務:
 
 ```go
diff --git a/routers/img.go b/routers/img.go
new file mode 100644
index 0000000..ab8930a
--- /dev/null
+++ b/routers/img.go
@@ -0,0 +1,66 @@
+package routers
+
+import (
+	"log"
+	"main/configs"
+	"main/models"
+	"net/http"
+	"regexp"
+	"strconv"
+)
+
+func WebpGet(w http.ResponseWriter, r *http.Request) {
+	// 獲取請求參數(id, version, width, height, fit, format)
+	reg := regexp.MustCompile(`^/img/([0-9a-zA-Z]+)-([0-9a-zA-Z]+)-([0-9]+)-([0-9]+)-([a-zA-Z]+).(jpg|jpeg|png|webp)$`)
+	matches := reg.FindStringSubmatch(r.URL.Path)
+	if len(matches) != 7 {
+		log.Println("URL 格式错误", matches)
+		w.WriteHeader(http.StatusNotFound)
+		return
+	}
+
+	version, fit, format := matches[2], matches[5], matches[6]
+
+	// id 轉換爲數字
+	id, err := strconv.ParseInt(matches[1], 10, 64)
+	if err != nil {
+		log.Println("ID 格式错误", matches[1])
+		w.WriteHeader(http.StatusNotFound)
+		return
+	}
+
+	// width 轉換爲數字
+	width, err := strconv.Atoi(matches[3])
+	if err != nil {
+		log.Println("width 格式错误", matches[3])
+		w.WriteHeader(http.StatusNotFound)
+		return
+	}
+
+	// height 轉換爲數字
+	height, err := strconv.Atoi(matches[4])
+	if err != nil {
+		log.Println("height 格式错误", matches[4])
+		w.WriteHeader(http.StatusNotFound)
+		return
+	}
+
+	// 從數據庫中獲取圖片信息
+	var image models.Image = models.Image{ID: int(id)}
+	if err := configs.ORMDB().Where("id = ?", id).First(&image).Error; err != nil {
+		log.Println("获取图片失败", version, format, err)
+		w.WriteHeader(http.StatusNotFound)
+		return
+	}
+
+	data, err := image.ToWebP(width, height, fit)
+	if err != nil {
+		log.Println("转换图片失败", err)
+		w.WriteHeader(http.StatusBadRequest)
+		return
+	}
+
+	w.Header().Set("Content-Type", "image/webp")
+	w.Header().Set("Cache-Control", "max-age=31536000")
+	w.Write(data)
+}