-- Copyright (c) 2016-present, Facebook, Inc. -- All rights reserved. -- -- This source code is licensed under the BSD-style license found in the -- LICENSE file in the root directory of this source tree. {-# LANGUAGE GADTs #-} {-# LANGUAGE LambdaCase #-} {-# LANGUAGE NamedFieldPuns #-} {-# LANGUAGE NoRebindableSyntax #-} {-# LANGUAGE OverloadedStrings #-} module Duckling.Time.BG.Rules ( rules ) where import Control.Applicative ((<|>)) import Data.Maybe import Data.Text (Text) import Prelude import qualified Data.Text as Text import Duckling.Dimensions.Types import Duckling.Duration.Helpers (duration) import Duckling.Duration.Types (DurationData (..)) import Duckling.Numeral.Helpers (isNatural, parseInt) import Duckling.Numeral.Types (NumeralData (..)) import Duckling.Ordinal.Types (OrdinalData (..)) import Duckling.Regex.Types import Duckling.Time.Computed import Duckling.Time.Helpers import Duckling.Time.Types (TimeData (..), TimeIntervalType (..)) import Duckling.Types import qualified Duckling.Duration.Types as TDuration import qualified Duckling.Numeral.Types as TNumeral import qualified Duckling.Ordinal.Types as TOrdinal import qualified Duckling.Time.Types as TTime import qualified Duckling.TimeGrain.Types as TG ruleIntersect :: Rule ruleIntersect = Rule { name = "intersect" , pattern = [ Predicate $ and . sequence [isNotLatent, isGrainFinerThan TG.Year] , Predicate $ or . sequence [isNotLatent, isGrainOfTime TG.Year] ] , prod = \case (Token Time td1:Token Time td2:_) -> Token Time <$> intersect td1 td2 _ -> Nothing } ruleIntersectOf :: Rule ruleIntersectOf = Rule { name = "intersect by \",\", \"of\", \"from\", \"'s\"" , pattern = [ Predicate isNotLatent , regex "от|за|на|," , Predicate isNotLatent ] , prod = \case (Token Time td1:_:Token Time td2:_) -> Token Time . notLatent <$> intersect td1 td2 _ -> Nothing } ruleIntersectYear :: Rule ruleIntersectYear = Rule { name = "intersect by \",\", \"of\", \"from\" for year" , pattern = [ Predicate isNotLatent , regex "на|," , Predicate $ isGrainOfTime TG.Year ] , prod = \case (Token Time td1:_:Token Time td2:_) -> Token Time . notLatent <$> intersect td1 td2 _ -> Nothing } ruleAbsorbOnDay :: Rule ruleAbsorbOnDay = Rule { name = "on " , pattern = [ regex "на" , Predicate $ isGrainOfTime TG.Day ] , prod = \case (_:token:_) -> Just token _ -> Nothing } ruleAbsorbOnADOW :: Rule ruleAbsorbOnADOW = Rule { name = "on a " , pattern = [ regex "в" , Predicate isADayOfWeek ] , prod = \case (_:token:_) -> Just token _ -> Nothing } ruleAbsorbInMonthYear :: Rule ruleAbsorbInMonthYear = Rule { name = "in|during |year" , pattern = [ regex "в|през" , Predicate $ or . sequence [isAMonth, isGrainOfTime TG.Year] ] , prod = \case (_:Token Time td:_) -> tt $ notLatent td _ -> Nothing } ruleAbsorbCommaTOD :: Rule ruleAbsorbCommaTOD = Rule { name = "absorption of , after named day" , pattern = [ Predicate isADayOfWeek , regex "," ] , prod = \case (token:_) -> Just token _ -> Nothing } ruleInstants :: [Rule] ruleInstants = mkRuleInstants [ ("right now" , TG.Second, 0 , "((точно\\s+)?сега)|веднага") , ("today" , TG.Day , 0 , "днес|(по това време)" ) , ("tomorrow" , TG.Day , 1 , "утре" ) , ("yesterday" , TG.Day , - 1, "вчера" ) , ( "after tomorrow" , TG.Day , 2 , "(в\\s*)?другиден" ) , ( "before yesterday", TG.Day , - 2, "(оня ден)|завчера" ) ] ruleNow :: Rule ruleNow = Rule { name = "now" , pattern = [ regex "сега" ] , prod = \_ -> tt now } ruleNextDOW :: Rule ruleNextDOW = Rule { name = "this|next " , pattern = [ regex "(т(о|а)зи)|следващ((ия(т)?)|ата)" , Predicate isADayOfWeek ] , prod = \case (_:Token Time td:_) -> tt $ predNth 0 True td _ -> Nothing } ruleThisTime :: Rule ruleThisTime = Rule { name = "this