phomemo-printer/main.go

127 lines
3.4 KiB
Go

package main
import (
phomemofilter "brainbaking.com/phomemoprinter/filter"
"log"
"os"
"os/exec"
"strconv"
"time"
"tinygo.org/x/bluetooth"
)
var adapter = bluetooth.DefaultAdapter
var phomemoFolder = "/Downloads/phomemo"
func main() {
home, _ := os.UserHomeDir()
inDir := home + phomemoFolder
files, err := os.ReadDir(inDir)
must(phomemoFolder, err)
if len(files) == 0 {
log.Printf("No files to print found in %s. Quitting.\n", inDir)
os.Exit(0)
}
// Enable BLE interface.
must("enable BLE stack", adapter.Enable())
// Start scanning. Use this to figure out your device's UUID. Phomemo uses a non-random one.
println("scanning...")
var phomemoAddress bluetooth.ScanResult
err = adapter.Scan(func(adapter *bluetooth.Adapter, device bluetooth.ScanResult) {
if device.LocalName() == "Mr.in_M02" {
phomemoAddress = device
println("found device: ", device.Address.String(), device.RSSI, device.LocalName())
adapter.StopScan()
}
})
must("start scan", err)
tryToPrint(phomemoAddress, inDir, files)
}
func tryToPrint(phomemoAddress bluetooth.ScanResult, dir string, files []os.DirEntry) {
var phomemo *bluetooth.Device
phomemo, err := adapter.Connect(phomemoAddress.Address, bluetooth.ConnectionParams{})
must("failed to connect to adapter", err)
log.Println("connected to ", phomemoAddress.Address.String())
defer func() {
if err := phomemo.Disconnect(); err != nil {
log.Println(err)
} else {
log.Println("disconnected")
}
}()
srvcs, err := phomemo.DiscoverServices(nil)
must("failed to discover service", err)
log.Println("Discovering all services on device...")
srvc := srvcs[0] // there's only one (0000ff00-...) anyway
log.Println("- service", srvc.UUID().String())
chars, err := srvc.DiscoverCharacteristics(nil)
must("failed to discover characteristics of service", err)
dumpFilter("phomemo-filter.py")
defer func() {
os.RemoveAll("phomemo-filter.py")
}()
// 3 characteristics to discover 0000ff01, 0000ff02, and 0000ff03
for _, file := range files {
filePath := dir + "/" + file.Name()
writeData(filePath, chars[1]) // 0000ff02-...
readResult(chars[0]) // 0000ff01-...
os.RemoveAll(filePath)
}
log.Println("-- DONE")
}
func dumpFilter(filename string) {
err := os.WriteFile(filename, phomemofilter.Pyfilter, 0666)
must("unable to dump python filter for use", err)
}
func writeData(path string, char bluetooth.DeviceCharacteristic) {
// first, it needs to be prepared for the printer
pathpho := path + ".pho"
argstr := []string{"-c", "python phomemo-filter.py " + path + " > " + pathpho}
_, err := exec.Command("/bin/zsh", argstr...).Output()
must("something went wrong while filtering phomemo data py", err)
data, _ := os.ReadFile(pathpho)
log.Printf("-- writing file %s to characteristic %s\n", path, char.UUID().String())
_, err = char.WriteWithoutResponse(data)
if err != nil {
log.Println(" ", err.Error())
}
time.Sleep(1 * time.Second)
os.RemoveAll(pathpho)
}
func readResult(char bluetooth.DeviceCharacteristic) {
log.Println("-- reading from to characteristic", char.UUID().String())
buf := make([]byte, 28)
n, err := char.Read(buf)
if err != nil {
log.Println(" ", err.Error())
} else {
log.Println(" data bytes", strconv.Itoa(n))
log.Println(" value =", buf[:n])
}
time.Sleep(1 * time.Second)
}
func must(action string, err error) {
if err != nil {
panic("failed to " + action + ": " + err.Error())
}
}