144 lines
2.7 KiB
Go
144 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"errors"
|
|
"io/fs"
|
|
|
|
"github.com/alexflint/go-arg"
|
|
|
|
"git.zorchenhimer.com/Zorchenhimer/go-studybox/rom"
|
|
)
|
|
|
|
type Arguments struct {
|
|
Pack *ArgPack `arg:"subcommand:pack"`
|
|
UnPack *ArgUnPack `arg:"subcommand:unpack"`
|
|
}
|
|
|
|
type ArgPack struct {
|
|
Input string `arg:"positional,required"`
|
|
Force bool `arg:"--force"`
|
|
Output string `arg:"--output,-o"`
|
|
}
|
|
|
|
type ArgUnPack struct {
|
|
Input string `arg:"positional,required" help:".json metadata file"`
|
|
NoAudio bool `arg:"--no-audio" help:"Do not unpack the audio portion"`
|
|
OutDir string `arg:"--dir" help:"Base directory to unpack into (json file will be here)"`
|
|
}
|
|
|
|
func main() {
|
|
args := &Arguments{}
|
|
arg.MustParse(args)
|
|
var err error
|
|
|
|
switch {
|
|
case args.Pack != nil:
|
|
err = pack(args.Pack)
|
|
case args.UnPack != nil:
|
|
err = unpack(args.UnPack)
|
|
default:
|
|
fmt.Fprintln(os.Stderr, "Missing command")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
os.Exit(2)
|
|
}
|
|
}
|
|
|
|
func pack(args *ArgPack) error {
|
|
if !strings.HasSuffix(args.Input, ".json") {
|
|
return fmt.Errorf("Pack needs a json file as input")
|
|
}
|
|
|
|
//fmt.Println("-- Processing " + args.Input)
|
|
sb, err := rom.Import(args.Input)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
//outDir := filepath.Base(args.Input)
|
|
//outDir = strings.ReplaceAll(outDir, ".json", "_output")
|
|
|
|
//err = os.MkdirAll(outDir, 0777)
|
|
//if err != nil {
|
|
// return err
|
|
//}
|
|
|
|
//err = sb.Export(outDir)
|
|
//if err != nil {
|
|
// return err
|
|
//}
|
|
|
|
// TODO: put this in the json file?
|
|
|
|
if args.Output != "" {
|
|
sb.Filename = args.Output
|
|
}
|
|
|
|
// outname := args.Input[:len(args.Input)-len(".json")]+".studybox"
|
|
if exists(sb.Filename) && !args.Force {
|
|
return fmt.Errorf("%s already exists or cannot be written to", sb.Filename)
|
|
}
|
|
|
|
fmt.Println(sb.Filename)
|
|
err = sb.Write(sb.Filename)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func unpack(args *ArgUnPack) error {
|
|
//fmt.Println("-- Processing " + file)
|
|
if !strings.HasSuffix(args.Input, ".studybox") {
|
|
return fmt.Errorf("Input needs to be a .studybox file.")
|
|
}
|
|
|
|
//outDir := filepath.Base(args.Input)
|
|
outbase := filepath.Base(args.Input[:len(args.Input)-len(".studybox")])
|
|
outdir := filepath.Dir(args.Input)
|
|
if args.OutDir != "" {
|
|
outdir = args.OutDir
|
|
}
|
|
outname := filepath.Join(outdir, outbase)
|
|
fmt.Println(outname)
|
|
//outDir = strings.ReplaceAll(outDir, ".studybox", "")
|
|
|
|
err := os.MkdirAll(outname, 0777)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
sb, err := rom.ReadFile(args.Input)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = sb.Export(outname, !args.NoAudio)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func exists(filename string) bool {
|
|
_, err := os.Stat(filename)
|
|
if err == nil {
|
|
return true
|
|
}
|
|
|
|
if errors.Is(err, fs.ErrNotExist) {
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|