DBRIS: Add load/occupancy data

2 files changed, 20 insertions, 5 deletions

diff --git a/lib/Travelynx.pm b/lib/Travelynx.pm
index 98ba4aa..cdfc84c 100755
--- a/lib/Travelynx.pm
+++ b/lib/Travelynx.pm
@@ -419,6 +419,14 @@ sub startup {
 			my $first  = $load->{FIRST}  // 0;
 			my $second = $load->{SECOND} // 0;
 
+			# DBRIS
+			if ( $first == 99 ) {
+				$first = 4;
+			}
+			if ( $second == 99 ) {
+				$second = 4;
+			}
+
 			my @symbols
 			  = (
 				qw(help_outline person_outline people priority_high not_interested)
diff --git a/lib/Travelynx/Model/InTransit.pm b/lib/Travelynx/Model/InTransit.pm
index 2b9832c..9f02aba 100644
--- a/lib/Travelynx/Model/InTransit.pm
+++ b/lib/Travelynx/Model/InTransit.pm
@@ -209,9 +209,12 @@ sub add {
 						rt_dep    => _epoch( $j_stop->rt_dep ),
 						arr_delay => $j_stop->arr_delay,
 						dep_delay => $j_stop->dep_delay,
-						load      => undef,
-						lat       => $j_stop->lat,
-						lon       => $j_stop->lon,
+						load      => {
+							FIRST  => $j_stop->occupancy_first,
+							SECOND => $j_stop->occupancy_second
+						},
+						lat => $j_stop->lat,
+						lon => $j_stop->lon,
 					}
 				]
 			);
@@ -905,8 +908,12 @@ sub update_arrival_dbris {
 					rt_dep    => _epoch( $j_stop->rt_dep ),
 					arr_delay => $j_stop->arr_delay,
 					dep_delay => $j_stop->dep_delay,
-					lat       => $j_stop->lat,
-					lon       => $j_stop->lon,
+					load      => {
+						FIRST  => $j_stop->occupancy_first,
+						SECOND => $j_stop->occupancy_second
+					},
+					lat => $j_stop->lat,
+					lon => $j_stop->lon,
 				}
 			]
 		);