This commit is contained in:
Evan Burkey 2025-04-07 14:04:30 -07:00
parent f197492cf4
commit b3014466ac
6 changed files with 123 additions and 20 deletions

2
.gitignore vendored
View File

@ -1,4 +1,4 @@
WoWChatLog.txt WoWChatLog.txt
.idea .idea
config.toml config.toml
testloot.csv *.csv

View File

@ -27,8 +27,8 @@ func gambo() {
} }
for _, game := range games { for _, game := range games {
//err = uploadGambo(game, "https://forcek.in/game") err = uploadGambo(game, "https://forcek.in/game")
err = uploadGamboTest(game) //err = uploadGamboTest(game)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }

116
loot.go
View File

@ -7,6 +7,7 @@ import (
"errors" "errors"
"fmt" "fmt"
"github.com/sqweek/dialog" "github.com/sqweek/dialog"
"io"
"log" "log"
"net/http" "net/http"
"os" "os"
@ -15,6 +16,11 @@ import (
"time" "time"
) )
type LootUpload struct {
Loot []LootRecord `json:"loot"`
Players []string `json:"players"`
}
type LootRecord struct { type LootRecord struct {
Name string `json:"player"` Name string `json:"player"`
ItemID int `json:"item_id"` ItemID int `json:"item_id"`
@ -24,25 +30,62 @@ type LootRecord struct {
} }
func loot() { func loot() {
loots, err := parseLoot()
if err != nil {
log.Fatal(err)
}
fmt.Print("Enter the matching warcraft logs report id:")
var reportID string
_, err = fmt.Scanln(&reportID)
if err != nil {
log.Fatal(err)
}
players, err := getAttendance(reportID)
if err != nil {
log.Fatal(err)
}
upload := LootUpload{
Loot: loots,
Players: players,
}
err = uploadLoot(upload, "https://forcek.in/loot")
//err = uploadLootTest(upload)
if err != nil {
log.Fatal(err)
}
}
func parseLoot() ([]LootRecord, error) {
csvPath, err := dialog.File().Title("Select the csv file").Load() csvPath, err := dialog.File().Title("Select the csv file").Load()
if err != nil { if err != nil {
if errors.Is(err, dialog.ErrCancelled) { if errors.Is(err, dialog.ErrCancelled) {
log.Fatalf("Cancelled dialog box, exiting") return nil, fmt.Errorf("cancelled dialog box, exiting")
} else { } else {
log.Fatal(err) return nil, err
} }
} }
csvFile, err := os.Open(csvPath) csvFile, err := os.Open(csvPath)
if err != nil { if err != nil {
log.Fatal(err) return nil, err
} }
defer csvFile.Close() defer csvFile.Close()
csvReader := csv.NewReader(csvFile) data, err := io.ReadAll(csvFile)
if err != nil {
return nil, err
}
data = bytes.ReplaceAll(data, []byte("[\"Bullet-Proof\" Vestplate]"), []byte("[Bullet-Proof Vestplate]"))
csvReader := csv.NewReader(bytes.NewReader(data))
records, err := csvReader.ReadAll() records, err := csvReader.ReadAll()
if err != nil { if err != nil {
log.Fatalf("Failed to parse csv: %v\n", err) return nil, fmt.Errorf("failed to parse csv: %v\n", err)
} }
loots := make([]LootRecord, 0) loots := make([]LootRecord, 0)
@ -80,16 +123,61 @@ func loot() {
loots = append(loots, l) loots = append(loots, l)
} }
for _, l := range loots { return loots, nil
err = uploadLoot(l, "https://forcek.in/loot")
//err = uploadLootTest(l)
if err != nil {
log.Fatal(err)
}
}
} }
func uploadLoot(l LootRecord, endpoint string) error { type WarcraftLog struct {
Lang string `json:"lang"`
Fights any `json:"fights"`
CompleteRaids any `json:"complete_raids"`
Friendlies any `json:"friendlies"`
Enemies any `json:"enemies"`
FriendlyPets any `json:"friendly_pets"`
EnemyPets any `json:"enemyPets"`
LogVersion int `json:"logVersion"`
GameVersion int `json:"gameVersion"`
Phases any `json:"phases"`
Title string `json:"title"`
Owner string `json:"owner"`
Start int64 `json:"start"`
End int64 `json:"end"`
Zone int `json:"zone"`
ExportedCharacters []struct {
ID int `json:"id"`
Name string `json:"name"`
Server string `json:"server"`
Region string `json:"region"`
} `json:"exportedCharacters"`
}
func getAttendance(reportID string) ([]string, error) {
logUrl := fmt.Sprintf("https://www.warcraftlogs.com:443/v1/report/fights/%s?api_key=%s", reportID, config.WarcraftLogsApiKey)
resp, err := http.Get(logUrl)
if err != nil {
return nil, err
}
defer resp.Body.Close()
body, err := io.ReadAll(resp.Body)
if err != nil {
return nil, err
}
var wlog WarcraftLog
err = json.Unmarshal(body, &wlog)
if err != nil {
return nil, err
}
players := make([]string, 0)
for _, p := range wlog.ExportedCharacters {
players = append(players, p.Name)
}
return players, nil
}
func uploadLoot(l LootUpload, endpoint string) error {
marshalled, err := json.Marshal(l) marshalled, err := json.Marshal(l)
if err != nil { if err != nil {
return err return err
@ -113,6 +201,6 @@ func uploadLoot(l LootRecord, endpoint string) error {
return nil return nil
} }
func uploadLootTest(l LootRecord) error { func uploadLootTest(l LootUpload) error {
return uploadLoot(l, "http://localhost:3000/loot") return uploadLoot(l, "http://localhost:3000/loot")
} }

14
loot_test.go Normal file
View File

@ -0,0 +1,14 @@
package main
import "testing"
func TestWarcraftLogs(t *testing.T) {
setupConfig()
players, err := getAttendance("VFy1BaAJ2pjKG8N6")
if err != nil {
t.Fatal(err)
}
if len(players) != 17 {
t.Fatalf("Wrong number of players: %d", len(players))
}
}

View File

@ -12,6 +12,7 @@ import (
type Config struct { type Config struct {
Log string Log string
Apikey string Apikey string
WarcraftLogsApiKey string
} }
var config Config var config Config

View File

@ -26,7 +26,7 @@ type Game struct {
} }
var ( var (
reTimeStamp = regexp.MustCompile(`(\d/\d \d+:\d+:\d+.\d+)`) reTimeStamp = regexp.MustCompile(`(\d+/\d+ \d+:\d+:\d+\.\d+)`)
reGameStart = regexp.MustCompile(`WoWGoldGambler: A new game has been started`) reGameStart = regexp.MustCompile(`WoWGoldGambler: A new game has been started`)
reWager = regexp.MustCompile(`Game Mode - ([A-Z]+) - Wager - ([\d,]+)g$`) reWager = regexp.MustCompile(`Game Mode - ([A-Z]+) - Wager - ([\d,]+)g$`)
reSignup = regexp.MustCompile(`([\p{L}']+)-[\p{L}'0-9]+: (1|-1)`) reSignup = regexp.MustCompile(`([\p{L}']+)-[\p{L}'0-9]+: (1|-1)`)