版本重构，加web管理方式

controllers/base.go

@@ -0,0 +1,117 @@
+package controllers
+
+import (
+	"github.com/astaxie/beego"
+	"github.com/cnlh/easyProxy/lib"
+	"strconv"
+	"strings"
+)
+
+type BaseController struct {
+	beego.Controller
+	controllerName string
+	actionName     string
+}
+
+//初始化参数
+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)
+	}
+}
+
+//加载模板
+func (s *BaseController) display(tpl ...string) {
+	var tplname string
+	if len(tpl) > 0 {
+		tplname = strings.Join([]string{tpl[0], "html"}, ".")
+	} else {
+		tplname = s.controllerName + "/" + s.actionName + ".html"
+	}
+	s.Data["menu"] = s.actionName
+	ip := s.Ctx.Request.Host
+	s.Data["ip"] = lib.Gethostbyname(ip[0:strings.LastIndex(ip, ":")])
+	s.Data["p"] = *lib.TcpPort
+	s.Data["proxyPort"] = beego.AppConfig.String("hostPort")
+	s.Layout = "public/layout.html"
+	s.TplName = tplname
+}
+
+//错误
+func (s *BaseController) error() {
+	s.Layout = "public/layout.html"
+	s.TplName = "public/error.html"
+}
+
+//去掉没有err返回值的int
+func (s *BaseController) GetIntNoErr(key string, def ...int) int {
+	strv := s.Ctx.Input.Query(key)
+	if len(strv) == 0 && len(def) > 0 {
+		return def[0]
+	}
+	val, _ := strconv.Atoi(strv)
+	return val
+}
+
+//获取去掉错误的bool值
+func (s *BaseController) GetBoolNoErr(key string, def ...bool) bool {
+	strv := s.Ctx.Input.Query(key)
+	if len(strv) == 0 && len(def) > 0 {
+		return def[0]
+	}
+	val, _ := strconv.ParseBool(strv)
+	return val
+}
+
+//ajax正确返回
+func (s *BaseController) AjaxOk(str string) {
+	s.Data["json"] = ajax(str, 1)
+	s.ServeJSON()
+	s.StopRun()
+}
+
+//ajax错误返回
+func (s *BaseController) AjaxErr(str string) {
+	s.Data["json"] = ajax(str, 0)
+	s.ServeJSON()
+	s.StopRun()
+}
+
+//组装ajax
+func ajax(str string, status int) (map[string]interface{}) {
+	json := make(map[string]interface{})
+	json["status"] = status
+	json["msg"] = str
+	return json
+}
+
+//ajax table返回
+func (s *BaseController) AjaxTable(list interface{}, cnt int, recordsTotal int) {
+	json := make(map[string]interface{})
+	json["data"] = list
+	json["draw"] = s.GetIntNoErr("draw")
+	json["err"] = ""
+	json["recordsTotal"] = recordsTotal
+	json["recordsFiltered"] = cnt
+	s.Data["json"] = json
+	s.ServeJSON()
+	s.StopRun()
+}
+
+//ajax table参数
+func (s *BaseController) GetAjaxParams() (start, limit int) {
+	s.Ctx.Input.Bind(&start, "start")
+	s.Ctx.Input.Bind(&limit, "length")
+	return
+}
+
+func (s *BaseController) SetInfo(name string) {
+	s.Data["name"] = name
+}
+
+func (s *BaseController) SetType(name string) {
+	s.Data["type"] = name
+}

controllers/index.go

