{-# LANGUAGE AllowAmbiguousTypes, UndecidableInstances, LambdaCase #-} module IHP.RouterSupport ( CanRoute (..) , HasPath (..) , AutoRoute (..) , runAction , get , post , startPage , frontControllerToWAIApp , withPrefix , FrontController (..) , defaultRouter , parseRoute , catchAll , mountFrontController , createAction , updateAction , urlTo , parseUUID , parseId , parseIntegerId , remainingText , parseText , webSocketApp , webSocketAppWithCustomPath , webSocketAppWithHTTPFallback , onlyAllowMethods , getMethod , routeParam ) where import qualified Prelude import ClassyPrelude hiding (index, delete, take) import qualified IHP.ModelSupport as ModelSupport import IHP.FrameworkConfig import IHP.ApplicationContext hiding (frameworkConfig) import Data.UUID import Network.HTTP.Types.Method import IHP.Controller.RequestContext import Network.Wai import IHP.ControllerSupport import Data.Attoparsec.ByteString.Char8 (string, Parser, parseOnly, take, endOfInput, choice, takeTill, takeByteString) import qualified Data.Attoparsec.ByteString.Char8 as Attoparsec import GHC.TypeLits import Data.Data import qualified Control.Monad.State.Strict as State import qualified Data.Text as Text import Network.HTTP.Types.URI import qualified Data.List as List import Unsafe.Coerce import IHP.HaskellSupport hiding (get) import qualified Data.Typeable as Typeable import qualified Data.ByteString.Char8 as ByteString import qualified Data.Char as Char import Control.Monad.Fail import Data.String.Conversions (ConvertibleStrings (convertString), cs) import qualified Text.Blaze.Html5 as Html5 import qualified IHP.ErrorController as ErrorController import qualified Control.Exception as Exception import qualified Data.List.Split as List import qualified Network.URI.Encode as URI import qualified Data.Text.Encoding as Text import Data.Dynamic import IHP.Router.Types import IHP.WebSocket (WSApp) import qualified IHP.WebSocket as WS import GHC.TypeLits as T import IHP.Controller.Context import IHP.Controller.Param import qualified Data.TMap as TMap import qualified IHP.ApplicationContext as ApplicationContext import Data.Kind runAction' :: forall application controller . ( Controller controller , ?applicationContext :: ApplicationContext , InitControllerContext application , ?application :: application , Typeable application , Typeable controller ) => controller -> Application runAction' controller request respond = do let ?modelContext = ApplicationContext.modelContext ?applicationContext requestContext <- createRequestContext ?applicationContext request respond let ?context = requestContext let ?requestContext = requestContext contextOrErrorResponse <- newContextForAction controller case contextOrErrorResponse of Left res -> res Right context -> let ?context = context in runAction controller {-# INLINABLE runAction' #-} class FrontController application where controllers :: (?applicationContext :: ApplicationContext, ?application :: application, ?context :: RequestContext) => [Parser Application] router :: (?applicationContext :: ApplicationContext, ?application :: application, ?context :: RequestContext) => [Parser Application] -> Parser Application router = defaultRouter {-# INLINABLE router #-} defaultRouter :: (?applicationContext :: ApplicationContext, ?application :: application, ?context :: RequestContext, FrontController application) => [Parser Application] -> Parser Application defaultRouter additionalControllers = do let allControllers = controllers <> additionalControllers applications <- choice $ map (\r -> r <* endOfInput) allControllers pure applications {-# INLINABLE defaultRouter #-} class HasPath controller where -- | Returns the path to a given action -- -- >>> pathTo UsersAction -- "/Users" -- -- >>> pathTo ShowUserAction { userId = "a32913dd-ef80-4f3e-9a91-7879e17b2ece" } -- "/ShowUser?userId=a32913dd-ef80-4f3e-9a91-7879e17b2ece" pathTo :: controller -> Text -- | Returns the url to a given action. -- -- Uses the baseUrl configured in @Config/Config.hs@. When no @baseUrl@ -- is configured in development mode, it will automatically detect the -- correct @baseUrl@ value. -- -- >>> urlTo UsersAction -- "http://localhost:8000/Users" -- -- >>> urlTo ShowUserAction { userId = "a32913dd-ef80-4f3e-9a91-7879e17b2ece" } -- "http://localhost:8000/ShowUser?userId=a32913dd-ef80-4f3e-9a91-7879e17b2ece" urlTo :: (?context :: context, ConfigProvider context, HasPath action) => action -> Text urlTo action = ?context.frameworkConfig.baseUrl <> pathTo action {-# INLINE urlTo #-} class HasPath controller => CanRoute controller where parseRoute' :: (?context :: RequestContext) => Parser controller -- | Each of these is tried when trying to parse an argument to a controller constructor (i.e. in IHP, an action). -- The type @d@ is an the type of the argument, and all we know about this type that its conforms to @Data@. -- We cannot cast @d@ to some arbitrary type, since adding additional constraints to @d@ (such as Read) -- will break the @fromConstrM@ function which actually constructs the action. -- -- The approach taken here is to make use of the type equality operator @:~:@ -- to check and see if @d@ happens to be a certain type. If it is, -- by matching on Just Refl, we are able to use @d@ as the type we matched it to. -- -- Please consult your doctor before engaging in Haskell type programming. parseFuncs :: forall d idType. (Data d, Data idType) => (ByteString -> Maybe idType) -> [Maybe ByteString -> Either TypedAutoRouteError d] parseFuncs parseIdType = [ -- Try and parse @Int@ or @Maybe Int@ \case Just queryValue -> case eqT :: Maybe (d :~: Int) of Just Refl -> readMay (cs queryValue :: String) |> \case Just int -> Right int Nothing -> Left BadType { field = "", value = Just queryValue, expectedType = "Int" } Nothing -> case eqT :: Maybe (d :~: Maybe Int) of Just Refl -> Right $ readMay (cs queryValue :: String) Nothing -> Left NotMatched Nothing -> case eqT :: Maybe (d :~: Maybe Int) of Just Refl -> Right Nothing Nothing -> Left NotMatched, \case Just queryValue -> case eqT :: Maybe (d :~: Integer) of Just Refl -> readMay (cs queryValue :: String) |> \case Just int -> Right int Nothing -> Left BadType { field = "", value = Just queryValue, expectedType = "Integer" } Nothing -> case eqT :: Maybe (d :~: Maybe Integer) of Just Refl -> Right $ readMay (cs queryValue :: String) Nothing -> Left NotMatched Nothing -> case eqT :: Maybe (d :~: Maybe Integer) of Just Refl -> Right Nothing Nothing -> Left NotMatched, -- Try and parse @Text@ or @Maybe Text@ \case Just queryValue -> case eqT :: Maybe (d :~: Text) of Just Refl -> Right $ cs queryValue Nothing -> case eqT :: Maybe (d :~: Maybe Text) of Just Refl -> Right $ Just $ cs queryValue Nothing -> Left NotMatched Nothing -> case eqT :: Maybe (d :~: Maybe Text) of Just Refl -> Right Nothing Nothing -> Left NotMatched, \case Just queryValue -> case parseIdType queryValue of Just idValue -> case eqT :: Maybe (d :~: idType) of Just Refl -> Right idValue Nothing -> Left NotMatched Nothing -> Left NotMatched Nothing -> Left NotMatched, -- Try and parse @[Text]@. If value is not present then default to empty list. \queryValue -> case eqT :: Maybe (d :~: [Text]) of Just Refl -> case queryValue of Just queryValue -> Right $ Text.splitOn "," (cs queryValue) Nothing -> Right [] Nothing -> Left NotMatched, -- Try and parse @[Int]@. If value is not present then default to empty list. \queryValue -> case eqT :: Maybe (d :~: [Int]) of Just Refl -> case queryValue of Just queryValue -> Text.splitOn "," (cs queryValue) |> map readMay |> catMaybes |> Right Nothing -> Right [] Nothing -> Left NotMatched, \queryValue -> case eqT :: Maybe (d :~: [Integer]) of Just Refl -> case queryValue of Just queryValue -> Text.splitOn "," (cs queryValue) |> map readMay |> catMaybes |> Right Nothing -> Right [] Nothing -> Left NotMatched, -- Try and parse a raw [UUID] \queryValue -> case eqT :: Maybe (d :~: [UUID]) of Just Refl -> case queryValue of Just queryValue -> queryValue |> cs |> Text.splitOn "," |> map (fromASCIIBytes . cs) |> catMaybes |> Right Nothing -> Right [] Nothing -> Left NotMatched, -- Try and parse a raw UUID \queryValue -> case eqT :: Maybe (d :~: UUID) of Just Refl -> case queryValue of Just queryValue -> queryValue |> fromASCIIBytes |> \case Just uuid -> uuid |> Right Nothing -> Left BadType { field = "", value = Just queryValue, expectedType = "UUID" } Nothing -> Left NotMatched Nothing -> Left NotMatched, -- This has to be last parser in the list -- -- Try and parse a UUID wrapped with a Id. In IHP types these are wrapped in a newtype @Id@ such as @Id User@. -- Since @Id@ is a newtype wrapping a UUID, it has the same data representation in GHC. -- Therefore, we're able to safely cast it to its @Id@ type with @unsafeCoerce@. -- -- We cannot use 'eqT' here for checking the types, as it's wrapped inside the @Id@ type. We expect -- that if it looks like a UUID, we can just treat it like an @Id@ type. For that to not overshadow other -- parsers, we need to have this last. \queryValue -> case queryValue of Just queryValue -> queryValue |> fromASCIIBytes |> \case Just uuid -> uuid |> unsafeCoerce |> Right Nothing -> Left BadType { field = "", value = Just queryValue, expectedType = "UUID" } Nothing -> Left NotMatched ] {-# INLINABLE parseFuncs #-} -- | As we fold over a constructor, we want the values parsed from the query string -- to be in the same order as they are in the constructor. -- This function uses the field labels from the constructor to sort the values from -- the query string. As a consequence, constructors with basic record syntax will not work with auto types. -- -- @data MyController = MyAction Text Int@ -- -- does not work. Instead use, -- -- @data MyController = MyAction { textArg :: Text, intArg :: Int }@ querySortedByFields :: Query -> Constr -> Query querySortedByFields query constructor = constrFields constructor |> map cs |> map (\field -> (field, join $ List.lookup field query)) {-# INLINABLE querySortedByFields #-} -- | Given a constructor and a parsed query string, attempt to construct a value of the constructor's type. -- For example, given the controller -- -- @data MyController = MyAction { textArg :: Text, intArg :: Int }@ -- -- this function will receive a representation of the @MyAction@ constructor as well as some query string -- @[("textArg", "some text"), ("intArg", "123")]@. -- -- By iterating through the query and attempting to match the type of each constructor argument -- with some transformation of the query string, we attempt to call @MyAction@. applyConstr :: (Data controller, Data idType) => (ByteString -> Maybe idType) -> Constr -> Query -> Either TypedAutoRouteError controller applyConstr parseIdType constructor query = let -- | Given some query item (key, optional value), try to parse into the current expected type -- by iterating through the available parse functions. attemptToParseArg :: forall d. (Data d) => (ByteString, Maybe ByteString) -> [Maybe ByteString -> Either TypedAutoRouteError d] -> State.StateT Query (Either TypedAutoRouteError) d attemptToParseArg queryParam@(queryName, queryValue) [] = State.lift (Left NoConstructorMatched { field = queryName , value = queryValue , expectedType = (dataTypeOf (Prelude.undefined :: d)) |> dataTypeName |> cs }) attemptToParseArg queryParam@(k, v) (parseFunc:restFuncs) = case parseFunc v of Right result -> pure result -- BadType will be returned if, for example, a text is passed to a query parameter typed as int. Left badType@BadType{} -> State.lift (Left badType { field = k }) -- otherwise, safe to assume the match just failed, so recurse on the rest of the functions and try to find one that matches. Left _ -> attemptToParseArg queryParam restFuncs -- | Attempt to parse the current expected type, and return its value. -- For the example @MyController@ this is called twice by @fromConstrM@. -- Once, it is called for @textArg@ where @d :: Text@. Then it is called -- for @intArg@ with @d ::: Int@. With both of these values parsed from the query string, -- the controller action is able to be created. nextField :: forall d. (Data d) => State.StateT Query (Either TypedAutoRouteError) d nextField = do queryParams <- State.get case queryParams of [] -> State.lift (Left TooFewArguments) (p@(key, value):rest) -> do State.put rest attemptToParseArg p (parseFuncs parseIdType) in case State.runStateT (fromConstrM nextField constructor) (querySortedByFields query constructor) of Right (x, []) -> pure x Right (_) -> Left TooFewArguments Left e -> Left e -- runtime type error {-# INLINABLE applyConstr #-} class Data controller => AutoRoute controller where autoRouteWithIdType :: (?context :: RequestContext, Data idType) => (ByteString -> Maybe idType) -> Parser controller autoRouteWithIdType parseIdFunc = let allConstructors :: [Constr] allConstructors = dataTypeConstrs (dataTypeOf (Prelude.undefined :: controller)) query :: Query query = queryString ?context.request paramValues :: [ByteString] paramValues = catMaybes $ map snd query parseAction :: Constr -> Parser controller parseAction constr = let prefix :: ByteString prefix = ByteString.pack (actionPrefix @controller) actionName = ByteString.pack (showConstr constr) actionPath :: ByteString actionPath = stripActionSuffixByteString actionName allowedMethods = allowedMethodsForAction @controller actionName checkRequestMethod action = do method <- getMethod unless (allowedMethods |> includes method) (Exception.throw UnexpectedMethodException { allowedMethods, method }) pure action action = case applyConstr parseIdFunc constr query of Right parsedAction -> pure parsedAction Left e -> Exception.throw e in do parsedAction <- string prefix >> (string actionPath <* endOfInput) *> action checkRequestMethod parsedAction in choice (map parseAction allConstructors) {-# INLINABLE autoRouteWithIdType #-} autoRoute :: (?context :: RequestContext) => Parser controller autoRoute = autoRouteWithIdType (\_ -> Nothing :: Maybe Integer) {-# INLINABLE autoRoute #-} -- | Specifies the allowed HTTP methods for a given action -- -- The default implementation does a smart guess based on the -- usual naming conventions for controllers. -- -- __Example (for default implementation):__ -- -- >>> allowedMethodsForAction @ProjectsController "DeleteProjectAction" -- [DELETE] -- -- >>> allowedMethodsForAction @ProjectsController "UpdateProjectAction" -- [POST, PATCH] -- -- >>> allowedMethodsForAction @ProjectsController "CreateProjectAction" -- [POST] -- -- >>> allowedMethodsForAction @ProjectsController "ShowProjectAction" -- [GET, HEAD] -- -- >>> allowedMethodsForAction @ProjectsController "HelloAction" -- [GET, POST, HEAD] -- allowedMethodsForAction :: ByteString -> [StdMethod] allowedMethodsForAction actionName = case actionName of a | "Delete" `ByteString.isPrefixOf` a -> [DELETE] a | "Update" `ByteString.isPrefixOf` a -> [POST, PATCH] a | "Create" `ByteString.isPrefixOf` a -> [POST] a | "Show" `ByteString.isPrefixOf` a -> [GET, HEAD] _ -> [GET, POST, HEAD] {-# INLINE allowedMethodsForAction #-} -- | Returns the url prefix for a controller. The prefix is based on the -- module where the controller is defined. -- -- All controllers defined in the `Web/` directory don't have a prefix at all. -- -- E.g. controllers in the `Admin/` directory are prefixed with @/admin/@. actionPrefix :: forall (controller :: Type). Typeable controller => String actionPrefix = case moduleName of ('W':'e':'b':'.':_) -> "/" ('I':'H':'P':'.':_) -> "/" ("") -> "/" moduleName -> "/" <> let prefix = getPrefix "" moduleName in map Char.toLower prefix <> "/" where moduleName :: String moduleName = Typeable.typeOf (error "unreachable" :: controller) |> Typeable.typeRepTyCon |> Typeable.tyConModule -- E.g. getPrefix "" "Admin.User" == "Admin" getPrefix prefix ('.':_) = prefix getPrefix prefix (x:xs) = getPrefix (prefix <> [x]) xs getPrefix prefix [] = prefix {-# INLINE actionPrefix #-} -- | Strips the "Action" at the end of action names -- -- >>> stripActionSuffixString "ShowUserAction" -- "ShowUser" -- -- >>> stripActionSuffixString "UsersAction" -- "UsersAction" -- -- >>> stripActionSuffixString "User" -- "User" stripActionSuffixString :: String -> String stripActionSuffixString string = case string of "Action" -> "" (x:xs) -> x : stripActionSuffixString xs "" -> "" {-# INLINE stripActionSuffixString #-} -- | Like 'stripActionSuffixString' but for ByteStrings stripActionSuffixByteString :: ByteString -> ByteString stripActionSuffixByteString actionName = fromMaybe actionName (ByteString.stripSuffix "Action" actionName) {-# INLINE stripActionSuffixByteString #-} -- | Returns the create action for a given controller. -- Example: `createAction @UsersController == Just CreateUserAction` createAction :: forall controller. AutoRoute controller => Maybe controller createAction = fmap fromConstr createConstructor where createConstructor :: Maybe Constr createConstructor = find isCreateConstructor allConstructors allConstructors :: [Constr] allConstructors = dataTypeConstrs (dataTypeOf (Prelude.undefined :: controller)) isCreateConstructor :: Constr -> Bool isCreateConstructor constructor = "Create" `isPrefixOf` showConstr constructor && ClassyPrelude.null (constrFields constructor) {-# INLINE createAction #-} -- | Returns the update action when given a controller and id. -- Example: `updateAction @UsersController == Just (\id -> UpdateUserAction id)` updateAction :: forall controller id. AutoRoute controller => Maybe (id -> controller) updateAction = case updateConstructor of Just constructor -> Just $ \id -> buildInstance constructor id Nothing -> Nothing where updateConstructor :: Maybe Constr updateConstructor = find isUpdateConstructor allConstructors buildInstance :: Constr -> id -> controller buildInstance constructor id = State.evalState ((fromConstrM (do i <- State.get State.modify (+1) pure (unsafeCoerce id) )) constructor) 0 allConstructors :: [Constr] allConstructors = dataTypeConstrs (dataTypeOf (Prelude.undefined :: controller)) isUpdateConstructor :: Constr -> Bool isUpdateConstructor constructor = "Update" `isPrefixOf` (showConstr constructor) && (length (constrFields constructor) == 1) {-# INLINE updateAction #-} instance {-# OVERLAPPABLE #-} (AutoRoute controller, Controller controller) => CanRoute controller where parseRoute' = autoRoute {-# INLINABLE parseRoute' #-} -- | Instances of the @QueryParam@ type class can be represented in URLs as query parameters. -- Currently this is only Int, Text, and both wrapped in List and Maybe. -- IDs also are representable in a URL, but we are unable to match on polymorphic types using reflection, -- so we fall back to the default "show" for these. class Data a => QueryParam a where showQueryParam :: a -> String instance QueryParam Text where showQueryParam text = Text.unpack text instance QueryParam Int where showQueryParam = show instance QueryParam Integer where showQueryParam = show instance QueryParam UUID where showQueryParam = show instance QueryParam a => QueryParam (Maybe a) where showQueryParam (Just val) = showQueryParam val showQueryParam Nothing = "" instance QueryParam a => QueryParam [a] where showQueryParam = List.intercalate "," . map showQueryParam instance {-# OVERLAPPABLE #-} (Show controller, AutoRoute controller) => HasPath controller where {-# INLINABLE pathTo #-} pathTo !action = Text.pack (appPrefix <> actionName <> arguments) where appPrefix :: String !appPrefix = actionPrefix @controller actionName :: String !actionName = stripActionSuffixString $! showConstr constructor constructor = toConstr action stripQuotes ('"':rest) = List.init rest stripQuotes otherwise = otherwise -- | The @gmapQ@ function allows us to iterate over each term in a constructor function and -- build a list of results from performing some function on each term. -- Here we send each term through @constrShow@, giving us our preferred representation for -- use in URLs. showTerms :: controller -> [Maybe String] showTerms = gmapQ (constrShow typeShows) -- | @constrShow@ tries to convert each value @d@ into a String representation. -- If one passes, return it immediately, otherwise try all the defined @typeShow@ functions. constrShow :: Data d => [(d -> Maybe String)] -> d -> Maybe String constrShow [] _ = Nothing constrShow (f:fs) d = case f d of Just str -> Just str Nothing -> constrShow fs d -- | Try and match some value to all of the types we can represent in a URL. -- Only type not contained in here is the "Id" type, since we cannot match -- on polymorphic types. typeShows :: forall d. Data d => [(d -> Maybe String)] typeShows = [ \val -> (eqT :: Maybe (d :~: Text)) >>= \Refl -> Just (showQueryParam val), \val -> (eqT :: Maybe (d :~: [Text])) >>= \Refl -> Just (showQueryParam (val :: [Text])), \val -> (eqT :: Maybe (d :~: Maybe Text)) >>= \Refl -> Just (showQueryParam val), \val -> (eqT :: Maybe (d :~: Int)) >>= \Refl -> Just (showQueryParam val), \val -> (eqT :: Maybe (d :~: [Int])) >>= \Refl -> Just (showQueryParam val), \val -> (eqT :: Maybe (d :~: Maybe Int)) >>= \Refl -> Just (showQueryParam val), \val -> (eqT :: Maybe (d :~: Integer)) >>= \Refl -> Just (showQueryParam val), \val -> (eqT :: Maybe (d :~: [Integer])) >>= \Refl -> Just (showQueryParam val), \val -> (eqT :: Maybe (d :~: Maybe Integer)) >>= \Refl -> Just (showQueryParam val), \val -> (eqT :: Maybe (d :~: UUID)) >>= \Refl -> Just (showQueryParam val), \val -> (eqT :: Maybe (d :~: Maybe UUID)) >>= \Refl -> Just (showQueryParam val), \val -> (eqT :: Maybe (d :~: [UUID])) >>= \Refl -> Just (showQueryParam val) ] arguments :: String !arguments = show action -- `SomeRecord { a = b, c = d }` |> List.filter (/= ' ') |> List.break (== '{') |> snd |> List.drop 1 |> List.break (== '}') |> fst -- `a=b,c=d` |> List.splitWhen (== ',') -- ["a=b", "c=d"] |> map (\s -> let (key, value) = List.break (== '=') s in (key, List.drop 1 value)) |> map (\(k ,v) -> (k, stripQuotes v)) -- "value" -> value |> filter (\(k, v) -> (not . List.null) k && (not . List.null) v) -- At this point we have a list of keys and values as represented by @show@. -- For Lists and Maybe types, we want to represent these in a different way, -- so we construct another list of values using type reflection and the QueryParam type class. |> \(kvs :: [(String, String)]) -> zip (showTerms action) kvs -- If an Id type was present in the action, it will be returned as Nothing by @showTerms@ -- as we are not able to match on the type using reflection. -- In this case we default back to the @show@ representation. |> map (\(v1, (k, v2)) -> (k, fromMaybe v2 v1)) |> map (\(k, v) -> if isEmpty v then "" else k <> "=" <> URI.encode v) |> List.filter (not . isEmpty) |> List.intercalate "&" |> (\q -> if List.null q then q else '?':q) -- | Parses the HTTP Method from the request and returns it. getMethod :: (?context :: RequestContext) => Parser StdMethod getMethod = case parseMethod ?context.request.requestMethod of Left error -> fail (ByteString.unpack error) Right method -> pure method {-# INLINABLE getMethod #-} -- | Routes a given path to an action when requested via GET. -- -- __Example:__ -- -- > instance FrontController WebApplication where -- > controllers = [ -- > get "/my-custom-page" NewSessionAction -- > ] -- -- The request @GET \/my-custom-page@ is now executing NewSessionAction -- -- Also see 'post'. get :: (Controller action , InitControllerContext application , ?application :: application , ?applicationContext :: ApplicationContext , ?context :: RequestContext , Typeable application , Typeable action ) => ByteString -> action -> Parser Application get path action = do method <- getMethod case method of GET -> do string path pure (runAction' action) _ -> fail "Invalid method, expected GET" {-# INLINABLE get #-} -- | Routes a given path to an action when requested via POST. -- -- __Example:__ -- -- > instance FrontController WebApplication where -- > controllers = [ -- > post "/do-something" DoSomethingAction -- > ] -- -- The request @POST \/do-something@ is now executing DoSomethingAction -- -- Also see 'get'. post :: (Controller action , InitControllerContext application , ?application :: application , ?applicationContext :: ApplicationContext , ?context :: RequestContext , Typeable application , Typeable action ) => ByteString -> action -> Parser Application post path action = do method <- getMethod case method of POST -> do string path pure (runAction' action) _ -> fail "Invalid method, expected POST" {-# INLINABLE post #-} -- | Filter methods when writing a custom routing parser -- -- __Example:__ -- -- > instance CanRoute ApiController where -- > parseRoute' = do -- > string "/api/" -- > let -- > createRecordAction = do -- > onlyAllowMethods [POST] -- > -- > table <- parseText -- > endOfInput -- > pure CreateRecordAction { table } -- > -- > updateRecordAction = do -- > onlyAllowMethods [PATCH] -- > -- > table <- parseText -- > string "/" -- > id <- parseUUID -- > pure UpdateRecordAction { table, id } -- > -- > createRecordAction <|> updateRecordAction -- onlyAllowMethods :: (?context :: RequestContext) => [StdMethod] -> Parser () onlyAllowMethods methods = do method <- getMethod unless (method `elem` methods) (fail ("Invalid method, expected one of: " <> show methods)) {-# INLINABLE onlyAllowMethods #-} -- | Routes to a given WebSocket app if the path matches the WebSocket app name -- -- __Example:__ -- -- > instance FrontController WebApplication where -- > controllers = [ -- > webSocketApp @AutoRefreshWSApp -- > ] -- -- The request @\/AutoRefreshWSApp@ will call the AutoRefreshWSApp -- webSocketApp :: forall webSocketApp application. ( WSApp webSocketApp , InitControllerContext application , ?application :: application , ?applicationContext :: ApplicationContext , ?context :: RequestContext , Typeable application , Typeable webSocketApp ) => Parser Application webSocketApp = webSocketAppWithCustomPath @webSocketApp typeName where typeName :: ByteString typeName = Typeable.typeOf (error "unreachable" :: webSocketApp) |> show |> ByteString.pack {-# INLINABLE webSocketApp #-} webSocketAppWithHTTPFallback :: forall webSocketApp application. ( WSApp webSocketApp , InitControllerContext application , ?application :: application , ?applicationContext :: ApplicationContext , ?context :: RequestContext , Typeable application , Typeable webSocketApp , Controller webSocketApp ) => Parser Application webSocketAppWithHTTPFallback = webSocketAppWithCustomPathAndHTTPFallback @webSocketApp @application typeName where typeName :: ByteString typeName = Typeable.typeOf (error "unreachable" :: webSocketApp) |> show |> ByteString.pack {-# INLINABLE webSocketAppWithHTTPFallback #-} -- | Routes to a given WebSocket app if the path matches -- -- __Example:__ -- -- > instance FrontController WebApplication where -- > controllers = [ -- > webSocketAppWithCustomPath @AutoRefreshWSApp "my-ws-app" -- > ] -- -- The request @\/my-ws-app@ will call the AutoRefreshWSApp -- webSocketAppWithCustomPath :: forall webSocketApp application. ( WSApp webSocketApp , InitControllerContext application , ?application :: application , ?applicationContext :: ApplicationContext , ?context :: RequestContext , Typeable application , Typeable webSocketApp ) => ByteString -> Parser Application webSocketAppWithCustomPath path = do Attoparsec.char '/' string path pure (startWebSocketAppAndFailOnHTTP (WS.initialState @webSocketApp)) {-# INLINABLE webSocketAppWithCustomPath #-} webSocketAppWithCustomPathAndHTTPFallback :: forall webSocketApp application. ( WSApp webSocketApp , InitControllerContext application , ?application :: application , ?applicationContext :: ApplicationContext , ?context :: RequestContext , Typeable application , Typeable webSocketApp , Controller webSocketApp ) => ByteString -> Parser Application webSocketAppWithCustomPathAndHTTPFallback path = do Attoparsec.char '/' string path let action = WS.initialState @webSocketApp pure (startWebSocketApp action (runActionWithNewContext action)) {-# INLINABLE webSocketAppWithCustomPathAndHTTPFallback #-} -- | Defines the start page for a router (when @\/@ is requested). startPage :: forall action application. (Controller action, InitControllerContext application, ?application::application, ?applicationContext::ApplicationContext, ?context::RequestContext, Typeable application, Typeable action) => action -> Parser Application startPage action = get (ByteString.pack (actionPrefix @action)) action {-# INLINABLE startPage #-} withPrefix prefix routes = string prefix >> choice (map (\r -> r <* endOfInput) routes) {-# INLINABLE withPrefix #-} frontControllerToWAIApp :: forall app (autoRefreshApp :: Type). (?applicationContext :: ApplicationContext, FrontController app, WSApp autoRefreshApp, Typeable autoRefreshApp, InitControllerContext ()) => Middleware -> app -> Application -> Application frontControllerToWAIApp middleware application notFoundAction request respond = do let requestContext = RequestContext { request, respond, requestBody = FormBody { params = [], files = [] }, frameworkConfig = ?applicationContext.frameworkConfig } let ?context = requestContext let path = request.rawPathInfo handleException :: SomeException -> IO (Either String Application) handleException exception = pure $ Right $ ErrorController.handleRouterException exception routes = let ?application = application in router [let ?application = () in webSocketApp @autoRefreshApp] routedAction :: Either String Application <- (do res <- evaluate $ parseOnly (routes <* endOfInput) path case res of Left s -> pure $ Left s Right action -> do pure $ Right action ) `Exception.catch` handleException case routedAction of Left message -> notFoundAction request respond Right action -> (middleware action) request respond {-# INLINABLE frontControllerToWAIApp #-} mountFrontController :: forall frontController. (?applicationContext :: ApplicationContext, ?context :: RequestContext, FrontController frontController) => frontController -> Parser Application mountFrontController application = let ?application = application in router [] {-# INLINABLE mountFrontController #-} parseRoute :: forall controller application. (?applicationContext :: ApplicationContext, ?context :: RequestContext, Controller controller, CanRoute controller, InitControllerContext application, ?application :: application, Typeable application, Typeable controller) => Parser Application parseRoute = do action <- parseRoute' @controller pure $ runAction' @application action {-# INLINABLE parseRoute #-} parseUUIDOrTextId :: ByteString -> Maybe Dynamic parseUUIDOrTextId queryVal = queryVal |> fromASCIIBytes |> \case Just uuid -> uuid |> toDyn |> Just Nothing -> Nothing parseRouteWithId :: forall controller application. (?applicationContext :: ApplicationContext, ?context :: RequestContext, Controller controller, CanRoute controller, InitControllerContext application, ?application :: application, Typeable application, Data controller) => Parser Application parseRouteWithId = do action <- parseRoute' @controller pure (runAction' @application action) catchAll :: forall action application. (?applicationContext :: ApplicationContext, ?context :: RequestContext, Controller action, InitControllerContext application, Typeable action, ?application :: application, Typeable application, Data action) => action -> Parser Application catchAll action = do string (ByteString.pack (actionPrefix @action)) _ <- takeByteString pure (runAction' @application action) {-# INLINABLE catchAll #-} -- | This instances makes it possible to write @@ in HSX instance {-# OVERLAPPABLE #-} (HasPath action) => ConvertibleStrings action Html5.AttributeValue where convertString action = Html5.textValue (pathTo action) {-# INLINE convertString #-} -- | Parses and returns an UUID parseUUID :: Parser UUID parseUUID = do uuid <- take 36 case fromASCIIBytes uuid of Just theUUID -> pure theUUID Nothing -> fail "not uuid" {-# INLINABLE parseUUID #-} -- | Parses an UUID, afterwards wraps it in an Id parseId :: ((ModelSupport.PrimaryKey table) ~ UUID) => Parser (ModelSupport.Id' table) parseId = ModelSupport.Id <$> parseUUID {-# INLINABLE parseId #-} -- | Returns all the remaining text until the end of the input remainingText :: Parser Text remainingText = Text.decodeUtf8 <$> takeByteString {-# INLINABLE remainingText #-} -- | Parses until the next @/@ parseText :: Parser Text parseText = Text.decodeUtf8 <$> takeTill ('/' ==) {-# INLINABLE parseText #-} parseIntegerId :: (Data idType) => ByteString -> Maybe idType parseIntegerId queryVal = let rawValue :: Maybe Integer = readMay (cs queryVal :: String) in rawValue >>= Just . unsafeCoerce -- | Parses and returns an integer -- parseRational :: (Integral a) => Parser a -- parseRational = Attoparsec.decimal -- | Parses a route query parameter -- -- __Example:__ -- -- > let showPost = do -- > string "/post" -- > let postId = routeParam "postId" -- > pure ShowPostAction { .. } -- Will parse the `postId` query in `/post?postId=09b545dd-9744-4ef8-87b8-8d227f4faa1e` -- routeParam :: (?context::RequestContext, ParamReader paramType) => ByteString -> paramType routeParam paramName = let requestContext = ?context in let ?context = FrozenControllerContext { requestContext = requestContext, customFields = mempty } in param paramName -- | Display a better error when the user missed to pass an argument to an action. -- -- E.g. when you forgot to pass a projectId to the ShowProjectAction: -- -- > Show project -- -- The correct code would be this: -- -- > Show project -- -- See https://github.com/digitallyinduced/ihp/issues/840 instance ((T.TypeError (T.Text "Looks like you forgot to pass a " :<>: (T.ShowType argument) :<>: T.Text " to this " :<>: (T.ShowType controller))), Data argument, Data controller, Data (argument -> controller)) => AutoRoute (argument -> controller) where