Added Backup Update

2022-05-24 15:07:31 -05:00
parent eab7a20533
commit 880b858d01
5 changed files with 80 additions and 7 deletions
--- a/.gitignore
+++ b/.gitignore
@@ -0,0 +1,2 @@
+data
+vrisingd
--- a/README.md
+++ b/README.md
@@ -1,2 +1,6 @@
 # VRisingD (V Rising Daemon)
 This tool allows for the simple configuration, backup, and starting of a V Rising Server to better assist for Docker deployments using a web page
+
+- [x] Backup Saves
+- [x] Update
+- [] Upload Config Files
--- a/handlers.go
+++ b/handlers.go
@@ -1,18 +1,27 @@
 package main

 import (
+	"archive/zip"
 	"bytes"
+	"io"
 	"net/http"
+	"os"
 	"os/exec"
+	"path/filepath"
+	"strings"

 	"github.com/gin-gonic/gin"
 )

-func index(c *gin.Context) {
-	index_page, err := indexHTML.ReadFile("html/index.html")
+func check(err error) {
 	if err != nil {
 		panic(err)
 	}
+}
+
+func index(c *gin.Context) {
+	index_page, err := indexHTML.ReadFile("html/index.html")
+	check(err)
 	c.Data(http.StatusOK, "text/html", index_page)
 }

@@ -30,3 +39,55 @@ func update(c *gin.Context) {
 		c.String(http.StatusOK, out.String())
 	}
 }
+
+func RecursiveZip(pathToZip, destinationPath string) error {
+
+	os.MkdirAll("data", 0755)
+
+	destinationFile, err := os.Create(destinationPath)
+	if err != nil {
+		return err
+	}
+	myZip := zip.NewWriter(destinationFile)
+	err = filepath.Walk(pathToZip, func(filePath string, info os.FileInfo, err error) error {
+		if info.IsDir() {
+			return nil
+		}
+		if err != nil {
+			return err
+		}
+		relPath := strings.TrimPrefix(filePath, filepath.Dir(pathToZip))
+		zipFile, err := myZip.Create(relPath)
+		if err != nil {
+			return err
+		}
+		fsFile, err := os.Open(filePath)
+		if err != nil {
+			return err
+		}
+		_, err = io.Copy(zipFile, fsFile)
+		if err != nil {
+			return err
+		}
+		return nil
+	})
+	if err != nil {
+		return err
+	}
+	err = myZip.Close()
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func backup(c *gin.Context) {
+
+	err := RecursiveZip("/root/.wine/drive_c/users/root/AppData/LocalLow/Stunlock Studios/VRisingServer/Saves", "data/backup.zip")
+	check(err)
+
+	backup, err := os.ReadFile("data/backup.zip")
+	check(err)
+
+	c.Data(http.StatusOK, "application/zip", backup)
+}
--- a/html/index.html
+++ b/html/index.html
@@ -1,15 +1,20 @@
 <!DOCTYPE html>
 <header>
-    <title>Rod Biren</title>
+    <title>Rod</title>
 </header>

 <body>
    <p id="console"> Why Hello There</p>
-    <button id="update" type="button" onclick="update()">Update</button>
+    <button id="update" type="button">Update</button>
+    <form method="get" action="backup.zip">
+        <button type="submit">Download!</button>
+    </form>
+
 </body>

 <script>
    let update_button = document.getElementById("update")
+    let start_button = document.getElementById("start")
    let console = document.getElementById("console")

    update_button.addEventListener("click", update)
--- a/main.go
+++ b/main.go
@@ -19,6 +19,7 @@ func main() {

 	r.GET("/", index)
 	r.GET("/api/update", update)
+	r.GET("/backup.zip", backup)
 	srv := &http.Server{
 		Addr:    ":2169",
 		Handler: r,