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
7 changed files with 275 additions and 32 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 ./...
```

98
cmd/main.go Normal file
View File

@@ -0,0 +1,98 @@
package main
import (
"bytes"
"encoding/json"
"fmt"
"os"
"path/filepath"
"gitea.tbdevent.eu/ilbinek/reforger_utils"
)
const DEBUG = true
func main() {
addonID := "BADC0DEDABBEDA5E"
version := "0.12.4573"
assetsReply, err := reforger_utils.DoAssetsRequest(addonID, version)
if err != nil {
panic(err)
}
downloadDir := filepath.Join("downloads", addonID)
if DEBUG {
if err := os.MkdirAll(downloadDir, os.ModePerm); err != nil {
panic(err)
}
}
bytesReply, err := json.MarshalIndent(assetsReply, "", " ")
if err != nil {
panic(err)
}
if DEBUG {
if err := os.WriteFile(filepath.Join(downloadDir, "assets.json"), bytesReply, os.ModePerm); err != nil {
panic(err)
}
}
for _, asset := range assetsReply {
for _, file := range asset.Files {
manifest, err := reforger_utils.GetManifest(file.ManifestPath)
if err != nil {
panic(err)
}
if DEBUG {
// save manifest
var manifestBuf bytes.Buffer
manifestEncoder := json.NewEncoder(&manifestBuf)
manifestEncoder.SetIndent("", " ")
if err := manifestEncoder.Encode(manifest); err != nil {
panic(err)
}
if err := os.WriteFile(filepath.Join(downloadDir, file.Name+".manifest.json"), manifestBuf.Bytes(), os.ModePerm); err != nil {
panic(err)
}
}
downedFile, err := reforger_utils.Download(manifest, downloadDir, file.Name)
if err != nil {
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")
if file.Name[len(file.Name)-4:] != ".pak" {
continue
}
pakInfo, err := reforger_utils.GetPakFileInfo(&downedFile)
if err != nil {
panic(err)
}
for _, entry := range pakInfo {
fmt.Printf(" - %s (offset: %d, size: %d)\n", entry.Name, entry.Offset, entry.Size)
if DEBUG {
entryData := downedFile[entry.Offset : entry.Offset+entry.Size]
path := filepath.Join("downloads", addonID, "extracted", entry.Name)
if err := os.MkdirAll(filepath.Dir(path), os.ModePerm); err != nil {
panic(err)
}
if err := os.WriteFile(path, entryData, os.ModePerm); err != nil {
panic(err)
}
}
}
}
}
}

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
}

View File

@@ -61,11 +61,13 @@ type Manifest struct {
Size int `json:"size"` Size int `json:"size"`
Offsets []int `json:"offsets"` Offsets []int `json:"offsets"`
} `json:"remainder"` } `json:"remainder"`
Fragments []struct { Fragments []Fragment `json:"fragments"`
}
type Fragment struct {
Sha512 string `json:"sha512"` Sha512 string `json:"sha512"`
Size int `json:"size"` Size int `json:"size"`
Offsets []int `json:"offsets"` Offsets []int `json:"offsets"`
} `json:"fragments"`
} }
type RDBHeader struct { type RDBHeader struct {

81
util.go
View File

@@ -9,10 +9,25 @@ import (
"path/filepath" "path/filepath"
"slices" "slices"
"strings" "strings"
"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"
@@ -28,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)
} }
@@ -45,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
@@ -58,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
} }
@@ -71,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
@@ -96,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
@@ -112,28 +134,28 @@ 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
for _, fragment := range manifest.Fragments { lastLoggedPercent := float32(-1)
log := getLogger().With("file", filen)
log.Info("download started", "size_bytes", manifest.Size)
// Download all fragments serially to keep execution simple and deterministic.
for idx, fragment := range manifest.Fragments {
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 {
return nil, err return nil, err
} }
downloaded += len(content) * len(fragment.Offsets) downloaded += len(*content) * len(fragment.Offsets)
for _, offset := range fragment.Offsets { for _, offset := range fragment.Offsets {
trackFrag = append(trackFrag, Frag{Offset: offset, Size: len(content)}) trackFrag = append(trackFrag, Frag{Offset: offset, Size: len(*content)})
copy(ret[offset:offset+len(content)], content) copy(ret[offset:offset+len(*content)], *content)
} }
if Debug {
// save fragment to file in the fragments/ directory
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 {
return nil, err
}
}
percent := float32(downloaded) / float32(manifest.Size) * 100.0 percent := float32(downloaded) / float32(manifest.Size) * 100.0
downedMB := float32(downloaded) / 1024.0 / 1024.0 downedMB := float32(downloaded) / 1024.0 / 1024.0
fmt.Printf("\r\033[32mDownloaded\033[0m %fMB/%fMB (\033[36m%.2f%%\033[0m) of \033[33m%s\033[0m", downedMB, totalMB, percent, filen) if percent-lastLoggedPercent >= 5 || idx == len(manifest.Fragments)-1 {
log.Info("download progress", "downloaded_mb", downedMB, "total_mb", totalMB, "percent", percent)
lastLoggedPercent = percent
}
} }
// sort trackFrag by Offset // sort trackFrag by Offset
@@ -149,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:]
@@ -159,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)
} }
@@ -180,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 {