use channel for status messages
This commit is contained in:
@@ -10,142 +10,158 @@ import (
|
|||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"regexp"
|
"regexp"
|
||||||
|
"slices"
|
||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
type UpdateStats struct {
|
type UpdateStats struct {
|
||||||
Updated int
|
Updated int
|
||||||
Current int
|
Current int
|
||||||
Outdated int
|
Outdated int
|
||||||
LogMessages []string
|
Error error
|
||||||
Error error
|
MessageBuf []string
|
||||||
}
|
}
|
||||||
|
|
||||||
func Update(wowdir string, force bool, removeUnknown bool, skipDownload bool) UpdateStats {
|
func Update(wowdir string, force bool, removeUnknown bool, skipDownload bool) UpdateStats {
|
||||||
stats := UpdateStats{
|
stats := UpdateStats{
|
||||||
LogMessages: make([]string, 0),
|
Error: nil,
|
||||||
Error: nil,
|
MessageBuf: make([]string, 0),
|
||||||
}
|
}
|
||||||
|
|
||||||
manifest, err := GetManifest()
|
msgChan := make(chan string)
|
||||||
if err != nil {
|
done := make(chan bool)
|
||||||
stats.Error = fmt.Errorf("Failed to get manifest: %v\n", err)
|
|
||||||
return stats
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, file := range manifest.Files {
|
go func() {
|
||||||
path := strings.ReplaceAll(file.Path, `\`, `/`)
|
manifest, err := GetManifest()
|
||||||
path = strings.TrimLeft(path, `\`)
|
if err != nil {
|
||||||
|
stats.Error = fmt.Errorf("Failed to get manifest: %v\n", err)
|
||||||
localPath := filepath.Join(wowdir, path)
|
done <- true
|
||||||
localDir := filepath.Dir(localPath)
|
return
|
||||||
if _, err = os.Stat(localDir); os.IsNotExist(err) {
|
|
||||||
err = os.MkdirAll(localDir, 0755)
|
|
||||||
if err != nil {
|
|
||||||
stats.Error = fmt.Errorf("failed to create directory %s: %v", localDir, err)
|
|
||||||
return stats
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if !force {
|
|
||||||
if _, err = os.Stat(localPath); err == nil {
|
|
||||||
data, err := os.ReadFile(localPath)
|
|
||||||
if err != nil {
|
|
||||||
stats.Error = fmt.Errorf("failed to read %s: %v", localPath, err)
|
|
||||||
return stats
|
|
||||||
}
|
|
||||||
hashBytes := md5.Sum(data)
|
|
||||||
hash := hex.EncodeToString(hashBytes[:])
|
|
||||||
if hash == file.Hash {
|
|
||||||
stats.LogMessages = append(stats.LogMessages, fmt.Sprintf("File %s is up to date", localPath))
|
|
||||||
stats.Current += 1
|
|
||||||
continue
|
|
||||||
} else {
|
|
||||||
stats.Outdated += 1
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if !skipDownload {
|
|
||||||
outFile, err := os.Create(localPath)
|
|
||||||
if err != nil {
|
|
||||||
stats.Error = fmt.Errorf("failed to create file %s: %v", localPath, err)
|
|
||||||
return stats
|
|
||||||
}
|
|
||||||
|
|
||||||
downloadSuccess := false
|
|
||||||
for _, url := range []string{file.Urls.Cloudflare, file.Urls.Digitalocean, file.Urls.None} {
|
|
||||||
resp, err := http.Get(url)
|
|
||||||
if err != nil {
|
|
||||||
if resp != nil {
|
|
||||||
resp.Body.Close()
|
|
||||||
}
|
|
||||||
stats.LogMessages = append(stats.LogMessages, fmt.Sprintf("Failed to download %s: %v", url, err))
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if resp.StatusCode != http.StatusOK {
|
|
||||||
resp.Body.Close()
|
|
||||||
stats.LogMessages = append(stats.LogMessages, fmt.Sprintf("HTTP Status %d", resp.StatusCode))
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
_, err = io.Copy(outFile, resp.Body)
|
|
||||||
if err != nil {
|
|
||||||
stats.LogMessages = append(stats.LogMessages, fmt.Sprintf("Failed to write file %s: %v", localPath, err))
|
|
||||||
resp.Body.Close()
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
resp.Body.Close()
|
|
||||||
downloadSuccess = true
|
|
||||||
stats.LogMessages = append(stats.LogMessages, fmt.Sprintf("Successfully downloaded %s", localPath))
|
|
||||||
break
|
|
||||||
}
|
|
||||||
|
|
||||||
outFile.Close()
|
|
||||||
if !downloadSuccess {
|
|
||||||
stats.Error = fmt.Errorf("Failed to download updates, see above messages")
|
|
||||||
return stats
|
|
||||||
}
|
|
||||||
|
|
||||||
stats.Updated += 1
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if removeUnknown {
|
|
||||||
patches := make([]string, 0)
|
|
||||||
patchreg := regexp.MustCompile(`patch-[A-Za-z].MPQ`)
|
|
||||||
|
|
||||||
for _, file := range manifest.Files {
|
for _, file := range manifest.Files {
|
||||||
if patchreg.MatchString(file.Path) {
|
path := strings.ReplaceAll(file.Path, `\`, `/`)
|
||||||
patches = append(patches, strings.Split(file.Path, "Data\\")[1])
|
path = strings.TrimLeft(path, `\`)
|
||||||
|
|
||||||
|
localPath := filepath.Join(wowdir, path)
|
||||||
|
localDir := filepath.Dir(localPath)
|
||||||
|
if _, err = os.Stat(localDir); os.IsNotExist(err) {
|
||||||
|
err = os.MkdirAll(localDir, 0755)
|
||||||
|
if err != nil {
|
||||||
|
stats.Error = fmt.Errorf("failed to create directory %s: %v", localDir, err)
|
||||||
|
done <- true
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !force {
|
||||||
|
if _, err = os.Stat(localPath); err == nil {
|
||||||
|
data, err := os.ReadFile(localPath)
|
||||||
|
if err != nil {
|
||||||
|
stats.Error = fmt.Errorf("failed to read %s: %v", localPath, err)
|
||||||
|
done <- true
|
||||||
|
return
|
||||||
|
}
|
||||||
|
hashBytes := md5.Sum(data)
|
||||||
|
hash := hex.EncodeToString(hashBytes[:])
|
||||||
|
if hash == file.Hash {
|
||||||
|
msgChan <- fmt.Sprintf("File %s is up to date", localPath)
|
||||||
|
stats.Current += 1
|
||||||
|
continue
|
||||||
|
} else {
|
||||||
|
stats.Outdated += 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !skipDownload {
|
||||||
|
outFile, err := os.Create(localPath)
|
||||||
|
if err != nil {
|
||||||
|
stats.Error = fmt.Errorf("failed to create file %s: %v", localPath, err)
|
||||||
|
done <- true
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
downloadSuccess := false
|
||||||
|
for _, url := range []string{file.Urls.Cloudflare, file.Urls.Digitalocean, file.Urls.None} {
|
||||||
|
resp, err := http.Get(url)
|
||||||
|
if err != nil {
|
||||||
|
if resp != nil {
|
||||||
|
resp.Body.Close()
|
||||||
|
}
|
||||||
|
msgChan <- fmt.Sprintf("Failed to download %s: %v", url, err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if resp.StatusCode != http.StatusOK {
|
||||||
|
resp.Body.Close()
|
||||||
|
msgChan <- fmt.Sprintf("HTTP Status %d", resp.StatusCode)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = io.Copy(outFile, resp.Body)
|
||||||
|
if err != nil {
|
||||||
|
msgChan <- fmt.Sprintf("Failed to write file %s: %v", localPath, err)
|
||||||
|
resp.Body.Close()
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
resp.Body.Close()
|
||||||
|
downloadSuccess = true
|
||||||
|
msgChan <- fmt.Sprintf("Successfully downloaded %s", localPath)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
outFile.Close()
|
||||||
|
if !downloadSuccess {
|
||||||
|
stats.Error = fmt.Errorf("Failed to download updates, see above messages")
|
||||||
|
done <- true
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
stats.Updated += 1
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
err = filepath.WalkDir(filepath.Join(wowdir, "Data"), func(path string, d fs.DirEntry, err error) error {
|
if removeUnknown {
|
||||||
if !d.IsDir() && patchreg.MatchString(d.Name()) {
|
patches := make([]string, 0)
|
||||||
del := true
|
patchreg := regexp.MustCompile(`patch-[A-Za-z].MPQ`)
|
||||||
for _, patch := range patches {
|
|
||||||
if patch == d.Name() {
|
for _, file := range manifest.Files {
|
||||||
del = false
|
if patchreg.MatchString(file.Path) {
|
||||||
break
|
patches = append(patches, strings.Split(file.Path, "Data\\")[1])
|
||||||
}
|
|
||||||
}
|
|
||||||
if del {
|
|
||||||
err = os.Remove(path)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
stats.LogMessages = append(stats.LogMessages, fmt.Sprintf("Removed unknown patch %s", d.Name()))
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return nil
|
|
||||||
})
|
|
||||||
|
|
||||||
if err != nil {
|
err = filepath.WalkDir(filepath.Join(wowdir, "Data"), func(path string, d fs.DirEntry, err error) error {
|
||||||
stats.Error = fmt.Errorf("failed to delete unknown patches: %v", err)
|
if !d.IsDir() && patchreg.MatchString(d.Name()) {
|
||||||
|
del := true
|
||||||
|
if slices.Contains(patches, d.Name()) {
|
||||||
|
del = false
|
||||||
|
}
|
||||||
|
if del {
|
||||||
|
err = os.Remove(path)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
msgChan <- fmt.Sprintf("Removed unknown patch %s", d.Name())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
stats.Error = fmt.Errorf("failed to delete unknown patches: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
done <- true
|
||||||
|
}()
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case msg := <-msgChan:
|
||||||
|
fmt.Println(msg)
|
||||||
|
stats.MessageBuf = append(stats.MessageBuf, msg)
|
||||||
|
case <-done:
|
||||||
|
return stats
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return stats
|
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user