package webServer

import (
	"context"
	"io/ioutil"
	"net/http"

	"serfNode/config"
	"serfNode/ipc"
	"serfNode/serf"

	"github.com/gin-gonic/gin"
)

func Serve(port string) {

	r := gin.Default()
	r.GET("/serf/config", getSerfState)

	r.GET("/serf/state", getSerfState)
	r.GET("/serf/member", getSerfMembers)
	r.POST("/serf/join", joinCluster)
	r.GET("/serf/leave", leaveCluster)
	r.POST("/serf/plc", transData)

	r.Run(":" + port)
}

func resetConfig(c *gin.Context) {
	nodeName := config.SerfConf.NodeName

	err := ipc.GetConfigFromNodeRed()
	if err == nil {
		if nodeName != config.SerfConf.NodeName {
			ctx := context.Background()
			serf.InitAgent(ctx, *config.SerfConf)
		}
	}

}

func getSerfState(c *gin.Context) {
	c.JSON(http.StatusOK, gin.H{
		"success": true,
		"data":    serf.ClusterState(),
	})
}

func getSerfMembers(c *gin.Context) {
	c.JSON(http.StatusOK, gin.H{
		"success": true,
		"data":    serf.Members(),
	})
}

type joinParam struct {
	NodeAddr string `json:"nodeAddr" binding:"required"`
}

func joinCluster(c *gin.Context) {
	var query joinParam
	err := c.BindJSON(&query)
	if err != nil {
		c.JSON(http.StatusBadRequest, gin.H{
			"success": false,
			"message": "参数错误",
		})

		return
	}

	err = serf.JoinCluster(query.NodeAddr)
	if err != nil {
		c.JSON(http.StatusOK, gin.H{
			"success": false,
			"message": err.Error(),
		})

		return
	}

	c.JSON(http.StatusOK, gin.H{
		"success": true,
		"message": "加入成功",
	})
}

func leaveCluster(c *gin.Context) {
	err := serf.LeaveCluster()
	if err != nil {
		c.JSON(http.StatusOK, gin.H{
			"success": false,
			"message": err.Error(),
		})

		return
	}

	c.JSON(http.StatusOK, gin.H{
		"success": true,
		"message": "加入成功",
	})
}

func transData(c *gin.Context) {
	bodyBytes, _ := ioutil.ReadAll(c.Request.Body)

	err := serf.QueryMsg2Master(serf.PLCDATA, bodyBytes)
	if err != nil {
		c.JSON(http.StatusOK, gin.H{
			"success": false,
			"message": err.Error(),
		})

		return
	}

	c.JSON(http.StatusOK, gin.H{
		"success": true,
		"message": "发送成功",
	})
}