{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE OverloadedStrings #-}
module Lackey
( rubyForAPI
) where
import qualified Data.Char as Char
import Data.Function ((&))
import qualified Data.Maybe as Maybe
import qualified Data.Proxy as Proxy
import qualified Data.Text as Text
import qualified Data.Text.Encoding as Text
import qualified Servant.Foreign as Servant
type Language = Servant.NoTypes
languageProxy :: Proxy.Proxy Language
languageProxy :: Proxy Language
languageProxy = Proxy Language
forall k (t :: k). Proxy t
Proxy.Proxy
type Request = Servant.NoContent
requestProxy :: Proxy.Proxy Request
requestProxy :: Proxy Request
requestProxy = Proxy Request
forall k (t :: k). Proxy t
Proxy.Proxy
renderRequests :: [Servant.Req Request] -> Text.Text
renderRequests :: [Req Request] -> Text
renderRequests [Req Request]
requests = [Req Request]
requests [Req Request] -> ([Req Request] -> [Text]) -> [Text]
forall a b. a -> (a -> b) -> b
& (Req Request -> Text) -> [Req Request] -> [Text]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap Req Request -> Text
renderRequest [Text] -> ([Text] -> Text) -> Text
forall a b. a -> (a -> b) -> b
& Text -> [Text] -> Text
Text.intercalate Text
";"
functionName :: Servant.Req Request -> Text.Text
functionName :: Req Request -> Text
functionName Req Request
request = Req Request
request Req Request -> (Req Request -> FunctionName) -> FunctionName
forall a b. a -> (a -> b) -> b
& Req Request -> FunctionName
forall ftype. Req ftype -> FunctionName
Servant._reqFuncName FunctionName -> (FunctionName -> Text) -> Text
forall a b. a -> (a -> b) -> b
& FunctionName -> Text
Servant.snakeCase
hasBody :: Servant.Req Request -> Bool
hasBody :: Req Request -> Bool
hasBody Req Request
request = case Req Request -> Maybe Request
forall ftype. Req ftype -> Maybe ftype
Servant._reqBody Req Request
request of
Maybe Request
Nothing -> Bool
False
Just Request
_ -> Bool
True
bodyArgument :: Text.Text
bodyArgument :: Text
bodyArgument = Text
"body"
underscore :: Text.Text -> Text.Text
underscore :: Text -> Text
underscore Text
text =
Text
text Text -> (Text -> Text) -> Text
forall a b. a -> (a -> b) -> b
& Text -> Text
Text.toLower Text -> (Text -> Text) -> Text
forall a b. a -> (a -> b) -> b
& (Char -> Char) -> Text -> Text
Text.map (\Char
c -> if Char -> Bool
Char.isAlphaNum Char
c then Char
c else Char
'_')
getHeaders :: Servant.Req Request -> [Text.Text]
Req Request
request =
Req Request
request
Req Request
-> (Req Request -> [HeaderArg Request]) -> [HeaderArg Request]
forall a b. a -> (a -> b) -> b
& Req Request -> [HeaderArg Request]
forall ftype. Req ftype -> [HeaderArg ftype]
Servant._reqHeaders
[HeaderArg Request]
-> ([HeaderArg Request] -> [Arg Request]) -> [Arg Request]
forall a b. a -> (a -> b) -> b
& (HeaderArg Request -> Maybe (Arg Request))
-> [HeaderArg Request] -> [Arg Request]
forall a b. (a -> Maybe b) -> [a] -> [b]
Maybe.mapMaybe
(\HeaderArg Request
h -> case HeaderArg Request
h of
Servant.HeaderArg Arg Request
x -> Arg Request -> Maybe (Arg Request)
forall a. a -> Maybe a
Just Arg Request
x
Servant.ReplaceHeaderArg Arg Request
_ Text
_ -> Maybe (Arg Request)
forall a. Maybe a
Nothing
)
[Arg Request] -> ([Arg Request] -> [Text]) -> [Text]
forall a b. a -> (a -> b) -> b
& (Arg Request -> Text) -> [Arg Request] -> [Text]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (PathSegment -> Text
Servant.unPathSegment (PathSegment -> Text)
-> (Arg Request -> PathSegment) -> Arg Request -> Text
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Arg Request -> PathSegment
forall ftype. Arg ftype -> PathSegment
Servant._argName)
getURLPieces :: Servant.Req Request -> [Either Text.Text Text.Text]
getURLPieces :: Req Request -> [Either Text Text]
getURLPieces Req Request
request =
let
url :: Url Request
url = Req Request
request Req Request -> (Req Request -> Url Request) -> Url Request
forall a b. a -> (a -> b) -> b
& Req Request -> Url Request
forall ftype. Req ftype -> Url ftype
Servant._reqUrl
path :: [Text]
path =
Url Request
url
Url Request -> (Url Request -> Path Request) -> Path Request
forall a b. a -> (a -> b) -> b
& Url Request -> Path Request
forall ftype. Url ftype -> Path ftype
Servant._path
Path Request -> (Path Request -> [Arg Request]) -> [Arg Request]
forall a b. a -> (a -> b) -> b
& (Segment Request -> Maybe (Arg Request))
-> Path Request -> [Arg Request]
forall a b. (a -> Maybe b) -> [a] -> [b]
Maybe.mapMaybe
((\SegmentType Request
segment -> case SegmentType Request
segment of
Servant.Static PathSegment
_ -> Maybe (Arg Request)
forall a. Maybe a
Nothing
Servant.Cap Arg Request
arg -> Arg Request -> Maybe (Arg Request)
forall a. a -> Maybe a
Just Arg Request
arg
)
(SegmentType Request -> Maybe (Arg Request))
-> (Segment Request -> SegmentType Request)
-> Segment Request
-> Maybe (Arg Request)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Segment Request -> SegmentType Request
forall ftype. Segment ftype -> SegmentType ftype
Servant.unSegment
)
[Arg Request] -> ([Arg Request] -> [Text]) -> [Text]
forall a b. a -> (a -> b) -> b
& (Arg Request -> Text) -> [Arg Request] -> [Text]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (PathSegment -> Text
Servant.unPathSegment (PathSegment -> Text)
-> (Arg Request -> PathSegment) -> Arg Request -> Text
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Arg Request -> PathSegment
forall ftype. Arg ftype -> PathSegment
Servant._argName)
query :: [Text]
query = Url Request
url Url Request
-> (Url Request -> [QueryArg Request]) -> [QueryArg Request]
forall a b. a -> (a -> b) -> b
& Url Request -> [QueryArg Request]
forall ftype. Url ftype -> [QueryArg ftype]
Servant._queryStr [QueryArg Request] -> ([QueryArg Request] -> [Text]) -> [Text]
forall a b. a -> (a -> b) -> b
& (QueryArg Request -> Text) -> [QueryArg Request] -> [Text]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap
(PathSegment -> Text
Servant.unPathSegment (PathSegment -> Text)
-> (QueryArg Request -> PathSegment) -> QueryArg Request -> Text
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Arg Request -> PathSegment
forall ftype. Arg ftype -> PathSegment
Servant._argName (Arg Request -> PathSegment)
-> (QueryArg Request -> Arg Request)
-> QueryArg Request
-> PathSegment
forall b c a. (b -> c) -> (a -> b) -> a -> c
. QueryArg Request -> Arg Request
forall ftype. QueryArg ftype -> Arg ftype
Servant._queryArgName)
in (Text -> Either Text Text) -> [Text] -> [Either Text Text]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap Text -> Either Text Text
forall a b. a -> Either a b
Left [Text]
path [Either Text Text] -> [Either Text Text] -> [Either Text Text]
forall a. Semigroup a => a -> a -> a
<> (Text -> Either Text Text) -> [Text] -> [Either Text Text]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap Text -> Either Text Text
forall a b. b -> Either a b
Right [Text]
query
functionArguments :: Servant.Req Request -> Text.Text
functionArguments :: Req Request -> Text
functionArguments Req Request
request = [Text] -> Text
Text.concat
[ Text
"("
, [ [Text -> Maybe Text
forall a. a -> Maybe a
Just Text
"excon"]
, Req Request
request Req Request
-> (Req Request -> [Either Text Text]) -> [Either Text Text]
forall a b. a -> (a -> b) -> b
& Req Request -> [Either Text Text]
getURLPieces [Either Text Text]
-> ([Either Text Text] -> [Maybe Text]) -> [Maybe Text]
forall a b. a -> (a -> b) -> b
& (Either Text Text -> Maybe Text)
-> [Either Text Text] -> [Maybe Text]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap
(\Either Text Text
piece -> Text -> Maybe Text
forall a. a -> Maybe a
Just (Text -> Maybe Text) -> Text -> Maybe Text
forall a b. (a -> b) -> a -> b
$ case Either Text Text
piece of
Left Text
capture -> Text -> Text
underscore Text
capture
Right Text
param -> Text -> Text
underscore Text
param Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text
": nil"
)
, Req Request
request Req Request -> (Req Request -> [Text]) -> [Text]
forall a b. a -> (a -> b) -> b
& Req Request -> [Text]
getHeaders [Text] -> ([Text] -> [Maybe Text]) -> [Maybe Text]
forall a b. a -> (a -> b) -> b
& (Text -> Maybe Text) -> [Text] -> [Maybe Text]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (Text -> Maybe Text
forall a. a -> Maybe a
Just (Text -> Maybe Text) -> (Text -> Text) -> Text -> Maybe Text
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text
": nil") (Text -> Text) -> (Text -> Text) -> Text -> Text
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Text -> Text
underscore)
, [if Req Request -> Bool
hasBody Req Request
request then Text -> Maybe Text
forall a. a -> Maybe a
Just Text
bodyArgument else Maybe Text
forall a. Maybe a
Nothing]
]
[[Maybe Text]] -> ([[Maybe Text]] -> [Maybe Text]) -> [Maybe Text]
forall a b. a -> (a -> b) -> b
& [[Maybe Text]] -> [Maybe Text]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
concat
[Maybe Text] -> ([Maybe Text] -> [Text]) -> [Text]
forall a b. a -> (a -> b) -> b
& [Maybe Text] -> [Text]
forall a. [Maybe a] -> [a]
Maybe.catMaybes
[Text] -> ([Text] -> Text) -> Text
forall a b. a -> (a -> b) -> b
& Text -> [Text] -> Text
Text.intercalate Text
","
, Text
")"
]
requestMethod :: Servant.Req Request -> Text.Text
requestMethod :: Req Request -> Text
requestMethod Req Request
request =
Req Request
request Req Request -> (Req Request -> Method) -> Method
forall a b. a -> (a -> b) -> b
& Req Request -> Method
forall ftype. Req ftype -> Method
Servant._reqMethod Method -> (Method -> Text) -> Text
forall a b. a -> (a -> b) -> b
& Method -> Text
Text.decodeUtf8 Text -> (Text -> Text) -> Text
forall a b. a -> (a -> b) -> b
& Text -> Text
Text.toLower Text -> (Text -> Text) -> Text
forall a b. a -> (a -> b) -> b
& Char -> Text -> Text
Text.cons Char
':'
requestPath :: Servant.Req Request -> Text.Text
requestPath :: Req Request -> Text
requestPath Req Request
request =
let
path :: Text
path =
Req Request
request
Req Request -> (Req Request -> Url Request) -> Url Request
forall a b. a -> (a -> b) -> b
& Req Request -> Url Request
forall ftype. Req ftype -> Url ftype
Servant._reqUrl
Url Request -> (Url Request -> Path Request) -> Path Request
forall a b. a -> (a -> b) -> b
& Url Request -> Path Request
forall ftype. Url ftype -> Path ftype
Servant._path
Path Request -> (Path Request -> [Text]) -> [Text]
forall a b. a -> (a -> b) -> b
& (Segment Request -> Text) -> Path Request -> [Text]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap
((\SegmentType Request
x -> case SegmentType Request
x of
Servant.Static PathSegment
y -> PathSegment -> Text
Servant.unPathSegment PathSegment
y
Servant.Cap Arg Request
y ->
let
z :: Text
z =
Arg Request
y Arg Request -> (Arg Request -> PathSegment) -> PathSegment
forall a b. a -> (a -> b) -> b
& Arg Request -> PathSegment
forall ftype. Arg ftype -> PathSegment
Servant._argName PathSegment -> (PathSegment -> Text) -> Text
forall a b. a -> (a -> b) -> b
& PathSegment -> Text
Servant.unPathSegment Text -> (Text -> Text) -> Text
forall a b. a -> (a -> b) -> b
& Text -> Text
underscore
in Text
"#{" Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text
z Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text
"}"
)
(SegmentType Request -> Text)
-> (Segment Request -> SegmentType Request)
-> Segment Request
-> Text
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Segment Request -> SegmentType Request
forall ftype. Segment ftype -> SegmentType ftype
Servant.unSegment
)
[Text] -> ([Text] -> Text) -> Text
forall a b. a -> (a -> b) -> b
& Text -> [Text] -> Text
Text.intercalate Text
"/"
query :: Text
query =
Req Request
request
Req Request -> (Req Request -> Url Request) -> Url Request
forall a b. a -> (a -> b) -> b
& Req Request -> Url Request
forall ftype. Req ftype -> Url ftype
Servant._reqUrl
Url Request
-> (Url Request -> [QueryArg Request]) -> [QueryArg Request]
forall a b. a -> (a -> b) -> b
& Url Request -> [QueryArg Request]
forall ftype. Url ftype -> [QueryArg ftype]
Servant._queryStr
[QueryArg Request] -> ([QueryArg Request] -> [Text]) -> [Text]
forall a b. a -> (a -> b) -> b
& (QueryArg Request -> Text) -> [QueryArg Request] -> [Text]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap
((\Text
x -> Text
x Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text
"=#{" Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text -> Text
underscore Text
x Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text
"}")
(Text -> Text)
-> (QueryArg Request -> Text) -> QueryArg Request -> Text
forall b c a. (b -> c) -> (a -> b) -> a -> c
. PathSegment -> Text
Servant.unPathSegment
(PathSegment -> Text)
-> (QueryArg Request -> PathSegment) -> QueryArg Request -> Text
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Arg Request -> PathSegment
forall ftype. Arg ftype -> PathSegment
Servant._argName
(Arg Request -> PathSegment)
-> (QueryArg Request -> Arg Request)
-> QueryArg Request
-> PathSegment
forall b c a. (b -> c) -> (a -> b) -> a -> c
. QueryArg Request -> Arg Request
forall ftype. QueryArg ftype -> Arg ftype
Servant._queryArgName
)
[Text] -> ([Text] -> Text) -> Text
forall a b. a -> (a -> b) -> b
& Text -> [Text] -> Text
Text.intercalate Text
"&"
url :: Text
url = Text
"/" Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text
path Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> (if Text -> Bool
Text.null Text
query then Text
"" else Text
"?" Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text
query)
in Text
"\"" Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text
url Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text
"\""
requestHeaders :: Servant.Req Request -> Text.Text
Req Request
request =
[ [Text
"{"]
, Req Request
request Req Request -> (Req Request -> [Text]) -> [Text]
forall a b. a -> (a -> b) -> b
& Req Request -> [Text]
getHeaders [Text] -> ([Text] -> [Text]) -> [Text]
forall a b. a -> (a -> b) -> b
& (Text -> Text) -> [Text] -> [Text]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (\Text
x -> Text
"\"" Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text
x Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text
"\"=>" Text -> Text -> Text
forall a. Semigroup a => a -> a -> a
<> Text -> Text
underscore Text
x)
, [Text
"}"]
]
[[Text]] -> ([[Text]] -> [Text]) -> [Text]
forall a b. a -> (a -> b) -> b
& [[Text]] -> [Text]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
concat
[Text] -> ([Text] -> Text) -> Text
forall a b. a -> (a -> b) -> b
& [Text] -> Text
Text.concat
requestBody :: Servant.Req Request -> Text.Text
requestBody :: Req Request -> Text
requestBody Req Request
request = if Req Request -> Bool
hasBody Req Request
request then Text
bodyArgument else Text
"nil"
functionBody :: Servant.Req Request -> Text.Text
functionBody :: Req Request -> Text
functionBody Req Request
request = [Text] -> Text
Text.concat
[ Text
"excon.request("
, Text
":method=>"
, Req Request -> Text
requestMethod Req Request
request
, Text
","
, Text
":path=>"
, Req Request -> Text
requestPath Req Request
request
, Text
","
, Text
":headers=>"
, Req Request -> Text
requestHeaders Req Request
request
, Text
","
, Text
":body=>"
, Req Request -> Text
requestBody Req Request
request
, Text
")"
]
renderRequest :: Servant.Req Request -> Text.Text
renderRequest :: Req Request -> Text
renderRequest Req Request
request = [Text] -> Text
Text.concat
[ Text
"def "
, Req Request -> Text
functionName Req Request
request
, Req Request -> Text
functionArguments Req Request
request
, Req Request -> Text
functionBody Req Request
request
, Text
"end"
]
requestsForAPI
:: ( Servant.HasForeign Language Request api
, Servant.GenerateList Request (Servant.Foreign Request api)
)
=> Proxy.Proxy api
-> [Servant.Req Request]
requestsForAPI :: Proxy api -> [Req Request]
requestsForAPI Proxy api
api = Proxy api
api Proxy api -> (Proxy api -> [Req Request]) -> [Req Request]
forall a b. a -> (a -> b) -> b
& Proxy Language -> Proxy Request -> Proxy api -> [Req Request]
forall k (lang :: k) ftype api.
(HasForeign lang ftype api,
GenerateList ftype (Foreign ftype api)) =>
Proxy lang -> Proxy ftype -> Proxy api -> [Req ftype]
Servant.listFromAPI Proxy Language
languageProxy Proxy Request
requestProxy
rubyForAPI
:: ( Servant.HasForeign Language Request api
, Servant.GenerateList Request (Servant.Foreign Request api)
)
=> Proxy.Proxy api
-> Text.Text
rubyForAPI :: Proxy api -> Text
rubyForAPI Proxy api
api = Proxy api
api Proxy api -> (Proxy api -> [Req Request]) -> [Req Request]
forall a b. a -> (a -> b) -> b
& Proxy api -> [Req Request]
forall api.
(HasForeign Language Request api,
GenerateList Request (Foreign Request api)) =>
Proxy api -> [Req Request]
requestsForAPI [Req Request] -> ([Req Request] -> Text) -> Text
forall a b. a -> (a -> b) -> b
& [Req Request] -> Text
renderRequests