package main

import (
	"flag"
	"fmt"
	"log/slog"
	"os"
	"os/exec"
	"path/filepath"
	"sync"
	"time"

	"github.com/theater/picomap/lib/client"
	"github.com/theater/picomap/lib/picotool"
)

var target = flag.String("target", "all", "which firmware to load: picomap, picomap_test, or all")

func main() {
	flag.Parse()
	wd, err := os.Getwd()
	if err != nil {
		slog.Error("fatal", "err", err)
		os.Exit(1)
	}
	buildDir := filepath.Join(wd, "firmware", "build")

	if err := run(buildDir); err != nil {
		slog.Error("fatal", "err", err)
		os.Exit(1)
	}
}

func build(buildDir string) error {
	slog.Info("configuring")
	cmake := exec.Command("cmake", "-S", filepath.Join(filepath.Dir(buildDir)), "-B", buildDir)
	cmake.Stdout = os.Stdout
	cmake.Stderr = os.Stderr
	if err := cmake.Run(); err != nil {
		return fmt.Errorf("cmake failed: %w", err)
	}

	slog.Info("building")
	cmd := exec.Command("make", "-C", buildDir)
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr
	if err := cmd.Run(); err != nil {
		return fmt.Errorf("build failed: %w", err)
	}
	return nil
}

func boardSerial(id [8]byte) string {
	return fmt.Sprintf("%02X%02X%02X%02X%02X%02X%02X%02X",
		id[0], id[1], id[2], id[3], id[4], id[5], id[6], id[7])
}

type deviceInfo struct {
	dev    string
	serial string
	uf2    string
	name   string
}

func run(buildDir string) error {
	if err := build(buildDir); err != nil {
		return err
	}

	devs, err := client.ListSerial()
	if err != nil {
		return err
	}

	allTargets := []struct {
		name string
		uf2  string
	}{
		{"picomap", filepath.Join(buildDir, "picomap.uf2")},
		{"picomap_test", filepath.Join(buildDir, "picomap_test.uf2")},
	}

	var targets []struct {
		name string
		uf2  string
	}
	switch *target {
	case "all":
		targets = allTargets
	case "picomap":
		targets = allTargets[:1]
	case "picomap_test":
		targets = allTargets[1:]
	default:
		return fmt.Errorf("unknown target %q", *target)
	}

	if len(devs) < len(targets) {
		return fmt.Errorf("need %d device(s), found %d", len(targets), len(devs))
	}

	devices := make([]deviceInfo, len(targets))
	errs := make([]error, len(targets))

	var wg sync.WaitGroup
	for i := range targets {
		log := slog.With("dev", devs[i])
		wg.Go(func() {
			log.Info("connecting for info")
			c, err := client.NewSerial(devs[i], 2*time.Second)
			if err != nil {
				errs[i] = err
				return
			}
			info, err := c.Info()
			c.Close()
			if err != nil {
				errs[i] = err
				return
			}
			devices[i] = deviceInfo{
				dev:    devs[i],
				serial: boardSerial(info.BoardID),
				uf2:    targets[i].uf2,
				name:   targets[i].name,
			}
			log.Info("got info", "serial", devices[i].serial, "firmware", info.FirmwareName)
		})
	}
	wg.Wait()
	for i, err := range errs {
		if err != nil {
			return fmt.Errorf("[%s] info: %w", devs[i], err)
		}
	}

	for i := range devices {
		log := slog.With("serial", devices[i].serial)
		wg.Go(func() {
			log.Info("sending PICOBOOT")
			c, err := client.NewSerial(devices[i].dev, 2*time.Second)
			if err != nil {
				errs[i] = err
				return
			}
			err = c.PICOBOOT()
			c.Close()
			if err != nil {
				errs[i] = fmt.Errorf("PICOBOOT %s: %w", devices[i].serial, err)
				return
			}
			log.Info("PICOBOOT sent")
		})
	}
	wg.Wait()
	for i, err := range errs {
		if err != nil {
			return fmt.Errorf("[%s] %w", devices[i].serial, err)
		}
	}

	for i := range devices {
		log := slog.With("serial", devices[i].serial)
		wg.Go(func() {
			log.Info("loading", "uf2", devices[i].name)
			errs[i] = picotool.Load(devices[i].uf2, devices[i].serial, 10*time.Second)
			if errs[i] == nil {
				log.Info("loaded", "uf2", devices[i].name)
			}
		})
	}
	wg.Wait()
	for i, err := range errs {
		if err != nil {
			return fmt.Errorf("[%s] load: %w", devices[i].serial, err)
		}
	}

	slog.Info("done")
	return nil
}