-- 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. An additional grant -- of patent rights can be found in the PATENTS file in the same directory. {-# LANGUAGE GADTs #-} {-# LANGUAGE NoRebindableSyntax #-} {-# LANGUAGE OverloadedStrings #-} module Duckling.Time.SV.Rules ( rules ) where import Data.Text (Text) import Prelude import Duckling.Dimensions.Types import Duckling.Numeral.Helpers (parseInt) import Duckling.Ordinal.Types (OrdinalData (..)) import Duckling.Regex.Types import Duckling.Time.Helpers import Duckling.Time.Types (TimeData (..)) import Duckling.Types import qualified Duckling.Ordinal.Types as TOrdinal import qualified Duckling.Time.Types as TTime import qualified Duckling.TimeGrain.Types as TG daysOfWeek :: [(Text, String)] daysOfWeek = [ ( "Mandag" , "måndag(en)?s?|mån\\.?" ) , ( "Tisdag" , "tisdag(en)?s?|tis?\\.?" ) , ( "Onsdag" , "onsdag(en)?s?|ons\\.?" ) , ( "Torsdag" , "torsdag(en)?s?|tors?\\.?" ) , ( "Fredag" , "fredag(en)?s?|fre\\.?" ) , ( "Lordag" , "lördag(en)?s?|lör\\.?" ) , ( "Sondag" , "söndag(en)?s?|sön\\.?" ) ] ruleDaysOfWeek :: [Rule] ruleDaysOfWeek = zipWith go daysOfWeek [1..7] where go (name, regexPattern) i = Rule { name = name , pattern = [regex regexPattern] , prod = \_ -> tt $ dayOfWeek i } months :: [(Text, String)] months = [ ("Januari" , "januari|jan\\.?" ) , ("Februari" , "februari|feb\\.?" ) , ("Mars" , "mars|mar\\.?" ) , ("April" , "april|apr\\.?" ) , ("Maj" , "maj" ) , ("Juni" , "juni|jun\\.?" ) , ("Juli" , "juli|jul\\.?" ) , ("Augusti" , "augusti|aug\\.?" ) , ("September" , "september|sept?\\.?" ) , ("Oktober" , "oktober|okt\\.?" ) , ("November" , "november|nov\\.?" ) , ("December" , "december|dec\\.?" ) ] ruleMonths :: [Rule] ruleMonths = zipWith go months [1..12] where go (name, regexPattern) i = Rule { name = name , pattern = [regex regexPattern] , prod = \_ -> tt $ month i } ruleTheDayAfterTomorrow :: Rule ruleTheDayAfterTomorrow = Rule { name = "the day after tomorrow" , pattern = [ regex "i överimorgon" ] , prod = \_ -> tt $ cycleNth TG.Day 2 } ruleRelativeMinutesTotillbeforeIntegerHourofday :: Rule ruleRelativeMinutesTotillbeforeIntegerHourofday = Rule { name = "relative minutes to|till|before (hour-of-day)" , pattern = [ Predicate $ isIntegerBetween 1 59 , regex "i" , Predicate isAnHourOfDay ] , prod = \tokens -> case tokens of (token:_:Token Time td:_) -> do n <- getIntValue token t <- minutesBefore n td Just $ Token Time t _ -> Nothing } ruleRelativeMinutesAfterpastIntegerHourofday :: Rule ruleRelativeMinutesAfterpastIntegerHourofday = Rule { name = "relative minutes after|past (hour-of-day)" , pattern = [ Predicate $ isIntegerBetween 1 59 , regex "över" , Predicate isAnHourOfDay ] , prod = \tokens -> case tokens of (token: _: Token Time TimeData {TTime.form = Just (TTime.TimeOfDay (Just hours) _)}: _) -> do n <- getIntValue token tt $ hourMinute True hours n _ -> Nothing } ruleHourofdayIntegerAsRelativeMinutes :: Rule ruleHourofdayIntegerAsRelativeMinutes = Rule { name = " (as relative minutes)" , pattern = [ Predicate $ and . sequence [isNotLatent, isAnHourOfDay] , Predicate $ isIntegerBetween 1 59 ] , prod = \tokens -> case tokens of (Token Time TimeData {TTime.form = Just (TTime.TimeOfDay (Just hours) _)}: token: _) -> do n <- getIntValue token tt $ hourMinute True hours n _ -> Nothing } ruleQuarterTotillbeforeIntegerHourofday :: Rule ruleQuarterTotillbeforeIntegerHourofday = Rule { name = "quarter to|till|before (hour-of-day)" , pattern = [ regex "(en)? ?(kvart)(er)?" , regex "i" , Predicate isAnHourOfDay ] , prod = \tokens -> case tokens of (_:_:Token Time td:_) -> Token Time <$> minutesBefore 15 td _ -> Nothing } ruleQuarterAfterpastIntegerHourofday :: Rule ruleQuarterAfterpastIntegerHourofday = Rule { name = "quarter after|past (hour-of-day)" , pattern = [ regex "(en)? ?(kvart)(er)?" , regex "över" , Predicate isAnHourOfDay ] , prod = \tokens -> case tokens of (_: _: Token Time TimeData {TTime.form = Just (TTime.TimeOfDay (Just hours) _)}: _) -> tt $ hourMinute True hours 15 _ -> Nothing } ruleHourofdayQuarter :: Rule ruleHourofdayQuarter = Rule { name = " quarter (as relative minutes)" , pattern = [ Predicate isAnHourOfDay , regex "(en)? ?(kvart)(er)?" ] , prod = \tokens -> case tokens of (Token Time TimeData {TTime.form = Just (TTime.TimeOfDay (Just hours) _)}: _) -> tt $ hourMinute True hours 15 _ -> Nothing } ruleHalfTotillbeforeIntegerHourofday :: Rule ruleHalfTotillbeforeIntegerHourofday = Rule { name = "half to|till|before (hour-of-day)" , pattern = [ regex "halvtimme" , regex "i" , Predicate isAnHourOfDay ] , prod = \tokens -> case tokens of (_:_:Token Time td:_) -> Token Time <$> minutesBefore 30 td _ -> Nothing } ruleHalfAfterpastIntegerHourofday :: Rule ruleHalfAfterpastIntegerHourofday = Rule { name = "half after|past (hour-of-day)" , pattern = [ regex "halvtimme" , regex "över" , Predicate isAnHourOfDay ] , prod = \tokens -> case tokens of (_: _: Token Time TimeData {TTime.form = Just (TTime.TimeOfDay (Just hours) _)}: _) -> tt $ hourMinute True hours 30 _ -> Nothing } ruleHourofdayHalf :: Rule ruleHourofdayHalf = Rule { name = " half (as relative minutes)" , pattern = [ Predicate isAnHourOfDay , regex "halvtimme" ] , prod = \tokens -> case tokens of (Token Time TimeData {TTime.form = Just (TTime.TimeOfDay (Just hours) _)}: _) -> tt $ hourMinute True hours 30 _ -> Nothing } ruleTheOrdinalCycleOfTime :: Rule ruleTheOrdinalCycleOfTime = Rule { name = "the of