package main

import (
	"context"
	"flag"
	"fmt"
	"log"
	"strings"
	"time"

	"forge.cadoles.com/Pyxis/orion/emlid"
	"forge.cadoles.com/Pyxis/orion/emlid/updater"
)

const stepSleep = 5

const (
	phaseConfigureWifi    = "configure-wifi"
	phaseUpdateThenReboot = "update-then-reboot"
)

var (
	phase    = phaseConfigureWifi
	host     = "192.168.42.1"
	ssid     = ""
	security = string(updater.SecurityWPAPSK)
	password = ""
)

func init() {
	phases := []string{phaseConfigureWifi, phaseUpdateThenReboot}
	flag.StringVar(
		&phase, "phase", phase,
		fmt.Sprintf("The configuration phase. Available: %v", strings.Join(phases, ", ")),
	)
	flag.StringVar(&host, "host", host, "ReachRS module host")
	flag.StringVar(&ssid, "ssid", ssid, "The WiFi SSID to connect the module to")
	flag.StringVar(&security, "security", security, "The WiFi security algorithm")
	flag.StringVar(&password, "password", password, "The WiFi password")
}

func main() {

	flag.Parse()

	switch phase {
	case phaseConfigureWifi:
		configureWifi()
	case phaseUpdateThenReboot:
		updateThenReboot()
	default:
		log.Fatalf("unknown phase '%s'", phase)
	}

	log.Println("done")

}

func connect() *updater.Client {

	c := updater.NewClient(
		emlid.WithEndpoint(host, 80),
	)

	log.Printf("connecting to module '%s'", host)
	if err := c.Connect(); err != nil {
		log.Fatal(err)
	}

	log.Println("connected")

	return c

}

func configureWifi() {

	if ssid == "" {
		log.Fatal("you must provide a WiFi SSID with the -ssid flag")
	}

	c := connect()
	defer c.Close()

	ctx := context.Background()

	log.Println("checking module status")

	ctx, testResultsCancel := context.WithTimeout(ctx, 5*time.Second)
	defer testResultsCancel()
	results, err := c.TestResults(ctx)
	if err != nil {
		log.Fatal(err)
	}

	log.Printf("device: '%s'", results.Device)
	log.Printf("lora activated ? %v", results.Lora)
	log.Printf("mpu activated ? %v", results.MPU)
	log.Printf("stc activated ? %v", results.STC)
	log.Printf("ublox activated ? %v", results.UBlox)

	sleep(stepSleep)

	log.Printf("adding wifi network '%s'", ssid)

	ctx, addWifiCancel := context.WithTimeout(ctx, 5*time.Second)
	defer addWifiCancel()
	done, err := c.AddWifiNetwork(ctx, ssid, updater.WifiSecurity(security), password)
	if err != nil {
		log.Fatal(err)
	}

	if !done {
		log.Fatal("couldnt add wifi network")
	}

	sleep(stepSleep)

	log.Println("connecting module to wifi network")
	ctx, joinWifiCancel := context.WithTimeout(ctx, 5*time.Second)
	defer joinWifiCancel()
	if err := c.JoinWifiNetwork(ctx, ssid, true); err != nil {
		log.Fatal(err)
	}
	log.Printf("you can now switch to the wifi network and start phase '%s'", phaseUpdateThenReboot)

}

func updateThenReboot() {

	c := connect()
	defer c.Close()

	log.Println("checking module status")
	ctx, testResultsCancel := context.WithTimeout(context.Background(), 20*time.Second)
	defer testResultsCancel()
	results, err := c.TestResults(ctx)
	if err != nil {
		log.Fatalf("error while checking device state: %s", err)
	}

	log.Printf("device: '%s'", results.Device)
	log.Printf("lora activated ? %v", results.Lora)
	log.Printf("mpu activated ? %v", results.MPU)
	log.Printf("stc activated ? %v", results.STC)
	log.Printf("ublox activated ? %v", results.UBlox)

	sleep(stepSleep)

	log.Println("checking time sync")
	ctx, timeSyncedCancel := context.WithTimeout(context.Background(), 20*time.Second)
	defer timeSyncedCancel()
	synced, err := c.TimeSynced(ctx)
	if err != nil {
		log.Fatalf("error while checking time sync status: %s", err)
	}
	log.Printf("time synced ? %v", synced)

	sleep(stepSleep)

	log.Println("checking for upgrade")
	ctx, upgradeAvailableCancel := context.WithTimeout(context.Background(), 20*time.Second)
	defer upgradeAvailableCancel()
	_, _, err = c.ReceiverUpgradeAvailable(ctx)
	if err != nil {
		log.Fatalf("error while checking for upgrade: %s", err)
	}

	sleep(stepSleep)

	log.Println("checking reachview version")
	ctx, reachviewVersionCancel := context.WithTimeout(context.Background(), 20*time.Second)
	defer reachviewVersionCancel()
	version, err := c.ReachViewVersion(ctx)
	if err != nil {
		log.Fatalf("error while checking version: %s", err)
	}
	log.Printf("reachview version ? '%s'", version)

	sleep(stepSleep)

	log.Println("skipping update")
	if err := c.SkipUpdate(); err != nil {
		log.Fatalf("error while skipping update: %s", err)
	}

	sleep(stepSleep)

	log.Println("rebooting device")
	ctx, rebootCancel := context.WithTimeout(context.Background(), 20*time.Second)
	defer rebootCancel()
	if err := c.RebootNow(ctx, true); err != nil {
		log.Fatalf("error while rebooting: %s", err)
	}

}

func sleep(seconds int) {
	log.Printf("sleeping for %d seconds...", seconds)
	time.Sleep(time.Duration(seconds) * time.Second)
}