ArmA-3-web-server/backend/internal/services/steamcmd.go

package services

import (
	"context"
	"fmt"
	"os"
	"os/exec"
	"path/filepath"
	"sync/atomic"
	"time"
)

func CheckWorkshopMod(serverfileDir, modID string) bool {
	if modID == "" {
		return false
	}
	p := filepath.Join(serverfileDir, "steamapps", "workshop", "content", "107410", modID)
	fi, err := os.Stat(p)
	return err == nil && fi.IsDir()
}

const arma3AppID = "233780"

type SteamCmdManager struct {
	serverfileDir string
	streamer      *LogStreamer
	running       atomic.Bool
	cancel        context.CancelFunc
}

func NewSteamCmdManager(serverfileDir string, streamer *LogStreamer) *SteamCmdManager {
	return &SteamCmdManager{
		serverfileDir: serverfileDir,
		streamer:      streamer,
	}
}

func (s *SteamCmdManager) IsRunning() bool {
	return s.running.Load()
}

func (s *SteamCmdManager) UpdateGame(branch, user string) error {
	if !s.running.CompareAndSwap(false, true) {
		return fmt.Errorf("steamcmd already running")
	}

	args := []string{
		"+force_install_dir", s.serverfileDir,
		"+login", user,
		"+app_update", arma3AppID,
	}
	if branch != "" && branch != "stable" {
		args = append(args, "-beta", branch)
	}
	args = append(args, "validate", "+quit")

	return s.run("Game update", args)
}

func (s *SteamCmdManager) DownloadMod(modID string) error {
	return s.DownloadMods([]string{modID})
}

func (s *SteamCmdManager) DownloadMods(modIDs []string) error {
	if !s.running.CompareAndSwap(false, true) {
		return fmt.Errorf("steamcmd already running")
	}

	if len(modIDs) == 0 {
		return fmt.Errorf("no mod ids provided")
	}

	args := []string{
		"+force_install_dir", s.serverfileDir,
		"+login", "anonymous",
	}
	for _, id := range modIDs {
		args = append(args, "+workshop_download_item", "107410", id)
	}
	args = append(args, "+quit")

	label := fmt.Sprintf("Download %d mod(s)", len(modIDs))
	return s.run(label, args)
}

const steamcmdTimeout = 10 * time.Minute

func (s *SteamCmdManager) run(label string, args []string) error {
	ctx, cancel := context.WithTimeout(context.Background(), steamcmdTimeout)

	s.streamer.Broadcast("steamcmd", "[STEAMCMD] "+label+" starting...")

	steamcmdPath := os.Getenv("STEAMCMD_PATH")
	if steamcmdPath == "" {
		steamcmdPath = "steamcmd"
	}
	cmd := exec.CommandContext(ctx, steamcmdPath, args...)

	stdout, err := cmd.StdoutPipe()
	if err != nil {
		cancel()
		s.running.Store(false)
		return fmt.Errorf("stdout pipe: %w", err)
	}
	stderr, err := cmd.StderrPipe()
	if err != nil {
		cancel()
		s.running.Store(false)
		return fmt.Errorf("stderr pipe: %w", err)
	}

	if err := cmd.Start(); err != nil {
		cancel()
		s.running.Store(false)
		return fmt.Errorf("start steamcmd: %w", err)
	}

	s.cancel = cancel

	go s.streamer.Stream("steamcmd", stdout, "")
	go s.streamer.Stream("steamcmd", stderr, "")

	go func() {
		err := cmd.Wait()
		s.running.Store(false)
		s.cancel = nil
		if err == nil {
			s.streamer.Broadcast("steamcmd", "[STEAMCMD] SUCCESS: "+label+" finished")
		} else {
			s.streamer.Broadcast("steamcmd", "[STEAMCMD] ERROR: "+label+" failed: "+err.Error())
		}
	}()

	return nil
}