* colouring tool changes

1 files changed, 7 insertions, 4 deletions

diff --git a/tools/colouriser/colour.hs b/tools/colouriser/colour.hs
index 03ae8d51f4e0..3e6e39ba45f6 100644
--- a/tools/colouriser/colour.hs
+++ b/tools/colouriser/colour.hs
@@ -9,13 +9,16 @@ defineOptions "MainOptions" $ do
 	stringOption "optFile" "file" ""
 		"Name of the .hs file. Will be used for the HTML file as well"
 
-colorCode :: String -> String -> IO ()
-colorCode input output = do
+colorCode :: String -> IO ()
+colorCode input = do
 	code <- readFile input
-	writeFile output $ hscolour False code
+	putStr $ concat [ "<div class=\"code\">"
+		     		, hscolour False code
+				    , "</div>"
+				    ]
 
 main :: IO ()
 main = runCommand $ \opts args -> do
 	let file = optFile opts
 	unless (file == "") $
-		colorCode (file ++ ".hs") (file ++ ".html")
\ No newline at end of file
+		colorCode file
\ No newline at end of file