miflorad: add draft daemon implementation, no network submission yet

.gitignore

@@ -1,3 +1,4 @@
 .envrc
+cmd/miflorad/miflorad
 cmd/munin-miflora/munin-miflora
 cmd/munin-miflora-gatt/munin-miflora-gatt

Makefile

@@ -2,25 +2,40 @@
 
 MIFLORA_ADDR?=00:00:00:00:00:00
 
+RUN_COMMAND=miflorad
+RUN_OPTIONS=$(MIFLORA_ADDR)
+
 .PHONY: all
 all: clean build test
 
 .PHONY: run
 run: clean build test
-	sudo cmd/munin-miflora/munin-miflora xyz $(MIFLORA_ADDR)
+	sudo cmd/$(RUN_COMMAND)/$(RUN_COMMAND) $(RUN_OPTIONS)
 
 .PHONY: clean
 clean:
+	rm -f cmd/munin-miflora/miflorad
 	rm -f cmd/munin-miflora/munin-miflora
 	rm -f cmd/munin-miflora/munin-miflora-gatt
 
 .PHONY: build
-build: cmd/munin-miflora/munin-miflora cmd/munin-miflora/munin-miflora-gatt
+build: cmd/munin-miflora/miflorad cmd/munin-miflora/munin-miflora cmd/munin-miflora/munin-miflora-gatt
 
 .PHONY: test
 test: build
 	cd common && go test -v -race && cd ..
 
+.PHONY: remote-run
+remote-run: clean
+	cd cmd/$(RUN_COMMAND) && CGO_ENABLED=0 GOOS=linux GOARCH=arm go build -ldflags="-s -w" && cd ../..
+	file cmd/$(RUN_COMMAND)/$(RUN_COMMAND)
+	scp cmd/$(RUN_COMMAND)/$(RUN_COMMAND) extzero:$(RUN_COMMAND)
+	ssh extzero "./$(RUN_COMMAND) $(RUN_OPTIONS)"
+
+.PHONY: cmd/munin-miflora/miflorad
+cmd/munin-miflora/miflorad:
+	cd cmd/miflorad && CGO_ENABLED=0 go build && cd ../..
+
 .PHONY: cmd/munin-miflora/munin-miflora
 cmd/munin-miflora/munin-miflora:
 	cd cmd/munin-miflora && CGO_ENABLED=0 go build && cd ../..

cmd/miflorad/main.go

