Compare commits
4 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
7422bcd737 | ||
|
|
93b73d7bea | ||
|
|
110d8c073d | ||
|
|
2fa2b48611 |
2
go.mod
2
go.mod
@@ -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
2
go.sum
Normal 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
util.go
36
util.go
@@ -9,9 +9,25 @@ import (
|
|||||||
"path/filepath"
|
"path/filepath"
|
||||||
"slices"
|
"slices"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
|
"golang.org/x/net/proxy"
|
||||||
)
|
)
|
||||||
|
|
||||||
var Debug = false
|
var Debug = false
|
||||||
|
var Tor = false
|
||||||
|
var DownThreads = 8
|
||||||
|
|
||||||
|
func getHttpClient() (*http.Client, error) {
|
||||||
|
if Tor {
|
||||||
|
dialer, err := proxy.SOCKS5("tcp", "127.0.0.1:9050", 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"
|
||||||
@@ -27,7 +43,11 @@ 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
|
||||||
}
|
}
|
||||||
@@ -57,7 +77,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
|
||||||
}
|
}
|
||||||
@@ -109,7 +133,6 @@ func Download(manifest Manifest, downloadDir string, filen string) ([]byte, erro
|
|||||||
trackFrag := make([]Frag, 0)
|
trackFrag := make([]Frag, 0)
|
||||||
|
|
||||||
// start downloading chunks
|
// start downloading chunks
|
||||||
//currentOffset := 0
|
|
||||||
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 {
|
for _, fragment := range manifest.Fragments {
|
||||||
@@ -143,6 +166,7 @@ func Download(manifest Manifest, downloadDir string, filen string) ([]byte, erro
|
|||||||
|
|
||||||
// check for gaps in trackFrag and fill with remainder content
|
// check for gaps in trackFrag and fill with remainder content
|
||||||
curPos := 0
|
curPos := 0
|
||||||
|
|
||||||
for _, frag := range trackFrag {
|
for _, frag := range trackFrag {
|
||||||
if frag.Offset > curPos {
|
if frag.Offset > curPos {
|
||||||
// gap detected
|
// gap detected
|
||||||
@@ -164,7 +188,11 @@ func Download(manifest Manifest, downloadDir string, filen string) ([]byte, erro
|
|||||||
}
|
}
|
||||||
|
|
||||||
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
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user