Add download button

handler/routes.go

@@ -4,6 +4,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"net/http"
+	"strings"
 	"time"
 
 	"github.com/gorilla/sessions"
@@ -178,7 +179,34 @@ func SetClientStatus() echo.HandlerFunc {
 		db.Write("clients", clientID, &client)
 		log.Infof("Changed client %s enabled status to %v", client.ID, status)
 
-		return c.JSON(http.StatusOK, jsonHTTPResponse{true, "ok"})
+		return c.JSON(http.StatusOK, jsonHTTPResponse{true, "Changed client status successfully"})
+	}
+}
+
+// DownloadClient handler
+func DownloadClient() echo.HandlerFunc {
+	return func(c echo.Context) error {
+		clientID := c.QueryParam("clientid")
+		if clientID == "" {
+			return c.JSON(http.StatusNotFound, jsonHTTPResponse{false, "Missing clientid parameter"})
+		}
+
+		client, err := util.GetClientByID(clientID)
+		if err != nil {
+			log.Errorf("Cannot generate client id %s config file for downloading: %v", clientID, err)
+			return c.JSON(http.StatusNotFound, jsonHTTPResponse{false, "Client not found"})
+		}
+		// build config
+		server, _ := util.GetServer()
+		globalSettings, _ := util.GetGlobalSettings()
+		config := util.BuildClientConfig(client, server, globalSettings)
+
+		// create io reader from string
+		reader := strings.NewReader(config)
+
+		// set response header for downloading
+		c.Response().Header().Set(echo.HeaderContentDisposition, "attachment; filename=wg0.conf")
+		return c.Stream(http.StatusOK, "text/plain", reader)
 	}
 }
 

main.go

@@ -25,6 +25,7 @@ func main() {
 	app.POST("/new-client", handler.NewClient())
 	app.POST("/client/set-status", handler.SetClientStatus())
 	app.POST("/remove-client", handler.RemoveClient())
+	app.GET("/download", handler.DownloadClient())
 	app.GET("/wg-server", handler.WireGuardServer())
 	app.POST("wg-server/interfaces", handler.WireGuardServerInterfaces())
 	app.POST("wg-server/keypair", handler.WireGuardServerKeyPair())

templates/clients.html

@@ -36,8 +36,9 @@ Wireguard Clients
                         src="{{ .QRCode }}" />
                     <div class="info-box-content">
                         <div class="btn-group">
-                            <button type="button" class="btn btn-outline-success btn-sm">Download</button>
-                            <button type="button" class="btn btn-outline-primary btn-sm">Edit</button>
+                            <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>

util/db.go

@@ -224,3 +224,20 @@ func GetClients(hasQRCode bool) ([]model.ClientData, error) {
 
 	return clients, nil
 }
+
+// GetClientByID func to query a client from the database
+func GetClientByID(clientID string) (model.Client, error) {
+	client := model.Client{}
+
+	db, err := DBConn()
+	if err != nil {
+		return client, err
+	}
+
+	// read client information
+	if err := db.Read("clients", clientID, &client); err != nil {
+		return client, err
+	}
+
+	return client, nil
+}