From e3fac242bb9fc43b8e64acf441790c45b0bb3da1 Mon Sep 17 00:00:00 2001
From: Khanh Ngo <k@ndk.name>
Date: Mon, 1 Jun 2020 17:59:57 +0700
Subject: [PATCH] Client page adjustment

---
 handler/routes.go      |  15 ++++
 main.go                |   1 +
 templates/base.html    |  21 ++---
 templates/clients.html | 182 +++++++++++++++++++++++++++++++----------
 4 files changed, 164 insertions(+), 55 deletions(-)

diff --git a/handler/routes.go b/handler/routes.go
index 156fecc..a3f5494 100644
--- a/handler/routes.go
+++ b/handler/routes.go
@@ -92,6 +92,21 @@ func WireGuardClients() echo.HandlerFunc {
 	}
 }
 
+// GetClients handler return a list of Wireguard client data
+func GetClients() echo.HandlerFunc {
+	return func(c echo.Context) error {
+		// access validation
+		validSession(c)
+
+		clientDataList, err := util.GetClients(true)
+		if err != nil {
+			return c.JSON(http.StatusInternalServerError, jsonHTTPResponse{false, fmt.Sprintf("Cannot get client list: %v", err)})
+		}
+
+		return c.JSON(http.StatusOK, clientDataList)
+	}
+}
+
 // NewClient handler
 func NewClient() echo.HandlerFunc {
 	return func(c echo.Context) error {
diff --git a/main.go b/main.go
index 0022951..ed15b7f 100644
--- a/main.go
+++ b/main.go
@@ -57,6 +57,7 @@ func main() {
 	app.POST("wg-server/keypair", handler.WireGuardServerKeyPair())
 	app.GET("/global-settings", handler.GlobalSettings())
 	app.POST("/global-settings", handler.GlobalSettingSubmit())
+	app.GET("/api/clients", handler.GetClients())
 	app.GET("/api/machine-ips", handler.MachineIPAddresses())
 	app.GET("/api/suggest-client-ips", handler.SuggestIPAllocation())
 	app.GET("/api/apply-wg-config", handler.ApplyServerConfig(tmplBox))
diff --git a/templates/base.html b/templates/base.html
index 542aca7..75e0efe 100644
--- a/templates/base.html
+++ b/templates/base.html
@@ -56,16 +56,17 @@
                 </div>
             </form>
 
-        <!-- Right navbar links -->
-        <div class="navbar-nav ml-auto">
-            <button style="margin-left: 0.5em;" type="button" class="btn btn-outline-primary btn-sm" data-toggle="modal"
-                data-target="#modal_new_client"><i class="nav-icon fas fa-plus"></i> New
-                Client</button>
-            <button style="margin-left: 0.5em;" type="button" class="btn btn-outline-danger btn-sm" data-toggle="modal"
-                data-target="#modal_apply_config"><i class="nav-icon fas fa-check"></i> Apply
-                Config</button>
-            <button onclick="location.href='/logout';" style="margin-left: 0.5em;" type="button"
-                class="btn btn-outline-danger btn-sm"><i class="nav-icon fas fa-sign-out-alt"></i> Logout</button>
+            <!-- Right navbar links -->
+            <div class="navbar-nav ml-auto">
+                <button style="margin-left: 0.5em;" type="button" class="btn btn-outline-primary btn-sm" data-toggle="modal"
+                    data-target="#modal_new_client"><i class="nav-icon fas fa-plus"></i> New
+                    Client</button>
+                <button style="margin-left: 0.5em;" type="button" class="btn btn-outline-danger btn-sm" data-toggle="modal"
+                    data-target="#modal_apply_config"><i class="nav-icon fas fa-check"></i> Apply
+                    Config</button>
+                <button onclick="location.href='/logout';" style="margin-left: 0.5em;" type="button"
+                    class="btn btn-outline-danger btn-sm"><i class="nav-icon fas fa-sign-out-alt"></i> Logout</button>
+            </div>
         </nav>
         <!-- /.navbar -->
 
diff --git a/templates/clients.html b/templates/clients.html
index 9bf097f..38ab97c 100644
--- a/templates/clients.html
+++ b/templates/clients.html
@@ -24,55 +24,61 @@ Wireguard Clients
 <section class="content">
     <div class="container-fluid">
         <!-- <h5 class="mt-4 mb-2">Wireguard Clients</h5> -->
-        <div class="row">
-            {{range .clientDataList}}
-            <div class="col-sm-6" id="client_{{.Client.ID}}">
-                <div class="info-box">
-                    <div class="overlay" id="paused_{{.Client.ID}}"
-                        {{if eq .Client.Enabled true}}style="visibility: hidden;" {{end}}>
-                        <i class="paused-client fas fa-3x fa-play" onclick="resumeClient('{{.Client.ID}}')"></i>
-                    </div>
-                    <img
-                        src="{{ .QRCode }}" />
-                    <div class="info-box-content">
-                        <div class="btn-group">
-                            <button onclick="location.href='/download?clientid={{ .Client.ID }}'" type="button"
-                                class="btn btn-outline-success btn-sm">Download</button>
-                            <!-- <button type="button" class="btn btn-outline-primary btn-sm">Edit</button> -->
-                            <button type="button" class="btn btn-outline-warning btn-sm" data-toggle="modal"
-                                data-target="#modal_pause_client" data-clientid="{{ .Client.ID }}"
-                                data-clientname="{{ .Client.Name }}">Disable</button>
-                            <button type="button" class="btn btn-outline-danger btn-sm" data-toggle="modal"
-                                data-target="#modal_remove_client" data-clientid="{{ .Client.ID }}"
-                                data-clientname="{{ .Client.Name }}">Remove</button>
-                        </div>
-                        <hr>
-                        <span class="info-box-text"><i class="fas fa-user"></i> {{ .Client.Name }}</span>
-                        <span class="info-box-text"><i class="fas fa-envelope"></i> {{ .Client.Email }}</span>
-                        <span class="info-box-text"><i class="fas fa-clock"></i>
-                            {{ .Client.CreatedAt.Format "2 Jan 2006 15:04" }}</span>
-                        <span class="info-box-text"><i class="fas fa-history"></i>
-                            {{ .Client.UpdatedAt.Format "2 Jan 2006 15:04" }}</span>
-                        <span class="info-box-text"><strong>IP Allocation</strong></span>
-                        {{range .Client.AllocatedIPs}}
-                        <small class="badge badge-secondary">{{.}}</small>
-                        {{end}}
-                        <span class="info-box-text"><strong>Allowed IPs</strong></span>
-                        {{range .Client.AllowedIPs}}
-                        <small class="badge badge-secondary">{{.}}</small>
-                        {{end}}
-                    </div>
-                    <!-- /.info-box-content -->
-                </div>
-                <!-- /.info-box -->
-            </div>
-            <!-- /.col -->
-            {{end}}
+        <div class="row" id="client-list">
         </div>
         <!-- /.row -->
     </div>
 </section>
 
+<div class="modal fade" id="modal_edit_client">
+    <div class="modal-dialog">
+        <div class="modal-content">
+            <div class="modal-header">
+                <h4 class="modal-title">Edit Wireguard Client</h4>
+                <button type="button" class="close" data-dismiss="modal" aria-label="Close">
+                    <span aria-hidden="true">&times;</span>
+                </button>
+            </div>
+            <form name="frm_new_client" id="frm_new_client">
+                <div class="modal-body">
+                    <div class="form-group">
+                        <label for="client_name" class="control-label">Name</label>
+                        <input type="text" class="form-control" id="client_name" name="client_name">
+                    </div>
+                    <div class="form-group">
+                        <label for="client_email" class="control-label">Email</label>
+                        <input type="text" class="form-control" id="client_email" name="client_email">
+                    </div>
+                    <div class="form-group">
+                        <label for="client_allocated_ips" class="control-label">IP Allocation</label>
+                        <input type="text" data-role="tagsinput" class="form-control" id="client_allocated_ips">
+                    </div>
+                    <div class="form-group">
+                        <label for="client_allowed_ips" class="control-label">Allowed IPs</label>
+                        <input type="text" data-role="tagsinput" class="form-control" id="client_allowed_ips"
+                            value="0.0.0.0/0">
+                    </div>
+                    <div class="form-group">
+                        <div class="icheck-primary d-inline">
+                            <input type="checkbox" id="enabled" checked>
+                            <label for="enabled">
+                                Enable after creation
+                            </label>
+                        </div>
+                    </div>
+                </div>
+                <div class="modal-footer justify-content-between">
+                    <button type="button" class="btn btn-default" data-dismiss="modal">Cancel</button>
+                    <button type="submit" class="btn btn-primary">Submit</button>
+                </div>
+            </form>
+        </div>
+        <!-- /.modal-content -->
+    </div>
+    <!-- /.modal-dialog -->
+</div>
+<!-- /.modal -->
+
 <div class="modal fade" id="modal_pause_client">
     <div class="modal-dialog">
         <div class="modal-content bg-warning">
@@ -119,7 +125,88 @@ Wireguard Clients
 {{end}}
 
 {{define "bottom_js"}}
+<script type="text/html" id="template">
+
+
+</script>
+
     <script>
+        function renderClient(data) {
+            $.each(data, function(index, obj) {
+                // render client status css tag style
+                let clientStatusHtml = '>'
+                if (obj.Client.enabled) {
+                    clientStatusHtml = `style="visibility: hidden;">`
+                }
+
+                // render client allocated ip addresses
+                let allocatedIpsHtml = "";
+                $.each(obj.Client.allocated_ips, function(index, obj) {
+                    allocatedIpsHtml += `<small class="badge badge-secondary">${obj}</small>`;
+                })
+
+                // render client allowed ip addresses
+                let allowedIpsHtml = "";
+                $.each(obj.Client.allowed_ips, function(index, obj) {
+                    allowedIpsHtml += `<small class="badge badge-secondary">${obj}</small>`;
+                })
+
+                // render client html content
+                let html = `<div class="col-sm-6" id="client_${obj.Client.id}">
+                                <div class="info-box">
+                                    <div class="overlay" id="paused_${obj.Client.id}"` + clientStatusHtml
+                                        + `<i class="paused-client fas fa-3x fa-play" onclick="resumeClient('${obj.Client.id}')"></i>
+                                    </div>
+                                    <img src="${obj.QRCode}" />
+                                    <div class="info-box-content">
+                                        <div class="btn-group">
+                                            <button onclick="location.href='/download?clientid=${obj.Client.id}'" type="button"
+                                                class="btn btn-outline-success btn-sm">Download</button>
+                                            <button type="button" class="btn btn-outline-primary btn-sm" data-toggle="modal"
+                                                    data-target="#modal_edit_client">Edit</button>
+                                            <button type="button" class="btn btn-outline-warning btn-sm" data-toggle="modal"
+                                                data-target="#modal_pause_client" data-clientid="${obj.Client.id}"
+                                                data-clientname="${obj.Client.name}">Disable</button>
+                                            <button type="button" class="btn btn-outline-danger btn-sm" data-toggle="modal"
+                                                data-target="#modal_remove_client" data-clientid="${obj.Client.id}"
+                                                data-clientname="${obj.Client.name}">Remove</button>
+                                        </div>
+                                        <hr>
+                                        <span class="info-box-text"><i class="fas fa-user"></i> ${obj.Client.name}</span>
+                                        <span class="info-box-text"><i class="fas fa-envelope"></i> ${obj.Client.email}</span>
+                                        <span class="info-box-text"><i class="fas fa-clock"></i>
+                                            ${obj.Client.created_at}</span>
+                                        <span class="info-box-text"><i class="fas fa-history"></i>
+                                            ${obj.Client.updated_at}</span>
+                                        <span class="info-box-text"><strong>IP Allocation</strong></span>`
+                                        + allocatedIpsHtml
+                                        + `<span class="info-box-text"><strong>Allowed IPs</strong></span>`
+                                        + allowedIpsHtml
+                                    +`</div>
+                                </div>
+                            </div>`
+
+                // add the client html elements to the list
+                $('#client-list').append(html);
+            });
+        }
+        function populateClientList() {
+            $.ajax({
+                cache: false,
+                method: 'GET',
+                url: '/api/clients',
+                dataType: 'json',
+                contentType: "application/json",
+                success: function (data) {
+                    renderClient(data);
+                },
+                error: function (jqXHR, exception) {
+                    const responseJson = jQuery.parseJSON(jqXHR.responseText);
+                    toastr.error(responseJson['message']);
+                }
+            });
+        }
+
         function setClientStatus(clientID, status) {
             const data = {"id": clientID, "status": status};
             $.ajax({
@@ -130,7 +217,7 @@ Wireguard Clients
                 contentType: "application/json",
                 data: JSON.stringify(data),
                 success: function (data) {
-                    console.log("Set client " + clientID + " status to " + status)
+                    console.log("Set client " + clientID + " status to " + status);
                 },
                 error: function (jqXHR, exception) {
                     const responseJson = jQuery.parseJSON(jqXHR.responseText);
@@ -152,6 +239,11 @@ Wireguard Clients
         }
     </script>
     <script>
+        // load client list
+        $(document).ready(function () {
+            populateClientList();
+        })
+
         // modal_pause_client modal event
         $("#modal_pause_client").on('show.bs.modal', function (event) {
             const button = $(event.relatedTarget);