Separate TSCrunch into its own library and cli tools

cmd/multitonycrunch/main.go

@@ -0,0 +1,110 @@
+/*
+TonyCrunch is a fork of Antonio Savona's TSCrunch.
+Refactoring, including fast mode and multi-hack by burg.
+*/
+package main
+
+import (
+	"flag"
+	"fmt"
+	"log"
+	"os"
+	"runtime/pprof"
+	"sync"
+	"time"
+
+	"github.com/staD020/TSCrunch"
+)
+
+func usage() {
+	fmt.Printf("TSCrunch %s - binary cruncher, by Antonio Savona\n", TSCrunch.Version)
+	fmt.Println("Multi-hack and fast mode by burg, quickly compile multiple files")
+	fmt.Println("Usage: tscrunch [-p] [-i] [-f] [-q] infile infile infile")
+	fmt.Println(" -p  : input file is a prg, first 2 bytes are discarded.")
+	fmt.Println(" -i  : inplace crunching (forces -p)")
+	fmt.Println(" -q  : quiet mode")
+	fmt.Println(" -f  : fast mode")
+}
+
+func main() {
+	if err := run(); err != nil {
+		log.Printf("error: %v\n", err)
+		usage()
+		return
+	}
+}
+
+func run() error {
+	t0 := time.Now()
+	opt := TSCrunch.Options{STATS: true}
+	var cpuProfile string
+	flag.StringVar(&cpuProfile, "cpuprofile", "", "write cpu profile to `file`")
+	flag.BoolVar(&opt.PRG, "p", false, "")
+	flag.BoolVar(&opt.QUIET, "q", false, "")
+	flag.BoolVar(&opt.INPLACE, "i", false, "")
+	flag.BoolVar(&opt.Fast, "f", false, "")
+	flag.Usage = usage
+	flag.Parse()
+
+	if cpuProfile != "" {
+		f, err := os.Create(cpuProfile)
+		if err != nil {
+			return fmt.Errorf("could not create CPU profile %q: %w", cpuProfile, err)
+		}
+		defer f.Close()
+		if err := pprof.StartCPUProfile(f); err != nil {
+			return fmt.Errorf("could not start CPU profile: %w", err)
+		}
+		defer pprof.StopCPUProfile()
+	}
+
+	inFiles := flag.Args()
+	if len(inFiles) < 1 {
+		return fmt.Errorf("not enough args")
+	}
+
+	crunchFiles(opt, inFiles)
+
+	if !opt.QUIET {
+		fmt.Printf("elapsed: %s\n", time.Since(t0))
+	}
+	return nil
+}
+
+func crunchFiles(opt TSCrunch.Options, ff []string) {
+	wg := &sync.WaitGroup{}
+	wg.Add(len(ff))
+	for _, file := range ff {
+		go func(file string) {
+			defer wg.Done()
+			t1 := time.Now()
+			in, err := os.Open(file)
+			if err != nil {
+				log.Printf("error: %v\n", err)
+				return
+			}
+			defer in.Close()
+			t, err := TSCrunch.New(opt, in)
+			if err != nil {
+				log.Printf("error: %v\n", err)
+				return
+			}
+			f, err := os.Create(file + ".lz")
+			if err != nil {
+				log.Printf("error: %v\n", err)
+				return
+			}
+			defer f.Close()
+			_, err = t.WriteTo(f)
+			if err != nil {
+				log.Printf("error: %v\n", err)
+				return
+			}
+
+			if !opt.QUIET {
+				fmt.Printf("crunching %q took %s\n\n", file, time.Since(t1))
+			}
+		}(file)
+	}
+	wg.Wait()
+}

cmd/tonycrunch/main.go

