新功能+bug修复

web/controllers/index.go

@@ -160,11 +160,40 @@ func (s *IndexController) AddHost() {
 		s.display("index/hadd")
 	} else {
 		h := &server.HostList{
-			Vkey:   s.GetString("vkey"),
-			Host:   s.GetString("host"),
-			Target: s.GetString("target"),
+			Vkey:         s.GetString("vkey"),
+			Host:         s.GetString("host"),
+			Target:       s.GetString("target"),
+			HeaderChange: s.GetString("header"),
+			HostChange:   s.GetString("hostchange"),
 		}
 		server.CsvDb.NewHost(h)
 		s.AjaxOk("添加成功")
 	}
 }
+
+func (s *IndexController) EditHost() {
+	if s.Ctx.Request.Method == "GET" {
+		host := s.GetString("host")
+		if h, t, err := server.GetKeyByHost(host); err != nil {
+			s.error()
+		} else {
+			s.Data["t"] = t
+			s.Data["h"] = h
+		}
+		s.SetInfo("修改")
+		s.display("index/hedit")
+	} else {
+		host := s.GetString("host")
+		if h, _, err := server.GetKeyByHost(host); err != nil {
+			s.error()
+		} else {
+			h.Vkey = s.GetString("vkey")
+			h.Host = s.GetString("host")
+			h.Target = s.GetString("target")
+			h.HeaderChange = s.GetString("header")
+			h.HostChange = s.GetString("hostchange")
+			server.CsvDb.UpdateHost(h)
+		}
+		s.AjaxOk("修改成功")
+	}
+}

web/views/index/hadd.html

@@ -13,6 +13,15 @@
                         <label class="control-label">内网目标</label>
                         <input class="form-control" type="text" name="target" placeholder="内网隧道目标，例如10.1.50.203:22">
                     </div>
+                    <div class="form-group" id="header">
+                        <label class="control-label">header头修改(冒号分隔，多个请换行填写)</label>
+                        <textarea class="form-control" rows="4" type="text" name="header" placeholder="Cache-Control: no-cache"></textarea>
+                    </div>
+                    <div class="form-group" id="hostchange">
+                        <label class="control-label">host修改</label>
+                        <input class="form-control" value="" type="text" name="hostchange"
+                               placeholder="host修改">
+                    </div>
                 </form>
             </div>
             <div class="tile-footer">

web/views/index/hedit.html

@@ -0,0 +1,54 @@
+<div class="row tile">
+    <div class="col-md-6 col-md-auto">
+        <div>
+            <h3 class="tile-title">修改</h3>
+            <div class="tile-body">
+                <form>
+                    <input type="hidden" name="vkey" value="{{.t.VerifyKey}}">
+                    <div class="form-group">
+                        <label class="control-label">域名</label>
+                        <input class="form-control" value="{{.h.Host}}" type="text" name="host" placeholder="域名">
+                    </div>
+                    <div class="form-group">
+                        <label class="control-label">内网目标</label>
+                        <input class="form-control" value="{{.h.Target}}" type="text" name="target"
+                               placeholder="内网隧道目标，例如10.1.50.203:22">
+                    </div>
+                    <div class="form-group" id="header">
+                        <label class="control-label">header头修改(冒号分隔，多个请换行填写)</label>
+                        <textarea class="form-control" rows="4" type="text" name="header"
+                                  placeholder="Cache-Control: no-cache">{{.h.HeaderChange}}</textarea>
+                    </div>
+                    <div class="form-group" id="hostchange">
+                        <label class="control-label">host修改</label>
+                        <input class="form-control" value="{{.h.HostChange}}" type="text" name="hostchange"
+                               placeholder="host修改">
+                    </div>
+                </form>
+            </div>
+            <div class="tile-footer">
+                &nbsp;&nbsp;<button class="btn btn-success" href="#" id="add"><i
+                    class="fa fa-fw fa-lg fa-eye"></i>保存
+            </button>
+            </div>
+        </div>
+    </div>
+</div>
+</main>
+<script>
+    $(function () {
+        $("#add").on("click", function () {
+            $.ajax({
+                type: "POST",
+                url: "/index/edithost",
+                data: $("form").serializeArray(),
+                success: function (res) {
+                    alert(res.msg)
+                    if (res.status) {
+                        history.back(-1)
+                    }
+                }
+            })
+        })
+    })
+</script>

web/views/index/hlist.html

@@ -42,6 +42,10 @@
         window.location.href = "/index/addhost?vkey={{.vkey}}"
     }
 
+    function edit(host) {
+        window.location.href = "/index/edithost?host=" + host
+    }
+
     $(document).ready(function () {
         var table = $('#sampleTable').DataTable({
             dom: 'Bfrtip',
@@ -67,7 +71,8 @@
 
                     return '<div class="btn-group" role="group" aria-label="..."> ' +
                             '<button onclick="del(\'' + row.Host + '\')" type="button"   class="btn btn-danger btn-sm">删除</button>' +
-                            ' </div>'
+                            '<button onclick="edit(\'' + row.Host + '\')" type="button"   class="btn btn-primary btn-sm">编辑查看</button> '
+                            + ' </div>'
                 }
             }
             ],