From b15a00458aead818e7e3a8f12c76cd7417de4fe0 Mon Sep 17 00:00:00 2001
From: Etan Reisner <deryni@unreliablesource.net>
Date: Sun, 26 Apr 2015 19:23:47 -0400
Subject: [PATCH] Add a JUnit output format style.

This code is almost entirely based on the checkstyle format code.
---
 shellcheck.hs | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/shellcheck.hs b/shellcheck.hs
index 7111c83..e9505e4 100644
--- a/shellcheck.hs
+++ b/shellcheck.hs
@@ -89,6 +89,7 @@ formats = Map.fromList [
     ("json", forJson),
     ("gcc", forGcc),
     ("checkstyle", forCheckstyle),
+    ("junit", forJunit),
     ("tty", forTty)
     ]
 
@@ -228,6 +229,47 @@ forCheckstyle options files = do
         "/>\n"
         ]
 
+-- JUnit compatible output. A bit of a hack to avoid XML dependencies
+forJunit :: AnalysisOptions -> [FilePath] -> IO Status
+forJunit options files = do
+    putStrLn "<?xml version='1.0' encoding='UTF-8'?>"
+    putStrLn "<testsuites>"
+    statuses <- mapM process files
+    putStrLn "</testsuites>"
+    return $ mconcat statuses
+  where
+    process file = catchExceptions $ do
+        comments <- commentsFor options file
+        putStrLn (formatFile file comments)
+        return $ checkComments comments
+
+    severity "error" = "error"
+    severity "warning" = "warning"
+    severity _ = "info"
+    attr s v = concat [ s, "='", escape v, "' " ]
+    escape = concatMap escape'
+    escape' c = if isOk c then [c] else "&#" ++ show (ord c) ++ ";"
+    isOk x = any ($x) [isAsciiUpper, isAsciiLower, isDigit, (`elem` " ./")]
+
+    formatFile name comments = concat [
+        "<testsuite>\n<properties>\n",
+        "<property ",
+        attr "name" "filename",
+        attr "value" name,
+        "/>\n</properties>\n",
+            concatMap format comments,
+        "</testsuite>"
+        ]
+
+    format c = concat [
+        "<testcase ", attr "time" "0.0", attr "name" $ "ShellCheck.SC" ++ show (scCode c), ">\n",
+        "<failure ",
+        attr "type" $ severity . scSeverity $ c,
+        attr "message" $ "Line " ++ show (scLine c) ++ ", char " ++ show (scColumn c) ++ ": " ++ scMessage c,
+        "/>\n</testcase>\n"
+        ]
+
+
 commentsFor options file = liftM (getComments options) $ readContents file
 
 getComments = shellCheck