From d49f3f5aaf5617b31b0f21d946680ae502831f48 Mon Sep 17 00:00:00 2001
From: Michael Snoyman <michael@snoyman.com>
Date: Wed, 25 May 2016 17:53:38 +0300
Subject: [PATCH] Fix the Haddock mangling code (fixes #176)

---
 Handler/Haddock.hs    | 38 ++++++++++++++++++++------------------
 stackage-server.cabal |  1 +
 2 files changed, 21 insertions(+), 18 deletions(-)

diff --git a/Handler/Haddock.hs b/Handler/Haddock.hs
index 36a8aa8..500055a 100644
--- a/Handler/Haddock.hs
+++ b/Handler/Haddock.hs
@@ -5,8 +5,9 @@ module Handler.Haddock
 
 import Import
 import Stackage.Database
-import Text.HTML.TagStream.Types (Token' (..))
-import Text.HTML.TagStream.ByteString (tokenStream, showToken)
+import Text.HTML.DOM (eventConduit)
+import Text.XML (fromEvents)
+import Data.XML.Types (Event (..), Content (..))
 
 makeURL :: SnapName -> [Text] -> Text
 makeURL slug rest = concat
@@ -15,7 +16,7 @@ makeURL slug rest = concat
     : map (cons '/') rest
 
 shouldRedirect :: Bool
-shouldRedirect = True
+shouldRedirect = False
 
 getHaddockR :: SnapName -> [Text] -> Handler TypedContent
 getHaddockR slug rest
@@ -23,20 +24,20 @@ getHaddockR slug rest
   | final:_ <- reverse rest, ".html" `isSuffixOf` final = do
       render <- getUrlRender
 
-      let stylesheet = encodeUtf8 $ render $ StaticR haddock_style_css
-          script = encodeUtf8 $ render $ StaticR haddock_script_js
+      let stylesheet = render' $ StaticR haddock_style_css
+          script = render' $ StaticR haddock_script_js
+          render' = return . ContentText . render
 
-          addExtra t@(TagClose "head") =
-            [ TagOpen "link"
-                [ ("rel", "stylesheet")
+          addExtra t@(EventEndElement "head") =
+            [ EventBeginElement "link"
+                [ ("rel", [ContentText "stylesheet"])
                 , ("href", stylesheet)
                 ]
-                False
-            , TagOpen "script"
+            , EventEndElement "link"
+            , EventBeginElement "script"
                 [ ("src", script)
                 ]
-                False
-            , TagClose "script"
+            , EventEndElement "script"
             , t
             ]
           addExtra t = [t]
@@ -44,12 +45,13 @@ getHaddockR slug rest
       req <- parseUrl $ unpack $ makeURL slug rest
       (_, res) <- acquireResponse req >>= allocateAcquire
 
-      respondSource typeHtml
-         $ responseBody res
-        $= tokenStream
-        $= concatMapC addExtra
-        -- FIXME showToken does not encode HTML entities
-        $= mapC (Chunk . showToken id)
+      doc <- responseBody res
+          $$ eventConduit
+          =$ concatMapC addExtra
+          =$ mapC (Nothing, )
+          =$ fromEvents
+
+      sendResponse $ toHtml doc
   | otherwise = redirect $ makeURL slug rest
 
 getHaddockBackupR :: [Text] -> Handler ()
diff --git a/stackage-server.cabal b/stackage-server.cabal
index a2a2601..f478a15 100644
--- a/stackage-server.cabal
+++ b/stackage-server.cabal
@@ -132,6 +132,7 @@ library
                   , wai-logger                    >= 2.2         &&   < 2.3
                   , warp                          >= 3.2         &&   < 3.3
                   , xml-conduit                   >= 1.3         &&   < 1.4
+                  , xml-types
                   , yaml                          >= 0.8         &&   < 0.9
                   , yesod                         >= 1.4         &&   < 1.5
                   , yesod-auth                    >= 1.4         &&   < 1.5