-- SPDX-FileCopyrightText: Transitous Contributors
-- SPDX-License-Identifier: AGPL-3.0-or-later

function process_route(route)
  if string.find(route:get_short_name(), "ZKA") then
    route:set_route_type(714) --RAIL_REPLACEMENT_BUS_SERVICE in future motis
  elseif route:get_short_name() == "EIP" then
    route:set_route_type(101) --HIGH_SPEED_RAIL_SERVICE in future motis
  elseif route:get_short_name() == "EN" then
    route:set_route_type(105) --SLEEPER_RAIL_SERVICE in future motis
  else
    route:set_route_type(102) --LONG_DISTANCE_TRAINS_SERVICE in future motis
  end
end

function process_trip(trip)
  trip:set_display_name(trip:get_route():get_short_name() .. " " .. trip:get_short_name())
end