package main

import (
	"strings"
	"testing"
)

const miscInput string = `{"something":{"nested":"Here is my test value"},"array":["Hello","world","these","are","values"],"people":[{"first_name":"Charlie","last_name":"Johnson","age":22},{"first_name":"Tom","last_name":"Johnson","age":18},{"first_name":"Charlie","last_name":"Chaplin","age":122},{"first_name":"John","last_name":"Johnson","age":48}]}`
const mixedArray string = `{"array":["first",null,3,{"name":"second"},"third"]}`
const mixedArray2 string = `{"array":["first",null,3,"second",{"name":"third"}]}`

func TestSpecificCases(t *testing.T) {
	type test struct {
		name string
		program string
		quiet bool
		input string
		expected string
	}

	tests := []test {
		{
			name: "Verbose Extract",
			program: `s/#(~(people)~>_@(1>_#(~(first_name)~>_.|(..)>_*)-|(..)>_*)-|(..)>_*)-/p`,
			quiet: true,
			input: miscInput,
			expected: `"Tom"`,
		},
		{
			name: "Extract",
			program: `s/#("people">_ @(1>_#("first_name">_ .)-)-)-/p`,
			quiet: true,
			input: miscInput,
			expected: `"Tom"`,
		},
		{
			name: "Simple Extract",
			program: "s/#(\"people\" @(1 #(\"first_name\" (.>a))-)-)->_ <a/p",
			quiet: true,
			input: miscInput,
			expected: `"Tom"`,
		},
		{
			name: "Larger Extract",
			program: "s/#(\"people\" @(2 (.>a))-)->_ `<a`/p",
			quiet: true,
			input: miscInput,
			expected: `{"first_name":"Charlie","last_name":"Chaplin","age":122}`,
		},
		{
			name: "Extract ages",
			program: "s/#(\"people\">_ :(#(\"age\">_ .)-):)-/p",
			quiet: true,
			input: miscInput,
			expected: `[22,18,122,48]`,
		},
		{
			name: "Low memory count people",
			program: "aX/#(\"people\" :(#()#):)#>_ `1`/o es/#()#/{ xs/.*%+/p }",
			quiet: true,
			input: miscInput,
			expected: "4",
		},
		{
			name: "Get full names",
			program: "s/#(\"people\">_ .)-/{ s/:():/p as/:(#()#):/{ xdx } s/:(#((\"first_name\"|\"last_name\") .)#)-/X es/@(.#()-)-/{ xs/(#(\"first_name\"\".*>a\")#|#(\"last_name\"\".*>b\")#|.)*>_`\"<a <b\"`/Xxs/-(..)@/p } }",
			quiet: true,
			input: miscInput,
			expected: `["Charlie Johnson","Tom Johnson","Charlie Chaplin","John Johnson"]`,
		},
		{
			name: "Get full names 2",
			program: "s/#(\"people\">_.)-/{ s/:():/p as/:(#()#):/{ xdx } X/:(#((\"first_name\"|\"last_name\") .)#)-/o es/@(.#()-)-/{ xX/(#(\"first_name\"\".*>a\")#|#(\"last_name\"\".*>b\")#|.)*_`\"<a <b\"`/xs/-(..)@/p } }",
			quiet: true,
			input: miscInput,
			expected: `["Charlie Johnson","Tom Johnson","Charlie Chaplin","John Johnson"]`,
		},
		{
			name: "Change full names in place",
			program: "s/#(\"people\"@(.#(\"first_name\".)#)@)#/{ Nms/#(\"people\"@(.(#(\"first_name\"\".*>a\"\"last_name\"\".*>b\")#_)`#(\"name\"\"<a <b\")#`)@)#/ }",
			input: miscInput,
			expected: `{"something":{"nested":"Here is my test value"},"array":["Hello","world","these","are","values"],"people":[{"name":"Charlie Johnson","age":22},{"name":"Tom Johnson","age":18},{"name":"Charlie Chaplin","age":122},{"name":"John Johnson","age":48}]}`,
		},
		{
			name: "Get full names with substitute next command",
			program: "s/#(\"people\"_:(#(\"first_name\"_.)-)-)-/{ N/#(\"people\"_:(#(\"last_name\"_.)-)-)-/{ s/-(-(~(.*` `)-~(.*)-)~):/p }}",
			quiet: true,
			input: miscInput,
			expected: `["Charlie Johnson","Tom Johnson","Charlie Chaplin","John Johnson"]`,
		},
		{
			name: "Get full names with merge full command",
			program: "s/#(\"people\"_:():)-/p M/#(\"people\"@(.#()#)@)#/{ s/#(\"people\"_@(.#[(\"first_name\"\".*>a\"|\"last_name\"\".*>b\"|..)_]-`\"<a <b\"`)@)-/p }",
			quiet: true,
			input: miscInput,
			expected: `["Charlie Johnson","Tom Johnson","Charlie Chaplin","John Johnson"]`,
		},
		{
			name: "Verbose concat array values",
			program: "as/#(\"array\"_:():)-/{ :s N/#(._.)-/{ es/.*:():/be mbs } :em s/:(-(~(.*` `)-*~(.*)-)~)-/p }",
			quiet: true,
			input: miscInput,
			expected: `"Hello world these are values"`,
		},
		{
			name: "Short concat array values",
			 program: "M/#(\"array\":():)#/{ s/#(\"array\"_:(.*)-)-/ s/-(~(.*` `)-*~(.*)-)~/p }",
			quiet: true,
			input: miscInput,
			expected: `"Hello world these are values"`,
		},
		{
			name: "Drop first element of array",
			program: `s/#("people"@(0 .)@)#/d`,
			input: miscInput,
			expected: `{"something":{"nested":"Here is my test value"},"array":["Hello","world","these","are","values"],"people":[{"first_name":"Tom","last_name":"Johnson","age":18},{"first_name":"Charlie","last_name":"Chaplin","age":122},{"first_name":"John","last_name":"Johnson","age":48}]}`,
		},
		{
			name: "Drop last element of array",
			program: `M/#("people"@(.,)@)#/{ Ed }`,
			input: miscInput,
			expected: `{"something":{"nested":"Here is my test value"},"array":["Hello","world","these","are","values"],"people":[{"first_name":"Charlie","last_name":"Johnson","age":22},{"first_name":"Tom","last_name":"Johnson","age":18},{"first_name":"Charlie","last_name":"Chaplin","age":122}]}`,
		},
		{
			name: "Drop last element of simple array",
			program: `s/#("array"@(..)@)#/{ Ed }`,
			input: miscInput,
			expected: `{"something":{"nested":"Here is my test value"},"array":["Hello","world","these","are"],"people":[{"first_name":"Charlie","last_name":"Johnson","age":22},{"first_name":"Tom","last_name":"Johnson","age":18},{"first_name":"Charlie","last_name":"Chaplin","age":122},{"first_name":"John","last_name":"Johnson","age":48}]}`,
		},
		{
			name: "Drop last element of mixed array",
			program: `M/#("array"@(.,)@)#/{ Ed }`,
			input: mixedArray,
			expected: `{"array":["first",null,3,{"name":"second"}]}`,
		},
		{
			name: "Drop last element of mixed array 2",
			program: `M/#("array"@(.,)@)#/{ Ed }`,
			input: mixedArray2,
			expected: `{"array":["first",null,3,"second"]}`,
		},
		{
			name: "Prepend to array",
			program: "as/#(\"array\":(`\"First\"`):)#/",
			input: miscInput,
			expected: `{"something":{"nested":"Here is my test value"},"array":["First","Hello","world","these","are","values"],"people":[{"first_name":"Charlie","last_name":"Johnson","age":22},{"first_name":"Tom","last_name":"Johnson","age":18},{"first_name":"Charlie","last_name":"Chaplin","age":122},{"first_name":"John","last_name":"Johnson","age":48}]}`,
		},
		{
			name: "Append to array",
			program: "es/#(\"array\":(`\"Last\"`):)#/",
			input: miscInput,
			expected: `{"something":{"nested":"Here is my test value"},"array":["Hello","world","these","are","values","Last"],"people":[{"first_name":"Charlie","last_name":"Johnson","age":22},{"first_name":"Tom","last_name":"Johnson","age":18},{"first_name":"Charlie","last_name":"Chaplin","age":122},{"first_name":"John","last_name":"Johnson","age":48}]}`,
		},
	}

	for _, test := range tests {
		t.Logf("Running test: %s", test.name)

		var output strings.Builder
		run(config {
			quiet: test.quiet,
			program: test.program,
			in: strings.NewReader(test.input),
			out: &output,
		})
		
		if output.String() != test.expected {
			t.Errorf("Ran '%s' and expected %s but got %s", test.program, test.expected, output.String())
		}
	}
}