package main

import (
	"cli-mon/can"
	"cli-mon/ui"
	"cli-mon/yabl"
	flag "flag"
	"fmt"
	"golang.org/x/sys/unix"
	"os"
	"time"
)

func main() {
	filename := flag.String("f", "", "Candump filename")
	canbus := flag.String("i", "can0", "CAN bus interface")
	stdin := flag.Bool("s", false, "Read from stdin")
	gui := flag.Bool("gui", false, "Pseudo-GUI CLI interface")
	amqp := flag.String("amqp", "", "Run output to AMQP broker url")
	interval := time.Duration(*flag.Uint("t", 50, "Override data loss check interval, ms"))
	flag.Parse()

	var fromBus = make(chan *can.CanFrame)
	var toBus = make(chan *can.CanFrame)
	var fromProducer = make(chan *yabl.Event)
	var toConsumer = make(chan *yabl.Event)

	defer func() {
		//close(fromBus)
		//close(toConsumer)
		close(fromProducer)
		close(toBus)
	}()

	switch {
	case *stdin:
		can.ReadStdin(fromBus, os.Stdin)
	case len(*filename) > 0:
		can.Readfile(fromBus, filename)
	case len(*canbus) > 0:
		filter := []unix.CanFilter{
			{Id: yabl.CanIdContIntState, Mask: 0xFFF},
			{Id: yabl.CanIdContIntErr, Mask: 0xFFF},
			{Id: yabl.CanIdContIntForce, Mask: 0xFFF},
			{Id: yabl.CanIdContIntDebug, Mask: 0xFFF},
			{Id: yabl.CanIdPuPresentEnergy, Mask: 0xFFF},
			{Id: yabl.CanIdPuPeriphery, Mask: 0xFFF},
			{Id: yabl.CanIdCpuPresentEnergy, Mask: 0xFFF},
			{Id: yabl.CanIdCpuPeriphery, Mask: 0xFFF},
			{Id: yabl.CanIdCpuEnergySettings, Mask: 0xFFF},
			{Id: yabl.CanIdCpuErrors, Mask: 0xFFF},
			{Id: yabl.CanIdCpuDebug, Mask: 0xFFF},
			{Id: yabl.CanIdPuErrors, Mask: 0xFFF},
			{Id: yabl.CanIdPuDebug, Mask: 0xFFF},
			{Id: yabl.CanIdSeccTargetEnergy, Mask: 0xFFF},
			{Id: yabl.CanIdSeccErrors, Mask: 0xFFF},
			{Id: yabl.CanIdLogicAuth, Mask: 0xFFF},
			{Id: yabl.CanIdLogicEnergyMode, Mask: 0xFFF},
			{Id: yabl.CanIdLogicErrors, Mask: 0xFFF},
			{Id: yabl.CanIdLogicWorkingMode, Mask: 0xFFF},
			{Id: yabl.CanIdLogicCommunicationMode, Mask: 0xFFF},
			{Id: yabl.CanIdPeripheryState, Mask: 0xFFF},
			{Id: yabl.CanIdPeripheryInfo, Mask: 0xFFF},
			{Id: yabl.CanIdPeripheryDebug, Mask: 0xFFF},
			{Id: yabl.CanIdConverterPresentEnergy, Mask: 0xFFF},
			{Id: yabl.CanIdConverterErrors, Mask: 0xFFF},
			{Id: yabl.CanIdConverterDebug, Mask: 0xFFF},
		}

		var err error
		if err = can.StartCan(fromBus, toBus, *canbus, filter); err != nil {
			fmt.Println(err)
			return
		}
	default:
		flag.Usage()
	}

	tag := os.Getenv("HOSTNAME")
	if len(tag) == 0 {
		tag, _ = os.Hostname()
	}

	isOffline := len(*filename) > 0 || *stdin

	yabl.NewChanProtocolConverter(tag, fromBus, fromProducer, toBus, toConsumer, interval, isOffline)

	var consumer ui.Consumer

	if *gui {
		consumer = ui.NewUI()
	} else if len(*amqp) > 0 {
		consumer = ui.NewAmqpWithEventProducer(*amqp, tag, fromProducer)
	} else {
		consumer = ui.NewLog()
	}

	consumer.Start()

	for event := range toConsumer {
		consumer.Consume(event)
	}

	consumer.Stop()
}