package main

import (
	"fmt"
	"log"
	"math/rand"
	"net"
	"time"

	"github.com/go-ping/ping"

	"github.com/milosgajdos/tenus"
)

func init() {

	ZoreideBridge.initializeHierarchy()

}

func (b *AbstractBridge) RefreshArp() {

	// we want the program to recover in case of issues
	defer func() {
		if r := recover(); r != nil {

			fmt.Println("An error happened in <RefreshARP()>, but Zoreide recovered. ")
			fmt.Println("Error was: ", r)
		}
	}()

	b.GArp = new(Gratuitous)

	var err error
	b.GArp.IP, _, err = net.ParseCIDR(b.BridgeIpCIDR)
	if err != nil {
		log.Println("Error parsing CIDR: ", err.Error())
	}
	b.GArp.IfaceName = b.ExistingInterface
	log.Println("ARP Interface name: ", b.GArp.IfaceName)

	SendGratuitous(b.GArp)

}

func (b *AbstractBridge) initializeHierarchy() {

	b.hIerarchyNumber = rand.Int63()
	log.Println("Initialized host number: ", b.hIerarchyNumber)

}

func (b *AbstractBridge) configureIpAndBridgeUp() {
	// we want the program to recover in case of issues
	defer func() {
		if r := recover(); r != nil {

			fmt.Println("An error happened in <configureIpAndBridgeUp()>, but Zoreide recovered. ")
			fmt.Println("Error was: ", r)
		}
	}()

	// first we check the IP is free. Something weird could have happened in some
	// other server
	brIp, brIpNet, err := net.ParseCIDR(b.BridgeIpCIDR)
	if err != nil {
		log.Println("Error parsing CIDR: ", err.Error())
	}

	br, err := tenus.NewLinkFrom(b.ExistingInterface)
	if err != nil {
		log.Println("Error creating Ethernet Alias: ", err.Error())
		log.Println("Problematic interface: ", b.ExistingInterface)
	}

	if err := br.SetLinkIp(brIp, brIpNet); err != nil {
		log.Println("Error setting UP the IP: ", err.Error())
	} else {
		log.Printf("%s configured with %s\n", b.ExistingInterface, brIp.String())
	}

	// refresh ARP , should not be necessary, but we know, Cisco FA
	b.RefreshArp()
}

func (b *AbstractBridge) removeIPandBridgeInt() {

	// we want the program to recover in case of issues
	defer func() {
		if r := recover(); r != nil {

			fmt.Println("An error happened in <removeIPandBridgeInt()>, but Zoreide recovered. ")
			fmt.Println("Error was: ", r)
		}
	}()

	br, err := tenus.NewLinkFrom(b.ExistingInterface)
	if err != nil {
		log.Println("Error creating Ethernet Handle: ", err.Error())
		log.Println("Problematic interface: ", b.ExistingInterface)

	}

	brIp, brIpNet, err := net.ParseCIDR(b.BridgeIpCIDR)
	if err != nil {
		log.Println("Error parsing CIDR: ", err.Error())
	}
	if err := br.UnsetLinkIp(brIp, brIpNet); err != nil {
		log.Println("Error setting DOWN the IP: ", err.Error())
	}

}

func isActive(bridgeip string) bool {

	log.Println("Check for active IP: ", bridgeip)

	pinger, err := ping.NewPinger(bridgeip)
	if err != nil {
		log.Println("Ping error: " + err.Error())
	}
	// just in case it doesn't stops alone
	defer pinger.Stop()
	pinger.Count = 5
	pinger.Interval = time.Duration(10 * time.Millisecond)
	pinger.Timeout = time.Duration(1 * time.Second)
	pinger.Run() // blocks until finished
	stats := pinger.Statistics()
	log.Println("Ping results for: ", bridgeip)
	log.Printf("%d packet sent, %d packet recv\n", pinger.Count, stats.PacketsRecv)
	log.Println("IsACTIVE: ", stats.PacketsRecv == pinger.Count)

	return stats.PacketsRecv == pinger.Count

}