From aca5d23bfb40e32402194e62b431433a12be3c93 Mon Sep 17 00:00:00 2001
From: sabadev <saba@sabadev.xyz>
Date: Sun, 11 Apr 2021 20:07:30 -0400
Subject: [PATCH] Added 404 error codes.

---
 src/Html.hs | 23 +++++++++++++----------
 1 file changed, 13 insertions(+), 10 deletions(-)

diff --git a/src/Html.hs b/src/Html.hs
index 3a18ed2..fee5c34 100644
--- a/src/Html.hs
+++ b/src/Html.hs
@@ -3,6 +3,7 @@ module Html where
 import ApiTypes
 import Control.Exception.Safe (SomeException)
 import Control.Monad (void)
+import Control.Monad.Error.Class (MonadError(..))
 import Control.Monad.IO.Class (MonadIO(..), liftIO)
 import Data.ByteString.Lazy (ByteString(..))
 import Data.List (sort)
@@ -74,17 +75,19 @@ blogListItem theme (blogLink -> (Just file)) = li_ [class_ "blog-link"] $ a_ [hr
 blogLink :: T.Text -> Maybe T.Text
 blogLink = T.stripSuffix markdownExtension
 
-imageNotFound :: (MonadIO m) => SomeException -> m ByteString
-imageNotFound _ = pure mempty
+imageNotFound :: (MonadError ServerError m) => SomeException -> m a
+imageNotFound _ = throwError $ err404 { errBody = "No image found." }
 
-blogNotFound :: (MonadIO m) => Maybe Theme -> BlogId -> SomeException -> m (Html ())
-blogNotFound theme blogId _ = htmlContainer theme Nothing $ do
-  div_ [class_ "not-found"] $ do
-    h1_ $ toHtml @T.Text "Blog not found"
-    p_ $ do
-      toHtml @T.Text "Blog post "
-      em_ $ toHtml $ T.pack blogId
-      toHtml @T.Text " could not found."
+blogNotFound :: (MonadIO m, MonadError ServerError m) => Maybe Theme -> BlogId -> SomeException -> m a
+blogNotFound theme blogId _ = do
+  body <- htmlContainer theme Nothing $ do
+    div_ [class_ "not-found"] $ do
+      h1_ $ toHtml @T.Text "Blog not found"
+      p_ $ do
+        toHtml @T.Text "Blog post "
+        em_ $ toHtml $ T.pack blogId
+        toHtml @T.Text " could not found."
+  throwError $ err404 { errBody = renderBS body }
 
 siteTitle :: T.Text
 siteTitle = "My Site"
-- 
2.20.1