Web api

web/controllers/base.go

@@ -19,8 +19,11 @@ func (s *BaseController) Prepare() {
 	controllerName, actionName := s.GetControllerAndAction()
 	s.controllerName = strings.ToLower(controllerName[0 : len(controllerName)-10])
 	s.actionName = strings.ToLower(actionName)
-	if s.GetSession("auth") != true {
-		s.Redirect("/login/index", 302)
+	arr := strings.Split(s.Ctx.Request.RemoteAddr, ":")
+	if len(arr) > 0 && arr[0] != beego.AppConfig.String("authip") {
+		if s.GetSession("auth") != true {
+			s.Redirect("/login/index", 302)
+		}
 	}
 }
 

web/controllers/client.go

@@ -9,7 +9,7 @@ type ClientController struct {
 	BaseController
 }
 
-func (s *ClientController) Client() {
+func (s *ClientController) List() {
 	if s.Ctx.Request.Method == "GET" {
 		s.Data["menu"] = "client"
 		s.SetInfo("客户端管理")
@@ -32,7 +32,7 @@ func (s *ClientController) Add() {
 			VerifyKey: utils.GetRandomString(16),
 			Id:        server.CsvDb.GetClientId(),
 			Status:    true,
-			Remark:    s.GetString("Remark"),
+			Remark:    s.GetString("remark"),
 			Cnf: &utils.Config{
 				U:        s.GetString("u"),
 				P:        s.GetString("p"),
@@ -55,12 +55,26 @@ func (s *ClientController) Add() {
 		s.AjaxOk("添加成功")
 	}
 }
+func (s *ClientController) GetClient() {
+	if s.Ctx.Request.Method == "POST" {
+		id := s.GetIntNoErr("id")
+		data := make(map[string]interface{})
+		if c, err := server.CsvDb.GetClient(id); err != nil {
+			data["code"] = 0
+		} else {
+			data["code"] = 1
+			data["data"] = c
+		}
+		s.Data["json"] = data
+		s.ServeJSON()
+	}
+}
 
 //修改客户端
 func (s *ClientController) Edit() {
+	id := s.GetIntNoErr("id")
 	if s.Ctx.Request.Method == "GET" {
 		s.Data["menu"] = "client"
-		id := s.GetIntNoErr("id")
 		if c, err := server.CsvDb.GetClient(id); err != nil {
 			s.error()
 		} else {
@@ -69,11 +83,10 @@ func (s *ClientController) Edit() {
 		s.SetInfo("修改")
 		s.display()
 	} else {
-		id := s.GetIntNoErr("Id")
 		if c, err := server.CsvDb.GetClient(id); err != nil {
 			s.error()
 		} else {
-			c.Remark = s.GetString("Remark")
+			c.Remark = s.GetString("remark")
 			c.Cnf.U = s.GetString("u")
 			c.Cnf.P = s.GetString("p")
 			c.Cnf.Compress = s.GetString("compress")

web/controllers/index.go

@@ -101,7 +101,18 @@ func (s *IndexController) Add() {
 		}
 	}
 }
-
+func (s *IndexController) GetOneTunnel() {
+	id := s.GetIntNoErr("id")
+	data := make(map[string]interface{})
+	if t, err := server.CsvDb.GetTask(id); err != nil {
+		data["code"] = 0
+	} else {
+		data["code"] = 1
+		data["data"] = t
+	}
+	s.Data["json"] = data
+	s.ServeJSON()
+}
 func (s *IndexController) Edit() {
 	id := s.GetIntNoErr("id")
 	if s.Ctx.Request.Method == "GET" {
@@ -175,6 +186,20 @@ func (s *IndexController) HostList() {
 	}
 }
 
+func (s *IndexController) GetHost() {
+	if s.Ctx.Request.Method == "POST" {
+		data := make(map[string]interface{})
+		if h, err := server.GetInfoByHost(s.GetString("host")); err != nil {
+			data["code"] = 0
+		} else {
+			data["data"] = h
+			data["code"] = 1
+		}
+		s.Data["json"] = data
+		s.ServeJSON()
+	}
+}
+
 func (s *IndexController) DelHost() {
 	host := s.GetString("host")
 	if err := server.CsvDb.DelHost(host); err != nil {