@@ -0,0 +1,165 @@
+package controllers
+
+import (
+	"github.com/cnlh/easyProxy/lib"
+)
+
+type IndexController struct {
+	BaseController
+}
+
+func (s *IndexController) Index() {
+	s.SetInfo("使用说明")
+	s.display("index/index")
+}
+
+func (s *IndexController) Tcp() {
+	s.SetInfo("tcp隧道管理")
+	s.SetType("tunnelServer")
+	s.display("index/list")
+}
+
+func (s *IndexController) Udp() {
+	s.SetInfo("udp隧道管理")
+	s.SetType("udpServer")
+	s.display("index/list")
+}
+
+func (s *IndexController) Socks5() {
+	s.SetInfo("socks5管理")
+	s.SetType("sock5Server")
+	s.display("index/list")
+}
+
+func (s *IndexController) Http() {
+	s.SetInfo("http代理管理")
+	s.SetType("httpProxyServer")
+	s.display("index/list")
+}
+
+func (s *IndexController) Host() {
+	s.SetInfo("host模式管理")
+	s.SetType("hostServer")
+	s.display("index/list")
+}
+
+func (s *IndexController) GetTaskList() {
+	start, length := s.GetAjaxParams()
+	taskType := s.GetString("type")
+	list, cnt := lib.CsvDb.GetTaskList(start, length, taskType)
+	s.AjaxTable(list, cnt, cnt)
+}
+
+func (s *IndexController) Add() {
+	if s.Ctx.Request.Method == "GET" {
+		s.Data["type"] = s.GetString("type")
+		s.SetInfo("新增")
+		s.display()
+	} else {
+		t := &lib.TaskList{
+			TcpPort:   s.GetIntNoErr("port"),
+			Mode:      s.GetString("type"),
+			Target:    s.GetString("target"),
+			VerifyKey: lib.GetRandomString(16),
+			U:         s.GetString("u"),
+			P:         s.GetString("p"),
+			Compress:  s.GetString("compress"),
+			IsRun:     0,
+		}
+		lib.CsvDb.NewTask(t)
+		if err := lib.AddTask(t); err != nil {
+			s.AjaxErr(err.Error())
+		} else {
+			s.AjaxOk("添加成功")
+		}
+	}
+}
+
+func (s *IndexController) Edit() {
+	if s.Ctx.Request.Method == "GET" {
+		vKey := s.GetString("vKey")
+		if t, err := lib.CsvDb.GetTask(vKey); err != nil {
+			s.error()
+		} else {
+			s.Data["t"] = t
+		}
+		s.SetInfo("修改")
+		s.display()
+	} else {
+		vKey := s.GetString("vKey")
+		if t, err := lib.CsvDb.GetTask(vKey); err != nil {
+			s.error()
+		} else {
+			t.TcpPort = s.GetIntNoErr("port")
+			t.Mode = s.GetString("type")
+			t.Target = s.GetString("target")
+			t.U = s.GetString("u")
+			t.P = s.GetString("p")
+			t.Compress = s.GetString("compress")
+			lib.CsvDb.UpdateTask(t)
+			lib.StopServer(t.VerifyKey)
+			lib.StartTask(t.VerifyKey)
+		}
+		s.AjaxOk("修改成功")
+	}
+}
+
+func (s *IndexController) Stop() {
+	vKey := s.GetString("vKey")
+	if err := lib.StopServer(vKey); err != nil {
+		s.AjaxErr("停止失败")
+	}
+	s.AjaxOk("停止成功")
+}
+func (s *IndexController) Del() {
+	vKey := s.GetString("vKey")
+	if err := lib.DelTask(vKey); err != nil {
+		s.AjaxErr("删除失败")
+	}
+	s.AjaxOk("删除成功")
+}
+
+func (s *IndexController) Start() {
+	vKey := s.GetString("vKey")
+	if err := lib.StartTask(vKey); err != nil {
+		s.AjaxErr("开启失败")
+	}
+	s.AjaxOk("开启成功")
+}
+
+func (s *IndexController) HostList() {
+	if s.Ctx.Request.Method == "GET" {
+		s.Data["vkey"] = s.GetString("vkey")
+		s.SetInfo("域名列表")
+		s.display("index/hlist")
+	} else {
+		start, length := s.GetAjaxParams()
+		vkey := s.GetString("vkey")
+		list, cnt := lib.CsvDb.GetHostList(start, length, vkey)
+		s.AjaxTable(list, cnt, cnt)
+	}
+}
+
+func (s *IndexController) DelHost() {
+	host := s.GetString("host")
+	if err := lib.CsvDb.DelHost(host); err != nil {
+		s.AjaxErr("删除失败")
+	}
+	s.AjaxOk("删除成功")
+}
+
+func (s *IndexController) AddHost() {
+	if s.Ctx.Request.Method == "GET" {
+		s.Data["vkey"] = s.GetString("vkey")
+		s.SetInfo("新增")
+		s.display("index/hadd")
+	} else {
+		h := &lib.HostList{
+			Vkey:   s.GetString("vkey"),
+			Host:   s.GetString("host"),
+			Target: s.GetString("target"),
+		}
+		lib.CsvDb.NewHost(h)
+		s.AjaxOk("添加成功")
+	}
+}

controllers/login.go

@@ -0,0 +1,27 @@
+package controllers
+
+import (
+	"github.com/astaxie/beego"
+)
+
+type LoginController struct {
+	beego.Controller
+}
+
+func (self *LoginController) Index() {
+	self.TplName = "login/index.html"
+}
+func (self *LoginController) Verify() {
+	if self.GetString("psd") == beego.AppConfig.String("password") {
+		self.SetSession("auth", true)
+		self.Data["json"] = map[string]interface{}{"status": 1, "msg": "验证成功"}
+		self.ServeJSON()
+	} else {
+		self.Data["json"] = map[string]interface{}{"status": 0, "msg": "验证失败"}
+		self.ServeJSON()
+	}
+}
+func (self *LoginController) Out() {
+	self.SetSession("auth", false)
+	self.Redirect("/login/index", 302)
+}