Switch on language for highlighting. Ignore non-Haskell for now

Co-authored-by: Daan Rijks <[email protected]>

Author: TauOmicronMu

message-index/site.hs

@@ -18,6 +18,7 @@ import Data.Maybe (fromMaybe, listToMaybe, mapMaybe)
 import Data.Monoid (mappend)
 import qualified Data.Text as T
 import Data.Traversable
+import Debug.Trace
 import Hakyll
 import Lens.Micro (_1, _2, _3)
 import Lens.Micro.Extras (view)
@@ -68,19 +69,37 @@ main = hakyll $ do
           <&> \ident ->
             fromFilePath $ takeDirectory (takeDirectory (toFilePath ident)) </> "index.md"
       bread <- breadcrumbField ["index.html", thisMessage]
+
       pandocCompiler
         >>= loadAndApplyTemplate
           "templates/example.html"
           ( mconcat
               [ listField
                   "files"
                   ( mconcat
-                      [ indexlessUrlField "url",
-                        field "name" (pure . view _1 . itemBody),
-                        -- TODO: pick the right language
-                        field "beforeHighlighted" (maybe (pure "<not present>") (fmap (T.unpack . highlight "haskell" . T.pack) . fmap itemBody . load . itemIdentifier) . view _2 . itemBody),
-                        field "afterHighlighted" (maybe (pure "<not present>") (fmap (T.unpack . highlight "haskell" . T.pack) . fmap itemBody . load . itemIdentifier) . view _3 . itemBody)
-                      ]
+                      ( let getName = view _1 . itemBody
+                            nameField = field "name" (pure . getName)
+
+                            highlightField ident lens = field ident $ \item -> do
+                              let name = getName item
+                              case view lens $ itemBody item of
+                                Nothing -> pure "<not present>"
+                                Just exampleItem -> do
+                                  exampleText <- fmap itemBody $ load $ itemIdentifier exampleItem
+                                  let language =
+                                        case takeExtension name of
+                                          ".hs" -> "haskell"
+                                          _ -> ""
+                                  pure $ T.unpack $ highlight language $ T.pack $ exampleText
+
+                            beforeField = highlightField "beforeHighlighted" _2
+                            afterField = highlightField "afterHighlighted" _3
+                         in [ indexlessUrlField "url",
+                              nameField,
+                              beforeField,
+                              afterField
+                            ]
+                      )
                   )
                   (return files),
                 defaultContext

message-index/templates/example.html

@@ -14,5 +14,9 @@
       $afterHighlighted$
     </div>
   </div>
+  //<script> 
+  //  console.log("$name$");
+  //</script>
 </div>
 $endfor$
+