package lexers

import (
	"testing"

	assert "github.com/alecthomas/assert/v2"
	"github.com/alecthomas/chroma/v2"
)

func TestGoHTMLTemplateIssue126(t *testing.T) {
	for _, source := range []string{
		`<rss version="2.0" xmlns:atom="http://www.w3.org/2005/Atom">
  <channel>
    <title>{{ if eq  .Title  .Site.Title }}{{ .Site.Title }}{{ else }}{{ with .Title }}{{.}} on {{ end }}{{ .Site.Title }}{{ end }}</title>
    <link>{{ .Permalink }}</link>
    <description>Recent content {{ if ne  .Title  .Site.Title }}{{ with .Title }}in {{.}} {{ end }}{{ end }}on {{ .Site.Title }}</description>
    <generator>Hugo -- gohugo.io</generator>{{ with .Site.LanguageCode }}
    <language>{{.}}</language>{{end}}{{ with .Site.Author.email }}
    <managingEditor>{{.}}{{ with $.Site.Author.name }} ({{.}}){{end}}</managingEditor>{{end}}{{ with .Site.Author.email }}
    <webMaster>{{.}}{{ with $.Site.Author.name }} ({{.}}){{end}}</webMaster>{{end}}{{ with .Site.Copyright }}
    <copyright>{{.}}</copyright>{{end}}{{ if not .Date.IsZero }}
    <lastBuildDate>{{ .Date.Format "Mon, 02 Jan 2006 15:04:05 -0700" | safeHTML }}</lastBuildDate>{{ end }}
    {{ with .OutputFormats.Get "RSS" }}
        {{ printf "<atom:link href=%q rel=\"self\" type=%q />" .Permalink .MediaType | safeHTML }}
    {{ end }}
	{{/*
		Print all pages
	*/}}
    {{ range .Data.Pages }}
    <item>
      <title>{{ .Title }}</title>
      <link>{{ .Permalink }}</link>
      <pubDate>{{ .Date.Format "Mon, 02 Jan 2006 15:04:05 -0700" | safeHTML }}</pubDate>
      {{ with .Site.Author.email }}<author>{{.}}{{ with $.Site.Author.name }} ({{.}}){{end}}</author>{{end}}
      <guid>{{ .Permalink }}</guid>
      <description>{{ .Summary | html }}</description>
    </item>
    {{ end }}
  </channel>
</rss>
`,
		`{{ $headless := .Site.GetPage "page" "some-headless-bundle" }}
{{ $reusablePages := $headless.Resources.Match "author*" }}
<h2>Authors</h2>
{{ range $reusablePages }}
    <h3>{{ .Title }}</h3>
    {{ .Content }}
{{ end }}`} {
		tokens, err := chroma.Tokenise(GoHTMLTemplate, nil, source)
		assert.NoError(t, err)
		assert.Equal(t, source, chroma.Stringify(tokens...))
	}
}

func TestGoHTMLTemplateMultilineComments(t *testing.T) {
	for _, source := range []string{
		`
{{/*
	This is a multiline comment
*/}}
`,
		`
{{- /*
	This is a multiline comment
*/}}
`,
		`
{{/*
	This is a multiline comment
*/ -}}
`,
		`
{{- /*
	This is a multiline comment
*/ -}}
`,
	} {
		tokens, err := chroma.Tokenise(GoHTMLTemplate, nil, source)
		assert.NoError(t, err)
		assert.Equal(t, source, chroma.Stringify(tokens...))

		// Make sure that there are no errors
		for _, token := range tokens {
			assert.NotEqual(t, chroma.Error, token.Type)
		}

		// Make sure that multiline comments are printed
		found := false
		for _, token := range tokens {
			if token.Type == chroma.CommentMultiline {
				found = true
			}
		}
		assert.True(t, found)
	}
}

func TestGoHTMLTemplateNegativeNumber(t *testing.T) {
	for _, source := range []string{
		`
{{ fn -3 }}
`,
	} {
		tokens, err := chroma.Tokenise(GoHTMLTemplate, nil, source)
		assert.NoError(t, err)
		assert.Equal(t, source, chroma.Stringify(tokens...))

		// Make sure that there are no errors
		for _, token := range tokens {
			assert.NotEqual(t, chroma.Error, token.Type)
		}

		// Make sure that negative number is found
		found := false
		for _, token := range tokens {
			if token.Type == chroma.LiteralNumberInteger {
				found = true
			}
		}
		assert.True(t, found)
	}
}