package jd

import (
	"testing"
)

func TestUnmarshal(t *testing.T) {
	checkUnmarshal(t, ``, voidNode{})
	checkUnmarshal(t, `null`, jsonNull{})
	checkUnmarshal(t, `true`, jsonBool(true))
	checkUnmarshal(t, `"a"`, jsonString("a"))
	checkUnmarshal(t, `1.0`, jsonNumber(1.0))
	checkUnmarshal(t, `1`, jsonNumber(1.0))
	checkUnmarshal(t, `{}`, jsonObject{})
	checkUnmarshal(t, `[]`, jsonArray{})
}

func TestYamlV3BooleanKeys(t *testing.T) {
	// These tests would have failed with yaml.v2 (YAML 1.1) but pass with yaml.v3 (YAML 1.2)
	// because in YAML 1.2, only "true"/"false" are boolean literals, not "on"/"off"/"yes"/"no"
	//
	// With yaml.v2, these would fail with "unsupported key type bool" because:
	// - "on:" would be parsed as boolean key `true`
	// - "off:" would be parsed as boolean key `false`
	// - "yes:" would be parsed as boolean key `true`
	// - "no:" would be parsed as boolean key `false`
	//
	// With yaml.v3, these are parsed as string keys, which is the correct YAML 1.2 behavior.

	tests := []struct {
		name     string
		yaml     string
		expected JsonNode
	}{
		{
			name: "GitHub Actions on key",
			yaml: `on:
  push:
    branches: [main]`,
			expected: jsonObject{
				"on": jsonObject{
					"push": jsonObject{
						"branches": jsonArray{jsonString("main")},
					},
				},
			},
		},
		{
			name: "off key as string",
			yaml: `off: disabled`,
			expected: jsonObject{
				"off": jsonString("disabled"),
			},
		},
		{
			name: "yes key as string",
			yaml: `yes: confirmed`,
			expected: jsonObject{
				"yes": jsonString("confirmed"),
			},
		},
		{
			name: "no key as string",
			yaml: `no: rejected`,
			expected: jsonObject{
				"no": jsonString("rejected"),
			},
		},
		{
			name: "multiple YAML 1.1 boolean-like keys",
			yaml: `on: enabled
off: disabled
yes: confirmed
no: rejected`,
			expected: jsonObject{
				"on":  jsonString("enabled"),
				"off": jsonString("disabled"),
				"yes": jsonString("confirmed"),
				"no":  jsonString("rejected"),
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			node, err := ReadYamlString(tt.yaml)
			if err != nil {
				t.Errorf("ReadYamlString(%q) failed: %v", tt.yaml, err)
				return
			}

			if !tt.expected.Equals(node) {
				t.Errorf("ReadYamlString(%q) = %v, want %v", tt.yaml, node, tt.expected)
			}
		})
	}
}

func checkUnmarshal(t *testing.T, s string, n JsonNode) {
	node, err := ReadJsonString(s)
	if err != nil {
		t.Fatalf("%v", err.Error())
	}
	if !n.Equals(node) {
		t.Errorf("ReadJsonString(%v) = %v. Want %v.", s, node, n)
	}
	node, err = ReadYamlString(s)
	if err != nil {
		t.Fatalf("%v", err.Error())
	}
	if !n.Equals(node) {
		t.Errorf("ReadYamlString(%v) = %v. Want %v.", s, node, n)
	}
}

func TestReadDiff(t *testing.T) {
	checkReadDiff(t,
		Diff{
			DiffElement{
				Path:   p("a"),
				Remove: []JsonNode{jsonNumber(1)},
				Add:    []JsonNode{jsonNumber(2)},
			},
		},
		`@ ["a"]`,
		`- 1`,
		`+ 2`)
	checkReadDiff(t,
		Diff{
			DiffElement{
				Path:   p("a", 1.0, "b"),
				Remove: []JsonNode{jsonNumber(1)},
				Add:    []JsonNode{jsonNumber(2)},
			},
		},
		`@ ["a",1,"b"]`,
		`- 1`,
		`+ 2`)
	checkReadDiff(t,
		Diff{
			DiffElement{
				Path:   p(),
				Remove: []JsonNode{jsonNumber(1)},
				Add:    []JsonNode{jsonNumber(2)},
			},
			DiffElement{
				Path:   p(),
				Remove: []JsonNode{jsonNumber(2)},
				Add:    []JsonNode{jsonNumber(3)},
			},
		},
		`@ []`,
		`- 1`,
		`+ 2`,
		`@ []`,
		`- 2`,
		`+ 3`)
	checkReadDiff(t,
		Diff{
			DiffElement{
				Path:   p(0),
				Remove: []JsonNode{jsonNumber(1)},
			},
			DiffElement{
				Path:   p(2),
				Remove: []JsonNode{jsonNumber(4)},
			},
		},
		`@ [0]`,
		`[`,
		`- 1`,
		`  2`,
		`@ [2]`,
		`  3`,
		`- 4`,
		`]`,
	)
}

func TestReadDiffError(t *testing.T) {
	checkReadDiffError(t, `- 1`)
	checkReadDiffError(t, `+ 1`)
	checkReadDiffError(t,
		`@ ["a"]`,
		`@ ["b"]`)
	checkReadDiffError(t,
		`@ ["a"]`,
		`+ 1`,
		`- 2`)
	checkReadDiffError(t,
		`@ ["a"]`,
		`- 1`,
		`- 1`)
	checkReadDiffError(t,
		`@ ["a"]`,
		`+ 2`,
		`+ 2`)
	checkReadDiffError(t,
		`@ ["a"]`,
		`- 1`,
		`@ ["b"]`)
	checkReadDiffError(t,
		`@ `,
		`- 1`)
}

func checkReadDiff(t *testing.T, d Diff, diffLines ...string) {
	t.Helper()
	want := ""
	for _, dl := range diffLines {
		want += dl + "\n"
	}
	actual, err := readDiff(want)
	if err != nil {
		t.Errorf("%v", err.Error())
	}
	got := actual.Render()
	if got != want {
		t.Errorf("readDiff got %v. Want %v.", got, want)
	}
}

func checkReadDiffError(t *testing.T, diffLines ...string) {
	t.Helper()
	diff := ""
	for _, dl := range diffLines {
		diff += dl + "\n"
	}
	actual, err := readDiff(diff)
	if actual != nil {
		t.Errorf("readDiff(%v) = %v. Want nil.", diff, actual)
	}
	if err == nil {
		t.Errorf("Expected error for readDiff(%v).", diff)
	}
}

func p(elements ...interface{}) Path {
	var path jsonArray
	for _, e := range elements {
		n, err := NewJsonNode(e)
		if err != nil {
			panic(err)
		}
		path = append(path, n)
	}
	p, err := NewPath(path)
	if err != nil {
		panic(err)
	}
	return p
}