From fb923cdb7885cc75de960e11e527c6c0b1405923 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Vytautas=20=C5=A0altenis?= <Vytautas.Shaltenis@gmail.com>
Date: Sat, 13 Apr 2013 22:57:16 +0300
Subject: [PATCH] Add an option to strip <script> elements

Partially addresses issue #11.
---
 html.go        |  4 ++++
 inline_test.go | 14 +++++++++++++-
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/html.go b/html.go
index cf176cc..182208d 100644
--- a/html.go
+++ b/html.go
@@ -28,6 +28,7 @@ const (
 	HTML_SKIP_STYLE                           // skip embedded <style> elements
 	HTML_SKIP_IMAGES                          // skip embedded images
 	HTML_SKIP_LINKS                           // skip all links
+	HTML_SKIP_SCRIPT                          // skip embedded <script> elements
 	HTML_SAFELINK                             // only link to trusted protocols
 	HTML_TOC                                  // generate a table of contents
 	HTML_OMIT_CONTENTS                        // skip the main contents (for a standalone table of contents)
@@ -460,6 +461,9 @@ func (options *Html) RawHtmlTag(out *bytes.Buffer, text []byte) {
 	if options.flags&HTML_SKIP_IMAGES != 0 && isHtmlTag(text, "img") {
 		return
 	}
+	if options.flags&HTML_SKIP_SCRIPT != 0 && isHtmlTag(text, "script") {
+		return
+	}
 	out.Write(text)
 }
 
diff --git a/inline_test.go b/inline_test.go
index 0403339..510cf3d 100644
--- a/inline_test.go
+++ b/inline_test.go
@@ -70,8 +70,20 @@ func TestRawHtmlTag(t *testing.T) {
 
 		"zz <STYLE>p {}</STYLE>\n",
 		"<p>zz p {}</p>\n",
+
+		"<SCRIPT>alert()</SCRIPT>\n",
+		"<p>alert()</p>\n",
+
+		"zz <SCRIPT>alert()</SCRIPT>\n",
+		"<p>zz alert()</p>\n",
+
+		"zz <script>alert()</script>\n",
+		"<p>zz alert()</p>\n",
+
+		" <script>alert()</script>\n",
+		"<p>alert()</p>\n",
 	}
-	doTestsInlineParam(t, tests, 0, HTML_SKIP_STYLE)
+	doTestsInlineParam(t, tests, 0, HTML_SKIP_STYLE|HTML_SKIP_SCRIPT)
 }
 
 func TestEmphasis(t *testing.T) {