package list

import (
	"bytes"
	"os"
	"strings"

	"testing"

	warewulfconf "github.com/warewulf/warewulf/internal/pkg/config"

	"github.com/stretchr/testify/assert"
	"github.com/warewulf/warewulf/internal/pkg/node"
	"github.com/warewulf/warewulf/internal/pkg/testenv"
	"github.com/warewulf/warewulf/internal/pkg/warewulfd"
	"github.com/warewulf/warewulf/internal/pkg/wwlog"
)

func Test_List(t *testing.T) {
	tests := []struct {
		name    string
		args    []string
		wantErr bool
		stdout  string
		inDb    string
	}{
		{
			name:    "single node list",
			args:    []string{},
			wantErr: false,
			stdout: `
NODE NAME  PROFILES   NETWORK
---------  --------   -------
n01        [default]  --
`,
			inDb: `WW_INTERNAL: 45
nodeprofiles:
  default: {}
nodes:
  n01:
    profiles:
    - default
`,
		},
		{
			name:    "multiple nodes list",
			args:    []string{},
			wantErr: false,
			stdout: `
NODE NAME  PROFILES   NETWORK
---------  --------   -------
n01        [default]  --
n02        [default]  --
`,
			inDb: `WW_INTERNAL: 45
nodeprofiles:
  default: {}
nodes:
  n01:
    profiles:
    - default
  n02:
   profiles:
   - default
`,
		},
		{
			name:    "node list returns multiple nodes",
			args:    []string{"n01,n02"},
			wantErr: false,
			stdout: `
NODE NAME  PROFILES   NETWORK
---------  --------   -------
n01        [default]  --
n02        [default]  --
`,
			inDb: `WW_INTERNAL: 45
nodeprofiles:
  default: {}
nodes:
  n01:
    profiles:
    - default
  n02:
   profiles:
   - default
`,
		},
		{
			name:    "node list returns multiple nodes (case 2)",
			args:    []string{"n01,n03"},
			wantErr: false,
			stdout: `
NODE NAME  PROFILES   NETWORK
---------  --------   -------
n01        [default]  --
n03        [default]  --
`,
			inDb: `WW_INTERNAL: 45
nodeprofiles:
  default: {}
nodes:
  n01:
    profiles:
    - default
  n02:
   profiles:
   - default
  n03:
   profiles:
   - default
  n04:
   profiles:
   - default
  n05:
   profiles:
   - default
`,
		},
		{
			name:    "node list returns one node",
			args:    []string{"n01,"},
			wantErr: false,
			stdout: `
NODE NAME  PROFILES   NETWORK
---------  --------   -------
n01        [default]  --
`,
			inDb: `WW_INTERNAL: 45
nodeprofiles:
  default: {}
nodes:
  n01:
    profiles:
    - default
  n02:
   profiles:
   - default
`,
		},
		{
			name:    "node list profile with network",
			args:    []string{},
			wantErr: false,
			stdout: `
NODE NAME  PROFILES   NETWORK
---------  --------   -------
n01        [default]  default
`,
			inDb: `WW_INTERNAL: 45
nodeprofiles:
  default:
    network devices:
      default:
        device: eth0
nodes:
  n01:
    profiles:
    - default
`},
		{
			name:    "node list profile with comment",
			args:    []string{"-a"},
			wantErr: false,
			stdout: `
NODE  FIELD     PROFILE  VALUE
----  -----     -------  -----
n01   Comment   default  profilecomment
n01   Profiles  --       default
`,
			inDb: `WW_INTERNAL: 45
nodeprofiles:
  default:
    comment: profilecomment
nodes:
  n01:
    profiles:
    - default
`},
		{
			name:    "node list profile with comment superseded",
			args:    []string{"-a"},
			wantErr: false,
			stdout: `
NODE  FIELD     PROFILE     VALUE
----  -----     -------     -----
n01   Comment   SUPERSEDED  nodecomment
n01   Profiles  --          default
`,
			inDb: `WW_INTERNAL: 45
nodeprofiles:
  default:
    comment: profilecomment
nodes:
  n01:
    comment: nodecomment
    profiles:
    - default
`},
		{
			name:    "node list profile with ipmi user",
			args:    []string{"-i"},
			wantErr: false,
			stdout: `
NODE  IPMI IPADDR  IPMI PORT  IPMI USERNAME  IPMI INTERFACE
----  -----------  ---------  -------------  --------------
n01   <nil>        --         admin          --
`,
			inDb: `WW_INTERNAL: 45
nodeprofiles:
  default:
    ipmi:
      username: admin
nodes:
  n01:
    profiles:
    - default
`},
		{
			name:    "node list profile with ipmi user superseded",
			args:    []string{"-i"},
			wantErr: false,
			stdout: `
NODE  IPMI IPADDR  IPMI PORT  IPMI USERNAME  IPMI INTERFACE
----  -----------  ---------  -------------  --------------
n01   <nil>        --         user           --
`,
			inDb: `WW_INTERNAL: 45
nodeprofiles:
  default:
    ipmi:
      username: admin
nodes:
  n01:
    ipmi:
      username: user
    profiles:
    - default
`},
		{
			name:    "multiple profiles list",
			args:    []string{},
			wantErr: false,
			stdout: `
NODE NAME  PROFILES  NETWORK
---------  --------  -------
n01        [p1 p2]   --
`,
			inDb: `WW_INTERNAL: 45
nodeprofiles:
  p1: {}
  p2: {}
nodes:
  n01:
    profiles:
    - p1
    - p2
`},
		{
			name:    "multiple profiles list all",
			args:    []string{"-a"},
			wantErr: false,
			stdout: `
NODE  FIELD     PROFILE  VALUE
----  -----     -------  -----
n01   Profiles  --       p1,p2
`,
			inDb: `WW_INTERNAL: 45
nodeprofiles:
  p1: {}
  p2: {}
nodes:
  n01:
    profiles:
    - p1
    - p2
`},
		{
			name:    "multiple overlays list long with negation",
			args:    []string{"-l"},
			wantErr: false,
			stdout: `
NODE NAME  KERNEL OVERRIDE  CONTAINER  OVERLAYS (S/R)
---------  ---------------  ---------  --------------
n01        --               --         /rop1,rop2
`,
			inDb: `WW_INTERNAL: 45
nodeprofiles:
  p1:
    runtime overlay:
    - rop1
    - rop2
nodes:
  n01:
    profiles:
    - p1
`},
		{
			name:    "multiple overlays list long",
			args:    []string{"-l"},
			wantErr: false,
			stdout: `
NODE NAME  KERNEL OVERRIDE  CONTAINER  OVERLAYS (S/R)
---------  ---------------  ---------  --------------
n01        --               --         sop1/nop1,~rop1,rop1,rop2
`,
			inDb: `WW_INTERNAL: 45
nodeprofiles:
  p1:
    system overlay:
    - sop1
    runtime overlay:
    - rop1
    - rop2
nodes:
  n01:
    profiles:
    - p1
    runtime overlay:
    - nop1
    - ~rop1
`},
		{
			name:    "multiple overlays list all with negation",
			args:    []string{"-a"},
			wantErr: false,
			stdout: `
NODE  FIELD           PROFILE  VALUE
----  -----           -------  -----
n01   Profiles        --       p1
n01   RuntimeOverlay  p1+      nop1,~rop1,rop1,rop2
n01   SystemOverlay   p1       sop1
`,
			inDb: `WW_INTERNAL: 45
nodeprofiles:
  p1:
    system overlay:
    - sop1
    runtime overlay:
    - rop1
    - rop2
nodes:
  n01:
    profiles:
    - p1
    runtime overlay:
    - nop1
    - ~rop1
`},
		{
			name:    "multiple overlays list all",
			args:    []string{"-a"},
			wantErr: false,
			stdout: `
NODE  FIELD           PROFILE  VALUE
----  -----           -------  -----
n01   Profiles        --       p1
n01   RuntimeOverlay  p1+      nop1,rop1,rop2
`,
			inDb: `WW_INTERNAL: 45
nodeprofiles:
  p1:
    runtime overlay:
    - rop1
    - rop2
nodes:
  n01:
    profiles:
    - p1
    runtime overlay:
    - nop1
`},
		{
			name:    "network onboot",
			args:    []string{"-a"},
			wantErr: false,
			stdout: `
NODE  FIELD                    PROFILE  VALUE
----  -----                    -------  -----
n1    NetDevs[default].OnBoot  --       true
`,
			inDb: `WW_INTERNAL: 45
nodes:
  n1:
    network devices:
      default:
        onboot: true
`},
		{
			name:    "empty network device",
			args:    []string{"-a"},
			wantErr: false,
			stdout: `
NODE  FIELD  PROFILE  VALUE
----  -----  -------  -----
`,
			inDb: `WW_INTERNAL: 46
nodes:
  wwnode1:
    network devices:
      default: {}
`},
	}

	conf_yml := `WW_INTERNAL: 0`
	tempWarewulfConf, warewulfConfErr := os.CreateTemp("", "warewulf.conf-")
	assert.NoError(t, warewulfConfErr)
	defer os.Remove(tempWarewulfConf.Name())
	_, warewulfConfErr = tempWarewulfConf.Write([]byte(conf_yml))
	assert.NoError(t, warewulfConfErr)
	assert.NoError(t, tempWarewulfConf.Sync())
	assert.NoError(t, warewulfconf.New().Read(tempWarewulfConf.Name()))

	tempNodeConf, nodesConfErr := os.CreateTemp("", "nodes.conf-")
	assert.NoError(t, nodesConfErr)
	defer os.Remove(tempNodeConf.Name())
	node.ConfigFile = tempNodeConf.Name()
	warewulfd.SetNoDaemon()
	for _, tt := range tests {
		var err error
		_, err = tempNodeConf.Seek(0, 0)
		assert.NoError(t, err)
		assert.NoError(t, tempNodeConf.Truncate(0))
		_, err = tempNodeConf.Write([]byte(tt.inDb))
		assert.NoError(t, err)
		assert.NoError(t, tempNodeConf.Sync())

		t.Run(tt.name, func(t *testing.T) {
			baseCmd := GetCommand()
			baseCmd.SetArgs(tt.args)
			buf := new(bytes.Buffer)
			baseCmd.SetOut(buf)
			baseCmd.SetErr(buf)
			wwlog.SetLogWriter(buf)
			wwlog.SetLogWriterErr(buf)
			wwlog.SetLogWriterInfo(buf)
			err := baseCmd.Execute()
			assert.NoError(t, err)
			assert.Equal(t, strings.TrimSpace(tt.stdout), strings.TrimSpace(buf.String()))

		})
	}
}

