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) {
	// 接收上傳的圖片文件, 僅限一張
	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

	// 先檢查 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))
}