150 lines
3.9 KiB
Go
150 lines
3.9 KiB
Go
package routers
|
|
|
|
import (
|
|
"crypto/md5"
|
|
"encoding/json"
|
|
"fmt"
|
|
"image"
|
|
_ "image/gif"
|
|
_ "image/jpeg"
|
|
_ "image/png"
|
|
|
|
"io/ioutil"
|
|
"log"
|
|
"main/configs"
|
|
"main/models"
|
|
"main/utils"
|
|
"net/http"
|
|
"os"
|
|
|
|
"github.com/gorilla/mux"
|
|
)
|
|
|
|
func ImagesGet(w http.ResponseWriter, r *http.Request) {
|
|
var listview models.ListView
|
|
listview.Page = utils.ParamInt(r.URL.Query().Get("page"), 1)
|
|
listview.PageSize = utils.ParamInt(r.URL.Query().Get("pageSize"), 10)
|
|
|
|
var image_list []models.Image
|
|
db := configs.ORMDB()
|
|
db.Offset((listview.Page - 1) * listview.PageSize).Limit(listview.PageSize).Find(&image_list)
|
|
for _, image := range image_list {
|
|
listview.List = append(listview.List, image)
|
|
}
|
|
|
|
db.Model(&models.Image{}).Count(&listview.Total)
|
|
|
|
listview.Next = listview.Page*listview.PageSize < int(listview.Total)
|
|
listview.WriteJSON(w)
|
|
}
|
|
|
|
func ImagesPost(w http.ResponseWriter, r *http.Request) {
|
|
models.AccountRead(w, r, func(account *models.Account) {
|
|
// 接收上傳的圖片文件, 僅限一張
|
|
file, file_header, err := r.FormFile("file")
|
|
if err != nil {
|
|
log.Println(err)
|
|
return
|
|
}
|
|
defer file.Close()
|
|
|
|
// 圖片寬高
|
|
imgData, format, err := image.Decode(file)
|
|
if err != nil {
|
|
log.Println(err)
|
|
return
|
|
}
|
|
|
|
fmt.Println(format, imgData.Bounds().Dx(), imgData.Bounds().Dy())
|
|
|
|
// 將文件指針移回開頭
|
|
if _, err := file.Seek(0, 0); err != nil {
|
|
log.Println(err)
|
|
return
|
|
}
|
|
|
|
// 读取文件内容
|
|
content, err := ioutil.ReadAll(file)
|
|
if err != nil {
|
|
log.Println(err)
|
|
return
|
|
}
|
|
|
|
// 整理文件信息
|
|
var img models.Image
|
|
img.Name = file_header.Filename
|
|
img.Size = int(file_header.Size) // 數據大小
|
|
img.Hash = fmt.Sprintf("%x", md5.Sum(content)) // 计算哈希
|
|
img.Type = file_header.Header.Get("Content-Type") // 文件類型
|
|
img.Path = fmt.Sprintf("data/images/%s.%s", img.Hash, format) // 存儲路徑
|
|
img.Width = imgData.Bounds().Dx() // 圖片寬度
|
|
img.Height = imgData.Bounds().Dy() // 圖片高度
|
|
img.Format = format // 圖片格式
|
|
img.UserID = account.ID // 用戶ID
|
|
|
|
// 先檢查 data/images 目錄是否存在
|
|
if _, err := ioutil.ReadDir("data/images"); err != nil {
|
|
if err := os.Mkdir("data/images", 0777); err != nil {
|
|
log.Println(err)
|
|
return
|
|
}
|
|
}
|
|
|
|
// 將文件存儲到本地 data/images 目錄下
|
|
if err := ioutil.WriteFile(img.Path, content, 0666); err != nil {
|
|
log.Println(err)
|
|
return
|
|
}
|
|
|
|
// 存儲圖片信息到數據庫
|
|
if err := configs.ORMDB().Create(&img).Error; err != nil {
|
|
log.Println(err)
|
|
return
|
|
}
|
|
|
|
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
|
w.Write(utils.ToJSON(img))
|
|
})
|
|
}
|
|
|
|
func ImagesItemGet(w http.ResponseWriter, r *http.Request) {
|
|
image := models.Image{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
|
|
if err := configs.ORMDB().First(&image).Error; err != nil {
|
|
log.Println(err)
|
|
return
|
|
}
|
|
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
|
w.Write(utils.ToJSON(image))
|
|
}
|
|
|
|
func ImagesItemPatch(w http.ResponseWriter, r *http.Request) {
|
|
image := models.Image{}
|
|
body, err := ioutil.ReadAll(r.Body)
|
|
if err != nil {
|
|
log.Println(err)
|
|
return
|
|
}
|
|
defer r.Body.Close()
|
|
if err = json.Unmarshal(body, &image); err != nil {
|
|
log.Println(err)
|
|
return
|
|
}
|
|
image.ID = utils.ParamInt(mux.Vars(r)["id"], 0)
|
|
if err := configs.ORMDB().Model(&image).Updates(image).Error; err != nil {
|
|
log.Println(err)
|
|
return
|
|
}
|
|
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
|
w.Write(utils.ToJSON(image))
|
|
}
|
|
|
|
func ImagesItemDelete(w http.ResponseWriter, r *http.Request) {
|
|
image := models.Image{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
|
|
if err := configs.ORMDB().Delete(&image).Error; err != nil {
|
|
log.Println(err)
|
|
return
|
|
}
|
|
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
|
w.Write(utils.ToJSON(image))
|
|
}
|