src/Gargantext/API/Node/Types.hs

   1 {-# LANGUAGE TemplateHaskell    #-}
   2
   3 module Gargantext.API.Node.Types where
   4
   5 import Control.Lens hiding (elements, Empty)
   6 import Data.Aeson
   7 import qualified Data.ByteString as BS
   8 import qualified Data.ByteString.Base64 as BSB64
   9 import Data.Either
  10 import Data.Swagger
  11 import Data.Text (Text)
  12 import qualified Data.Text as T
  13 import qualified Data.Text.Encoding as TE
  14 import GHC.Generics (Generic)
  15 import Servant.Job.Utils (jsonOptions)
  16 import Web.FormUrlEncoded          (FromForm, ToForm)
  17
  18 import Gargantext.Core (Lang(..))
  19 import Gargantext.Core.Utils.Prefix (unPrefixSwagger)
  20 import Gargantext.Prelude
  21 import qualified Gargantext.Database.GargDB as GargDB
  22 import Gargantext.API.Node.Corpus.New.Types (FileType, FileFormat)
  23 import Gargantext.Core.Text.List.Social (FlowSocialListWith)
  24
  25 -------------------------------------------------------
  26 data NewWithForm = NewWithForm
  27   { _wf_filetype   :: !FileType
  28   , _wf_fileformat :: !FileFormat
  29   , _wf_data       :: !Text    -- NOTE for binary files, this represents base-64 data
  30   , _wf_lang       :: !(Maybe Lang)
  31   , _wf_name       :: !Text
  32   , _wf_selection  :: !FlowSocialListWith
  33   } deriving (Eq, Show, Generic)
  34
  35 makeLenses ''NewWithForm
  36 instance FromForm NewWithForm
  37 instance ToForm NewWithForm
  38 instance FromJSON NewWithForm where
  39   parseJSON = genericParseJSON $ jsonOptions "_wf_"
  40 instance ToJSON NewWithForm where
  41   toJSON = genericToJSON $ jsonOptions "_wf_"
  42 instance ToSchema NewWithForm where
  43   declareNamedSchema = genericDeclareNamedSchema (unPrefixSwagger "_wf_")
  44
  45 -------------------------------------------------------
  46
  47 data NewWithFile = NewWithFile
  48   { _wfi_b64_data  :: !Text
  49   , _wfi_lang      :: !(Maybe Lang)
  50   , _wfi_name      :: !Text
  51   } deriving (Eq, Show, Generic)
  52
  53 makeLenses ''NewWithFile
  54 instance FromForm NewWithFile
  55 instance ToForm NewWithFile
  56 instance FromJSON NewWithFile where
  57   parseJSON = genericParseJSON $ jsonOptions "_wfi_"
  58 instance ToJSON NewWithFile where
  59   toJSON = genericToJSON $ jsonOptions "_wfi_"
  60
  61
  62 instance ToSchema NewWithFile where
  63   declareNamedSchema = genericDeclareNamedSchema (unPrefixSwagger "_wfi_")
  64
  65 instance GargDB.SaveFile NewWithFile where
  66   saveFile' fp (NewWithFile b64d _ _) = do
  67     let eDecoded = BSB64.decode $ TE.encodeUtf8 b64d
  68     case eDecoded of
  69       Left err -> panic $ T.pack $ "Error decoding: " <> err
  70       Right decoded -> BS.writeFile fp decoded
  71     -- BS.writeFile fp $ BSB64.decodeLenient $ TE.encodeUtf8 b64d
  72
  73 --instance GargDB.ReadFile NewWithFile where
  74 --  readFile' = TIO.readFile