package config

import (
	"os"
	"testing"

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

func TestParse(t *testing.T) {
	tests := map[string]struct {
		input  string
		result string
	}{
		"default": {
			input: ``,
			result: `
warewulf:
  autobuild overlays: true
  grubboot: false
  host overlay: true
  port: 9873
  secure: true
  tls port: 9874
  update interval: 60
nfs:
  enabled: true
  systemd name: nfsd
dhcp:
  enabled: true
  systemd name: dhcpd
  template: default
image mounts:
- dest: /etc/resolv.conf
  source: /etc/resolv.conf
ssh:
  key types:
  - ed25519
  - ecdsa
  - rsa
  - dsa
tftp:
  enabled: true
  ipxe:
    "00:00": undionly.kpxe
    "00:07": ipxe-snponly-x86_64.efi
    "00:09": ipxe-snponly-x86_64.efi
    "00:0B": arm64-efi/snponly.efi
  systemd name: tftp
api:
  enabled: false
  allowed subnets:
  - 127.0.0.0/8
  - ::1/128
`,
		},
		"cidr": {
			input: `
ipaddr: 192.168.0.1/24
`,
			result: `
ipaddr: 192.168.0.1
network: 192.168.0.0
netmask: 255.255.255.0
warewulf:
  autobuild overlays: true
  grubboot: false
  host overlay: true
  port: 9873
  secure: true
  tls port: 9874
  update interval: 60
nfs:
  enabled: true
  systemd name: nfsd
dhcp:
  enabled: true
  systemd name: dhcpd
  template: default
image mounts:
- dest: /etc/resolv.conf
  source: /etc/resolv.conf
ssh:
  key types:
  - ed25519
  - ecdsa
  - rsa
  - dsa
tftp:
  enabled: true
  ipxe:
    "00:00": undionly.kpxe
    "00:07": ipxe-snponly-x86_64.efi
    "00:09": ipxe-snponly-x86_64.efi
    "00:0B": arm64-efi/snponly.efi
  systemd name: tftp
api:
  enabled: false
  allowed subnets:
  - 127.0.0.0/8
  - ::1/128
`,
		},
		"cidr with conflicts": {
			input: `
ipaddr: 192.168.1.1/24
network: 192.168.0.0
netmask: 255.255.0.0
`,
			result: `
ipaddr: 192.168.1.1
network: 192.168.0.0
netmask: 255.255.0.0
warewulf:
  autobuild overlays: true
  grubboot: false
  host overlay: true
  port: 9873
  secure: true
  tls port: 9874
  update interval: 60
nfs:
  enabled: true
  systemd name: nfsd
dhcp:
  enabled: true
  systemd name: dhcpd
  template: default
image mounts:
- dest: /etc/resolv.conf
  source: /etc/resolv.conf
ssh:
  key types:
  - ed25519
  - ecdsa
  - rsa
  - dsa
tftp:
  enabled: true
  ipxe:
    "00:00": undionly.kpxe
    "00:07": ipxe-snponly-x86_64.efi
    "00:09": ipxe-snponly-x86_64.efi
    "00:0B": arm64-efi/snponly.efi
  systemd name: tftp
api:
  enabled: false
  allowed subnets:
  - 127.0.0.0/8
  - ::1/128
`,
		},
		"ipv6 cidr": {
			input: `
ipaddr6: "2001:db8::1/64"
`,
			result: `
ipaddr6: "2001:db8::1"
prefixlen6: "64"
warewulf:
  autobuild overlays: true
  grubboot: false
  host overlay: true
  port: 9873
  secure: true
  tls port: 9874
  update interval: 60
nfs:
  enabled: true
  systemd name: nfsd
dhcp:
  enabled: true
  systemd name: dhcpd
  template: default
image mounts:
- dest: /etc/resolv.conf
  source: /etc/resolv.conf
ssh:
  key types:
  - ed25519
  - ecdsa
  - rsa
  - dsa
tftp:
  enabled: true
  ipxe:
    "00:00": undionly.kpxe
    "00:07": ipxe-snponly-x86_64.efi
    "00:09": ipxe-snponly-x86_64.efi
    "00:0B": arm64-efi/snponly.efi
  systemd name: tftp
api:
  enabled: false
  allowed subnets:
  - 127.0.0.0/8
  - ::1/128
`,
		},
		"example": {
			input: `
ipaddr: 192.168.200.1
netmask: 255.255.255.0
network: 192.168.200.0
warewulf:
  port: 9873
  secure: false
  update interval: 60
  autobuild overlays: true
  host overlay: true
dhcp:
  enabled: true
  range start: 192.168.200.50
  range end: 192.168.200.99
  systemd name: dhcpd
tftp:
  enabled: true
  systemd name: tftp
nfs:
  enabled: true
  export paths:
  - path: /home
    export options: rw,sync
  - path: /opt
    export options: ro,sync,no_root_squash
  systemd name: nfs-server
image mounts:
  - source: /etc/resolv.conf
    dest: /etc/resolv.conf
    readonly: true
`,
			result: `
ipaddr: 192.168.200.1
netmask: 255.255.255.0
network: 192.168.200.0
warewulf:
  autobuild overlays: true
  grubboot: false
  host overlay: true
  port: 9873
  secure: false
  tls port: 9874
  update interval: 60
nfs:
  enabled: true
  systemd name: nfs-server
  export paths:
  - path: /home
    export options: rw,sync
  - path: /opt
    export options: ro,sync,no_root_squash
dhcp:
  enabled: true
  systemd name: dhcpd
  template: default
  range end: 192.168.200.99
  range start: 192.168.200.50
image mounts:
- dest: /etc/resolv.conf
  readonly: true
  source: /etc/resolv.conf
ssh:
  key types:
  - ed25519
  - ecdsa
  - rsa
  - dsa
tftp:
  enabled: true
  ipxe:
    "00:00": undionly.kpxe
    "00:07": ipxe-snponly-x86_64.efi
    "00:09": ipxe-snponly-x86_64.efi
    "00:0B": arm64-efi/snponly.efi
  systemd name: tftp
api:
  enabled: false
  allowed subnets:
  - 127.0.0.0/8
  - ::1/128
`,
		},
	}

	for name, tt := range tests {
		t.Run(name, func(t *testing.T) {
			conf := New()
			err := conf.Parse([]byte(tt.input), false)
			assert.NoError(t, err)
			removePaths(conf)
			result, err := conf.Dump()
			assert.NoError(t, err)
			assert.YAMLEq(t, tt.result, string(result))
		})
	}
}

// These paths are subject to change based on the environment the test is being
// run in.
func removePaths(conf *WarewulfYaml) {
	conf.Paths = nil
	conf.TFTP.TftpRoot = ""
}

func TestInitializedFromFile(t *testing.T) {
	example_warewulf_conf := ""
	tempWarewulfConf, warewulfConfErr := os.CreateTemp("", "warewulf.conf-")
	assert.NoError(t, warewulfConfErr)
	defer func() { _ = os.Remove(tempWarewulfConf.Name()) }()
	_, warewulfConfErr = tempWarewulfConf.Write([]byte(example_warewulf_conf))
	assert.NoError(t, warewulfConfErr)
	assert.NoError(t, tempWarewulfConf.Sync())

	conf := New()
	assert.False(t, conf.InitializedFromFile())
	assert.NoError(t, conf.Read(tempWarewulfConf.Name(), false))
	assert.True(t, conf.InitializedFromFile())
	assert.Equal(t, conf.GetWarewulfConf(), tempWarewulfConf.Name())
}

func TestCache(t *testing.T) {
	confOrig := New()
	confCached := Get()

	assert.Equal(t, 9873, confOrig.Warewulf.Port)
	assert.Equal(t, 9873, confCached.Warewulf.Port)

	confOrig.Warewulf.Port = 9999
	assert.Equal(t, 9999, confCached.Warewulf.Port)
	assert.Equal(t, 9999, Get().Warewulf.Port)

	New()
	assert.NotEqual(t, 9999, Get().Warewulf.Port)
}

func TestIpCIDR(t *testing.T) {
	tests := map[string]struct {
		ipaddr  string
		netmask string
		cidr    string
	}{
		"blank": {
			ipaddr:  "",
			netmask: "",
			cidr:    "",
		},
		"ip only": {
			ipaddr:  "192.168.0.1",
			netmask: "",
			cidr:    "",
		},
		"netmask only": {
			ipaddr:  "",
			netmask: "255.255.255.0",
			cidr:    "",
		},
		"full": {
			ipaddr:  "192.168.0.1",
			netmask: "255.255.255.0",
			cidr:    "192.168.0.1/24",
		},
		"invalid ip": {
			ipaddr:  "asdf",
			netmask: "255.255.255.0",
			cidr:    "",
		},
		"invalid netmask": {
			ipaddr:  "192.168.0.1",
			netmask: "asdf",
			cidr:    "",
		},
	}

	for name, tt := range tests {
		t.Run(name, func(t *testing.T) {
			conf := New()
			conf.Ipaddr = tt.ipaddr
			conf.Netmask = tt.netmask
			assert.Equal(t, tt.cidr, conf.IpCIDR())
		})
	}
}

func TestNetworkCIDR(t *testing.T) {
	tests := map[string]struct {
		network string
		netmask string
		cidr    string
	}{
		"blank": {
			network: "",
			netmask: "",
			cidr:    "",
		},
		"network only": {
			network: "192.168.0.0",
			netmask: "",
			cidr:    "",
		},
		"netmask only": {
			network: "",
			netmask: "255.255.255.0",
			cidr:    "",
		},
		"full": {
			network: "192.168.0.0",
			netmask: "255.255.255.0",
			cidr:    "192.168.0.0/24",
		},
		"invalid network": {
			network: "asdf",
			netmask: "255.255.255.0",
			cidr:    "",
		},
		"invalid netmask": {
			network: "192.168.0.0",
			netmask: "asdf",
			cidr:    "",
		},
	}

	for name, tt := range tests {
		t.Run(name, func(t *testing.T) {
			conf := New()
			conf.Network = tt.network
			conf.Netmask = tt.netmask
			assert.Equal(t, tt.cidr, conf.NetworkCIDR())
		})
	}
}

func TestIpCIDR6(t *testing.T) {
	tests := map[string]struct {
		ipaddr6    string
		prefixlen6 string
		cidr       string
	}{
		"blank": {
			ipaddr6:    "",
			prefixlen6: "",
			cidr:       "",
		},
		"ip only": {
			ipaddr6:    "2001:db8::1",
			prefixlen6: "",
			cidr:       "",
		},
		"prefix only": {
			ipaddr6:    "",
			prefixlen6: "64",
			cidr:       "",
		},
		"full": {
			ipaddr6:    "2001:db8::1",
			prefixlen6: "64",
			cidr:       "2001:db8::1/64",
		},
		"invalid ip": {
			ipaddr6:    "asdf",
			prefixlen6: "64",
			cidr:       "",
		},
		"invalid prefix": {
			ipaddr6:    "2001:db8::1",
			prefixlen6: "asdf",
			cidr:       "",
		},
		"ipv4 address": {
			ipaddr6:    "192.168.0.1",
			prefixlen6: "24",
			cidr:       "",
		},
	}

	for name, tt := range tests {
		t.Run(name, func(t *testing.T) {
			conf := New()
			conf.Ipaddr6 = tt.ipaddr6
			conf.PrefixLen6 = tt.prefixlen6
			assert.Equal(t, tt.cidr, conf.IpCIDR6())
		})
	}
}