API-1 # Minor refactor

web/controllers/backup_controller.go

@@ -11,7 +11,7 @@ import (
 	"github.com/vwxyzjn/portwarden/web/models"
 )
 
-func EncryptBackupController(c *gin.Context) {
+func EncryptBackupHandler(c *gin.Context) {
 	var ebi models.EncryptBackupInfo
 	if err := c.ShouldBindJSON(&ebi); err != nil {
 		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error(), "message": ""})
@@ -32,7 +32,7 @@ func EncryptBackupController(c *gin.Context) {
 	}
 }
 
-func DecryptBackupController(c *gin.Context) {
+func DecryptBackupHandler(c *gin.Context) {
 	var dbi models.DecryptBackupInfo
 	var err error
 	if err = c.ShouldBind(&dbi); err != nil {

web/server.go

@@ -1,23 +1,12 @@
 package main
 
 import (
-	"net/http"
+	"github.com/vwxyzjn/portwarden/web/server"
-
-	"github.com/gin-contrib/cors"
-	"github.com/gin-gonic/gin"
-	"github.com/vwxyzjn/portwarden/web/controllers"
 )
 
 func main() {
-	r := gin.Default()
+	ps := server.PortwardenServer{
-	r.Use(cors.Default())
+		Port: 5000,
-
+	}
-	r.GET("/", func(c *gin.Context) {
+	ps.Run()
-		http.ServeFile(c.Writer, c.Request, "index.html")
-	})
-
-	r.POST("/encrypt", controllers.EncryptBackupController)
-	r.POST("/decrypt", controllers.DecryptBackupController)
-
-	r.Run(":5000")
 }

web/server/server.go

@@ -0,0 +1,29 @@
+package server
+
+import (
+	"net/http"
+	"strconv"
+
+	"github.com/gin-contrib/cors"
+	"github.com/gin-gonic/gin"
+	"github.com/vwxyzjn/portwarden/web/controllers"
+)
+
+type PortwardenServer struct {
+	Port   int
+	Router *gin.Engine
+}
+
+func (ps *PortwardenServer) Run() {
+	ps.Router = gin.Default()
+	ps.Router.Use(cors.Default())
+
+	ps.Router.GET("/", func(c *gin.Context) {
+		http.ServeFile(c.Writer, c.Request, "index.html")
+	})
+
+	ps.Router.POST("/encrypt", controllers.EncryptBackupHandler)
+	ps.Router.POST("/decrypt", controllers.DecryptBackupHandler)
+
+	ps.Router.Run(":" + strconv.Itoa(ps.Port))
+}