api: add unit-openvpnrw API

Store OpenVPN Road Warrior configuration to be used inside the VPN dashboard
NethServer · Sep 18, 2024 · b37a303 · b37a303
1 parent a213de9
commit b37a303
Show file tree

Hide file tree

Showing 4 changed files with 70 additions and 0 deletions.
diff --git a/api/main.go b/api/main.go
@@ -148,6 +148,7 @@ func main() {
 		reports.POST("/ovpnrw-connections", methods.UpdateOvpnConnections)
 		reports.POST("/dpi-stats", methods.UpdateDpiStats)
 		reports.POST("/unit-name", methods.SetUnitName)
+		reports.POST("/unit-openvpnrw", methods.SetUnitOpenVPNRW)
 	}
 
 	// handle missing endpoint

diff --git a/api/methods/report.go b/api/methods/report.go
@@ -66,6 +66,55 @@ func SetUnitName(c *gin.Context) {
 	}
 }
 
+func SetUnitOpenVPNRW(c *gin.Context) {
+	var req models.UnitOpenVPNRWRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		c.JSON(http.StatusBadRequest, structs.Map(response.StatusBadRequest{
+			Code:    400,
+			Message: "Invalid request",
+			Data:    err.Error(),
+		}))
+		return
+	}
+
+	if checkUnitId(c.MustGet("UnitId").(string)) != nil {
+		c.JSON(http.StatusInternalServerError, structs.Map(response.StatusInternalServerError{
+			Message: "Unit not found",
+			Data:    nil,
+			Code:    404,
+		}))
+		return
+	}
+
+	dbpool, dbctx := storage.ReportInstance()
+
+	// Remove all previous data
+	_, err := dbpool.Exec(dbctx, "DELETE FROM openvpn_rw WHERE uuid = $1", c.MustGet("UnitId").(string))
+	if err != nil {
+		logs.Logs.Println("[ERR][UNITOPENVPNRW] error deleting previous data: " + err.Error())
+		c.JSON(http.StatusInternalServerError, structs.Map(response.StatusInternalServerError{
+			Code:    500,
+			Message: "Error deleting previous data",
+			Data:    err.Error(),
+		}))
+		return
+	}
+
+	// insert inside OpenVPN table
+	for _, server := range req.Data {
+		_, err := dbpool.Exec(dbctx, "INSERT INTO openvpn_rw (uuid, instance, name, device) VALUES ($1, $2, $3, $4)", c.MustGet("UnitId").(string), server.Instance, server.Name, server.Device)
+		if err != nil {
+			logs.Logs.Println("[ERR][UNITOPENVPNRW] error inserting data: " + err.Error())
+			c.JSON(http.StatusInternalServerError, structs.Map(response.StatusInternalServerError{
+				Code:    500,
+				Message: "Error inserting data",
+				Data:    err.Error(),
+			}))
+			return
+		}
+	}
+}
+
 func checkUnitId(unitId string) error {
 	if unitId == "" {
 		return errors.New("uuid is empty")

diff --git a/api/models/report.go b/api/models/report.go
@@ -84,3 +84,15 @@ type DpiStatsRequest struct {
 type UnitNameRequest struct {
 	Name string `json:"name" binding:"required"`
 }
+
+type OpenVPNRWServer struct {
+	Instance string `json:"instance" binding:"required"`
+	Name     string `json:"name" binding:"required"`
+	Device   string `json:"device" binding:"required"`
+}
+
+type OpenVPNRWServers []OpenVPNRWServer
+
+type UnitOpenVPNRWRequest struct {
+	Data OpenVPNRWServers `json:"data" binding:"required"`
+}
diff --git a/api/storage/report_schema.sql.tmpl b/api/storage/report_schema.sql.tmpl
@@ -15,6 +15,14 @@ CREATE TABLE IF NOT EXISTS units (
     created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
 );
 
+CREATE TABLE IF NOT EXISTS openvpn_rw (
+    uuid UUID NOT NULL references units(uuid),
+    instance TEXT NOT NULL,
+    name TEXT,
+    device TEXT NOT NULL,
+    updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+);
+
 -- General retention policies
 
 -- Keep raw data for 30 days