package main

import (
	"strconv"
	"strings"
	"testing"

	c "github.com/Azareal/Gosora/common"
)

func TestPreparser(t *testing.T) {
	var msgList = &METriList{nil}

	// Note: The open tag is evaluated without knowledge of the close tag for efficiency and simplicity, so the parser autofills the associated close tag when it finds an open tag without a partner
	msgList.Add("", "")
	msgList.Add(" ", "")
	msgList.Add(" hi", "hi")
	msgList.Add("hi ", "hi")
	msgList.Add("hi", "hi")
	msgList.Add(":grinning:", "😀")
	msgList.Add("😀", "😀")
	msgList.Add(" ", "")
	msgList.Add("<p>", "")
	msgList.Add("</p>", "")
	msgList.Add("<p></p>", "")

	msgList.Add("<", "&lt;")
	msgList.Add(">", "&gt;")
	msgList.Add("<meow>", "&lt;meow&gt;")
	msgList.Add("&lt;", "&amp;lt;")
	msgList.Add("&", "&amp;")

	// Note: strings.TrimSpace strips newlines, if there's nothing before or after them
	msgList.Add("<br>", "")
	msgList.Add("<br />", "")
	msgList.Add("\\n", "\n", "")
	msgList.Add("\\n\\n", "\n\n", "")
	msgList.Add("\\n\\n\\n", "\n\n\n", "")
	msgList.Add("\\r\\n", "\r\n", "") // Windows style line ending
	msgList.Add("\\n\\r", "\n\r", "")

	msgList.Add("ho<br>ho", "ho\n\nho")
	msgList.Add("ho<br />ho", "ho\n\nho")
	msgList.Add("ho\\nho", "ho\nho", "ho\nho")
	msgList.Add("ho\\n\\nho", "ho\n\nho", "ho\n\nho")
	//msgList.Add("ho\\n\\n\\n\\nho", "ho\n\n\n\nho", "ho\n\n\nho")
	msgList.Add("ho\\r\\nho", "ho\r\nho", "ho\nho") // Windows style line ending
	msgList.Add("ho\\n\\rho", "ho\n\rho", "ho\nho")

	msgList.Add("<b></b>", "<strong></strong>")
	msgList.Add("<b>hi</b>", "<strong>hi</strong>")
	msgList.Add("<b>h</b>", "<strong>h</strong>")
	msgList.Add("<s>hi</s>", "<del>hi</del>")
	msgList.Add("<del>hi</del>", "<del>hi</del>")
	msgList.Add("<u>hi</u>", "<u>hi</u>")
	msgList.Add("<em>hi</em>", "<em>hi</em>")
	msgList.Add("<i>hi</i>", "<em>hi</em>")
	msgList.Add("<strong>hi</strong>", "<strong>hi</strong>")
	msgList.Add("<b><i>hi</i></b>", "<strong><em>hi</em></strong>")
	msgList.Add("<strong><em>hi</em></strong>", "<strong><em>hi</em></strong>")
	msgList.Add("<b><i><b>hi</b></i></b>", "<strong><em><strong>hi</strong></em></strong>")
	msgList.Add("<strong><em><strong>hi</strong></em></strong>", "<strong><em><strong>hi</strong></em></strong>")
	msgList.Add("<div>hi</div>", "&lt;div&gt;hi&lt;/div&gt;")
	msgList.Add("<span>hi</span>", "hi") // This is stripped since the editor (Trumbowyg) likes blasting useless spans
	msgList.Add("<span   >hi</span>", "hi")
	msgList.Add("<span style='background-color: yellow;'>hi</span>", "hi")
	msgList.Add("<span style='background-color: yellow;'>>hi</span>", "&gt;hi")
	msgList.Add("<b>hi", "<strong>hi</strong>")
	msgList.Add("hi</b>", "hi&lt;/b&gt;")
	msgList.Add("</b>", "&lt;/b&gt;")
	msgList.Add("</del>", "&lt;/del&gt;")
	msgList.Add("</strong>", "&lt;/strong&gt;")
	msgList.Add("<b>", "<strong></strong>")
	msgList.Add("<span style='background-color: yellow;'>hi", "hi")
	msgList.Add("hi</span>", "hi")
	msgList.Add("</span>", "")
	msgList.Add("<span></span>", "")
	msgList.Add("<span   ></span>", "")
	msgList.Add("<span><span></span></span>", "")
	msgList.Add("<span><b></b></span>", "<strong></strong>")
	msgList.Add("<h1>t</h1>", "<h2>t</h2>")
	msgList.Add("<h2>t</h2>", "<h3>t</h3>")
	msgList.Add("<h3>t</h3>", "<h4>t</h4>")
	msgList.Add("<></>", "&lt;&gt;&lt;/&gt;")
	msgList.Add("</><>", "&lt;/&gt;&lt;&gt;")
	msgList.Add("<>", "&lt;&gt;")
	msgList.Add("</>", "&lt;/&gt;")
	msgList.Add("<p>hi</p>", "hi")
	msgList.Add("<p></p>", "")
	msgList.Add("<blockquote>hi</blockquote>", "<blockquote>hi</blockquote>")
	msgList.Add("<blockquote><b>hi</b></blockquote>", "<blockquote><strong>hi</strong></blockquote>")
	msgList.Add("<blockquote><meow>hi</meow></blockquote>", "<blockquote>&lt;meow&gt;hi&lt;/meow&gt;</blockquote>")
	msgList.Add("\\<blockquote>hi</blockquote>", "&lt;blockquote&gt;hi&lt;/blockquote&gt;")
	//msgList.Add("\\\\<blockquote><meow>hi</meow></blockquote>", "\\<blockquote>&lt;meow&gt;hi&lt;/meow&gt;</blockquote>") // TODO: Double escapes should print a literal backslash
	//msgList.Add("&lt;blockquote&gt;hi&lt;/blockquote&gt;", "&lt;blockquote&gt;hi&lt;/blockquote&gt;") // TODO: Stop double-entitising this
	msgList.Add("\\<blockquote>hi</blockquote>\\<blockquote>hi</blockquote>", "&lt;blockquote&gt;hi&lt;/blockquote&gt;&lt;blockquote&gt;hi&lt;/blockquote&gt;")
	msgList.Add("\\<a itemprop=\"author\">Admin</a>", "&lt;a itemprop=&#34;author&#34;&gt;Admin&lt;/a&gt;")
	msgList.Add("<blockquote>\\<a itemprop=\"author\">Admin</a></blockquote>", "<blockquote>&lt;a itemprop=&#34;author&#34;&gt;Admin&lt;/a&gt;</blockquote>")
	msgList.Add("\n<blockquote>\\<a itemprop=\"author\">Admin</a></blockquote>\n", "<blockquote>&lt;a itemprop=&#34;author&#34;&gt;Admin&lt;/a&gt;</blockquote>")
	msgList.Add("tt\n<blockquote>\\<a itemprop=\"author\">Admin</a></blockquote>\ntt", "tt\n<blockquote>&lt;a itemprop=&#34;author&#34;&gt;Admin&lt;/a&gt;</blockquote>\ntt")
	msgList.Add("@", "@")
	msgList.Add("@Admin", "@1")
	msgList.Add("@Bah", "@Bah")
	msgList.Add(" @Admin", "@1")
	msgList.Add("\n@Admin", "@1")
	msgList.Add("@Admin\n", "@1")
	msgList.Add("@Admin\ndd", "@1\ndd")
	msgList.Add("d@Admin", "d@Admin")
	msgList.Add("\\@Admin", "@Admin")
	msgList.Add("@元気", "@元気")
	// TODO: More tests for unicode names?
	//msgList.Add("\\\\@Admin", "@1")
	//msgList.Add("byte 0", string([]byte{0}), "")
	msgList.Add("byte 'a'", string([]byte{'a'}), "a")
	//msgList.Add("byte 255", string([]byte{255}), "")
	//msgList.Add("rune 0", string([]rune{0}), "")
	// TODO: Do a test with invalid UTF-8 input

	for _, item := range msgList.Items {
		res := c.PreparseMessage(item.Msg)
		if res != item.Expects {
			if item.Name != "" {
				t.Error("Name: ", item.Name)
			}
			t.Error("Testing string '" + item.Msg + "'")
			t.Error("Bad output:", "'"+res+"'")
			//t.Error("Ouput in bytes:", []byte(res))
			t.Error("Expected:", "'"+item.Expects+"'")
		}
	}
}

