facebook/duckling
1
1
Fork 0
mirror of https://github.com/facebook/duckling.git synced 2024-09-11 21:27:13 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update custom dimension example to include pattern match on dimension data

Browse Source 
Summary: Pull Request resolved: https://github.com/facebook/duckling/pull/233

Reviewed By: watashi

Differential Revision: D9021889

Pulled By: chinmay87

fbshipit-source-id: 1e134cf6d20006e1ee8824fa2fcbb8f7cdc4f197
This commit is contained in:
Ziyang Liu 2018-08-13 16:35:38 -07:00 committed by Facebook Github Bot
parent 3de16f3fae
commit a365dec52a
1 changed files with 35 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
exe/CustomDimensionExample.hs
View File

@ -11,6 +11,8 @@
{-# LANGUAGE NoRebindableSyntax #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE RecordWildCards #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE TypeApplications #-}
{-# LANGUAGE TypeFamilies #-}
module Main (main) where
@ -30,6 +32,7 @@ import qualified TextShow as TS
import Duckling.Debug
import Duckling.Locale
import Duckling.Regex.Types (GroupMatch(..))
import Duckling.Resolve (Resolve(..))
import Duckling.Types
@ -37,7 +40,7 @@ data MyDimension = MyDimension deriving (Eq, Show, Typeable)
instance CustomDimension MyDimension where
type DimensionData MyDimension = MyData
dimRules _ = [myRule]
dimRules _ = [myDimensionRule, myDimensionRule']
dimLangRules _ _ = []
dimLocaleRules _ _ = []
dimDependents _ = HashSet.empty
@ -61,9 +64,15 @@ newtype MyValue = MyValue { value :: Text }
instance ToJSON MyValue where
toJSON (MyValue value) = object [ "value" .= value ]
myRule :: Rule
myRule = Rule
{ name = "my dimension"
myDimensionPredicate :: Predicate
myDimensionPredicate (Token (CustomDimension (dim :: a)) dimData)
| Just Refl <- eqT @a @MyDimension, MyData{..} <- dimData =
iField == 42 && bField
myDimensionPredicate _ = False
myDimensionRule :: Rule
myDimensionRule = Rule
{ name = "my dimension (simple)"
, pattern =
[ regex "my dimension"
]
@ -76,7 +85,28 @@ myRule = Rule
_ -> Nothing
}
myDimensionRule' :: Rule
myDimensionRule' = Rule
{ name = "my dimension (pattern match)"
, pattern =
[ Predicate myDimensionPredicate
, regex "pattern match"
]
, prod = \case
((Token (CustomDimension (dim :: a)) dimData):
Token RegexMatch (GroupMatch _):
_)
| Just Refl <- eqT @a @MyDimension, MyData{..} <- dimData ->
Just . Token (CustomDimension MyDimension) $ MyData
{ iField = iField * 10
, bField = not bField
, tField = "goodnight moon"
}
_ -> Nothing
}
main :: IO ()
main = do
let en = makeLocale EN Nothing
debug en "input for my dimension" [This (CustomDimension MyDimension)] >>= print
debug en "testing my dimension" [This (CustomDimension MyDimension)] >>= print
debug en "testing my dimension pattern match" [This (CustomDimension MyDimension)] >>= print