New daemon architecture

2026-02-06 13:47:06 -03:00
commit cbe18da598
18 changed files with 2435 additions and 0 deletions
--- a/internal/imageproc/imageproc.go
+++ b/internal/imageproc/imageproc.go
@@ -0,0 +1,189 @@
+package imageproc
+
+import (
+	"fmt"
+	"image"
+	"image/color"
+	_ "image/gif"
+	_ "image/jpeg"
+	_ "image/png"
+	"io"
+	"os"
+
+	"github.com/disintegration/imaging"
+	dither "github.com/makeworld-the-better-one/dither"
+
+	"bleh/internal/mxw01"
+)
+
+// DecodeImage loads an image from a given path or stdin ("-").
+func DecodeImage(path string) (image.Image, error) {
+	if path == "-" {
+		return DecodeImageFromReader(os.Stdin)
+	}
+	img, err := imaging.Open(path, imaging.AutoOrientation(true))
+	if err != nil {
+		return nil, fmt.Errorf("failed to open image %q: %v", path, err)
+	}
+	return img, nil
+}
+
+func DecodeImageFromReader(r io.Reader) (image.Image, error) {
+	img, _, err := image.Decode(r)
+	if err != nil {
+		return nil, fmt.Errorf("decode error: %v", err)
+	}
+	return img, nil
+}
+
+func PadImageToMinLines(img image.Image, minLines int) image.Image {
+	bounds := img.Bounds()
+	if bounds.Dy() >= minLines {
+		return img
+	}
+	dst := imaging.New(bounds.Dx(), minLines, color.White)
+	dst = imaging.Paste(dst, img, image.Pt(0, 0))
+	return dst
+}
+
+func LoadImageMonoFromImage(img image.Image, ditherType string) ([]byte, int, error) {
+	b := img.Bounds()
+	if b.Dx() <= 0 || b.Dy() <= 0 {
+		return nil, 0, fmt.Errorf("invalid image bounds: %v", b)
+	}
+	ratio := float64(b.Dx()) / float64(b.Dy())
+	height := int(float64(mxw01.LinePixels) / ratio)
+	if height <= 0 {
+		return nil, 0, fmt.Errorf("computed invalid height: %d", height)
+	}
+
+	img = imaging.Resize(img, mxw01.LinePixels, height, imaging.Lanczos)
+	img = imaging.Grayscale(img)
+
+	if ditherType != "none" {
+		palette := []color.Color{color.Black, color.White}
+		d := dither.NewDitherer(palette)
+		switch ditherType {
+		case "floyd":
+			d.Matrix = dither.FloydSteinberg
+		case "bayer2x2":
+			d.Mapper = dither.Bayer(2, 2, 1.0)
+		case "bayer4x4":
+			d.Mapper = dither.Bayer(4, 4, 1.0)
+		case "bayer8x8":
+			d.Mapper = dither.Bayer(8, 8, 1.0)
+		case "bayer16x16":
+			d.Mapper = dither.Bayer(16, 16, 1.0)
+		case "atkinson":
+			d.Matrix = dither.Atkinson
+		case "jjn":
+			d.Matrix = dither.JarvisJudiceNinke
+		default:
+			return nil, 0, fmt.Errorf("unknown dither type: %s", ditherType)
+		}
+		img = d.DitherCopy(img)
+	} else {
+		img = imaging.AdjustContrast(img, 10)
+	}
+
+	pixels := make([]byte, (mxw01.LinePixels*height)/8)
+	for y := 0; y < height; y++ {
+		for x := 0; x < mxw01.LinePixels; x++ {
+			gray := color.GrayModel.Convert(img.At(x, y)).(color.Gray)
+			if gray.Y < 128 {
+				idx := (y*mxw01.LinePixels + x) / 8
+				pixels[idx] |= 1 << (x % 8)
+			}
+		}
+	}
+
+	return pixels, height, nil
+}
+
+func LoadImage4BitFromImage(img image.Image, ditherType string) ([]byte, int, error) {
+	b := img.Bounds()
+	if b.Dx() <= 0 || b.Dy() <= 0 {
+		return nil, 0, fmt.Errorf("invalid image bounds: %v", b)
+	}
+	ratio := float64(b.Dx()) / float64(b.Dy())
+	height := int(float64(mxw01.LinePixels) / ratio)
+	if height <= 0 {
+		return nil, 0, fmt.Errorf("computed invalid height: %d", height)
+	}
+
+	img = imaging.Resize(img, mxw01.LinePixels, height, imaging.Lanczos)
+	img = imaging.Grayscale(img)
+
+	palette := make([]color.Color, 16)
+	for i := 0; i < 16; i++ {
+		v := uint8(i * 17)
+		palette[i] = color.Gray{Y: 255 - v}
+	}
+
+	if ditherType != "none" {
+		d := dither.NewDitherer(palette)
+		switch ditherType {
+		case "floyd":
+			d.Matrix = dither.FloydSteinberg
+		case "bayer2x2":
+			d.Mapper = dither.Bayer(2, 2, 0.2)
+		case "bayer4x4":
+			d.Mapper = dither.Bayer(4, 4, 0.2)
+		case "bayer8x8":
+			d.Mapper = dither.Bayer(8, 8, 0.2)
+		case "bayer16x16":
+			d.Mapper = dither.Bayer(16, 16, 0.2)
+		case "atkinson":
+			d.Matrix = dither.Atkinson
+		case "jjn":
+			d.Matrix = dither.JarvisJudiceNinke
+		default:
+			return nil, 0, fmt.Errorf("unknown dither type: %s", ditherType)
+		}
+		img = d.DitherCopy(img)
+	}
+
+	width := mxw01.LinePixels
+	pixels := make([]byte, (width*height)/2)
+	for y := 0; y < height; y++ {
+		for x := 0; x < width; x++ {
+			gray := color.GrayModel.Convert(img.At(x, y)).(color.Gray)
+			level := (255 - gray.Y) >> 4
+			idx := (y*width + x) >> 1
+			shift := uint(((x & 1) ^ 1) << 2)
+			pixels[idx] |= byte(level) << shift
+		}
+	}
+
+	return pixels, height, nil
+}
+
+func RenderPreviewFrom1bpp(pixels []byte, width, height int) image.Image {
+	img := image.NewGray(image.Rect(0, 0, width, height))
+	for y := 0; y < height; y++ {
+		for x := 0; x < width; x++ {
+			idx := (y*width + x) / 8
+			bit := uint(x % 8)
+			if pixels[idx]&(1<<bit) != 0 {
+				img.SetGray(x, y, color.Gray{Y: 0})
+			} else {
+				img.SetGray(x, y, color.Gray{Y: 255})
+			}
+		}
+	}
+	return img
+}
+
+func RenderPreviewFrom4bpp(pixels []byte, width, height int) image.Image {
+	img := image.NewGray(image.Rect(0, 0, width, height))
+	for y := 0; y < height; y++ {
+		for x := 0; x < width; x++ {
+			idx := (y*width + x) >> 1
+			shift := uint(((x & 1) ^ 1) << 2)
+			val := (pixels[idx] >> shift) & 0x0F
+			gray := 255 - val*17
+			img.SetGray(x, y, color.Gray{Y: gray})
+		}
+	}
+	return img
+}
--- a/internal/ipc/jsonl.go
+++ b/internal/ipc/jsonl.go
@@ -0,0 +1,55 @@
+package ipc
+
+import (
+	"bufio"
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
+)
+
+const (
+	// MaxLineBytes is the maximum size of a single JSON line accepted by the JSONL protocol.
+	// 1 MiB is enough for fairly large base64 pixel payloads while keeping memory bounded.
+	MaxLineBytes = 1 << 20
+)
+
+// ReadLine reads a single '\n'-terminated line from r with a hard size limit.
+func ReadLine(r *bufio.Reader) ([]byte, error) {
+	var buf bytes.Buffer
+	for {
+		chunk, isPrefix, err := r.ReadLine()
+		if err != nil {
+			if err == io.EOF && buf.Len() > 0 {
+				return buf.Bytes(), nil
+			}
+			return nil, err
+		}
+		buf.Write(chunk)
+		if buf.Len() > MaxLineBytes {
+			return nil, fmt.Errorf("jsonl line too large (> %d bytes)", MaxLineBytes)
+		}
+		if !isPrefix {
+			break
+		}
+	}
+	return buf.Bytes(), nil
+}
+
+func EncodeLine(w io.Writer, v any) error {
+	b, err := json.Marshal(v)
+	if err != nil {
+		return err
+	}
+	b = append(b, '\n')
+	_, err = w.Write(b)
+	return err
+}
+
+func DecodeRequestLine(line []byte) (*Request, error) {
+	var req Request
+	if err := json.Unmarshal(line, &req); err != nil {
+		return nil, err
+	}
+	return &req, nil
+}
--- a/internal/ipc/protocol.go
+++ b/internal/ipc/protocol.go
@@ -0,0 +1,22 @@
+package ipc
+
+// JSONL request/response protocol shared by bleh and blehd.
+
+type Request struct {
+	ID     string         `json:"id"`
+	Method string         `json:"method"`
+	Params map[string]any `json:"params,omitempty"`
+}
+
+type Error struct {
+	Code    string `json:"code"`
+	Message string `json:"message"`
+}
+
+type Response struct {
+	ID     string `json:"id,omitempty"`
+	OK     bool   `json:"ok"`
+	Result any    `json:"result,omitempty"`
+	Error  *Error `json:"error,omitempty"`
+	Event  string `json:"event,omitempty"`
+}
--- a/internal/mxw01/mxw01.go
+++ b/internal/mxw01/mxw01.go
@@ -0,0 +1,282 @@
+package mxw01
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"time"
+
+	ble "github.com/go-ble/ble"
+)
+
+var (
+	MainServiceUUID      = ble.MustParse("ae30")
+	PrintCharacteristic  = ble.MustParse("ae01")
+	NotifyCharacteristic = ble.MustParse("ae02")
+	DatCharacteristic    = ble.MustParse("ae03")
+	TargetPrinterName    = "MXW01"
+	ScanTimeout          = 10 * time.Second
+	PrintCommandHeader   = []byte{0x22, 0x21}
+	PrintCommandFooter   = byte(0xFF)
+)
+
+type PrintMode byte
+
+const (
+	Mode1bpp PrintMode = 0x00
+	Mode4bpp PrintMode = 0x02
+)
+
+const (
+	LinePixels   = 384
+	MinLines     = 86 // firmware refuses to print anything shorter
+	DefaultSleep = 6 * time.Millisecond
+)
+
+type Status struct {
+	OK         bool   `json:"ok"`
+	State      string `json:"state"`
+	BatteryPct int    `json:"battery"`
+	TempC      int    `json:"temp"`
+}
+
+type Version struct {
+	Version   string `json:"version"`
+	PrintType string `json:"printType"`
+}
+
+func FindPrinter(ctx context.Context, address string) (ble.Advertisement, error) {
+	var addr ble.Addr
+	var adv ble.Advertisement
+
+	if address != "" {
+		addr = ble.NewAddr(address)
+	}
+
+	ctxScan, cancel := context.WithTimeout(ctx, ScanTimeout)
+	defer cancel()
+
+	err := ble.Scan(ctxScan, false, func(a ble.Advertisement) {
+		if address != "" {
+			if a.Addr().String() == addr.String() {
+				adv = a
+				cancel()
+			}
+			return
+		}
+		if a.LocalName() == TargetPrinterName {
+			adv = a
+			cancel()
+		}
+	}, nil)
+	if err != nil && !errors.Is(err, context.Canceled) {
+		return nil, fmt.Errorf("scan error: %w", err)
+	}
+	if adv == nil {
+		return nil, fmt.Errorf("printer not found")
+	}
+	return adv, nil
+}
+
+func DiscoverChars(client ble.Client) (printChr, notifyChr, dataChr *ble.Characteristic, err error) {
+	services, err := client.DiscoverServices([]ble.UUID{MainServiceUUID})
+	if err != nil || len(services) == 0 {
+		return nil, nil, nil, fmt.Errorf("service discovery failed: %v", err)
+	}
+	svc := services[0]
+	chars, err := client.DiscoverCharacteristics(nil, svc)
+	if err != nil {
+		return nil, nil, nil, fmt.Errorf("characteristic discovery failed: %v", err)
+	}
+	for _, c := range chars {
+		switch c.UUID.String() {
+		case PrintCharacteristic.String():
+			printChr = c
+		case NotifyCharacteristic.String():
+			notifyChr = c
+		case DatCharacteristic.String():
+			dataChr = c
+		}
+	}
+	return printChr, notifyChr, dataChr, nil
+}
+
+func BuildCommand(cmdId byte, payload []byte) []byte {
+	cmd := append([]byte{}, PrintCommandHeader...)
+	cmd = append(cmd, cmdId)
+	cmd = append(cmd, 0x00) // reserved
+	cmd = append(cmd, byte(len(payload)&0xFF), byte(len(payload)>>8))
+	cmd = append(cmd, payload...)
+	cmd = append(cmd, CalculateCRC8(payload))
+	cmd = append(cmd, PrintCommandFooter)
+	return cmd
+}
+
+func SendSimpleCommand(client ble.Client, printChr *ble.Characteristic, cmdId byte) error {
+	cmd := BuildCommand(cmdId, []byte{0x00})
+	return client.WriteCharacteristic(printChr, cmd, true)
+}
+
+func SendLineCommand(client ble.Client, printChr *ble.Characteristic, cmdId byte, lines uint) error {
+	param := []byte{byte(lines & 0xFF), byte(lines >> 8)}
+	cmd := BuildCommand(cmdId, param)
+	return client.WriteCharacteristic(printChr, cmd, true)
+}
+
+func ParseNotification(data []byte) (cmd byte, payloadLen int, err error) {
+	if len(data) < 6 {
+		return 0, 0, fmt.Errorf("malformed notification (too short)")
+	}
+	if data[0] != 0x22 || data[1] != 0x21 {
+		return 0, 0, fmt.Errorf("invalid notification header")
+	}
+	cmd = data[2]
+	payloadLen = int(data[4]) | int(data[5])<<8
+	return cmd, payloadLen, nil
+}
+
+func DecodeStatusNotification(data []byte) (*Status, error) {
+	_, _, err := ParseNotification(data)
+	if err != nil {
+		return nil, err
+	}
+	if len(data) < 14 {
+		return nil, fmt.Errorf("malformed status notification")
+	}
+	battery := int(data[9])
+	temp := int(data[10])
+	statusOk := data[12] == 0
+	statusCode := data[6]
+	errCode := data[13]
+
+	statusMsg := "Unknown"
+	if statusOk {
+		switch statusCode {
+		case 0x0:
+			statusMsg = "Standby"
+		case 0x1:
+			statusMsg = "Printing"
+		case 0x2:
+			statusMsg = "Feeding paper"
+		case 0x3:
+			statusMsg = "Ejecting paper"
+		}
+	} else {
+		switch errCode {
+		case 0x1, 0x9:
+			statusMsg = "No paper"
+		case 0x4:
+			statusMsg = "Overheated"
+		case 0x8:
+			statusMsg = "Low battery"
+		}
+	}
+
+	return &Status{OK: statusOk, State: statusMsg, BatteryPct: battery, TempC: temp}, nil
+}
+
+func DecodeBatteryNotification(data []byte) (int, error) {
+	_, _, err := ParseNotification(data)
+	if err != nil {
+		return 0, err
+	}
+	if len(data) < 7 {
+		return 0, fmt.Errorf("malformed battery notification")
+	}
+	return int(data[6]), nil
+}
+
+func DecodePrintTypeNotification(data []byte) (string, error) {
+	_, _, err := ParseNotification(data)
+	if err != nil {
+		return "", err
+	}
+	if len(data) < 7 {
+		return "", fmt.Errorf("malformed printtype notification")
+	}
+	switch data[6] {
+	case 0x01:
+		return "High pressure", nil
+	case 0xFF:
+		return "Unknown", nil
+	default:
+		return "Low pressure", nil
+	}
+}
+
+func DecodeVersionNotification(data []byte) (*Version, error) {
+	_, dataLen, err := ParseNotification(data)
+	if err != nil {
+		return nil, err
+	}
+	if len(data) < 15 {
+		return nil, fmt.Errorf("malformed version notification (too short)")
+	}
+	if len(data) < 14+dataLen {
+		return nil, fmt.Errorf("malformed version notification")
+	}
+	ver := string(data[6 : 6+dataLen])
+	pt := "Unknown"
+	switch data[14] {
+	case 0x32:
+		pt = "High pressure"
+	case 0x31:
+		pt = "Low pressure"
+	}
+	return &Version{Version: ver, PrintType: pt}, nil
+}
+
+func DecodeQueryCountNotification(data []byte) ([]byte, error) {
+	_, _, err := ParseNotification(data)
+	if err != nil {
+		return nil, err
+	}
+	if len(data) < 12 {
+		return nil, fmt.Errorf("malformed querycount notification")
+	}
+	out := make([]byte, 6)
+	copy(out, data[6:12])
+	return out, nil
+}
+
+func CalculateCRC8(data []byte) byte {
+	table := [256]byte{
+		0x00, 0x07, 0x0e, 0x09, 0x1c, 0x1b, 0x12, 0x15,
+		0x38, 0x3f, 0x36, 0x31, 0x24, 0x23, 0x2a, 0x2d,
+		0x70, 0x77, 0x7e, 0x79, 0x6c, 0x6b, 0x62, 0x65,
+		0x48, 0x4f, 0x46, 0x41, 0x54, 0x53, 0x5a, 0x5d,
+		0xe0, 0xe7, 0xee, 0xe9, 0xfc, 0xfb, 0xf2, 0xf5,
+		0xd8, 0xdf, 0xd6, 0xd1, 0xc4, 0xc3, 0xca, 0xcd,
+		0x90, 0x97, 0x9e, 0x99, 0x8c, 0x8b, 0x82, 0x85,
+		0xa8, 0xaf, 0xa6, 0xa1, 0xb4, 0xb3, 0xba, 0xbd,
+		0xc7, 0xc0, 0xc9, 0xce, 0xdb, 0xdc, 0xd5, 0xd2,
+		0xff, 0xf8, 0xf1, 0xf6, 0xe3, 0xe4, 0xed, 0xea,
+		0xb7, 0xb0, 0xb9, 0xbe, 0xab, 0xac, 0xa5, 0xa2,
+		0x8f, 0x88, 0x81, 0x86, 0x93, 0x94, 0x9d, 0x9a,
+		0x27, 0x20, 0x29, 0x2e, 0x3b, 0x3c, 0x35, 0x32,
+		0x1f, 0x18, 0x11, 0x16, 0x03, 0x04, 0x0d, 0x0a,
+		0x57, 0x50, 0x59, 0x5e, 0x4b, 0x4c, 0x45, 0x42,
+		0x6f, 0x68, 0x61, 0x66, 0x73, 0x74, 0x7d, 0x7a,
+		0x89, 0x8e, 0x87, 0x80, 0x95, 0x92, 0x9b, 0x9c,
+		0xb1, 0xb6, 0xbf, 0xb8, 0xad, 0xaa, 0xa3, 0xa4,
+		0xf9, 0xfe, 0xf7, 0xf0, 0xe5, 0xe2, 0xeb, 0xec,
+		0xc1, 0xc6, 0xcf, 0xc8, 0xdd, 0xda, 0xd3, 0xd4,
+		0x69, 0x6e, 0x67, 0x60, 0x75, 0x72, 0x7b, 0x7c,
+		0x51, 0x56, 0x5f, 0x58, 0x4d, 0x4a, 0x43, 0x44,
+		0x19, 0x1e, 0x17, 0x10, 0x05, 0x02, 0x0b, 0x0c,
+		0x21, 0x26, 0x2f, 0x28, 0x3d, 0x3a, 0x33, 0x34,
+		0x4e, 0x49, 0x40, 0x47, 0x52, 0x55, 0x5c, 0x5b,
+		0x76, 0x71, 0x78, 0x7f, 0x6a, 0x6d, 0x64, 0x63,
+		0x3e, 0x39, 0x30, 0x37, 0x22, 0x25, 0x2c, 0x2b,
+		0x06, 0x01, 0x08, 0x0f, 0x1a, 0x1d, 0x14, 0x13,
+		0xae, 0xa9, 0xa0, 0xa7, 0xb2, 0xb5, 0xbc, 0xbb,
+		0x96, 0x91, 0x98, 0x9f, 0x8a, 0x8d, 0x84, 0x83,
+		0xde, 0xd9, 0xd0, 0xd7, 0xc2, 0xc5, 0xcc, 0xcb,
+		0xe6, 0xe1, 0xe8, 0xef, 0xfa, 0xfd, 0xf4, 0xf3,
+	}
+	crc := byte(0)
+	for _, b := range data {
+		crc = table[crc^b]
+	}
+	return crc
+}
--- a/internal/mxw01/print.go
+++ b/internal/mxw01/print.go
@@ -0,0 +1,56 @@
+package mxw01
+
+import (
+	"fmt"
+	"time"
+
+	ble "github.com/go-ble/ble"
+)
+
+func SendImageBufferToPrinter(client ble.Client, dataChr, printChr *ble.Characteristic, pixels []byte, height int, mode PrintMode, intensity byte, attMTU int) error {
+	cmd := BuildCommand(0xA2, []byte{intensity})
+	if err := client.WriteCharacteristic(printChr, cmd, true); err != nil {
+		return fmt.Errorf("intensity set failed: %v", err)
+	}
+
+	param := []byte{byte(height & 0xFF), byte(height >> 8), 0x30, byte(mode)}
+	cmd = BuildCommand(0xA9, param)
+	if err := client.WriteCharacteristic(printChr, cmd, true); err != nil {
+		return fmt.Errorf("print command failed: %v", err)
+	}
+
+	bytesPerLine := LinePixels / 8
+	if mode == Mode4bpp {
+		bytesPerLine = LinePixels / 2
+	}
+
+	// ATT_MTU includes 3 bytes of ATT overhead for Write Command/Request.
+	payloadMTU := 20
+	if attMTU > 3 {
+		payloadMTU = attMTU - 3
+	}
+	if payloadMTU < 1 {
+		payloadMTU = 1
+	}
+
+	for y := 0; y < height; y++ {
+		slice := pixels[y*bytesPerLine : (y+1)*bytesPerLine]
+		for offset := 0; offset < len(slice); offset += payloadMTU {
+			end := offset + payloadMTU
+			if end > len(slice) {
+				end = len(slice)
+			}
+			chunk := slice[offset:end]
+			if err := client.WriteCharacteristic(dataChr, chunk, true); err != nil {
+				return fmt.Errorf("line %d chunk write failed: %v", y, err)
+			}
+			time.Sleep(DefaultSleep)
+		}
+	}
+
+	cmd = BuildCommand(0xAD, []byte{0x00})
+	if err := client.WriteCharacteristic(printChr, cmd, true); err != nil {
+		return fmt.Errorf("flush failed: %v", err)
+	}
+	return nil
+}