6 Commits

Author SHA1 Message Date
ilbinek
1fe78a76ab Add README with install and logging docs 2026-04-10 20:49:18 +02:00
ilbinek
79b92a994c Add structured container-friendly logging 2026-04-10 20:37:42 +02:00
ilbinek
f05dd6ee26 Simplify Download to single-threaded flow 2026-04-10 17:58:07 +02:00
ilbinek
f90cf628e3 Add configurable SOCKS5 address with default 2026-04-10 17:17:44 +02:00
Sotirios Pupakis
d453604e67 multithread 2025-09-18 23:48:52 +02:00
Sotirios Pupakis
7422bcd737 tried 2025-09-18 22:38:24 +02:00
6 changed files with 185 additions and 84 deletions

76
README.md Normal file
View File

@@ -0,0 +1,76 @@
# reforger_utils
Utilities for downloading and processing Arma Reforger workshop content.
## Install
This module is hosted on a private Gitea instance.
Set private module settings once:
```powershell
go env -w GOPRIVATE=gitea.tbdevent.eu
```
Then install a version:
```powershell
go get gitea.tbdevent.eu/ilbinek/reforger_utils@v0.0.15
```
## Features
- Download asset metadata and manifests from BI APIs.
- Download and rebuild fragmented files.
- Parse PAK file entries.
- Optional SOCKS5/Tor support for HTTP requests.
## Basic Usage
```go
package main
import (
"fmt"
"gitea.tbdevent.eu/ilbinek/reforger_utils"
)
func main() {
assets, err := reforger_utils.DoAssetsRequest("BADC0DEDABBEDA5E", "0.12.4573")
if err != nil {
panic(err)
}
fmt.Printf("found %d assets\n", len(assets))
}
```
## Logging
The library uses structured slog logging and writes JSON logs to stdout by default.
This is container-friendly (including Dokploy) because logs are newline-based and do not overwrite previous output.
You can inject your own logger:
```go
package main
import (
"log/slog"
"os"
"gitea.tbdevent.eu/ilbinek/reforger_utils"
)
func main() {
logger := slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{Level: slog.LevelInfo}))
reforger_utils.SetLogger(logger)
}
```
## Development
```powershell
go test ./...
```

View File

