- separate TSCrunch into a library and a cli tool

- implements the io.WriterTo interface and accepts an io.Reader as input
- added multitscrunch cli tool, for crunching many files in parallel
- separated the boot binary into it's own embedded boot.prg file for easy future updates
- add go.mod and go.sum to support Go modules

Author: Enno

boot.prg

@@ -0,0 +1,105 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"log"
+	"os"
+	"runtime/pprof"
+	"sync"
+	"time"
+
+	"github.com/staD020/TSCrunch"
+)
+
+func usage() {
+	fmt.Println("TSCrunch 1.3 - binary cruncher, by Antonio Savona")
+	fmt.Println("Multi-hack by burg, quickly compile multiple files")
+	fmt.Println("Usage: tscrunch [-raw] [-i] [-q] infile infile infile")
+	fmt.Println(" -p  : input file is a prg, first 2 bytes are discarded.")
+	fmt.Println(" -i  : inplace crunching (disables -raw)")
+	fmt.Println(" -q  : quiet mode")
+}
+
+func main() {
+	err := run()
+	if err != nil {
+		log.Printf("error: %v\n", err)
+		usage()
+		os.Exit(1)
+	}
+}
+
+func run() error {
+	t0 := time.Now()
+	opt := TSCrunch.Options{}
+	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.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/multitscrunch/main.go

@@ -0,0 +1,87 @@
+/*
+TSCrunch binary cruncher, by Antonio Savona
+*/
+package main
+
+import (
+	"flag"
+	"fmt"
+	"log"
+	"os"
+	"runtime/pprof"
+	"time"
+
+	"github.com/staD020/TSCrunch"
+)
+
+func usage() {
+	fmt.Println("TSCrunch 1.3 - binary cruncher, by Antonio Savona")
+	fmt.Println("Usage: tscrunch [-p] [-i] [-q] [-x $addr] 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")
+}
+
+func main() {
+	if err := run(); err != nil {
+		log.Printf("error: %v\n", err)
+		usage()
+		os.Exit(1)
+	}
+}
+
+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.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()
+	opt.STATS = true
+	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
+}

cmd/tscrunch/main.go

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

go.mod

@@ -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=