gdritter repos shelob / master shelob / src / Network / Shelob.hs
master

Tree @master (Download .tar.gz)

Shelob.hs @master — raw · history · blame 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
{-# LANGUAGE DeriveFunctor #-}

module Network.Shelob where

import qualified Data.ByteString as BS
import qualified Data.ByteString.Char8 as BS8

import           Network.Shelob.Connection
import           Network.Shelob.Request
import           Network.Shelob.Response
import           Network.Shelob.Types

userAgent :: Header
userAgent = Header "User-Agent" "haskell/vriska"

request :: Method -> BS.ByteString -> [Header] -> Request
request m p hs = Request m p (hs' ++ hs) ""
  where hs' = [ Header "Content-Length" "0"
              , userAgent
              ]

head :: BS.ByteString -> [Header] -> Request
head p hs = Request MHead p (hs' ++ hs) ""
  where hs' = [ Header "Content-Length" "0"
              , userAgent
              ]

get :: BS.ByteString -> [Header] -> Request
get p hs = Request MGet p (hs' ++ hs) ""
  where hs' = [ Header "Content-Length" "0"
              , userAgent
              ]

post :: BS.ByteString -> [Header] -> BS.ByteString -> Request
post p hs body = Request MPost p (hs' ++ hs) body
  where hs' = [ Header "Content-Length" (BS8.pack (show (BS.length body)))
              , userAgent
              ]

makeRequest :: Connection -> Request -> IO (Either HTTPError Response)
makeRequest conn req = do
  let pReq = pRequest req
  connWrite conn pReq
  readResponse conn req