// Copyright 2014 Google Inc. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package parser

import (
	"bytes"
	"testing"
)

var validPrinterTestCases = []struct {
	name   string
	input  string
	output string
}{
	{
		input: `
foo {}
`,
		output: `
foo {}
`,
	},
	{
		input: `
foo(name= "abc",num= 4,)
`,
		output: `
foo {
    name: "abc",
    num: 4,
}
`,
	},
	{
		input: `
			foo {
				stuff: ["asdf", "jkl;", "qwert",
					"uiop", "bnm,"]
			}
			`,
		output: `
foo {
    stuff: [
        "asdf",
        "bnm,",
        "jkl;",
        "qwert",
        "uiop",
    ],
}
`,
	},
	{
		input: `
		        var = "asdf"
			foo {
				stuff: ["asdf"] + var,
			}`,
		output: `
var = "asdf"
foo {
    stuff: ["asdf"] + var,
}
`,
	},
	{
		input: `
		        var = "asdf"
			foo {
				stuff: [
				    "asdf"
				] + var,
			}`,
		output: `
var = "asdf"
foo {
    stuff: [
        "asdf",
    ] + var,
}
`,
	},
	{
		input: `
		        var = "asdf"
			foo {
				stuff: ["asdf"] + var + ["qwert"],
			}`,
		output: `
var = "asdf"
foo {
    stuff: ["asdf"] + var + ["qwert"],
}
`,
	},
	{
		input: `
		foo {
			stuff: {
				isGood: true,
				name: "bar",
				num: 4,
			}
		}
		`,
		output: `
foo {
    stuff: {
        isGood: true,
        name: "bar",
        num: 4,
    },
}
`,
	},
	{
		input: `
// comment1
foo {
	// comment2
	isGood: true,  // comment3
}
`,
		output: `
// comment1
foo {
    // comment2
    isGood: true, // comment3
}
`,
	},
	{
		input: `
foo {
	name: "abc",
	num: 4,
}

bar  {
	name: "def",
	num: 5,
}
		`,
		output: `
foo {
    name: "abc",
    num: 4,
}

bar {
    name: "def",
    num: 5,
}
`,
	},
	{
		input: `
foo {
    bar: "b" +
        "a" +
	"z",
}
`,
		output: `
foo {
    bar: "b" +
        "a" +
        "z",
}
`,
	},
	{
		input: `
foo = "stuff"
bar = foo
baz = foo + bar
baz += foo
`,
		output: `
foo = "stuff"
bar = foo
baz = foo + bar
baz += foo
`,
	},
	{
		input: `
foo = 100
bar = foo
baz = foo + bar
baz += foo
`,
		output: `
foo = 100
bar = foo
baz = foo + bar
baz += foo
`,
	},
	{
		input: `
foo = "bar " +
    "" +
    "baz"
`,
		output: `
foo = "bar " +
    "" +
    "baz"
`,
	},
	{
		input: `
//test
test /* test */ {
    srcs: [
        /*"bootstrap/bootstrap.go",
    "bootstrap/cleanup.go",*/
        "bootstrap/command.go",
        "bootstrap/doc.go", //doc.go
        "bootstrap/config.go", //config.go
    ],
    deps: ["libabc"],
    incs: []
} //test
//test
test2 {
}


//test3
`,
		output: `
//test
test /* test */ {
    srcs: [
        /*"bootstrap/bootstrap.go",
        "bootstrap/cleanup.go",*/
        "bootstrap/command.go",
        "bootstrap/config.go", //config.go
        "bootstrap/doc.go", //doc.go
    ],
    deps: ["libabc"],
    incs: [],
} //test
//test

test2 {
}

//test3
`,
	},
	{
		input: `
// test
module // test

 {
    srcs
   : [
        "src1.c",
        "src2.c",
    ],
//test
}
//test2
`,
		output: `
// test
module { // test
    srcs: [
        "src1.c",
        "src2.c",
    ],
    //test
}

//test2
`,
	},
	{
		input: `
/*test {
    test: true,
}*/

test {
/*test: true,*/
}

// This
/* Is *//* A */ // A
// A

// Multiline
// Comment

test {}

// This
/* Is */
// A
// Trailing

// Multiline
// Comment
`,
		output: `
/*test {
    test: true,
}*/

test {
    /*test: true,*/
}

// This
/* Is */ /* A */ // A
// A

// Multiline
// Comment

test {}

// This
/* Is */
// A
// Trailing

// Multiline
// Comment
`,
	},
	{
		input: `
test // test

// test
{
}
`,
		output: `
test { // test
// test
}
`,
	},
	{
		input: `
// test
stuff {
    namespace: "google",
    string_vars: [
      {
          var: "one",
          values: [ "one_a", "one_b",],
      },
      {
          var: "two",
          values: [ "two_a", "two_b", ],
      },
    ],
}`,
		output: `
// test
stuff {
    namespace: "google",
    string_vars: [
        {
            var: "one",
            values: [
                "one_a",
                "one_b",
            ],
        },
        {
            var: "two",
            values: [
                "two_a",
                "two_b",
            ],
        },
    ],
}
`,
	},
	{
		input: `
// test
stuff {
    namespace: "google",
    list_of_lists: [
        [ "a", "b" ],
        [ "c", "d" ],
    ],
}
`,
		output: `
// test
stuff {
    namespace: "google",
    list_of_lists: [
        [
            "a",
            "b",
        ],
        [
            "c",
            "d",
        ],
    ],
}
`,
	},
	{
		input: `
// test
stuff {
    namespace: "google",
    list_of_structs: [{ key1: "a", key2: "b" }],
}
`,
		output: `
// test
stuff {
    namespace: "google",
    list_of_structs: [
        {
            key1: "a",
            key2: "b",
        },
    ],
}
`,
	},
	{
		input: `
// test
foo {
    stuff: [
        "a", // great comment
        "b",
    ],
}
`,
		output: `
// test
foo {
    stuff: [
        "a", // great comment
        "b",
    ],
}
`,
	},
	{
		input: `
// test
foo {
    stuff: [
        "a",
        // b comment
        "b",
    ],
}
`,
		output: `
// test
foo {
    stuff: [
        "a",
        // b comment
        "b",
    ],
}
`,
	},
	{
		input: `
// test
foo {
    stuff: [
        "a", // a comment
        // b comment
        "b",
    ],
}
`,
		output: `
// test
foo {
    stuff: [
        "a", // a comment
        // b comment
        "b",
    ],
}
`,
	},
	{
		input: `
// test
foo {
    stuff: [
        "a",
        // b comment
        // on multiline
        "b",
    ],
}
`,
		output: `
// test
foo {
    stuff: [
        "a",
        // b comment
        // on multiline
        "b",
    ],
}
`,
	},
	{ // Line comment are treat as groups separator
		input: `
// test
foo {
    stuff: [
        "b",
        // a comment
        "a",
    ],
}
`,
		output: `
// test
foo {
    stuff: [
        "b",
        // a comment
        "a",
    ],
}
`,
	},
	{
		name: "Basic selects",
		input: `
// test
foo {
    stuff: select(soong_config_variable("my_namespace", "my_variable"), {
        "a": "a2",
        // test2
        "b": "b2",
        // test3
        default: "c2",
    }),
}
`,
		output: `
// test
foo {
    stuff: select(soong_config_variable("my_namespace", "my_variable"), {
        "a": "a2",
        // test2
        "b": "b2",
        // test3
        default: "c2",
    }),
}
`,
	},
	{
		name: "Remove select with only default",
		input: `
// test
foo {
    stuff: select(soong_config_variable("my_namespace", "my_variable"), {
        // test2
        default: "c2",
    }),
}
`,
		output: `
// test
foo {
    stuff: "c2", // test2
}
`,
	},
	{
		name: "Appended selects",
		input: `
// test
foo {
    stuff: select(soong_config_variable("my_namespace", "my_variable"), {
        "a": "a2",
        // test2
        "b": "b2",
        // test3
        default: "c2",
    }) + select(release_variable("RELEASE_TEST"), {
        "d": "d2",
        "e": "e2",
        default: "f2",
    }),
}
`,
		output: `
// test
foo {
    stuff: select(soong_config_variable("my_namespace", "my_variable"), {
        "a": "a2",
        // test2
        "b": "b2",
        // test3
        default: "c2",
    }) + select(release_variable("RELEASE_TEST"), {
        "d": "d2",
        "e": "e2",
        default: "f2",
    }),
}
`,
	},
	{
		name: "Select with unset property",
		input: `
foo {
    stuff: select(soong_config_variable("my_namespace", "my_variable"), {
        "foo": unset,
        default: "c2",
    }),
}
`,
		output: `
foo {
    stuff: select(soong_config_variable("my_namespace", "my_variable"), {
        "foo": unset,
        default: "c2",
    }),
}
`,
	},
	{
		name: "Multi-condition select",
		input: `
foo {
    stuff: select((arch(), os()), {
        ("x86", "linux"): "a",
        (default, default): "b",
    }),
}
`,
		output: `
foo {
    stuff: select((arch(), os()), {
        ("x86", "linux"): "a",
        (default, default): "b",
    }),
}
`,
	},
	{
		name: "Multi-condition select with conditions on new lines",
		input: `
foo {
    stuff: select((arch(), 
    os()), {
        ("x86", "linux"): "a",
        (default, default): "b",
    }),
}
`,
		output: `
foo {
    stuff: select((
        arch(),
        os(),
    ), {
        ("x86", "linux"): "a",
        (default, default): "b",
    }),
}
`,
	},
	{
		name: "Select with multiline inner expression",
		input: `
foo {
    cflags: [
        "-DPRODUCT_COMPATIBLE_PROPERTY",
        "-DRIL_SHLIB",
        "-Wall",
        "-Wextra",
        "-Werror",
    ] + select(soong_config_variable("sim", "sim_count"), {
        "2": [
            "-DANDROID_MULTI_SIM",
            "-DANDROID_SIM_COUNT_2",
        ],
        default: [],
    }),
}
`,
		output: `
foo {
    cflags: [
        "-DPRODUCT_COMPATIBLE_PROPERTY",
        "-DRIL_SHLIB",
        "-Wall",
        "-Werror",
        "-Wextra",
    ] + select(soong_config_variable("sim", "sim_count"), {
        "2": [
            "-DANDROID_MULTI_SIM",
            "-DANDROID_SIM_COUNT_2",
        ],
        default: [],
    }),
}
`,
	},
	{
		name: "Select with bindings",
		input: `
foo {
    stuff: select(arch(), {
        "x86": "a",
        any
          @ baz: "b" + baz,
    }),
}
`,
		output: `
foo {
    stuff: select(arch(), {
        "x86": "a",
        any @ baz: "b" + baz,
    }),
}
`,
	},
	{
		name: "Simplify select",
		input: `
foo {
    stuff: select(arch(), {
        default: "a",
    }),
}
`,
		output: `
foo {
    stuff: "a",
}
`,
	},
	{
		name: "Simplify select with append",
		input: `
foo {
    stuff: select(arch(), {
        default: "a",
    }) + "foo",
}
`,
		output: `
foo {
    stuff: "a" + "foo",
}
`,
	},
}

func TestPrinter(t *testing.T) {
	for _, testCase := range validPrinterTestCases {
		t.Run(testCase.name, func(t *testing.T) {
			in := testCase.input[1:]
			expected := testCase.output[1:]

			r := bytes.NewBufferString(in)
			file, errs := Parse("", r)
			if len(errs) != 0 {
				t.Errorf("test case: %s", in)
				t.Errorf("unexpected errors:")
				for _, err := range errs {
					t.Errorf("  %s", err)
				}
				t.FailNow()
			}

			SortLists(file)

			got, err := Print(file)
			if err != nil {
				t.Errorf("test case: %s", in)
				t.Errorf("unexpected error: %s", err)
				t.FailNow()
			}

			if string(got) != expected {
				t.Errorf("test case: %s", in)
				t.Errorf("  expected: %s", expected)
				t.Errorf("       got: %s", string(got))
			}
		})
	}
}