50 lines
2.0 KiB
Haskell
50 lines
2.0 KiB
Haskell
|
{-# 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)
|