package routers

import (
	"encoding/json"
	"io/ioutil"
	"log"
	"main/models"
	"main/utils"
	"net/http"
	"strconv"

	"github.com/gorilla/mux"
	"github.com/gorilla/websocket"
)

var manager = models.NewWebSocketManager()

func ModelsGet(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)
	listview.List = models.QueryModels(listview.Page, listview.PageSize)
	listview.Total = models.CountModels()
	listview.Next = listview.Page*listview.PageSize < listview.Total
	w.Header().Set("Content-Type", "application/json; charset=utf-8")
	w.Write(listview.ToJSON())
}

func ModelsPost(w http.ResponseWriter, r *http.Request) {
	var model models.Model
	body, err := ioutil.ReadAll(r.Body)
	if err != nil {
		log.Println(err)
		return
	}
	defer r.Body.Close()
	if err = json.Unmarshal(body, &model); err != nil {
		log.Println(err)
		return
	}
	model.Create()
	w.Header().Set("Content-Type", "application/json; charset=utf-8")
	w.Write(utils.ToJSON(model))
}

func ModelItemGet(w http.ResponseWriter, r *http.Request) {
	if r.Header.Get("Upgrade") == "websocket" {
		vars := mux.Vars(r)
		id, _ := strconv.Atoi(vars["id"])
		model := models.QueryModel(id)
		if model.ID == 0 {
			w.WriteHeader(http.StatusNotFound)
			return
		}
		upgrader := websocket.Upgrader{}
		conn, err := upgrader.Upgrade(w, r, nil)
		if err != nil {
			log.Println(err)
			return
		}
		defer conn.Close()
		wsid := manager.AddConnection(conn)
		defer manager.RemoveConnection(wsid)
		for {
			_, msg, err := conn.ReadMessage()
			if err != nil {
				log.Println(err)
				return
			}
			log.Println(string(msg))
			if string(msg) == "close" {
				break
			}
		}
		return
	}
	model := models.Model{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
	model.Get()
	w.Header().Set("Content-Type", "application/json; charset=utf-8")
	w.Write(utils.ToJSON(model))
}

func ModelItemPatch(w http.ResponseWriter, r *http.Request) {
	// 取出原始数据
	model := models.Model{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
	if err := model.Get(); err != nil {
		w.WriteHeader(http.StatusNotFound)
		return
	}

	// 取出更新数据
	var model_new models.Model
	body, err := ioutil.ReadAll(r.Body)
	if err != nil {
		log.Println(err)
		return
	}
	defer r.Body.Close()
	if err = json.Unmarshal(body, &model_new); err != nil {
		log.Println(err)
		return
	}

	// 字段不爲空且不等於原始數據時更新
	if model_new.Name != "" && model_new.Name != model.Name {
		model.Name = model_new.Name
	}
	if model_new.Type != "" && model_new.Type != model.Type {
		model.Type = model_new.Type
	}
	if model_new.Status != "" && model_new.Status != model.Status {
		model.Status = model_new.Status
		// 如果狀態被改變爲 ready, 將模型發送到訓練隊列
		if model.Status == "ready" {
			model.SendToTrain()
		}
	}
	if model_new.Image != "" && model_new.Image != model.Image {
		model.Image = model_new.Image
	}

	model.Update()
	w.Header().Set("Content-Type", "application/json; charset=utf-8")
	w.Write(utils.ToJSON(model))
}

func ModelItemDelete(w http.ResponseWriter, r *http.Request) {
	model := models.Model{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
	model.Delete()
	w.WriteHeader(http.StatusNoContent)
}