WIP: readwise.

2023-04-11 11:02:42 +02:00
parent b7e88fcfc9
commit e2f7ebe6de
4 changed files with 94 additions and 19 deletions
--- a/lib/Parsers/Readwise.hs
+++ b/lib/Parsers/Readwise.hs
@@ -0,0 +1,49 @@
+{-# LANGUAGE OverloadedStrings #-}
+
+module Parsers.Readwise where
+
+import Data.Csv
+import Data.Text
+import Data.Vector (toList)
+import Data.ByteString.Lazy (ByteString)
+import Api.Types (Quote(..))
+
+data RwHighlight = RwHighlight { rhHightlight :: Text
+                               , rhTitle :: Text
+                               , rhAuthor :: Text
+                               , rhBookID :: Text
+                               , rhNote :: Text
+                               , rhColor :: Text
+                               , rhTags :: Text
+                               , rhLocationType :: Text
+                               , rhLocation :: Text
+                               , rhHighlighedAt :: Text
+                               , rhDocumentTags :: Text
+                               } deriving (Show, Eq)
+
+instance FromNamedRecord RwHighlight where
+  parseNamedRecord m = RwHighlight <$> m .: "Highlight"
+                                    <*> m .: "Book Title"
+                                    <*> m .: "Book Author"
+                                    <*> m .: "Amazon Book ID"
+                                    <*> m .: "Note"
+                                    <*> m .: "Color"
+                                    <*> m .: "Tags"
+                                    <*> m .: "Location Type"
+                                    <*> m .: "Location"
+                                    <*> m .: "Highlighted at"
+                                    <*> m .: "Document tags"
+
+parseDocument :: ByteString -> [RwHighlight]
+parseDocument d = case decodeByName d of
+  Left _ -> []
+  Right (_, va) -> toList va
+
+parse :: ByteString -> [Quote]
+parse d = fmap (\r -> Quote { qQuote = rhHightlight r
+                                  , qAuthor = rhAuthor r
+                                  , qTitle = rhTitle r
+                                  , qPage = rhLocation r
+                                  , qChapter = Nothing
+                                  , qCreatedOn = Nothing
+                                }) (parseDocument d)