mirror of
https://github.com/ehang-io/nps.git
synced 2025-09-02 11:56:53 +00:00
结构调整、kcp支持
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
package controllers
|
||||
|
||||
import (
|
||||
"github.com/cnlh/nps/lib"
|
||||
"github.com/cnlh/nps/lib/file"
|
||||
"github.com/cnlh/nps/server"
|
||||
)
|
||||
|
||||
@@ -72,27 +72,27 @@ func (s *IndexController) Add() {
|
||||
s.SetInfo("新增")
|
||||
s.display()
|
||||
} else {
|
||||
t := &lib.Tunnel{
|
||||
t := &file.Tunnel{
|
||||
TcpPort: s.GetIntNoErr("port"),
|
||||
Mode: s.GetString("type"),
|
||||
Target: s.GetString("target"),
|
||||
Config: &lib.Config{
|
||||
Config: &file.Config{
|
||||
U: s.GetString("u"),
|
||||
P: s.GetString("p"),
|
||||
Compress: s.GetString("compress"),
|
||||
Crypt: s.GetBoolNoErr("crypt"),
|
||||
},
|
||||
Id: lib.GetCsvDb().GetTaskId(),
|
||||
Id: file.GetCsvDb().GetTaskId(),
|
||||
UseClientCnf: s.GetBoolNoErr("use_client"),
|
||||
Status: true,
|
||||
Remark: s.GetString("remark"),
|
||||
Flow: &lib.Flow{},
|
||||
Flow: &file.Flow{},
|
||||
}
|
||||
var err error
|
||||
if t.Client, err = lib.GetCsvDb().GetClient(s.GetIntNoErr("client_id")); err != nil {
|
||||
if t.Client, err = file.GetCsvDb().GetClient(s.GetIntNoErr("client_id")); err != nil {
|
||||
s.AjaxErr(err.Error())
|
||||
}
|
||||
lib.GetCsvDb().NewTask(t)
|
||||
file.GetCsvDb().NewTask(t)
|
||||
if err := server.AddTask(t); err != nil {
|
||||
s.AjaxErr(err.Error())
|
||||
} else {
|
||||
@@ -103,7 +103,7 @@ func (s *IndexController) Add() {
|
||||
func (s *IndexController) GetOneTunnel() {
|
||||
id := s.GetIntNoErr("id")
|
||||
data := make(map[string]interface{})
|
||||
if t, err := lib.GetCsvDb().GetTask(id); err != nil {
|
||||
if t, err := file.GetCsvDb().GetTask(id); err != nil {
|
||||
data["code"] = 0
|
||||
} else {
|
||||
data["code"] = 1
|
||||
@@ -115,7 +115,7 @@ func (s *IndexController) GetOneTunnel() {
|
||||
func (s *IndexController) Edit() {
|
||||
id := s.GetIntNoErr("id")
|
||||
if s.Ctx.Request.Method == "GET" {
|
||||
if t, err := lib.GetCsvDb().GetTask(id); err != nil {
|
||||
if t, err := file.GetCsvDb().GetTask(id); err != nil {
|
||||
s.error()
|
||||
} else {
|
||||
s.Data["t"] = t
|
||||
@@ -123,7 +123,7 @@ func (s *IndexController) Edit() {
|
||||
s.SetInfo("修改")
|
||||
s.display()
|
||||
} else {
|
||||
if t, err := lib.GetCsvDb().GetTask(id); err != nil {
|
||||
if t, err := file.GetCsvDb().GetTask(id); err != nil {
|
||||
s.error()
|
||||
} else {
|
||||
t.TcpPort = s.GetIntNoErr("port")
|
||||
@@ -137,10 +137,10 @@ func (s *IndexController) Edit() {
|
||||
t.Config.Crypt = s.GetBoolNoErr("crypt")
|
||||
t.UseClientCnf = s.GetBoolNoErr("use_client")
|
||||
t.Remark = s.GetString("remark")
|
||||
if t.Client, err = lib.GetCsvDb().GetClient(s.GetIntNoErr("client_id")); err != nil {
|
||||
if t.Client, err = file.GetCsvDb().GetClient(s.GetIntNoErr("client_id")); err != nil {
|
||||
s.AjaxErr("修改失败")
|
||||
}
|
||||
lib.GetCsvDb().UpdateTask(t)
|
||||
file.GetCsvDb().UpdateTask(t)
|
||||
}
|
||||
s.AjaxOk("修改成功")
|
||||
}
|
||||
@@ -179,7 +179,7 @@ func (s *IndexController) HostList() {
|
||||
} else {
|
||||
start, length := s.GetAjaxParams()
|
||||
clientId := s.GetIntNoErr("client_id")
|
||||
list, cnt := lib.GetCsvDb().GetHost(start, length, clientId)
|
||||
list, cnt := file.GetCsvDb().GetHost(start, length, clientId)
|
||||
s.AjaxTable(list, cnt, cnt)
|
||||
}
|
||||
}
|
||||
@@ -200,7 +200,7 @@ func (s *IndexController) GetHost() {
|
||||
|
||||
func (s *IndexController) DelHost() {
|
||||
host := s.GetString("host")
|
||||
if err := lib.GetCsvDb().DelHost(host); err != nil {
|
||||
if err := file.GetCsvDb().DelHost(host); err != nil {
|
||||
s.AjaxErr("删除失败")
|
||||
}
|
||||
s.AjaxOk("删除成功")
|
||||
@@ -213,19 +213,19 @@ func (s *IndexController) AddHost() {
|
||||
s.SetInfo("新增")
|
||||
s.display("index/hadd")
|
||||
} else {
|
||||
h := &lib.Host{
|
||||
h := &file.Host{
|
||||
Host: s.GetString("host"),
|
||||
Target: s.GetString("target"),
|
||||
HeaderChange: s.GetString("header"),
|
||||
HostChange: s.GetString("hostchange"),
|
||||
Remark: s.GetString("remark"),
|
||||
Flow: &lib.Flow{},
|
||||
Flow: &file.Flow{},
|
||||
}
|
||||
var err error
|
||||
if h.Client, err = lib.GetCsvDb().GetClient(s.GetIntNoErr("client_id")); err != nil {
|
||||
if h.Client, err = file.GetCsvDb().GetClient(s.GetIntNoErr("client_id")); err != nil {
|
||||
s.AjaxErr("添加失败")
|
||||
}
|
||||
lib.GetCsvDb().NewHost(h)
|
||||
file.GetCsvDb().NewHost(h)
|
||||
s.AjaxOk("添加成功")
|
||||
}
|
||||
}
|
||||
@@ -251,9 +251,9 @@ func (s *IndexController) EditHost() {
|
||||
h.HostChange = s.GetString("hostchange")
|
||||
h.Remark = s.GetString("remark")
|
||||
h.TargetArr = nil
|
||||
lib.GetCsvDb().UpdateHost(h)
|
||||
file.GetCsvDb().UpdateHost(h)
|
||||
var err error
|
||||
if h.Client, err = lib.GetCsvDb().GetClient(s.GetIntNoErr("client_id")); err != nil {
|
||||
if h.Client, err = file.GetCsvDb().GetClient(s.GetIntNoErr("client_id")); err != nil {
|
||||
s.AjaxErr("修改失败")
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user