package p2p

import (
	"encoding/json"
	"fmt"
	"git.sr.ht/~akilan1999/p2p-rendering-computation/config"
	"io/ioutil"
	"os"
	"time"
)

// Get IP table Data

type IpAddresses struct {
	IpAddress []IpAddress `json:"ip_address"`
}

type IpAddress struct {
	Ipv4   string `json:"ipv4"`
	Latency  time.Duration `json:"latency"`
	Download float64    `json:"download"`
	Upload float64 `json:"upload"`
}

// ReadIpTable Read data from Ip tables from json file
func ReadIpTable()(*IpAddresses ,error){
    // Get Path from config
	config, err := config.ConfigInit()
	if err != nil {
		return nil,err
	}
	jsonFile, err := os.Open(config.IPTable)
	// if we os.Open returns an error then handle it
	if err != nil {
		return nil,err
	}


	// defer the closing of our jsonFile so that we can parse it later on
	defer jsonFile.Close()

	// read our opened xmlFile as a byte array.
	byteValue, _ := ioutil.ReadAll(jsonFile)

	// we initialize our Users array
	var ipAddresses IpAddresses

	// we unmarshal our byteArray which contains our
	// jsonFile's content into 'users' which we defined above
	json.Unmarshal(byteValue, &ipAddresses)

    return &ipAddresses, nil
}

// WriteIpTable Write to IP table json file
func (i *IpAddresses) WriteIpTable() error {
	file, err := json.MarshalIndent(i, "", " ")
	if err != nil {
		return err
	}

	// Get Path from config
	config, err := config.ConfigInit()
	if err != nil {
		return err
	}

	err = ioutil.WriteFile(config.IPTable, file, 0644)
	if err != nil {
		return err
	}

	return nil
}

// PrintIpTable Print Ip table data for Cli
func PrintIpTable() error {
	table, err := ReadIpTable()

	if err != nil {
		return err
	}

	for i := 0; i < len(table.IpAddress); i++ {
		fmt.Printf("\nIP Address: %s\nLatency: %s\n-----------" +
			"-----------------\n",table.IpAddress[i].Ipv4,
			table.IpAddress[i].Latency)
	}
	return nil
}

// RemoveDuplicates This is a temporary fix current functions failing to remove
// Duplicate IP addresses from local IP table
func RemoveDuplicates() error {
	table, err := ReadIpTable()
	if err != nil {
		return err
	}

	var NoDuplicates IpAddresses
	for i, _:= range table.IpAddress {
		Exists := false
		for k := range NoDuplicates.IpAddress {
			if NoDuplicates.IpAddress[k].Ipv4 == table.IpAddress[i].Ipv4 {
				Exists = true
				break
			}
		}

		if Exists {
			continue
		}
		NoDuplicates.IpAddress = append(NoDuplicates.IpAddress, table.IpAddress[i])
	}

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

	return nil
}