package netconfig

import (
	"context"
	"net"
	"os"
	"path/filepath"
	"testing"

	"github.com/stretchr/testify/require"
)

// Validate the netplan configuration parser/writer functions.
func TestIfUpDown(t *testing.T) {
	// Setup test file.
	tmpDir, err := os.MkdirTemp("", "")
	require.NoError(t, err)
	configPath := filepath.Join(tmpDir, "interfaces")
	testDir, err := filepath.Abs("./tests/ifupdown")
	require.NoError(t, err)
	resultsDir := filepath.Join(testDir, "results")
	err = fileCopy(filepath.Join(testDir, "interfaces"), configPath)
	require.NoError(t, err)

	// Setup ifupdown and parse test file.
	i := new(ifUpDown)
	i.BaseConfig = configPath
	i.BackupDir = configPath + ".backup"
	err = i.ReadFile(i.BaseConfig)
	require.NoError(t, err)

	// Get the interfaces state.
	interfaces, err := i.GetInterfaces()
	require.NoError(t, err)

	// Verify interfaces read from file.
	err = testVerifyInterfaces(interfaces, resultsDir, 1)
	require.NoError(t, err)

	// Test setting the IP addresses on an interface.
	err = i.SetIfaceAddresses(context.Background(), "test_eth0", []*net.IPNet{
		{
			IP:   net.ParseIP("203.0.113.2"),
			Mask: net.CIDRMask(24, 32),
		},
		{
			IP:   net.ParseIP("203.0.113.3"),
			Mask: net.CIDRMask(24, 32),
		},
		{
			IP:   net.ParseIP("fc00::2"),
			Mask: net.CIDRMask(64, 128),
		},
	}, net.ParseIP("203.0.113.6"), net.ParseIP("fc00::1"))
	require.NoError(t, err)

	// Test setting routes on an interface.
	err = i.SetIfaceRoutes(context.Background(), "test_eth1", []*Route{
		{
			Destination: &net.IPNet{
				IP:   net.ParseIP("abcd:ef12:3455:3::"),
				Mask: net.CIDRMask(64, 128),
			},
			Gateway: net.ParseIP("abcd:ef12:3456:3::1"),
			Metric:  100,
		},
		{
			Destination: &net.IPNet{
				IP:   net.ParseIP("10.253.2.0"),
				Mask: net.CIDRMask(24, 32),
			},
			Gateway: net.ParseIP("1.2.3.1"),
			Metric:  100,
		},
	})
	require.NoError(t, err)

	// Verify we can read the modified files.
	i.Interfaces = nil
	err = i.ReadFile(i.BaseConfig)
	require.NoError(t, err)

	// Get the interfaces state.
	interfaces, err = i.GetInterfaces()
	require.NoError(t, err)

	// Verify interfaces read from file.
	err = testVerifyInterfaces(interfaces, resultsDir, 2)
	require.NoError(t, err)

	// Read the current file and expected state.
	err = testVerifyResults(resultsDir, tmpDir, 1)
	require.NoError(t, err)

	// Test setting the IP addresses on an interface.
	err = i.SetIfaceAddresses(context.Background(), "test_backend", []*net.IPNet{
		{
			IP:   net.ParseIP("1.2.10.4"),
			Mask: net.CIDRMask(24, 32),
		},
	}, net.ParseIP("1.2.10.254"), nil)
	require.NoError(t, err)

	// Test setting routes on an interface.
	err = i.SetIfaceRoutes(context.Background(), "test_eth0", []*Route{})
	require.NoError(t, err)

	// Verify we can read the modified files.
	i.Interfaces = nil
	err = i.ReadFile(i.BaseConfig)
	require.NoError(t, err)

	// Get the interfaces state.
	interfaces, err = i.GetInterfaces()
	require.NoError(t, err)

	// Verify interfaces read from file.
	err = testVerifyInterfaces(interfaces, resultsDir, 3)
	require.NoError(t, err)

	// Read the current file and expected state.
	err = testVerifyResults(resultsDir, tmpDir, 2)
	require.NoError(t, err)

	// Cleanup.
	err = os.RemoveAll(tmpDir)
	require.NoError(t, err)
}