feat(api): add unit-openvpn endpoint

NethServer · Sep 18, 2024 · f9d08a5 · f9d08a5
1 parent 4a9667b
commit f9d08a5
Show file tree

Hide file tree

Showing 4 changed files with 69 additions and 1 deletion.
diff --git a/api/main.go b/api/main.go
@@ -148,7 +148,8 @@ 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)
+		reports.POST("/unit-openvpn", methods.SetUnitOpenVPNRW)
+		reports.POST("/unit-wan", methods.SetUnitWan)
 	}
 
 	// handle missing endpoint

diff --git a/api/methods/report.go b/api/methods/report.go
@@ -115,6 +115,55 @@ func SetUnitOpenVPNRW(c *gin.Context) {
 	}
 }
 
+func SetUnitWan(c *gin.Context) {
+	// bind json
+	var req models.UnitWanRequest
+	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 wan_config WHERE uuid = $1", c.MustGet("UnitId").(string))
+	if err != nil {
+		logs.Logs.Println("[ERR][UNITWAN] 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 WAN table
+	for _, wan := range req.Data {
+		_, err := dbpool.Exec(dbctx, "INSERT INTO wan_config (uuid, interface, device) VALUES ($1, $2, $3)", c.MustGet("UnitId").(string), wan.Interface, wan.Device)
+		if err != nil {
+			logs.Logs.Println("[ERR][UNITWAN] 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
@@ -97,3 +97,14 @@ type OpenVPNConfigurations []OpenVPNConfiguration
 type UnitOpenVPNRWRequest struct {
 	Data OpenVPNConfigurations `json:"data" binding:"required"`
 }
+
+type Wan struct {
+	Interface string `json:"interface" binding:"required"`
+	Device    string `json:"device" binding:"required"`
+}
+
+type Wans []Wan
+
+type UnitWanRequest struct {
+	Data Wans `json:"data" binding:"required"`
+}
diff --git a/api/storage/report_schema.sql.tmpl b/api/storage/report_schema.sql.tmpl
@@ -24,6 +24,13 @@ CREATE TABLE IF NOT EXISTS openvpn_config (
     created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
 );
 
+CREATE TABLE IF NOT EXISTS wan_config (
+    uuid UUID NOT NULL references units(uuid),
+    interface TEXT NOT NULL,
+    device TEXT,
+    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+);
+
 -- General retention policies
 
 -- Keep raw data for 30 days