counting notes

lib/Bot.hs

@@ -12,7 +12,8 @@ import Web.Telegram.API.Bot (Chat (..), GetUpdatesRequest (..), Message (..),
                              Response (..), TelegramClient, Update (..),
                              User (..), getUpdatesM, getUpdatesRequest)
 
-import BotCommands (BotCmd (..), addNote, readCommand, showNew, showOld)
+import BotCommands (BotCmd (..), addNote, countNotes, readCommand, sendMessage,
+                    showNew, showOld)
 import Const (updateIdFile)
 import Tools (putLog, saveOffset, tshow)
 
@@ -44,7 +45,13 @@ handleMessage update =
                             showOld (fromIntegral chat_id) user_id
                         WrongCommand wrongCmd ->
                             putLog $ cmdErr wrongCmd
-                Nothing -> addNote user_id text
+                Nothing -> do
+                    addNote user_id text
+                    notes <- countNotes user_id
+                    sendMessage
+                        (fromIntegral chat_id)
+                        ("Добавлено. Всего " <> (tshow notes) <>
+                            " заметок.")
             saveOffset updateIdFile update_id
         Just msg ->
             putLog $ "unhandled " <> tshow msg

lib/BotCommands.hs

@@ -4,7 +4,9 @@
 module BotCommands
     ( BotCmd (..)
     , addNote
+    , countNotes
     , readCommand
+    , sendMessage
     , showNew
     , showOld
     ) where
@@ -24,6 +26,16 @@ import DB (EntityField (NoteId, NoteOwner), Note (..), User (..), runDB)
 
 data BotCmd = ShowNew | ShowOld | WrongCommand Text
 
+countNotes :: Int
+           -> TelegramClient (Int)
+countNotes userId = do
+    mUid <- runDB $ getKeyByValue DB.User{userTelegramId = fromIntegral userId}
+    case mUid of
+        Just uid -> do
+            notes <- runDB $ selectValList [NoteOwner ==. uid] []
+            pure $ length notes
+        Nothing -> pure 0
+
 sendMessage :: Integer
             -> Text
             -> TelegramClient ()