func TestParser(t *testing.T) {
	var msgList = &METriList{nil}

	url := "github.com/Azareal/Gosora"
	msgList.Add("", "")
	msgList.Add("haha", "haha")
	msgList.Add("<b>t</b>", "<b>t</b>")
	msgList.Add("//", "<red>[Invalid URL]</red>")
	msgList.Add("http://", "<red>[Invalid URL]</red>")
	msgList.Add("https://", "<red>[Invalid URL]</red>")
	msgList.Add("ftp://", "<red>[Invalid URL]</red>")
	msgList.Add("git://", "<red>[Invalid URL]</red>")
	msgList.Add("ssh://", "ssh://")

	msgList.Add("// ", "<red>[Invalid URL]</red> ")
	msgList.Add("// //", "<red>[Invalid URL]</red> <red>[Invalid URL]</red>")
	msgList.Add("http:// ", "<red>[Invalid URL]</red> ")
	msgList.Add("https:// ", "<red>[Invalid URL]</red> ")
	msgList.Add("ftp:// ", "<red>[Invalid URL]</red> ")
	msgList.Add("git:// ", "<red>[Invalid URL]</red> ")
	msgList.Add("ssh:// ", "ssh:// ")

	msgList.Add("// t", "<red>[Invalid URL]</red> t")
	msgList.Add("http:// t", "<red>[Invalid URL]</red> t")

	msgList.Add("http:", "http:")
	msgList.Add("https:", "https:")
	msgList.Add("ftp:", "ftp:")
	msgList.Add("git:", "git:")
	msgList.Add("ssh:", "ssh:")

	msgList.Add("http", "http")
	msgList.Add("https", "https")
	msgList.Add("ftp", "ftp")
	msgList.Add("git", "git")
	msgList.Add("ssh", "ssh")

	msgList.Add("ht", "ht")
	msgList.Add("htt", "htt")
	msgList.Add("ft", "ft")
	msgList.Add("gi", "gi")
	msgList.Add("ss", "ss")
	msgList.Add("haha\nhaha\nhaha", "haha<br>haha<br>haha")
	msgList.Add("//"+url, "<a href='//"+url+"'>//"+url+"</a>")
	msgList.Add("//a", "<a href='//a'>//a</a>")
	msgList.Add("https://"+url, "<a href='https://"+url+"'>https://"+url+"</a>")
	msgList.Add("https://t", "<a href='https://t'>https://t</a>")
	msgList.Add("http://"+url, "<a href='http://"+url+"'>http://"+url+"</a>")
	msgList.Add("#http://"+url, "#http://"+url)
	msgList.Add("@http://"+url, "<red>[Invalid Profile]</red>ttp://"+url)
	msgList.Add("//"+url+"\n", "<a href='//"+url+"'>//"+url+"</a><br>")
	msgList.Add("\n//"+url, "<br><a href='//"+url+"'>//"+url+"</a>")
	msgList.Add("\n//"+url+"\n", "<br><a href='//"+url+"'>//"+url+"</a><br>")
	msgList.Add("\n//"+url+"\n\n", "<br><a href='//"+url+"'>//"+url+"</a><br><br>")
	msgList.Add("//"+url+"\n//"+url, "<a href='//"+url+"'>//"+url+"</a><br><a href='//"+url+"'>//"+url+"</a>")
	msgList.Add("//"+url+"\n\n//"+url, "<a href='//"+url+"'>//"+url+"</a><br><br><a href='//"+url+"'>//"+url+"</a>")
	msgList.Add("//"+c.Site.URL, "<a href='//"+c.Site.URL+"'>//"+c.Site.URL+"</a>")
	msgList.Add("//"+c.Site.URL+"\n", "<a href='//"+c.Site.URL+"'>//"+c.Site.URL+"</a><br>")
	msgList.Add("//"+c.Site.URL+"\n//"+c.Site.URL, "<a href='//"+c.Site.URL+"'>//"+c.Site.URL+"</a><br><a href='//"+c.Site.URL+"'>//"+c.Site.URL+"</a>")

	var local = func(url string) {
		msgList.Add("//"+url, "<a href='//"+url+"'>//"+url+"</a>")
		msgList.Add("//"+url+"\n", "<a href='//"+url+"'>//"+url+"</a><br>")
		msgList.Add("//"+url+"\n//"+url, "<a href='//"+url+"'>//"+url+"</a><br><a href='//"+url+"'>//"+url+"</a>")
	}
	local("localhost")
	local("127.0.0.1")
	local("[::1]")

	msgList.Add("https://www.youtube.com/watch?v=lalalalala","<iframe class='postIframe' src='https://www.youtube-nocookie.com/embed/lalalalala' frameborder=0 allowfullscreen></iframe>")
	//msgList.Add("https://www.youtube.com/watch?v=;","<iframe class='postIframe' src='https://www.youtube-nocookie.com/embed/;' frameborder=0 allowfullscreen></iframe>")
	msgList.Add("https://www.youtube.com/watch?v=d;","<iframe class='postIframe' src='https://www.youtube-nocookie.com/embed/d' frameborder=0 allowfullscreen></iframe>")
	msgList.Add("https://www.youtube.com/watch?v=d;d","<iframe class='postIframe' src='https://www.youtube-nocookie.com/embed/d' frameborder=0 allowfullscreen></iframe>")
	msgList.Add("https://www.youtube.com/watch?v=alert()","<red>[Invalid URL]</red>()")
	msgList.Add("https://www.youtube.com/watch?v=js:alert()","<red>[Invalid URL]</red>()")
	msgList.Add("https://www.youtube.com/watch?v='+><script>alert(\"\")</script><+'","<red>[Invalid URL]</red>'+><script>alert(\"\")</script><+'")
	msgList.Add("https://www.youtube.com/watch?v='+onready='alert(\"\")'+'","<red>[Invalid URL]</red>'+onready='alert(\"\")'+'")
	msgList.Add(" https://www.youtube.com/watch?v=lalalalala"," <iframe class='postIframe' src='https://www.youtube-nocookie.com/embed/lalalalala' frameborder=0 allowfullscreen></iframe>")
	msgList.Add("https://www.youtube.com/watch?v=lalalalala tt","<iframe class='postIframe' src='https://www.youtube-nocookie.com/embed/lalalalala' frameborder=0 allowfullscreen></iframe> tt")
	msgList.Add("https://www.youtube.com/watch?v=lalalalala&d=haha","<iframe class='postIframe' src='https://www.youtube-nocookie.com/embed/lalalalala' frameborder=0 allowfullscreen></iframe>")
	msgList.Add("https://gaming.youtube.com/watch?v=lalalalala","<iframe class='postIframe' src='https://www.youtube-nocookie.com/embed/lalalalala' frameborder=0 allowfullscreen></iframe>")
	msgList.Add("https://gaming.youtube.com/watch?v=lalalalala&d=haha","<iframe class='postIframe' src='https://www.youtube-nocookie.com/embed/lalalalala' frameborder=0 allowfullscreen></iframe>")
	msgList.Add("https://m.youtube.com/watch?v=lalalalala","<iframe class='postIframe' src='https://www.youtube-nocookie.com/embed/lalalalala' frameborder=0 allowfullscreen></iframe>")
	msgList.Add("https://m.youtube.com/watch?v=lalalalala&d=haha","<iframe class='postIframe' src='https://www.youtube-nocookie.com/embed/lalalalala' frameborder=0 allowfullscreen></iframe>")
	msgList.Add("http://www.youtube.com/watch?v=lalalalala","<iframe class='postIframe' src='https://www.youtube-nocookie.com/embed/lalalalala' frameborder=0 allowfullscreen></iframe>")
	msgList.Add("//www.youtube.com/watch?v=lalalalala","<iframe class='postIframe' src='https://www.youtube-nocookie.com/embed/lalalalala' frameborder=0 allowfullscreen></iframe>")
	//msgList.Add("www.youtube.com/watch?v=lalalalala","<iframe class='postIframe' src='https://www.youtube-nocookie.com/embed/lalalalala' frameborder=0 allowfullscreen></iframe>")

	msgList.Add("#tid-1", "<a href='/topic/1'>#tid-1</a>")
	msgList.Add("##tid-1", "##tid-1")
	msgList.Add("# #tid-1", "# #tid-1")
	msgList.Add("@ #tid-1", "<red>[Invalid Profile]</red>#tid-1")
	msgList.Add("@#tid-1", "<red>[Invalid Profile]</red>tid-1")
	msgList.Add("@ #tid-@", "<red>[Invalid Profile]</red>#tid-@")
	msgList.Add("#tid-1 #tid-1", "<a href='/topic/1'>#tid-1</a> <a href='/topic/1'>#tid-1</a>")
	msgList.Add("#tid-0", "<red>[Invalid Topic]</red>")
	msgList.Add("https://"+url+"/#tid-1", "<a href='https://"+url+"/#tid-1'>https://"+url+"/#tid-1</a>")
	msgList.Add("https://"+url+"/?hi=2", "<a href='https://"+url+"/?hi=2'>https://"+url+"/?hi=2</a>")
	msgList.Add("#fid-1", "<a href='/forum/1'>#fid-1</a>")
	msgList.Add(" #fid-1", " <a href='/forum/1'>#fid-1</a>")
	msgList.Add("#fid-0", "<red>[Invalid Forum]</red>")
	msgList.Add(" #fid-0", " <red>[Invalid Forum]</red>")
	msgList.Add("#", "#")
	msgList.Add("# ", "# ")
	msgList.Add(" @", " @")
	msgList.Add(" #", " #")
	msgList.Add("#@", "#@")
	msgList.Add("#@ ", "#@ ")
	msgList.Add("#@1", "#@1")
	msgList.Add("#f", "#f")
	msgList.Add("#ff", "#ff")
	msgList.Add("#ffffid-0", "#ffffid-0")
	//msgList.Add("#ffffid-0", "#ffffid-0")
	msgList.Add("#nid-0", "#nid-0")
	msgList.Add("#nnid-0", "#nnid-0")
	msgList.Add("@@", "<red>[Invalid Profile]</red>")
	msgList.Add("@@ @@", "<red>[Invalid Profile]</red> <red>[Invalid Profile]</red>")
	msgList.Add("@@1", "<red>[Invalid Profile]</red>1")
	msgList.Add("@#1", "<red>[Invalid Profile]</red>1")
	msgList.Add("@##1", "<red>[Invalid Profile]</red>#1")
	msgList.Add("@2", "<red>[Invalid Profile]</red>")
	msgList.Add("@2t", "<red>[Invalid Profile]</red>t")
	msgList.Add("@2 t", "<red>[Invalid Profile]</red> t")
	msgList.Add("@2 ", "<red>[Invalid Profile]</red> ")
	msgList.Add("@2 @2", "<red>[Invalid Profile]</red> <red>[Invalid Profile]</red>")
	msgList.Add("@1", "<a href='/user/admin.1' class='mention'>@Admin</a>")
	msgList.Add(" @1", " <a href='/user/admin.1' class='mention'>@Admin</a>")
	msgList.Add("@1t", "<a href='/user/admin.1' class='mention'>@Admin</a>t")
	msgList.Add("@1 ", "<a href='/user/admin.1' class='mention'>@Admin</a> ")
	msgList.Add("@1 @1", "<a href='/user/admin.1' class='mention'>@Admin</a> <a href='/user/admin.1' class='mention'>@Admin</a>")
	msgList.Add("@0", "<red>[Invalid Profile]</red>")
	msgList.Add("@-1", "<red>[Invalid Profile]</red>1")

	for _, item := range msgList.Items {
		res := c.ParseMessage(item.Msg, 1, "forums")
		if res != item.Expects {
			if item.Name != "" {
				t.Error("Name: ", item.Name)
			}
			t.Error("Testing string '" + item.Msg + "'")
			t.Error("Bad output:", "'"+res+"'")
			t.Error("Expected:", "'"+item.Expects+"'")
			break
		}
	}

	c.AddHashLinkType("nnid-", func(sb *strings.Builder, msg string, i *int) {
		tid, intLen := c.CoerceIntString(msg[*i:])
		*i += intLen

		topic, err := c.Topics.Get(tid)
		if err != nil || !c.Forums.Exists(topic.ParentID) {
			sb.Write(c.InvalidTopic)
			return
		}
		c.WriteURL(sb, c.BuildTopicURL("", tid), "#nnid-"+strconv.Itoa(tid))
	})
	res := c.ParseMessage("#nnid-1", 1, "forums")
	expect := "<a href='/topic/1'>#nnid-1</a>"
	if res != expect {
		t.Error("Bad output:", "'"+res+"'")
		t.Error("Expected:", "'"+expect+"'")
	}

	c.AddHashLinkType("longidnameneedtooverflowhack-", func(sb *strings.Builder, msg string, i *int) {
		tid, intLen := c.CoerceIntString(msg[*i:])
		*i += intLen

		topic, err := c.Topics.Get(tid)
		if err != nil || !c.Forums.Exists(topic.ParentID) {
			sb.Write(c.InvalidTopic)
			return
		}
		c.WriteURL(sb, c.BuildTopicURL("", tid), "#longidnameneedtooverflowhack-"+strconv.Itoa(tid))
	})
	res = c.ParseMessage("#longidnameneedtooverflowhack-1", 1, "forums")
	expect = "<a href='/topic/1'>#longidnameneedtooverflowhack-1</a>"
	if res != expect {
		t.Error("Bad output:", "'"+res+"'")
		t.Error("Expected:", "'"+expect+"'")
	}
}