init

2019-08-08 17:13:34 +08:00
parent e8e5966a8c
commit f0bf2d5fec
2885 changed files with 1195993 additions and 12 deletions
@@ -0,0 +1,137 @@
+package controllers
+
+import (
+	"anytunnel/at-admin/app/modules/web/models"
+	"anytunnel/at-admin/app/utils"
+	"regexp"
+	"strings"
+	"time"
+
+	validation "github.com/go-ozzo/ozzo-validation"
+)
+
+type ServerController struct {
+	BaseController
+}
+
+func (this *ServerController) Reset() {
+	serverModel := models.Server{}
+	serverId := this.GetString("server_id")
+	err := serverModel.Offline(serverId)
+	if err != nil {
+		this.JsonError(err)
+	}
+	err = serverModel.Reset(serverId)
+	if err != nil {
+		this.JsonError(err)
+	}
+	this.JsonSuccess("")
+
+}
+func (this *ServerController) Delete() {
+	serverModel := models.Server{}
+	serverId := this.GetString("server_id")
+	hasTunnel, err := serverModel.HasTunnelRef(serverId)
+	if err != nil {
+		this.JsonError(err)
+	}
+	if hasTunnel {
+		this.JsonError("Tunnel引用,不能删除")
+	}
+	err = serverModel.Delete(serverId)
+	if err != nil {
+		this.JsonError(err)
+	}
+	this.JsonSuccess("")
+
+}
+func (this *ServerController) Add() {
+	serverModel := models.Server{}
+	if this.Ctx.Input.IsPost() {
+		_, data := this.getServerFromPost(false)
+		_, err := serverModel.Insert(data)
+		if err != nil {
+			this.JsonError(err)
+		}
+		this.JsonSuccess("")
+	} else {
+		this.Data["action"] = "add"
+		this.viewLayout("server/form", "form")
+	}
+}
+func (this *ServerController) Edit() {
+	serverModel := models.Server{}
+	serverId := this.GetString("server_id")
+	if this.Ctx.Input.IsPost() {
+		_, data := this.getServerFromPost(true)
+		_, err := serverModel.Update(serverId, data)
+		if err != nil {
+			this.JsonError(err)
+		}
+		this.JsonSuccess("")
+	} else {
+		server, err := serverModel.GetServerByServerId(serverId)
+		if err != nil {
+			this.JsonError(err, "")
+		}
+		if len(server) == 0 {
+			this.ViewError("Server不存在")
+		}
+		this.Data["server"] = server
+		this.Data["action"] = "edit"
+		this.viewLayout("server/form", "form")
+	}
+
+}
+func (this *ServerController) List() {
+
+	userID := strings.Trim(this.GetString("user_id"), " ")
+	page, err := this.GetInt("page", 1)
+	if err != nil {
+		page = 1
+	}
+	//每页的条数
+	pageSize := 10
+	limit := (page - 1) * pageSize
+
+	serverModel := models.Server{}
+	var servers = []map[string]string{}
+	var serverCount = 0
+	if userID == "" {
+		serverCount, err = serverModel.CountServers()
+		servers, err = serverModel.GetServersByLimit(limit, pageSize)
+	} else {
+		serverCount, err = serverModel.CountServersByUserID(userID)
+		servers, err = serverModel.GetServersByUserIDAndLimit(userID, limit, pageSize)
+	}
+	if err != nil {
+		this.ViewError(err.Error())
+	}
+
+	this.Data["userID"] = userID
+	this.Data["servers"] = servers
+	this.Data["page"] = utils.NewMisc().Page(serverCount, page, pageSize, "/web/server/list?page={page}&user_id="+userID)
+	this.viewLayoutTitle("Server列表", "server/list", "form")
+}
+
+func (this *ServerController) getServerFromPost(isUpdate bool) (serverId string, server map[string]interface{}) {
+	server = map[string]interface{}{
+		"name": this.GetString("name"),
+		// "user_id":   this.GetString("user_id"),
+		"user_id":   0,
+		"is_delete": 0,
+	}
+	err := validation.Validate(server["name"],
+		validation.Required.Error("名称不能为空"),
+		validation.Match(regexp.MustCompile("^.{1,15}$")).Error("名称长度必须是1-15字符"))
+	if err != nil {
+		this.JsonError(err.Error())
+	}
+	if isUpdate {
+		server["update_time"] = time.Now().Unix()
+	} else {
+		server["token"] = utils.NewMisc().RandString(32)
+		server["create_time"] = time.Now().Unix()
+	}
+	return
+}