added handler for creating mod packs and listing mod packs

Author: mroote

src/handlers.go

@@ -8,6 +8,7 @@ import (
 	"log"
 	"net/http"
 	"os"
+	"path/filepath"
 	"strconv"
 
 	"github.com/gorilla/mux"
@@ -18,6 +19,11 @@ type JSONResponse struct {
 	Data    interface{} `json:"data,string"`
 }
 
+type ModPack struct {
+	Mods  []string `json:"mods"`
+	Title string   `json:"title"`
+}
+
 // Returns JSON response of all mods installed in factorio/mods
 func ListInstalledMods(w http.ResponseWriter, r *http.Request) {
 	var err error
@@ -201,6 +207,76 @@ func DownloadMod(w http.ResponseWriter, r *http.Request) {
 	http.ServeFile(w, r, modFile)
 }
 
+func CreateModPackHandler(w http.ResponseWriter, r *http.Request) {
+	var err error
+	resp := JSONResponse{
+		Success: false,
+	}
+
+	w.Header().Set("Content-Type", "application/json;charset=UTF-8")
+
+	switch r.Method {
+	case "GET":
+		resp.Data = "Unsupported method"
+		resp.Success = false
+		if err = json.NewEncoder(w).Encode(resp); err != nil {
+			log.Printf("Get request to modpack handler: %s", err)
+		}
+	case "POST":
+		var mods ModPack
+		body, err := ioutil.ReadAll(r.Body)
+		if err != nil {
+			log.Printf("Error in create mod pack handler body: %s", err)
+			return
+		}
+
+		log.Printf("Creating modpack with contents: %v", string(body))
+
+		err = json.Unmarshal(body, &mods)
+		if err != nil {
+			log.Printf("Error unmarshaling server settings JSON: %s", err)
+			return
+		}
+
+		err = createModPack(mods.Title, mods.Mods...)
+		if err != nil {
+			log.Printf("Error in creating modpack handler: %s", err)
+			return
+		}
+
+		resp.Success = true
+		resp.Data = fmt.Sprintf("Created modpack: %s", mods.Title)
+		if err := json.NewEncoder(w).Encode(resp); err != nil {
+			log.Printf("Error listing saves: %s", err)
+		}
+	}
+}
+
+func ListModPacks(w http.ResponseWriter, r *http.Request) {
+	var err error
+	resp := JSONResponse{
+		Success: false,
+	}
+
+	w.Header().Set("Content-Type", "application/json;charset=UTF-8")
+
+	resp.Data, err = listModPacks(filepath.Join(config.FactorioDir, "modpacks"))
+	if err != nil {
+		resp.Success = false
+		resp.Data = fmt.Sprintf("Error listing modpack files: %s", err)
+		if err := json.NewEncoder(w).Encode(resp); err != nil {
+			log.Printf("Error listing modpacks: %s", err)
+		}
+		return
+	}
+
+	resp.Success = true
+
+	if err := json.NewEncoder(w).Encode(resp); err != nil {
+		log.Printf("Error listing saves: %s", err)
+	}
+}
+
 // Lists all save files in the factorio/saves directory
 func ListSaves(w http.ResponseWriter, r *http.Request) {
 	var err error

src/main.go

@@ -91,7 +91,7 @@ func main() {
 
 	router := NewRouter()
 	createModPackDir()
-	createModPack("test", "advanced-logistics-system_0.2.7.zip", "YARM_0.7.15.zip")
+
 	fmt.Printf("Starting server on: %s:%s", config.ServerIP, config.ServerPort)
 	log.Fatal(http.ListenAndServe(config.ServerIP+":"+config.ServerPort, router))
 }

src/mods.go

@@ -62,8 +62,8 @@ func rmMod(modName string) error {
 				log.Printf("Error removing mod %s: %s", mod, err)
 				return err
 			}
-			log.Printf("Removed mod: %s", mod)
 			removed = true
+			log.Printf("Removed mod: %s", mod)
 		}
 	}
 
@@ -132,6 +132,21 @@ func createModPack(title string, mods ...string) error {
 	return nil
 }
 
+func listModPacks(modDir string) ([]string, error) {
+	result := []string{}
+
+	files, err := ioutil.ReadDir(modDir)
+	if err != nil {
+		log.Printf("Error listing modpacks: %s", err)
+		return result, err
+	}
+	for _, f := range files {
+		result = append(result, f.Name())
+	}
+
+	return result, nil
+}
+
 // Parses mod-list.json file in factorio/mods
 // returns ModList struct
 func parseModList() (ModList, error) {

src/routes.go

@@ -199,5 +199,10 @@ var apiRoutes = Routes{
 		"POST",
 		"/user/remove",
 		RemoveUser,
+	}, {
+		"ListModPacks",
+		"GET",
+		"/mods/packs/list",
+		ListModPacks,
 	},
 }