@@ -0,0 +1,214 @@
+package main
+
+import (
+	"context"
+	"flag"
+	"fmt"
+	"os"
+	"os/signal"
+	"strings"
+	"syscall"
+	"time"
+
+	common "miflorad/common"
+	impl "miflorad/common/ble"
+
+	"github.com/go-ble/ble"
+	"github.com/go-ble/ble/examples/lib/dev"
+
+	"github.com/pkg/errors"
+)
+
+var (
+	scanTimeout = flag.Duration("scantimeout", 6*time.Second, "timeout after that a scan per peripheral will be aborted")
+	interval    = flag.Duration("interval", 15*time.Second, "metrics collection interval")
+	readRetries = flag.Int("readretries", 2, "number of times reading will be attempted per peripheral")
+)
+
+type peripheral struct {
+	id                string
+	lastMetaDataFetch time.Time
+	metaData          common.VersionBatteryResponse
+}
+
+var allPeripherals []*peripheral
+
+var (
+	countSuccess = 0
+	countFailure = 0
+)
+
+func checkTooShortInterval() error {
+	numPeripherals := int64(len(flag.Args()))
+	if (*scanTimeout).Nanoseconds()*int64(*readRetries)*numPeripherals >= (*interval).Nanoseconds() {
+		return errors.New(fmt.Sprintf(
+			"The interval of %s is too short given the scan timeout of %s for %d peripheral(s) with %d retries each! Exiting...\n",
+			*interval, *scanTimeout, numPeripherals, *readRetries))
+	}
+	return nil
+}
+
+func readData(peripheral *peripheral, client ble.Client) error {
+	if time.Since(peripheral.lastMetaDataFetch) >= 24*time.Hour {
+		metaData, err := impl.RequestVersionBattery(client, client.Profile())
+		if err != nil {
+			return errors.Wrap(err, "can't request version battery")
+		}
+		peripheral.metaData = metaData
+		peripheral.lastMetaDataFetch = time.Now()
+	}
+
+	if peripheral.metaData.RequiresModeChangeBeforeRead() {
+		err2 := impl.RequestModeChange(client, client.Profile())
+		if err2 != nil {
+			return errors.Wrap(err2, "can't request mode change")
+		}
+	}
+
+	sensorData, err3 := impl.RequestSensorData(client, client.Profile())
+	if err3 != nil {
+		return errors.Wrap(err3, "can't request sensor data")
+	}
+
+	fmt.Println(sensorData.Temperature, sensorData.Brightness)
+
+	return nil
+}
+
+func connectPeripheral(peripheral *peripheral) error {
+	fmt.Fprintf(os.Stderr, "Scanning for %s...\n", peripheral.id)
+
+	filter := func(adv ble.Advertisement) bool {
+		return strings.ToUpper(adv.Addr().String()) == strings.ToUpper(peripheral.id)
+	}
+
+	timeConnectStart := time.Now()
+
+	ctx := ble.WithSigHandler(context.WithTimeout(context.Background(), *scanTimeout))
+	client, err := ble.Connect(ctx, filter)
+	if err != nil {
+		return errors.Wrapf(err, "can't connect to %s", peripheral.id)
+	}
+
+	timeConnectTook := time.Since(timeConnectStart).Seconds()
+	fmt.Println(timeConnectTook)
+	// fmt.Fprintf(os.Stdout, "%s.miflora.%s.connect_time %.2f %d\n", prefix, id, timeConnectTook, time.Now().Unix())
+
+	// Source: https://github.com/go-ble/ble/blob/master/examples/basic/explorer/main.go#L53
+	// Make sure we had the chance to print out the message.
+	done := make(chan struct{})
+	// Normally, the connection is disconnected by us after our exploration.
+	// However, it can be asynchronously disconnected by the remote peripheral.
+	// So we wait(detect) the disconnection in the go routine.
+	go func() {
+		<-client.Disconnected()
+		close(done)
+	}()
+
+	timeReadoutStart := time.Now()
+
+	if _, err := client.DiscoverProfile(true); err != nil {
+		return errors.Wrap(err, "can't descover profile")
+	}
+
+	err2 := readData(peripheral, client)
+
+	timeReadoutTook := time.Since(timeReadoutStart).Seconds()
+	fmt.Println(timeReadoutTook)
+	// fmt.Fprintf(os.Stdout, "%s.miflora.%s.readout_time %.2f %d\n", prefix, id, timeReadoutTook, time.Now().Unix())
+
+	client.CancelConnection()
+
+	<-done
+
+	return err2
+}
+
+func readPeripheral(peripheral *peripheral) error {
+	var err error
+	for retry := 0; retry < *readRetries; retry++ {
+		err = connectPeripheral(peripheral)
+		// stop retrying once we have a success, last err will be returned (or nil)
+		if err == nil {
+			break
+		}
+	}
+	return err
+}
+
+func readAllPeripherals(quit chan struct{}) {
+	for _, peripheral := range allPeripherals {
+		select {
+		case <-quit:
+			return
+		default:
+		}
+
+		err := readPeripheral(peripheral)
+		if err != nil {
+			countFailure++
+			fmt.Fprintf(os.Stderr, "Failed to read peripheral %s, err: %s\n", peripheral.id, err)
+			continue
+		}
+		countSuccess++
+	}
+}
+
+func main() {
+	flag.Parse()
+	if len(flag.Args()) < 1 {
+		fmt.Fprintf(os.Stderr, "Usage: %s [options] peripheral-id [peripheral-ids...] \n", os.Args[0])
+		flag.PrintDefaults()
+		os.Exit(1)
+	}
+
+	if err := checkTooShortInterval(); err != nil {
+		fmt.Fprintf(os.Stderr, "%s\n", err)
+		os.Exit(1)
+	}
+
+	device, err := dev.NewDevice("default")
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Failed to open device, err: %s\n", err)
+		os.Exit(1)
+	}
+	ble.SetDefaultDevice(device)
+
+	intervalTicker := time.NewTicker(*interval)
+	quit := make(chan struct{})
+
+	go func() {
+		fmt.Fprintf(os.Stderr, "Starting miflorad loop with %s interval...\n", *interval)
+
+		allPeripherals = make([]*peripheral, len(flag.Args()))
+		for i, peripheralID := range flag.Args() {
+			allPeripherals[i] = &peripheral{
+				id:                peripheralID,
+				lastMetaDataFetch: time.Unix(0, 0),
+			}
+		}
+
+		readAllPeripherals(quit)
+		for range intervalTicker.C {
+			readAllPeripherals(quit)
+		}
+	}()
+
+	signals := make(chan os.Signal, 1)
+	signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM)
+
+	signal := <-signals
+	fmt.Fprintf(os.Stderr, "Received %s! Stopping...\n", signal)
+	intervalTicker.Stop()
+	close(quit)
+	// wait for last readPeripheral to finish (worst case)
+	time.Sleep(*scanTimeout * time.Duration(*readRetries))
+
+	fmt.Fprintf(os.Stderr, "Failures:  %d\n", countFailure)
+	fmt.Fprintf(os.Stderr, "Successes: %d\n", countSuccess)
+
+	if err := device.Stop(); err != nil {
+		fmt.Fprintf(os.Stderr, "Failed to close device, err: %s\n", err)
+		os.Exit(1)
+	}
+}