Run Auto Mod when submitting for rank

Author: Swan

files/zip.go

@@ -0,0 +1,61 @@
+package files
+
+import (
+	"archive/zip"
+	"fmt"
+	"io"
+	"os"
+	"path/filepath"
+	"strings"
+)
+
+// UnzipArchive Unzips an archive to an output path
+func UnzipArchive(zipPath string, outputPath string) error {
+	archive, err := zip.OpenReader(zipPath)
+
+	if err != nil {
+		return err
+	}
+
+	defer archive.Close()
+
+	for _, f := range archive.File {
+		filePath := filepath.Join(outputPath, f.Name)
+
+		if !strings.HasPrefix(filePath, filepath.Clean(outputPath)+string(os.PathSeparator)) {
+			return fmt.Errorf("invalid file path for: %v", filePath)
+		}
+
+		if f.FileInfo().IsDir() {
+			if err := os.MkdirAll(filePath, os.ModePerm); err != nil {
+				return err
+			}
+			continue
+		}
+
+		if err := os.MkdirAll(filepath.Dir(filePath), os.ModePerm); err != nil {
+			return err
+		}
+
+		dstFile, err := os.OpenFile(filePath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())
+
+		if err != nil {
+			return err
+		}
+
+		fileInArchive, err := f.Open()
+
+		if err != nil {
+			return err
+		}
+
+		if _, err := io.Copy(dstFile, fileInArchive); err != nil {
+			return err
+		}
+
+		dstFile.Close()
+		fileInArchive.Close()
+	}
+
+	return nil
+}

handlers/ranking.go

@@ -5,10 +5,14 @@ import (
 	"github.com/Quaver/api2/config"
 	"github.com/Quaver/api2/db"
 	"github.com/Quaver/api2/enums"
+	"github.com/Quaver/api2/files"
+	"github.com/Quaver/api2/tools"
 	"github.com/Quaver/api2/webhooks"
 	"github.com/gin-gonic/gin"
 	"gorm.io/gorm"
 	"net/http"
+	"os"
+	"path/filepath"
 	"strconv"
 	"time"
 )
@@ -120,13 +124,51 @@ func SubmitMapsetToRankingQueue(c *gin.Context) *APIError {
 		return APIErrorForbidden("You cannot submit any more mapsets to the queue at this time.")
 	}
 
+	automodOk, err := downloadAndRunAutomod(mapset)
+
+	if err != nil {
+		return APIErrorServerError("Error running automod", err)
+	}
+
+	if !automodOk {
+		return APIErrorBadRequest("Your mapset has Auto Mod issues. Please fix them, then try again.")
+	}
+
 	if existingQueueMapset == nil {
 		return addMapsetToRankingQueue(c, mapset)
 	} else {
 		return resubmitMapsetToRankingQueue(c, existingQueueMapset)
 	}
 }
 
+// Runs the automod and returns an error if the mapset is good
+func downloadAndRunAutomod(mapset *db.Mapset) (bool, error) {
+	archivePath, err := files.CacheMapset(mapset)
+
+	if err != nil {
+		return false, err
+	}
+
+	output, _ := filepath.Abs(fmt.Sprintf("%v/%v", files.GetTempDirectory(), time.Now().UnixMilli()))
+
+	if err := files.UnzipArchive(archivePath, output); err != nil {
+		return false, err
+	}
+
+	automod, err := tools.RunAutoMod(output)
+
+	if err != nil {
+		return false, err
+	}
+
+	if automod.HasIssues {
+		return false, nil
+	}
+
+	_ = os.Remove(output)
+	return true, nil
+}
+
 // RemoveFromRankingQueue Allows a user to remove their mapset from the ranking queue (self deny)
 // Endpoint: POST /v2/ranking/queue/:id/remove
 func RemoveFromRankingQueue(c *gin.Context) *APIError {

tools/automod.go

@@ -0,0 +1,39 @@
+package tools
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"github.com/Quaver/api2/config"
+	"os/exec"
+)
+
+type AutoMod struct {
+	HasIssues bool `json:"HasIssues"`
+}
+
+func RunAutoMod(directory string) (*AutoMod, error) {
+	cmd := exec.Command(
+		config.Instance.QuaverToolsPath,
+		"-automod",
+		directory)
+
+	var out bytes.Buffer
+	var stderr bytes.Buffer
+	cmd.Stdout = &out
+	cmd.Stderr = &stderr
+
+	err := cmd.Run()
+
+	if err != nil {
+		return nil, fmt.Errorf("%v\n\n```%v```", err, stderr.String())
+	}
+
+	var automod AutoMod
+
+	if err := json.Unmarshal(out.Bytes(), &automod); err != nil {
+		return nil, err
+	}
+
+	return &automod, nil
+}