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)
  17
  18 import Gargantext.Core (Lang(..){-, allLangs-})
  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.File (FileType)
  23
  24 -------------------------------------------------------
  25 data NewWithForm = NewWithForm
  26   { _wf_filetype :: !FileType
  27   , _wf_data     :: !Text    -- NOTE for binary files, this represents base-64 data
  28   , _wf_lang     :: !(Maybe Lang)
  29   , _wf_name     :: !Text
  30   } deriving (Eq, Show, Generic)
  31
  32 makeLenses ''NewWithForm
  33 instance FromForm NewWithForm
  34 instance FromJSON NewWithForm where
  35   parseJSON = genericParseJSON $ jsonOptions "_wf_"
  36 instance ToJSON NewWithForm where
  37   toJSON = genericToJSON $ jsonOptions "_wf_"
  38 instance ToSchema NewWithForm where
  39   declareNamedSchema = genericDeclareNamedSchema (unPrefixSwagger "_wf_")
  40
  41 -------------------------------------------------------
  42
  43 data NewWithFile = NewWithFile
  44   { _wfi_b64_data :: !Text
  45   , _wfi_lang     :: !(Maybe Lang)
  46   , _wfi_name     :: !Text
  47   } deriving (Eq, Show, Generic)
  48
  49 makeLenses ''NewWithFile
  50 instance FromForm NewWithFile
  51 instance FromJSON NewWithFile where
  52   parseJSON = genericParseJSON $ jsonOptions "_wfi_"
  53 instance ToJSON NewWithFile where
  54   toJSON = genericToJSON $ jsonOptions "_wfi_"
  55
  56
  57 instance ToSchema NewWithFile where
  58   declareNamedSchema = genericDeclareNamedSchema (unPrefixSwagger "_wfi_")
  59
  60 instance GargDB.SaveFile NewWithFile where
  61   saveFile' fp (NewWithFile b64d _ _) = do
  62     let eDecoded = BSB64.decode $ TE.encodeUtf8 b64d
  63     case eDecoded of
  64       Left err -> panic $ T.pack $ "Error decoding: " <> err
  65       Right decoded -> BS.writeFile fp decoded
  66     -- BS.writeFile fp $ BSB64.decodeLenient $ TE.encodeUtf8 b64d
  67
  68 --instance GargDB.ReadFile NewWithFile where
  69 --  readFile' = TIO.readFile