aboutsummaryrefslogtreecommitdiff
path: root/lib/Server/GTFSRT.hs
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--lib/Server/GTFSRT.hs84
1 files changed, 68 insertions, 16 deletions
diff --git a/lib/Server/GTFSRT.hs b/lib/Server/GTFSRT.hs
index 7035ccf..0a654c6 100644
--- a/lib/Server/GTFSRT.hs
+++ b/lib/Server/GTFSRT.hs
@@ -2,6 +2,7 @@
{-# LANGUAGE OverloadedLists #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE RecordWildCards #-}
+{-# LANGUAGE TupleSections #-}
{-# LANGUAGE TypeApplications #-}
module Server.GTFSRT (gtfsRealtimeServer) where
@@ -9,7 +10,8 @@ module Server.GTFSRT (gtfsRealtimeServer) where
import qualified Data.Sequence as Seq
import Data.Time.Clock.POSIX (getPOSIXTime)
import Data.Time.Clock.System (SystemTime (systemSeconds),
- getSystemTime)
+ getSystemTime,
+ utcToSystemTime)
import GTFS.Realtime.Alert as AL (Alert (..))
import GTFS.Realtime.Alert.Cause (Cause (CONSTRUCTION))
import GTFS.Realtime.Alert.Effect (Effect (DETOUR))
@@ -29,9 +31,11 @@ import Text.ProtocolBuffers (Utf8 (Utf8),
import Text.ProtocolBuffers.WireMessage (zzEncode64)
import API (GtfsRealtimeAPI)
+import Control.Monad (forM)
import Control.Monad.IO.Class (MonadIO (..))
import Data.ByteString.Lazy (fromStrict)
import Data.Functor ((<&>))
+import Data.Maybe (catMaybes)
import Data.Pool (Pool)
import Data.Sequence (Seq)
import Data.Text (Text)
@@ -39,14 +43,23 @@ import qualified Data.Text as T
import Data.Text.Encoding (encodeUtf8)
import Data.Time (Day)
import Data.Time.Calendar (Day, toGregorian)
+import Data.Time.Clock (UTCTime)
import qualified Data.UUID as UUID
+import Data.Word (Word64)
import Database.Persist (Entity (Entity),
- selectList)
+ PersistQueryRead (selectFirst),
+ selectList, (==.))
import Database.Persist.Postgresql (SqlBackend)
-import GTFS (GTFS)
+import GHC.Float (double2Float)
+import GTFS (GTFS, TripID)
+import GTFS.Realtime.Position as POS (Position (..))
+import GTFS.Realtime.VehicleDescriptor as VD (VehicleDescriptor (..))
+import GTFS.Realtime.VehiclePosition as VP (VehiclePosition (..))
import Persist (Announcement (..),
+ EntityField (TripPingToken),
Key (..),
- RunningTrip,
+ RunningTrip (..),
+ TripPing (..),
runSql)
import Servant.API ((:<|>) (..))
import Servant.Server (Handler (Handler),
@@ -69,6 +82,9 @@ toStupidDate date = toUtf8
else show num
where ndigits = length (show num)
+-- | basically unix timestamps, raw (because why not i guess)
+toStupidTime :: UTCTime -> Word64
+toStupidTime = fromIntegral . systemSeconds . utcToSystemTime
gtfsRealtimeServer :: GTFS -> Pool SqlBackend -> Server GtfsRealtimeAPI
gtfsRealtimeServer gtfs dbpool = handleServiceAlerts :<|> handleTripUpdates :<|> handleVehiclePositions
@@ -84,16 +100,8 @@ gtfsRealtimeServer gtfs dbpool = handleServiceAlerts :<|> handleTripUpdates :<|>
-- TODO: is this time range reasonable, needed, etc.?
, informed_entity =
[dEntitySelector
- { trip =
- Just (TripDescriptor
- { trip_id = Just (toUtf8 announcementTrip)
- , route_id = Nothing
- , direction_id = Nothing
- , start_time = Nothing
- , start_date = Just (toStupidDate announcementDay)
- , schedule_relationship = Nothing
- , TD.ext'field = defaultValue
- })
+ { ES.trip =
+ Just (dTripDescriptor announcementTrip (Just announcementDay))
}
]
, cause = Nothing
@@ -112,8 +120,41 @@ gtfsRealtimeServer gtfs dbpool = handleServiceAlerts :<|> handleTripUpdates :<|>
-- TODO: how to propagate delay values to next stops?
pure undefined
handleVehiclePositions = runSql dbpool $ do
- -- TODO: how to know which trips are currently running?
- pure undefined
+ (running :: [Entity RunningTrip]) <- selectList [] []
+ pings <- forM running $ \(Entity key entity) -> do
+ selectFirst [TripPingToken ==. key] [] <&> fmap (, entity)
+ dFeedMessage $ Seq.fromList $ fmap mkPosition $ catMaybes pings
+ where mkPosition (Entity (TripPingKey key) TripPing{..}, RunningTrip{..}) =
+ (dFeedEntity (toUtf8 . T.pack . show $ key))
+ { FE.vehicle = Just $ VehiclePosition
+ { trip = Just (dTripDescriptor runningTripTripNumber Nothing)
+ , VP.vehicle = case runningTripTrainset of
+ Nothing -> Nothing
+ Just trainset -> Just $ VehicleDescriptor
+ { VD.id = Nothing
+ , VD.label = (Just (toUtf8 trainset))
+ , VD.license_plate = Nothing
+ , VD.ext'field = defaultValue
+ }
+ , position = Just $ Position
+ { latitude = double2Float tripPingLat
+ , longitude = double2Float tripPingLong
+ , odometer = Nothing
+ , speed = Nothing
+ , bearing = Nothing
+ , POS.ext'field = defaultValue
+ }
+ , current_stop_sequence = Nothing
+ , stop_id = Nothing
+ , current_status = Nothing
+ , timestamp = Just (toStupidTime tripPingTimestamp)
+ , congestion_level = Nothing
+ , occupancy_status = Nothing
+ , occupancy_percentage = Nothing
+ , multi_carriage_details = []
+ , VP.ext'field = defaultValue
+ }
+ }
lang :: Utf8 -> Utf8 -> TranslatedString
@@ -153,3 +194,14 @@ dEntitySelector = EntitySelector
, direction_id = Nothing
, ES.ext'field = defaultValue
}
+
+dTripDescriptor :: TripID -> Maybe Day -> TripDescriptor
+dTripDescriptor tripID day = TripDescriptor
+ { trip_id = Just (toUtf8 tripID)
+ , route_id = Nothing
+ , direction_id = Nothing
+ , start_time = Nothing
+ , start_date = fmap toStupidDate day
+ , schedule_relationship = Nothing
+ , TD.ext'field = defaultValue
+ }