package main

import (
	"fmt"
	"log"

	"git.freifunk-franken.de/jkimmel/abbel/capture"
	"git.freifunk-franken.de/jkimmel/abbel/packet"
	"git.freifunk-franken.de/jkimmel/abbel/tlv"
	"inet.af/netaddr"
)

type options struct {
	Group      string
	Port       uint16
	Ifs        []string
	UseCapture bool
}

func parseOpts() (options, error) {
	var opt options
	var err error

	opt = options{
		Group:      "ff02:0:0:0:0:0:1:6",
		Port:       6696,
		Ifs:        []string{"babel"},
		UseCapture: false,
	}

	return opt, err
}

type updatesKey struct {
	prefix   netaddr.IPPrefix
	routerID tlv.RouterID
	nexthop  netaddr.IP
}

var (
	updates = map[updatesKey]tlv.Update{}
)

type BabelPacketConn interface {
	ReadFrom(b []byte) (body []byte, src netaddr.IP, ifindex int, err error)
	Close() error
}

func run(opt options) error {
	var err error
	var conn BabelPacketConn

	if opt.UseCapture {
		conn, err = capture.FromInterface(opt.Ifs[0])
	} else {
		conn, err = packet.Listen(opt.Group, opt.Port, opt.Ifs...)
	}
	if err != nil {
		return err
	}
	defer conn.Close()

	buf := [4096]byte{}
	var s tlv.PacketDecoder
	for {
		b, src, ifindex, err := conn.ReadFrom(buf[:])
		if err != nil {
			fmt.Println("Skipping packet:", err)
			continue
		}

		fmt.Print("\x1B[1m")
		fmt.Printf("Packet size %4d from %s", len(b), src)
		fmt.Println("\x1B[0m")

		s.Reset(b, src, ifindex)
		for s.Scan() {
			switch t := s.TLV().(type) {
			case tlv.NextHop:
				//fmt.Printf("%12s %s\n", t.T(), t.Address)
			case tlv.Hello:
				fmt.Printf("% 12s %+v\n", t.T(), t)
			case tlv.Update:
				key := updatesKey{t.Prefix, t.RouterID, t.NextHop}
				tmetric := t.Metric
				diff := int(tmetric) - int(updates[key].Metric)
				updates[key] = t
				if diff == 0 && tmetric != 0xFFFF {
					break
				}
				if t.Metric == 0xFFFF {
					fmt.Print("\x1B[31m")
				} else {
					fmt.Print("\x1B[36m")
				}
				fmt.Printf("% 12s %-43s%+d %s\x1B[0m\n", t.T(), t.Prefix, diff, t.FormatHeader())
			case tlv.RouterID:
				//fmt.Printf("%12s %s\n", t.T(), t)
			case tlv.RouteRequest:
				fmt.Printf("%12s %s\n", t.T(), t.Prefix)
			case tlv.Raw:
				fmt.Printf("   T: %12s (%2d), L: %3d\n", t.T(), t.T(), len(t.V()))
			case tlv.IHU:
				fmt.Printf("%12s %-43sRxcost %5d Interval %4d \n", t.T(), t.Address, t.Rxcost, t.Interval)
			default:
				if t != nil {
					fmt.Println("Unknown TLV", t.T(), t)
				} else {
					fmt.Println("got nil TLV")
				}
			}
			subtlv := s.SubTLV()
			for len(subtlv) > 0 {
				switch tlv.SubType(subtlv[0]) {
				case tlv.SubTypeSourcePrefix:
					var pfx tlv.SourcePrefix
					pfx, subtlv, err = tlv.SourcePrefixFromBytes(subtlv[2:])
					fmt.Print("\x1B[7;36m")
					fmt.Printf("%12s %s\x1B[0m\n", pfx.T(), pfx)
				case tlv.SubTypeDiversity, tlv.SubTypePad1, tlv.SubTypePadN, tlv.SubTypeTimestamp:
					fmt.Print("\x1B[7m")
					fmt.Printf("%12s: %v\x1B[0m\n", "Subtlv", subtlv)
					subtlv = nil
				default:
					panic("unknown subtlv type")
				}
			}
		}
		if s.Err() != nil {
			fmt.Println(err)
		}
	}
}

func main() {
	opt, err := parseOpts()
	if err != nil {
		log.Fatal(err)
	}
	if err := run(opt); err != nil {
		log.Fatal(err)
	}
}