@@ -10,11 +10,11 @@ import (
"gitea.tbdevent.eu/ilbinek/reforger_utils" "gitea.tbdevent.eu/ilbinek/reforger_utils"
) )
const DEBUG = false const DEBUG = true
func main() { func main() {
addonID := "61EB463A41FE5274" addonID := "BADC0DEDABBEDA5E"
version := "0.1.166" version := "0.12.4573"
assetsReply, err := reforger_utils.DoAssetsRequest(addonID, version) assetsReply, err := reforger_utils.DoAssetsRequest(addonID, version)
if err != nil { if err != nil {
@@ -65,6 +65,10 @@ func main() {
panic(err) panic(err)
} }
if err := os.WriteFile(filepath.Join(downloadDir, file.Name), downedFile, os.ModePerm); err != nil {
panic(err)
}
fmt.Println("Downloaded", file.Name, len(downedFile), "bytes") fmt.Println("Downloaded", file.Name, len(downedFile), "bytes")
if file.Name[len(file.Name)-4:] != ".pak" { if file.Name[len(file.Name)-4:] != ".pak" {

2
go.mod
View File

@@ -1,3 +1,5 @@
module gitea.tbdevent.eu/ilbinek/reforger_utils module gitea.tbdevent.eu/ilbinek/reforger_utils
go 1.25.0 go 1.25.0
require golang.org/x/net v0.44.0 // indirect

2
go.sum Normal file
View File

@@ -0,0 +1,2 @@
golang.org/x/net v0.44.0 h1:evd8IRDyfNBMBTTY5XRF1vaZlD+EmWx6x8PkhR04H/I=
golang.org/x/net v0.44.0/go.mod h1:ECOoLqd5U3Lhyeyo/QDCEVQ4sNgYsqvCZ722XogGieY=

36
logger.go Normal file
View File

@@ -0,0 +1,36 @@
package reforger_utils
import (
"log/slog"
"os"
"sync"
)
var (
loggerMu sync.RWMutex
logger = newDefaultLogger()
)
func newDefaultLogger() *slog.Logger {
return slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{Level: slog.LevelInfo}))
}
// SetLogger configures the package logger.
// Passing nil resets it to the default JSON logger writing to stdout.
func SetLogger(l *slog.Logger) {
loggerMu.Lock()
defer loggerMu.Unlock()
if l == nil {
logger = newDefaultLogger()
return
}
logger = l
}
func getLogger() *slog.Logger {
loggerMu.RLock()
defer loggerMu.RUnlock()
return logger
}

127
util.go
View File

@@ -9,11 +9,25 @@ import (
"path/filepath" "path/filepath"
"slices" "slices"
"strings" "strings"
"sync"
"golang.org/x/net/proxy"
) )
var Debug = false var Debug = false
var DownThreads = 8 var Tor = false
var Socks5Address = "127.0.0.1:9050"
func getHttpClient() (*http.Client, error) {
if Tor {
dialer, err := proxy.SOCKS5("tcp", Socks5Address, nil, proxy.Direct)
if err != nil {
return nil, err
}
transport := &http.Transport{Dial: dialer.Dial}
return &http.Client{Transport: transport}, nil
}
return http.DefaultClient, nil
}
func DoAssetsRequest(addonID, version string) (AssetsReply, error) { func DoAssetsRequest(addonID, version string) (AssetsReply, error) {
url := "https://api-ar-workshop.bistudio.com/workshop-api/api/v3.0/s2s/assets/download-list" url := "https://api-ar-workshop.bistudio.com/workshop-api/api/v3.0/s2s/assets/download-list"
@@ -29,16 +43,19 @@ func DoAssetsRequest(addonID, version string) (AssetsReply, error) {
req.Header.Add("user-agent", "Arma Reforger/1.4.0.53 (Headless; Windows)") req.Header.Add("user-agent", "Arma Reforger/1.4.0.53 (Headless; Windows)")
req.Header.Add("content-length", fmt.Sprintf("%d", len(body))) req.Header.Add("content-length", fmt.Sprintf("%d", len(body)))
res, err := http.DefaultClient.Do(req) client, err := getHttpClient()
if err != nil {
return AssetsReply{}, err
}
res, err := client.Do(req)
if err != nil { if err != nil {
return AssetsReply{}, err return AssetsReply{}, err
} }
defer res.Body.Close() defer res.Body.Close()
if res.StatusCode != 200 { if res.StatusCode != 200 {
// print entire body
body, _ := io.ReadAll(res.Body) body, _ := io.ReadAll(res.Body)
fmt.Println(string(body)) getLogger().Error("assets request failed", "status", res.Status, "addon_id", addonID, "version", version, "body", string(body))
return AssetsReply{}, fmt.Errorf("bad status: %s", res.Status) return AssetsReply{}, fmt.Errorf("bad status: %s", res.Status)
} }
@@ -46,7 +63,7 @@ func DoAssetsRequest(addonID, version string) (AssetsReply, error) {
decoder := json.NewDecoder(res.Body) decoder := json.NewDecoder(res.Body)
var assetsReply AssetsReply var assetsReply AssetsReply
if err := decoder.Decode(&assetsReply); err != nil { if err := decoder.Decode(&assetsReply); err != nil {
panic(err) return AssetsReply{}, fmt.Errorf("failed to decode assets response: %w", err)
} }
return assetsReply, nil return assetsReply, nil
@@ -59,7 +76,11 @@ func GetManifest(manifestUrl string) (Manifest, error) {
return Manifest{}, err return Manifest{}, err
} }
res, err := http.DefaultClient.Do(req) client, err := getHttpClient()
if err != nil {
return Manifest{}, err
}
res, err := client.Do(req)
if err != nil { if err != nil {
return Manifest{}, err return Manifest{}, err
} }
@@ -72,7 +93,7 @@ func GetManifest(manifestUrl string) (Manifest, error) {
// Decode the JSON response // Decode the JSON response
var manifest Manifest var manifest Manifest
if err := json.NewDecoder(res.Body).Decode(&manifest); err != nil { if err := json.NewDecoder(res.Body).Decode(&manifest); err != nil {
panic(err) return Manifest{}, fmt.Errorf("failed to decode manifest response: %w", err)
} }
return manifest, nil return manifest, nil
@@ -97,8 +118,8 @@ func Download(manifest Manifest, downloadDir string, filen string) ([]byte, erro
if err != nil { if err != nil {
return nil, err return nil, err
} }
remContent = content remContent = *content
downloaded += len(content) downloaded += len(*content)
if Debug { if Debug {
// save remainder to file in the fragments/ directory, prefix with gap // save remainder to file in the fragments/ directory, prefix with gap
@@ -113,74 +134,29 @@ func Download(manifest Manifest, downloadDir string, filen string) ([]byte, erro
// start downloading chunks // start downloading chunks
ret := make([]byte, manifest.Size) ret := make([]byte, manifest.Size)
totalMB := float32(manifest.Size) / 1024.0 / 1024.0 totalMB := float32(manifest.Size) / 1024.0 / 1024.0
lastLoggedPercent := float32(-1)
log := getLogger().With("file", filen)
log.Info("download started", "size_bytes", manifest.Size)
var wg sync.WaitGroup // Download all fragments serially to keep execution simple and deterministic.
var mu sync.Mutex for idx, fragment := range manifest.Fragments {
fragmentsChan := make(chan Fragment, len(manifest.Fragments))
errChan := make(chan error, DownThreads)
// Start worker goroutines
for i := 0; i < DownThreads; i++ {
wg.Go(func() {
for fragment := range fragmentsChan {
url := transformShaToURL(fragment.Sha512, fragment.Size) url := transformShaToURL(fragment.Sha512, fragment.Size)
content, err := getContent(url) content, err := getContent(url)
if err != nil { if err != nil {
// Handle error, maybe by sending it to an error channel return nil, err
// For now, we'll just skip and print.
fmt.Printf("Error downloading fragment %s: %v\n", fragment.Sha512, err)
errChan <- err
} }
downloaded += len(*content) * len(fragment.Offsets)
mu.Lock()
// Safely write to the shared slice and update progress
for _, offset := range fragment.Offsets { for _, offset := range fragment.Offsets {
copy(ret[offset:offset+len(content)], content) trackFrag = append(trackFrag, Frag{Offset: offset, Size: len(*content)})
copy(ret[offset:offset+len(*content)], *content)
} }
downloaded += len(content) * len(fragment.Offsets)
// Track downloaded fragments
for _, offset := range fragment.Offsets {
trackFrag = append(trackFrag, Frag{
Offset: offset,
Size: len(content),
})
}
// Print progress
downedMB := float32(downloaded) / 1024.0 / 1024.0
percent := float32(downloaded) / float32(manifest.Size) * 100.0 percent := float32(downloaded) / float32(manifest.Size) * 100.0
fmt.Printf("\r\033[32mDownloaded\033[0m %.2fMB/%.2fMB (\033[36m%.2f%%\033[0m) of \033[33m%s\033[0m", downedMB, totalMB, percent, filen) downedMB := float32(downloaded) / 1024.0 / 1024.0
if percent-lastLoggedPercent >= 5 || idx == len(manifest.Fragments)-1 {
mu.Unlock() log.Info("download progress", "downloaded_mb", downedMB, "total_mb", totalMB, "percent", percent)
lastLoggedPercent = percent
// The Debug section can be safely moved here as well
if os.Getenv("DEBUG") == "true" { // Using env var for a cleaner check
mu.Lock() // Re-lock for file write
if err := os.WriteFile(filepath.Join(downloadDir, "fragments", filen, fmt.Sprintf("%d.%s.%d.bytes", fragment.Offsets[0], fragment.Sha512, fragment.Size)), content, os.ModePerm); err != nil {
fmt.Printf("Error saving file: %v\n", err)
}
mu.Unlock()
} }
} }
})
}
// Send fragments to the channel
for _, fragment := range manifest.Fragments {
fragmentsChan <- fragment
}
close(fragmentsChan) // Close the channel to signal workers no more tasks are coming
// Wait for all workers to finish
wg.Wait()
// Check for errors
if len(errChan) > 0 {
return nil, <-errChan // Return the first error encountered
}
close(errChan)
// sort trackFrag by Offset // sort trackFrag by Offset
slices.SortFunc(trackFrag, func(a, b Frag) int { slices.SortFunc(trackFrag, func(a, b Frag) int {
@@ -195,7 +171,7 @@ func Download(manifest Manifest, downloadDir string, filen string) ([]byte, erro
// gap detected // gap detected
gapSize := frag.Offset - curPos gapSize := frag.Offset - curPos
if currentRem+gapSize > len(remContent) { if currentRem+gapSize > len(remContent) {
panic("not enough remainder content to fill gap") return nil, fmt.Errorf("not enough remainder content to fill gap in %s", filen)
} }
copy(ret[curPos:curPos+gapSize], remContent[currentRem:currentRem+gapSize]) copy(ret[curPos:curPos+gapSize], remContent[currentRem:currentRem+gapSize])
remContent = remContent[gapSize:] remContent = remContent[gapSize:]
@@ -205,19 +181,24 @@ func Download(manifest Manifest, downloadDir string, filen string) ([]byte, erro
// Append rest of remainder content // Append rest of remainder content
copy(ret[curPos:], remContent) copy(ret[curPos:], remContent)
fmt.Println() log.Info("download completed", "size_bytes", len(ret))
return ret, nil return ret, nil
} }
func getContent(url string) ([]byte, error) { func getContent(url string) (*[]byte, error) {
resp, err := http.Get(url) client, err := getHttpClient()
if err != nil {
return nil, err
}
resp, err := client.Get(url)
if err != nil { if err != nil {
return nil, err return nil, err
} }
defer resp.Body.Close() defer resp.Body.Close()
if resp.StatusCode != 200 { if resp.StatusCode != 200 {
getLogger().Error("content request failed", "status", resp.Status, "url", url)
return nil, fmt.Errorf("bad status: %s", resp.Status) return nil, fmt.Errorf("bad status: %s", resp.Status)
} }
@@ -226,7 +207,7 @@ func getContent(url string) ([]byte, error) {
return nil, err return nil, err
} }
return body, nil return &body, nil
} }
func transformShaToURL(sha512s string, size int) string { func transformShaToURL(sha512s string, size int) string {