server.go

package controllers

import (
	"strings"
	"strconv"
	"bytes"

	"github.com/lifei6671/go-git-webhook/models"
	"github.com/lifei6671/go-git-webhook/modules/pager"

	"github.com/astaxie/beego"
	"github.com/astaxie/beego/logs"
)
// ServerController 服务器控制器
type ServerController struct {
	BaseController
}
// Index 服务器列表
func (c *ServerController) Index() {
	c.Prepare()

	c.Layout = ""
	c.TplName = "server/index.html"


	pageIndex, _ := c.GetInt("page", 1)

	var servers []models.Server

	pageOptions := pager.PageOptions{
		TableName:  models.NewServer().TableName(),
		EnableFirstLastLink : true,
		CurrentPage : pageIndex,
		PageSize : 15,
		ParamName : "page",
		Conditions : " AND create_at = " + strconv.Itoa(c.Member.MemberId) + " order by server_id desc",
	}



	//返回分页信息,
	//第一个:为返回的当前页面数据集合,ResultSet类型
	//第二个:生成的分页链接
	//第三个:返回总记录数
	//第四个:返回总页数
	totalItem, totalCount, rs, pageHtml := pager.GetPagerLinks(&pageOptions, c.Ctx)

	_,err := rs.QueryRows(&servers)      //把当前页面的数据序列化进一个切片内

	if err != nil {
		logs.Error("",err.Error())
	}

	c.Data["lists"] = servers
	c.Data["html"] = pageHtml
	c.Data["totalItem"] = totalItem
	c.Data["totalCount"] = totalCount
	c.Data["Server"] = true
}

// Edit 编辑
func (c *ServerController) Edit()  {
	c.Prepare()
	c.Layout = ""
	c.TplName = "server/edit.html"

	if c.Ctx.Input.IsPost() {
		id,_ := c.GetInt("id",0)
		account := c.GetString("account", "")
		serverName := c.GetString("name", "");
		ipAddress := c.GetString("ip", "")
		port, err := c.GetInt("port", 22)
		serverType := c.GetString("type", "ssh")
		status,_ := c.GetInt("status",0)

		if status !=0 && status != 1 {
			status = 0
		}

		if err != nil {
			c.JsonResult(500, "端口号错误");
		}
		tag := c.GetString("tag", "")

		key := c.GetString("key", "")

		if serverName == "" {
			c.JsonResult(500, "Server Name is require.")
		}
		if ipAddress == "" {
			c.JsonResult(500, "Server Ip is require.")
		}
		if port <= 0 {
			c.JsonResult(500, "Port is require.")
		}
		if tag != "" {

		}
		if key == "" {
			c.JsonResult(500, "SSH Private Key or Account Password is require.")
		}

		if !strings.EqualFold(serverType, "ssh") && !strings.EqualFold(serverType, "client") {
			c.JsonResult(500, "Server Type error.")
		}
		server := models.NewServer()


		if id > 0{
			server.ServerId = id
			if err := server.Find();err != nil {
				c.JsonResult(500,err.Error())
			}
			//如果不是本人创建则返回403
			if server.CreateAt != c.Member.MemberId {
				c.Abort("403")
			}
		}

		server.Account = account
		server.CreateAt = c.Member.MemberId
		server.IpAddress = ipAddress
		server.Name = serverName
		server.Port = port
		server.Tag = tag
		server.PrivateKey = key
		server.Type = serverType
		server.Status = status

		if err := server.Save(); err != nil {
			c.JsonResult(500, "Save failed:" + err.Error())
		} else {
			data := make(map[string]interface{},5)

			if id <= 0 {

				var buf bytes.Buffer

				viewPath := c.ViewPath

				if c.ViewPath == "" {
					viewPath = beego.BConfig.WebConfig.ViewsPath

				}

				beego.ExecuteViewPathTemplate(&buf, "server/index_list.html",viewPath,server)

				data["view"] = buf.String()
			}



			data["errcode"] = 0
			data["message"] = "ok"

			data["data"] = server

			c.Data["json"] = data
			c.ServeJSON(true)
			c.StopRun()

		}
	}

	id,err := strconv.Atoi(c.Ctx.Input.Param(":id"))
	if err != nil {
		c.Abort("404")
	}
	server := models.NewServer()
	server.ServerId = id

	if err := server.Find(); err != nil {
		c.Abort("404")
	}
	//如果不是本人创建则返回403
	if server.CreateAt != c.Member.MemberId {
		c.Abort("403")
	}
	if c.Ctx.Input.IsAjax() {

		c.JsonResult(0,"ok",*server)
	}
	c.Data["Model"] = server
	c.Data["Server"] = true


}

// Delete 删除一个Server
func (c *ServerController) Delete() {
	serverId,_ := c.GetInt("id",0)

	if serverId <= 0 {
		c.JsonResult(500,"Server ID is require.")
	}
	server := models.NewServer()

	server.ServerId = serverId

	if err := server.Find();err != nil {
		c.JsonResult(500,err.Error())
	}
	if server.CreateAt != c.Member.MemberId {
		c.JsonResult(403,"Permission denied")
	}
	if err := server.Delete();err != nil {
		c.JsonResult(500,err.Error())
	}

	models.NewRelation().DeleteByWhere(" AND server_id = ?",serverId)

	models.NewScheduler().DeleteByWhere(" AND server_id  = ?",serverId)

	c.JsonResult(0,"ok")
}