// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you 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 header

import (
	"fmt"
	"regexp"
	"testing"

	"github.com/apache/skywalking-eyes/pkg/comments"
	"github.com/stretchr/testify/require"
)

var config = &ConfigHeader{
	License: LicenseConfig{
		Content: `Apache License 2.0
  http://www.apache.org/licenses/LICENSE-2.0
Apache License 2.0`,
	},
}

func TestFix(t *testing.T) {
	tests := []struct {
		filename string
		comments string
	}{
		{
			filename: "Test.java",
			comments: `/*
 * Apache License 2.0
 *   http://www.apache.org/licenses/LICENSE-2.0
 * Apache License 2.0
 */

`,
		},
		{
			filename: "test.py",
			comments: `# Apache License 2.0
#   http://www.apache.org/licenses/LICENSE-2.0
# Apache License 2.0

`,
		},
	}
	for _, test := range tests {
		t.Run(test.filename, func(t *testing.T) {
			style := comments.FileCommentStyle(test.filename)
			h, err := GenerateLicenseHeader(style, config)
			require.NoError(t, err, fmt.Sprintf("style: %+v", style))
			require.Equal(t, test.comments, h, fmt.Sprintf("style: %+v", style))
		})
	}
}

func TestRewriteContent(t *testing.T) {
	tests := []struct {
		name            string
		style           *comments.CommentStyle
		content         string
		licenseHeader   string
		licensePattern  string
		expectedContent string
	}{
		{
			name:  "Ocaml",
			style: comments.FileCommentStyle("test.ml"),
			content: `print_string "hello worlds!\n";;
`,
			licenseHeader: getLicenseHeader("test.ml", t.Error),
			expectedContent: `(* Apache License 2.0
(*   http://www.apache.org/licenses/LICENSE-2.0
(* Apache License 2.0

print_string "hello worlds!\n";;
`},
		{
			name:  "Python with interpreter binary",
			style: comments.FileCommentStyle("test.py"),
			content: `#!/usr/bin/env python3
if __name__ == '__main__':
    print('Hello World')
`,
			licenseHeader: getLicenseHeader("test.py", t.Error),
			expectedContent: `#!/usr/bin/env python3
# Apache License 2.0
#   http://www.apache.org/licenses/LICENSE-2.0
# Apache License 2.0

if __name__ == '__main__':
    print('Hello World')
`},
		{
			name:  "Python with interpreter binary and encoding",
			style: comments.FileCommentStyle("test.py"),
			content: `#!/usr/bin/env python3
# -*- coding: latin-1 -*-
if __name__ == '__main__':
    print('Hello World')
`,
			licenseHeader: getLicenseHeader("test.py", t.Error),
			expectedContent: `#!/usr/bin/env python3
# -*- coding: latin-1 -*-
# Apache License 2.0
#   http://www.apache.org/licenses/LICENSE-2.0
# Apache License 2.0

if __name__ == '__main__':
    print('Hello World')
`},
		{
			name:  "Python with encoding",
			style: comments.FileCommentStyle("test.py"),
			content: `# -*- coding: latin-1 -*-
if __name__ == '__main__':
    print('Hello World')
`,
			licenseHeader: getLicenseHeader("test.py", t.Error),
			expectedContent: `# -*- coding: latin-1 -*-
# Apache License 2.0
#   http://www.apache.org/licenses/LICENSE-2.0
# Apache License 2.0

if __name__ == '__main__':
    print('Hello World')
`},
		{
			name:  "Python",
			style: comments.FileCommentStyle("test.py"),
			content: `if __name__ == '__main__':
    print('Hello World')
`,
			licenseHeader: getLicenseHeader("test.py", t.Error),
			expectedContent: `# Apache License 2.0
#   http://www.apache.org/licenses/LICENSE-2.0
# Apache License 2.0

if __name__ == '__main__':
    print('Hello World')
`},
		{
			name:  "Python with Blank Line",
			style: comments.FileCommentStyle("test.py"),
			content: `if __name__ == '__main__':
    print('Hello World')
`,
			licenseHeader: getLicenseHeaderCustomConfig("test.py", t.Error, &ConfigHeader{
				License: LicenseConfig{
					Content: `Apache License 2.0
  http://www.apache.org/licenses/LICENSE-2.0
Apache License 2.0

`,
				},
			}),
			expectedContent: `# Apache License 2.0
#   http://www.apache.org/licenses/LICENSE-2.0
# Apache License 2.0

if __name__ == '__main__':
    print('Hello World')
`},
		{
			name:  "XML one line declaration",
			style: comments.FileCommentStyle("test.xml"),
			content: `
<?xml version="1.0" encoding="UTF-8"?>
<project>
  <modelVersion>4.0.0</modelVersion>
</project>
`,
			licenseHeader: getLicenseHeader("test.xml", t.Error),
			expectedContent: `<?xml version="1.0" encoding="UTF-8"?>
<!--
  ~ Apache License 2.0
  ~   http://www.apache.org/licenses/LICENSE-2.0
  ~ Apache License 2.0
-->

<project>
  <modelVersion>4.0.0</modelVersion>
</project>
`},
		{
			name:  "XML multi-line declaration",
			style: comments.FileCommentStyle("test.xml"),
			content: `
<?xml
  version="1.0"
  encoding="UTF-8"
?>
<project>
  <modelVersion>4.0.0</modelVersion>
</project>
`,
			licenseHeader: getLicenseHeader("test.xml", t.Error),
			expectedContent: `<?xml
  version="1.0"
  encoding="UTF-8"
?>
<!--
  ~ Apache License 2.0
  ~   http://www.apache.org/licenses/LICENSE-2.0
  ~ Apache License 2.0
-->

<project>
  <modelVersion>4.0.0</modelVersion>
</project>
`},
		{
			name:          "SQL",
			style:         comments.FileCommentStyle("test.sql"),
			content:       `select * from user;`,
			licenseHeader: getLicenseHeader("test.sql", t.Error),
			expectedContent: `-- Apache License 2.0
--   http://www.apache.org/licenses/LICENSE-2.0
-- Apache License 2.0

select * from user;`},
		{
			name:          "Haskell",
			style:         comments.FileCommentStyle("test.hs"),
			content:       `import Foundation.Hashing.Hashable`,
			licenseHeader: getLicenseHeader("test.hs", t.Error),
			expectedContent: `{-
 Apache License 2.0
   http://www.apache.org/licenses/LICENSE-2.0
 Apache License 2.0
-}

import Foundation.Hashing.Hashable`},
		{
			name:  "Vim",
			style: comments.FileCommentStyle("test.vim"),
			content: `echo 'Hello' | echo 'world!'
`,
			licenseHeader: getLicenseHeader("test.vim", t.Error),
			expectedContent: `" Apache License 2.0
"   http://www.apache.org/licenses/LICENSE-2.0
" Apache License 2.0

echo 'Hello' | echo 'world!'
`},
		{
			name:          "Php-1",
			style:         comments.FileCommentStyle("test.php"),
			content:       ``,
			licenseHeader: getLicenseHeader("test.php", t.Error),
			expectedContent: `<?php
/*
 * Apache License 2.0
 *   http://www.apache.org/licenses/LICENSE-2.0
 * Apache License 2.0
 */

?>`,
		}, {
			name:  "Php-2",
			style: comments.FileCommentStyle("test.php"),
			content: `<?php declare(strict_types=1);
echo "Test";
`,
			licenseHeader: getLicenseHeader("test.php", t.Error),
			expectedContent: `<?php declare(strict_types=1);
/*
 * Apache License 2.0
 *   http://www.apache.org/licenses/LICENSE-2.0
 * Apache License 2.0
 */

echo "Test";
`,
		}, {
			name:  "Php-3",
			style: comments.FileCommentStyle("test.php"),
			content: `<?php
echo "Test";
`,
			licenseHeader: getLicenseHeader("test.php", t.Error),
			expectedContent: `<?php
/*
 * Apache License 2.0
 *   http://www.apache.org/licenses/LICENSE-2.0
 * Apache License 2.0
 */

echo "Test";
`,
		}, {
			name:  "Php-4",
			style: comments.FileCommentStyle("test.php"),
			content: `<?php
`,
			licenseHeader: getLicenseHeader("test.php", t.Error),
			expectedContent: `<?php
/*
 * Apache License 2.0
 *   http://www.apache.org/licenses/LICENSE-2.0
 * Apache License 2.0
 */

`,
		}, {
			name:  "Php-5",
			style: comments.FileCommentStyle("test.php"),
			content: `<?php
/**
 * This is a php docblock
 */
namespace test\test2;
`,
			licenseHeader: getLicenseHeader("test.php", t.Error),
			expectedContent: `<?php
/*
 * Apache License 2.0
 *   http://www.apache.org/licenses/LICENSE-2.0
 * Apache License 2.0
 */

/**
 * This is a php docblock
 */
namespace test\test2;
`,
		}, {
			name:  "Php-6",
			style: comments.FileCommentStyle("test.php"),
			content: `<?
/**
 * This is a php docblock
 */
namespace test\test2;
`,
			licenseHeader: getLicenseHeader("test.php", t.Error),
			expectedContent: `<?
/*
 * Apache License 2.0
 *   http://www.apache.org/licenses/LICENSE-2.0
 * Apache License 2.0
 */

/**
 * This is a php docblock
 */
namespace test\test2;
`,
		},
	}
	for _, test := range tests {
		t.Run(test.name, func(t *testing.T) {
			var r *regexp.Regexp
			if len(test.licensePattern) > 0 {
				r = regexp.MustCompile(test.licensePattern)
			}
			content := rewriteContent(test.style, []byte(test.content), test.licenseHeader, r)
			require.Equal(t, test.expectedContent, string(content), fmt.Sprintf("style: %+v", test.style))
		})
	}
}

func getLicenseHeader(filename string, tError func(args ...interface{})) string {
	s, err := GenerateLicenseHeader(comments.FileCommentStyle(filename), config)
	if err != nil {
		tError(err)
	}
	return s
}

func getLicenseHeaderCustomConfig(filename string, tError func(args ...interface{}), c *ConfigHeader) string {
	s, err := GenerateLicenseHeader(comments.FileCommentStyle(filename), c)
	if err != nil {
		tError(err)
	}
	return s
}