@@ -0,0 +1,90 @@
+/*
+TonyCrunch is a fork of Antonio Savona's TSCrunch.
+Refactoring, including fast mode by burg.
+*/
+package main
+
+import (
+	"flag"
+	"fmt"
+	"log"
+	"os"
+	"runtime/pprof"
+	"time"
+
+	"github.com/staD020/TSCrunch"
+)
+
+func usage() {
+	fmt.Printf("TSCrunch %s - binary cruncher, by Antonio Savona\n", TSCrunch.Version)
+	fmt.Println("Fast mode by burg")
+	fmt.Println("Usage: tscrunch [-p] [-i] [-f] [-q] [-x $081f|0x081f|2079] infile outfile")
+	fmt.Println(" -p  : input file is a prg, first 2 bytes are discarded.")
+	fmt.Println(" -x  $addr: creates a self extracting file (forces -p)")
+	fmt.Println(" -i  : inplace crunching (forces -p)")
+	fmt.Println(" -q  : quiet mode")
+	fmt.Println(" -f  : fast mode")
+}
+
+func main() {
+	if err := run(); err != nil {
+		log.Printf("error: %v\n", err)
+		usage()
+		return
+	}
+}
+
+func run() error {
+	t0 := time.Now()
+	opt := TSCrunch.Options{STATS: true}
+	var cpuProfile string
+	flag.StringVar(&cpuProfile, "cpuprofile", "", "write cpu profile to `file`")
+	flag.BoolVar(&opt.PRG, "p", false, "")
+	flag.BoolVar(&opt.QUIET, "q", false, "")
+	flag.BoolVar(&opt.INPLACE, "i", false, "")
+	flag.BoolVar(&opt.Fast, "f", false, "")
+	flag.StringVar(&opt.JumpTo, "x", "", "")
+	flag.Usage = usage
+	flag.Parse()
+
+	if cpuProfile != "" {
+		f, err := os.Create(cpuProfile)
+		if err != nil {
+			return fmt.Errorf("could not create CPU profile %q: %w", cpuProfile, err)
+		}
+		defer f.Close()
+		if err := pprof.StartCPUProfile(f); err != nil {
+			return fmt.Errorf("could not start CPU profile: %w", err)
+		}
+		defer pprof.StopCPUProfile()
+	}
+
+	if flag.NArg() != 2 {
+		return fmt.Errorf("not enough args")
+	}
+
+	inFilename := flag.Args()[0]
+	outFilename := flag.Args()[1]
+	in, err := os.Open(inFilename)
+	if err != nil {
+		return err
+	}
+	defer in.Close()
+	t, err := TSCrunch.New(opt, in)
+	if err != nil {
+		return err
+	}
+	out, err := os.Create(outFilename)
+	if err != nil {
+		return err
+	}
+	defer out.Close()
+	_, err = t.WriteTo(out)
+	if err != nil {
+		return err
+	}
+	if !opt.QUIET {
+		fmt.Printf("elapsed: %s\n", time.Since(t0))
+	}
+	return nil
+}

go.mod

@@ -0,0 +1,5 @@
+module github.com/staD020/TSCrunch
+
+go 1.17
+
+require github.com/RyanCarrier/dijkstra v1.1.0

go.sum

@@ -0,0 +1,8 @@
+github.com/RyanCarrier/dijkstra v1.1.0 h1:/NDihjfJA3CxFaZz8EdzTwdFKFZDvvB881OVLdakRcI=
+github.com/RyanCarrier/dijkstra v1.1.0/go.mod h1:5agGUBNEtUAGIANmbw09fuO3a2htPEkc1jNH01qxCWA=
+github.com/RyanCarrier/dijkstra-1 v0.0.0-20170512020943-0e5801a26345 h1:fgSpoKViTSqRb4hjDNj10ig5wUvO0CayCzFdLf6fuRM=
+github.com/RyanCarrier/dijkstra-1 v0.0.0-20170512020943-0e5801a26345/go.mod h1:OK4EvWJ441LQqGzed5NGB6vKBAE34n3z7iayPcEwr30=
+github.com/albertorestifo/dijkstra v0.0.0-20160910063646-aba76f725f72 h1:uGeGZl8PxSq8VZGG4QK5njJTFA4/G/x5CYORvQVXtAE=
+github.com/albertorestifo/dijkstra v0.0.0-20160910063646-aba76f725f72/go.mod h1:o+JdB7VetTHjLhU0N57x18B9voDBQe0paApdEAEoEfw=
+github.com/mattomatic/dijkstra v0.0.0-20130617153013-6f6d134eb237 h1:acuCHBjzG7MFTugvx3buC4m5rLDLaKC9J8C9jtlraRc=
+github.com/mattomatic/dijkstra v0.0.0-20130617153013-6f6d134eb237/go.mod h1:UOnLAUmVG5paym8pD3C4B9BQylUDC2vXFJJpT7JrlEA=