86 lines
1.3 KiB
Go
86 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"io/fs"
|
|
|
|
"github.com/alexflint/go-arg"
|
|
|
|
"git.zorchenhimer.com/Zorchenhimer/go-studybox/script"
|
|
)
|
|
|
|
type Arguments struct {
|
|
BaseDir string `arg:"positional,required"`
|
|
Output string `arg:"positional,required"`
|
|
}
|
|
|
|
type Walker struct {
|
|
Found []string
|
|
}
|
|
|
|
func (w *Walker) WalkFunc(path string, info fs.DirEntry, err error) error {
|
|
if info.IsDir() {
|
|
return nil
|
|
}
|
|
|
|
if strings.HasSuffix(path, "_scriptData.dat") {
|
|
w.Found = append(w.Found, path)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func run(args *Arguments) error {
|
|
w := &Walker{Found: []string{}}
|
|
err := filepath.WalkDir(args.BaseDir, w.WalkFunc)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Printf("found %d scripts\n", len(w.Found))
|
|
|
|
stats := make(script.Stats)
|
|
|
|
for _, file := range w.Found {
|
|
fmt.Println(file)
|
|
scr, err := script.ParseFile(file, 0x0000)
|
|
if err != nil {
|
|
if scr != nil {
|
|
for _, token := range scr.Tokens {
|
|
fmt.Println(token.String(scr.Labels))
|
|
}
|
|
}
|
|
return err
|
|
}
|
|
|
|
stats.Add(scr.Stats())
|
|
}
|
|
|
|
outfile, err := os.Create(args.Output)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer outfile.Close()
|
|
|
|
_, err = stats.WriteTo(outfile)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func main() {
|
|
args := &Arguments{}
|
|
arg.MustParse(args)
|
|
|
|
err := run(args)
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
os.Exit(1)
|
|
}
|
|
}
|