Files
2019-08-08 17:13:34 +08:00

138 lines
3.6 KiB
Go

package controllers
import (
"anytunnel/at-admin/app/modules/web/models"
"anytunnel/at-admin/app/utils"
"fmt"
"strings"
"time"
validation "github.com/go-ozzo/ozzo-validation"
"github.com/go-ozzo/ozzo-validation/is"
)
type IpListController struct {
BaseController
}
func (this *IpListController) Delete() {
ip_listModel := models.IpList{}
ip_listId := this.GetString("ip_list_id")
err := ip_listModel.Delete(ip_listId)
if err != nil {
this.JsonError(err)
}
this.JsonSuccess("")
}
func (this *IpListController) Add() {
ip_listModel := models.IpList{}
if this.Ctx.Input.IsPost() {
_, data := this.getIpListFromPost(false)
_, err := ip_listModel.Insert(data)
if err != nil {
this.JsonError(err)
}
this.JsonSuccess("")
} else {
regionModel := models.Region{}
regions, err := regionModel.GetSubRegions()
if err != nil {
this.ViewError(err.Error())
}
this.Data["regions"] = regions
this.Data["action"] = "add"
this.viewLayout("ip_list/form", "form")
}
}
func (this *IpListController) Edit() {
ip_listModel := models.IpList{}
ip_listId := this.GetString("ip_list_id")
if this.Ctx.Input.IsPost() {
_, data := this.getIpListFromPost(true)
_, err := ip_listModel.Update(ip_listId, data)
if err != nil {
this.JsonError(err)
}
this.JsonSuccess("")
} else {
ip_list, err := ip_listModel.GetIpListByIpListId(ip_listId)
if err != nil {
this.JsonError(err, "")
}
if len(ip_list) == 0 {
this.ViewError("IpList不存在")
}
regionModel := models.Region{}
regions, err := regionModel.GetSubRegions()
if err != nil {
this.ViewError(err.Error())
}
this.Data["regions"] = regions
this.Data["ip_list"] = ip_list
this.Data["action"] = "edit"
this.viewLayout("ip_list/form", "form")
}
}
func (this *IpListController) List() {
column := strings.Trim(this.GetString("type", ""), " ")
keyword := strings.Trim(this.GetString("id", ""), " ")
page, err := this.GetInt("page", 1)
if err != nil {
page = 1
}
//每页的条数
pageSize := 10
limit := (page - 1) * pageSize
ip_listModel := models.IpList{}
var ip_lists = []map[string]string{}
var ip_listCount = 0
if keyword == "" {
ip_listCount, err = ip_listModel.CountIpLists()
ip_lists, err = ip_listModel.GetIpListsByLimit(limit, pageSize)
} else {
ip_listCount, err = ip_listModel.CountIpListsByID(column, keyword)
ip_lists, err = ip_listModel.GetIpListsByIDAndLimit(column, keyword, limit, pageSize)
}
if err != nil {
this.ViewError(err.Error())
}
this.Data["ip_lists"] = ip_lists
this.Data["page"] = utils.NewMisc().Page(ip_listCount, page, pageSize, fmt.Sprintf("/web/ip_list/list?page={page}&type=%s&id=%s", column, keyword))
this.viewLayoutTitle("用户列表", "ip_list/list", "form")
}
func (this *IpListController) getIpListFromPost(isUpdate bool) (ip_listId string, ip_list map[string]interface{}) {
ip_list = map[string]interface{}{
"ip": this.GetString("ip"),
"cs_type": this.GetString("cs_type"),
"is_forbidden": this.GetString("is_forbidden"),
}
errs := validation.Errors{
"IP": validation.Validate(ip_list["ip"],
validation.Required.Error("不能为空"),
is.IPv4.Error("格式错误")),
"类型": validation.Validate(ip_list["cs_type"],
validation.Required.Error("不能为空"),
validation.In("server", "client").Error("错误"),
),
"访问控制": validation.Validate(ip_list["is_forbidden"],
validation.Required.Error("不能为空"),
validation.In("0", "1").Error("错误"),
),
}
err := errs.Filter()
if err != nil {
this.JsonError(err)
}
if isUpdate {
ip_list["update_time"] = time.Now().Unix()
} else {
ip_list["create_time"] = time.Now().Unix()
}
return
}