|
|
@@ -1,12 +1,18 @@
|
|
|
package main
|
|
|
|
|
|
import (
|
|
|
+ "archive/zip"
|
|
|
"encoding/json"
|
|
|
+ "fmt"
|
|
|
+ "io"
|
|
|
"io/ioutil"
|
|
|
"log"
|
|
|
+ "net/http"
|
|
|
"os"
|
|
|
"path/filepath"
|
|
|
+ "strconv"
|
|
|
"strings"
|
|
|
+ "time"
|
|
|
|
|
|
"imuslab.com/zoraxy/mod/dynamicproxy"
|
|
|
"imuslab.com/zoraxy/mod/utils"
|
|
|
@@ -31,7 +37,7 @@ type Record struct {
|
|
|
|
|
|
func SaveReverseProxyConfig(proxyConfigRecord *Record) error {
|
|
|
//TODO: Make this accept new def types
|
|
|
- os.MkdirAll("conf", 0775)
|
|
|
+ os.MkdirAll("./conf/proxy/", 0775)
|
|
|
filename := getFilenameFromRootName(proxyConfigRecord.Rootname)
|
|
|
|
|
|
//Generate record
|
|
|
@@ -39,12 +45,12 @@ func SaveReverseProxyConfig(proxyConfigRecord *Record) error {
|
|
|
|
|
|
//Write to file
|
|
|
js, _ := json.MarshalIndent(thisRecord, "", " ")
|
|
|
- return ioutil.WriteFile(filepath.Join("conf", filename), js, 0775)
|
|
|
+ return ioutil.WriteFile(filepath.Join("./conf/proxy/", filename), js, 0775)
|
|
|
}
|
|
|
|
|
|
func RemoveReverseProxyConfig(rootname string) error {
|
|
|
filename := getFilenameFromRootName(rootname)
|
|
|
- removePendingFile := strings.ReplaceAll(filepath.Join("conf", filename), "\\", "/")
|
|
|
+ removePendingFile := strings.ReplaceAll(filepath.Join("./conf/proxy/", filename), "\\", "/")
|
|
|
log.Println("Config Removed: ", removePendingFile)
|
|
|
if utils.FileExists(removePendingFile) {
|
|
|
err := os.Remove(removePendingFile)
|
|
|
@@ -83,3 +89,149 @@ func getFilenameFromRootName(rootname string) string {
|
|
|
filename = filename + ".config"
|
|
|
return filename
|
|
|
}
|
|
|
+
|
|
|
+/*
|
|
|
+ Importer and Exporter of Zoraxy proxy config
|
|
|
+*/
|
|
|
+
|
|
|
+func ExportConfigAsZip(w http.ResponseWriter, r *http.Request) {
|
|
|
+ // Specify the folder path to be zipped
|
|
|
+ folderPath := "./conf/"
|
|
|
+
|
|
|
+ // Set the Content-Type header to indicate it's a zip file
|
|
|
+ w.Header().Set("Content-Type", "application/zip")
|
|
|
+ // Set the Content-Disposition header to specify the file name
|
|
|
+ w.Header().Set("Content-Disposition", "attachment; filename=\"config.zip\"")
|
|
|
+
|
|
|
+ // Create a zip writer
|
|
|
+ zipWriter := zip.NewWriter(w)
|
|
|
+ defer zipWriter.Close()
|
|
|
+
|
|
|
+ // Walk through the folder and add files to the zip
|
|
|
+ err := filepath.Walk(folderPath, func(filePath string, fileInfo os.FileInfo, err error) error {
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+
|
|
|
+ if folderPath == filePath {
|
|
|
+ //Skip root folder
|
|
|
+ return nil
|
|
|
+ }
|
|
|
+
|
|
|
+ // Create a new file in the zip
|
|
|
+ if !utils.IsDir(filePath) {
|
|
|
+ zipFile, err := zipWriter.Create(filePath)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+
|
|
|
+ // Open the file on disk
|
|
|
+ file, err := os.Open(filePath)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ defer file.Close()
|
|
|
+
|
|
|
+ // Copy the file contents to the zip file
|
|
|
+ _, err = io.Copy(zipFile, file)
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return nil
|
|
|
+ })
|
|
|
+
|
|
|
+ if err != nil {
|
|
|
+ // Handle the error and send an HTTP response with the error message
|
|
|
+ http.Error(w, fmt.Sprintf("Failed to zip folder: %v", err), http.StatusInternalServerError)
|
|
|
+ return
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func ImportConfigFromZip(w http.ResponseWriter, r *http.Request) {
|
|
|
+ // Check if the request is a POST with a file upload
|
|
|
+ if r.Method != http.MethodPost {
|
|
|
+ http.Error(w, "Invalid request method", http.StatusBadRequest)
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ // Max file size limit (10 MB in this example)
|
|
|
+ r.ParseMultipartForm(10 << 20)
|
|
|
+
|
|
|
+ // Get the uploaded file
|
|
|
+ file, handler, err := r.FormFile("file")
|
|
|
+ if err != nil {
|
|
|
+ http.Error(w, "Failed to retrieve uploaded file", http.StatusInternalServerError)
|
|
|
+ return
|
|
|
+ }
|
|
|
+ defer file.Close()
|
|
|
+
|
|
|
+ if filepath.Ext(handler.Filename) != ".zip" {
|
|
|
+ http.Error(w, "Upload file is not a zip file", http.StatusInternalServerError)
|
|
|
+ return
|
|
|
+ }
|
|
|
+ // Create the target directory to unzip the files
|
|
|
+ targetDir := "./conf"
|
|
|
+ if utils.FileExists(targetDir) {
|
|
|
+ //Backup the old config to old
|
|
|
+ os.Rename("./conf", "./conf.old_"+strconv.Itoa(int(time.Now().Unix())))
|
|
|
+ }
|
|
|
+
|
|
|
+ err = os.MkdirAll(targetDir, os.ModePerm)
|
|
|
+ if err != nil {
|
|
|
+ http.Error(w, fmt.Sprintf("Failed to create target directory: %v", err), http.StatusInternalServerError)
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ // Open the zip file
|
|
|
+ zipReader, err := zip.NewReader(file, handler.Size)
|
|
|
+ if err != nil {
|
|
|
+ http.Error(w, fmt.Sprintf("Failed to open zip file: %v", err), http.StatusInternalServerError)
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ // Extract each file from the zip archive
|
|
|
+ for _, zipFile := range zipReader.File {
|
|
|
+ // Open the file in the zip archive
|
|
|
+ rc, err := zipFile.Open()
|
|
|
+ if err != nil {
|
|
|
+ http.Error(w, fmt.Sprintf("Failed to open file in zip: %v", err), http.StatusInternalServerError)
|
|
|
+ return
|
|
|
+ }
|
|
|
+ defer rc.Close()
|
|
|
+
|
|
|
+ // Create the corresponding file on disk
|
|
|
+ zipFile.Name = strings.ReplaceAll(zipFile.Name, "../", "")
|
|
|
+ fmt.Println("Restoring: " + strings.ReplaceAll(zipFile.Name, "\\", "/"))
|
|
|
+ if !strings.HasPrefix(strings.ReplaceAll(zipFile.Name, "\\", "/"), "conf/") {
|
|
|
+ //Malformed zip file.
|
|
|
+ http.Error(w, fmt.Sprintf("Invalid zip file structure or version too old"), http.StatusInternalServerError)
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ //Check if parent dir exists
|
|
|
+ if !utils.FileExists(filepath.Dir(zipFile.Name)) {
|
|
|
+ os.MkdirAll(filepath.Dir(zipFile.Name), 0775)
|
|
|
+ }
|
|
|
+
|
|
|
+ //Create the file
|
|
|
+ newFile, err := os.Create(zipFile.Name)
|
|
|
+ if err != nil {
|
|
|
+ http.Error(w, fmt.Sprintf("Failed to create file: %v", err), http.StatusInternalServerError)
|
|
|
+ return
|
|
|
+ }
|
|
|
+ defer newFile.Close()
|
|
|
+
|
|
|
+ // Copy the file contents from the zip to the new file
|
|
|
+ _, err = io.Copy(newFile, rc)
|
|
|
+ if err != nil {
|
|
|
+ http.Error(w, fmt.Sprintf("Failed to extract file from zip: %v", err), http.StatusInternalServerError)
|
|
|
+ return
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // Send a success response
|
|
|
+ w.WriteHeader(http.StatusOK)
|
|
|
+ fmt.Fprintln(w, "Configuration restored")
|
|
|
+}
|
Toby Chui