package p2p

import "fmt"

// Runs a speed test and does updates IP tables accordingly
func (ip *IpAddresses)SpeedTest() error{

	for i, _ := range ip.IpAddress {

		// Ping Test
		err := ip.IpAddress[i].PingTest()
		if err != nil {
			// Remove IP address of element not pingable 
			ip.IpAddress = append(ip.IpAddress[:i], ip.IpAddress[i+1:]...)
			// Proceed to next element in the array  
			continue
		}

		//Upload Speed Test
		err = ip.IpAddress[i].UploadSpeed()
		if err != nil {
			return err
		}

		err = ip.IpAddress[i].DownloadSpeed()
		if err != nil {
			return err
		}
		fmt.Println(ip.IpAddress[i].Latency)
	}

	err := ip.WriteIpTable()
	if err != nil {
		return err
	}

	return nil
}

// Called when ip tables from client/server is also passed on
func (ip *IpAddresses)SpeedTestUpdatedIPTable() error{
	targets, err := ReadIpTable()

	if err != nil {
		return err
	}

    // Appends all IP addresses
	for i, _ := range targets.IpAddress {
		ip.IpAddress = append(ip.IpAddress, targets.IpAddress[i])
	}

	err = ip.SpeedTest()

	if err != nil {
		return err
	}

	return nil
}

// Runs speed test in iptables locally only
func LocalSpeedTestIpTable() error {
	targets, err := ReadIpTable()
	if err != nil {
		return err
	}

	err = targets.SpeedTest()
	if err != nil {
		return err
	}

	return nil
}