From 37e868b3f2d4bf0b80339c0d7be4921be20d6158 Mon Sep 17 00:00:00 2001
From: tastytea <tastytea@tastytea.de>
Date: Sat, 29 May 2021 18:49:35 +0200
Subject: [PATCH] Remove <style> and <script> snippets.

Closes: https://schlomp.space/tastytea/epubgrep/issues/8
---
 src/search.cpp | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/search.cpp b/src/search.cpp
index aa6d078..d409704 100644
--- a/src/search.cpp
+++ b/src/search.cpp
@@ -134,7 +134,7 @@ void cleanup_text(string &text)
         {
             replacement = "</H>";
         }
-        else if (text.substr(pos, 5) == "<span")
+        else if (text.substr(pos, 6) == "<span ")
         {
             auto endpos{text.find('>')};
             boost::match_results<const char *> match;
@@ -144,6 +144,12 @@ void cleanup_text(string &text)
                 replacement = "<PAGE " + match[2] + ">";
             }
         }
+        else if (text.substr(pos, 7) == "<style "
+                 || text.substr(pos, 8) == "<script ")
+        {
+            pos = text.find('>', pos) + 1;
+        }
+
         text.replace(pos, text.find('>', pos) + 1 - pos, replacement);
         pos += replacement.length();
     }