package router

import (
	"encoding/json"
	"fmt"
	"time"
)

func (router *Router) GenerateStatusJSON(version, encryption string) ([]byte, error) {
	return json.Marshal(struct {
		Version         string
		Encryption      string
		Name, Interface string
		Macs            *MacCache
		Peers           *Peers
		Routes          *Routes
	}{version, encryption, router.Ourself.Name.String(), fmt.Sprintf("%v", router.Iface), router.Macs, router.Peers, router.Routes})
	// leaving out ConectionMaker due to async complexities
}

func (cache *MacCache) MarshalJSON() ([]byte, error) {
	type cacheEntry struct {
		Mac      string
		PeerName PeerName
		LastSeen time.Time
	}
	entries := make([]*cacheEntry, 0)
	for key, entry := range cache.table {
		entries = append(entries, &cacheEntry{intmac(key).String(), entry.peer.Name, entry.lastSeen})
	}
	return json.Marshal(entries)
}

func (peers *Peers) MarshalJSON() ([]byte, error) {
	ps := make([]*Peer, 0)
	peers.ForEach(func(_ PeerName, peer *Peer) { ps = append(ps, peer) })
	return json.Marshal(ps)
}

func (routes *Routes) MarshalJSON() ([]byte, error) {
	routes.RLock()
	defer routes.RUnlock()
	type uni struct {
		Dest, Via PeerName
	}
	type broad struct {
		Source PeerName
		Via    []PeerName
	}
	var r struct {
		Unicast   []*uni
		Broadcast []*broad
	}
	for name, hop := range routes.unicast {
		r.Unicast = append(r.Unicast, &uni{name, hop})
	}
	for name, hops := range routes.broadcast {
		r.Broadcast = append(r.Broadcast, &broad{name, hops})
	}
	return json.Marshal(r)
}

func (peer *Peer) MarshalJSON() ([]byte, error) {
	conns := make([]Connection, 0)
	peer.ForEachConnection(func(remoteName PeerName, conn Connection) {
		conns = append(conns, conn)
	})
	return json.Marshal(struct {
		Name        string
		UID         uint64
		Version     uint64
		Connections []Connection
	}{peer.Name.String(), peer.UID, peer.version, conns})
}

func (conn *RemoteConnection) MarshalJSON() ([]byte, error) {
	return json.Marshal(struct{ RemoteName, TcpAddr string }{conn.Remote().Name.String(), conn.RemoteTCPAddr()})
}

func (name PeerName) MarshalJSON() ([]byte, error) {
	return json.Marshal(name.String())
}