First commit

cli/main.go

@@ -0,0 +1,164 @@
+package main
+
+import (
+	"fmt"
+	"git.massivebox.net/massivebox/go-catprinter"
+	"github.com/pkg/errors"
+	"log"
+	"os"
+	"time"
+
+	"github.com/urfave/cli/v2"
+)
+
+var flags = []cli.Flag{
+	&cli.StringFlag{
+		Name:  "mac",
+		Usage: "MAC address of the printer. Provide this or name.",
+	},
+	&cli.StringFlag{
+		Name:  "name",
+		Usage: "common name of the printer. Provide this or MAC.",
+	},
+	&cli.StringFlag{
+		Name:      "image",
+		Usage:     "path to the image file to be printed",
+		Required:  true,
+		TakesFile: true,
+	},
+	&cli.BoolFlag{
+		Name:  "lowerQuality",
+		Usage: "print with lower quality, but slightly faster speed",
+	},
+	&cli.BoolFlag{
+		Name:  "autoRotate",
+		Usage: "rotate image to fit printer",
+	},
+	&cli.BoolFlag{
+		Name:  "dontDither",
+		Usage: "don't dither the image",
+	},
+	&cli.Float64Flag{
+		Name:  "blackPoint",
+		Value: 0.5,
+		Usage: "regulate at which point a gray pixel is printed as black",
+	},
+	&cli.BoolFlag{
+		Name:  "debugLog",
+		Usage: "print debugging messages",
+	},
+	&cli.BoolFlag{
+		Name:  "dumpImage",
+		Usage: "save dithered image to ./image.png",
+	},
+	&cli.BoolFlag{
+		Name:  "dumpRequest",
+		Usage: "save raw data sent to printer to ./request.bin",
+	},
+	&cli.BoolFlag{
+		Name:  "dontPrint",
+		Usage: "don't actually print the image",
+	},
+}
+
+func findMac(name string, c *catprinter.Client) (string, error) {
+	fmt.Printf("Finding MAC by name (will take %d seconds)...", c.Timeout/time.Second)
+	devices, err := c.ScanDevices(name)
+	if err != nil {
+		return "", err
+	}
+	switch len(devices) {
+	case 0:
+		return "", errors.New("no devices found with name " + name)
+	case 1:
+		for k, _ := range devices {
+			return k, nil
+		}
+		break
+	default:
+		fmt.Println("Found multiple devices:")
+		for m, n := range devices {
+			fmt.Printf("%s\t%s", m, n)
+		}
+		return "", errors.New("multiple devices found with name " + name + ", please specify MAC directly")
+	}
+	return "", nil
+}
+
+func action(cCtx *cli.Context) error {
+
+	var (
+		mac          = cCtx.String("mac")
+		name         = cCtx.String("name")
+		imagePath    = cCtx.String("image")
+		lowerQuality = cCtx.Bool("lowerQuality")
+		autoRotate   = cCtx.Bool("autoRotate")
+		dontDither   = cCtx.Bool("dontDither")
+		blackPoint   = cCtx.Float64("blackPoint")
+		debugLog     = cCtx.Bool("debugLog")
+		dumpImage    = cCtx.Bool("dumpImage")
+		dumpRequest  = cCtx.Bool("dumpRequest")
+		dontPrint    = cCtx.Bool("dontPrint")
+	)
+
+	fmt.Println("Initializing...")
+	c, err := catprinter.NewClient()
+	if err != nil {
+		return err
+	}
+	defer c.Stop()
+
+	c.Debug.Log = debugLog
+	c.Debug.DumpImage = dumpImage
+	c.Debug.DumpRequest = dumpRequest
+	c.Debug.DontPrint = dontPrint
+
+	opts := catprinter.NewOptions().
+		SetBestQuality(!lowerQuality).
+		SetDither(!dontDither).
+		SetAutoRotate(autoRotate).
+		SetBlackPoint(float32(blackPoint))
+
+	if (mac != "") == (name != "") {
+		return errors.New("either mac or name must be provided")
+	}
+
+	if name != "" {
+		mac, err = findMac(name, c)
+		if err != nil {
+			return err
+		}
+	}
+
+	fmt.Println("Connecting...")
+	err = c.Connect(mac)
+	if err != nil {
+		return err
+	}
+	fmt.Println("Connected!")
+
+	fmt.Println("Printing...")
+	err = c.PrintFile(imagePath, opts)
+	if err != nil {
+		return err
+	}
+
+	fmt.Println("All done, exiting now.")
+
+	return nil
+}
+
+func main() {
+
+	app := &cli.App{
+		Name:   "catprinter",
+		Usage:  "print images to some BLE thermal printers",
+		Flags:  flags,
+		Action: action,
+	}
+
+	if err := app.Run(os.Args); err != nil {
+		log.Fatal(err)
+	}
+
+}