func TestListMultipleFormats(t *testing.T) {
	tests := []struct {
		name    string
		args    []string
		stdout  string
		inDb    string
		wantErr bool
	}{
		{
			name: "single node list yaml output",
			args: []string{"-y"},
			stdout: `
- profiles:
    - default
  kernel: {}
  ipmi: {}
`,
			inDb: `WW_INTERNAL: 43
nodeprofiles:
  default: {}
nodes:
  n01:
    profiles:
    - default
`,
		},
		{
			name: "single node list json output",
			args: []string{"-j"},
			stdout: `
[
  {
    "Discoverable": "",
    "AssetKey": "",
    "Profiles": [
      "default"
    ],
    "Comment": "",
    "ClusterName": "",
    "ContainerName": "",
    "Ipxe": "",
    "RuntimeOverlay": null,
    "SystemOverlay": null,
    "Kernel": {},
    "Ipmi": {
      "UserName": "",
      "Password": "",
      "Ipaddr": "",
      "Gateway": "",
      "Netmask": "",
      "Port": "",
      "Interface": "",
      "EscapeChar": "",
      "Write": "",
      "Tags": {}
    },
    "Init": "",
    "Root": "",
    "NetDevs": {},
    "Tags": {},
    "PrimaryNetDev": "",
    "Disks": null,
    "FileSystems": null
  }
]
`,
			inDb: `WW_INTERNAL: 43
nodeprofiles:
  default: {}
nodes:
  n01:
    profiles:
    - default
`,
		},
		{
			name: "multiple nodes list json output",
			args: []string{"-j"},
			stdout: `
[
  {
    "Discoverable": "",
    "AssetKey": "",
    "Profiles": [
      "default"
    ],
    "Comment": "",
    "ClusterName": "",
    "ContainerName": "",
    "Ipxe": "",
    "RuntimeOverlay": null,
    "SystemOverlay": null,
    "Kernel": {},
    "Ipmi": {
      "UserName": "",
      "Password": "",
      "Ipaddr": "",
      "Gateway": "",
      "Netmask": "",
      "Port": "",
      "Interface": "",
      "EscapeChar": "",
      "Write": "",
      "Tags": {}
    },
    "Init": "",
    "Root": "",
    "NetDevs": {},
    "Tags": {},
    "PrimaryNetDev": "",
    "Disks": null,
    "FileSystems": null
  },
  {
    "Discoverable": "",
    "AssetKey": "",
    "Profiles": [
      "default"
    ],
    "Comment": "",
    "ClusterName": "",
    "ContainerName": "",
    "Ipxe": "",
    "RuntimeOverlay": null,
    "SystemOverlay": null,
    "Kernel": {},
    "Ipmi": {
      "UserName": "",
      "Password": "",
      "Ipaddr": "",
      "Gateway": "",
      "Netmask": "",
      "Port": "",
      "Interface": "",
      "EscapeChar": "",
      "Write": "",
      "Tags": {}
    },
    "Init": "",
    "Root": "",
    "NetDevs": {},
    "Tags": {},
    "PrimaryNetDev": "",
    "Disks": null,
    "FileSystems": null
  }
]
`,
			inDb: `WW_INTERNAL: 43
nodeprofiles:
  default: {}
nodes:
  n01:
    profiles:
    - default
  n02:
    profiles:
    - default
`,
		},
		{
			name: "multiple nodes list yaml output",
			args: []string{"-y"},
			stdout: `
- profiles:
    - default
  kernel: {}
  ipmi: {}
- profiles:
    - default
  kernel: {}
  ipmi: {}
`,
			inDb: `WW_INTERNAL: 43
nodeprofiles:
  default: {}
nodes:
  n01:
    profiles:
    - default
  n02:
    profiles:
    - default
`,
		},
	}

	warewulfd.SetNoDaemon()
	for _, tt := range tests {
		env := testenv.New(t)
		env.WriteFile(t, "etc/warewulf/nodes.conf", tt.inDb)
		var err error
		t.Run(tt.name, func(t *testing.T) {
			buf := new(bytes.Buffer)
			wwlog.SetLogWriter(buf)
			baseCmd := GetCommand()
			baseCmd.SetArgs(tt.args)
			baseCmd.SetOut(buf)
			baseCmd.SetErr(buf)
			err = baseCmd.Execute()
			if tt.wantErr {
				assert.Error(t, err)
			} else {
				assert.NoError(t, err)
			}
			assert.Equal(t, strings.TrimSpace(tt.stdout), strings.TrimSpace(buf.String()))
		})
	}
}