From dcaaa9b5dc2b07d4a8d6d809787a77c7cdfa756a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Vytautas=20=C5=A0altenis?= <Vytautas.Shaltenis@gmail.com>
Date: Sat, 13 Apr 2013 23:24:30 +0300
Subject: [PATCH] More <script> stripping

Partially addresses issue #11.
---
 html.go        | 16 +++++++++++++++-
 inline_test.go |  3 +++
 2 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/html.go b/html.go
index 182208d..1e6b434 100644
--- a/html.go
+++ b/html.go
@@ -168,10 +168,24 @@ func (options *Html) BlockHtml(out *bytes.Buffer, text []byte) {
 	}
 
 	doubleSpace(out)
-	out.Write(text)
+	if options.flags&HTML_SKIP_SCRIPT != 0 {
+		out.Write(stripTag(string(text), "script", "p"))
+	} else {
+		out.Write(text)
+	}
 	out.WriteByte('\n')
 }
 
+// This is a trivial implementation for the simplest possible case
+func stripTag(text, tag, newTag string) []byte {
+	openTag := fmt.Sprintf("<%s>", tag)
+	closeTag := fmt.Sprintf("</%s>", tag)
+	openNewTag := fmt.Sprintf("<%s>", newTag)
+	closeNewTag := fmt.Sprintf("</%s>", newTag)
+	noOpen := strings.Replace(text, openTag, openNewTag, -1)
+	return []byte(strings.Replace(noOpen, closeTag, closeNewTag, -1))
+}
+
 func (options *Html) HRule(out *bytes.Buffer) {
 	doubleSpace(out)
 	out.WriteString("<hr")
diff --git a/inline_test.go b/inline_test.go
index 510cf3d..4804d82 100644
--- a/inline_test.go
+++ b/inline_test.go
@@ -82,6 +82,9 @@ func TestRawHtmlTag(t *testing.T) {
 
 		" <script>alert()</script>\n",
 		"<p>alert()</p>\n",
+
+		"<script>alert()</script>\n",
+		"<p>alert()</p>\n",
 	}
 	doTestsInlineParam(t, tests, 0, HTML_SKIP_STYLE|HTML_SKIP_SCRIPT)
 }