digitransit-ui/app/component/routepage/TripStopListContainer.js

import PropTypes from 'prop-types';
import React from 'react';
import { createFragmentContainer, graphql } from 'react-relay';
import cx from 'classnames';
import isEmpty from 'lodash/isEmpty';
import connectToStores from 'fluxible-addons-react/connectToStores';
import groupBy from 'lodash/groupBy';
import values from 'lodash/values';
import { configShape, tripShape, vehicleShape } from '../../util/shapes';
import { getTripOrRouteMode } from '../../util/modeUtils';
import TripRouteStop from './TripRouteStop';
import withBreakpoint from '../../util/withBreakpoint';

class TripStopListContainer extends React.PureComponent {
  static propTypes = {
    trip: tripShape.isRequired,
    className: PropTypes.string,
    vehicles: PropTypes.objectOf(vehicleShape),
    currentTime: PropTypes.number.isRequired,
    tripStart: PropTypes.string.isRequired,
    breakpoint: PropTypes.string.isRequired,
    keepTracking: PropTypes.bool,
    setHumanScrolling: PropTypes.func,
  };

  static defaultProps = {
    vehicles: {},
    className: undefined,
    keepTracking: false,
    setHumanScrolling: () => {},
  };

  static contextTypes = {
    config: configShape.isRequired,
  };

  getStops() {
    const {
      breakpoint,
      currentTime,
      trip,
      tripStart,
      vehicles: propVehicles,
    } = this.props;

    const mode = getTripOrRouteMode(trip, trip.route, this.context.config);

    const vehicles = groupBy(
      values(propVehicles).filter(
        vehicle => currentTime - vehicle.timestamp < 5 * 60,
      ),
      vehicle => vehicle.next_stop,
    );

    const matchingVehicles = Object.keys(propVehicles)
      .map(key => propVehicles[key])
      .filter(
        vehicle =>
          vehicle.direction === undefined ||
          trip.pattern.directionId === undefined ||
          trip.pattern.directionId === -1 ||
          vehicle.direction === trip.pattern.directionId,
      )
      .filter(
        vehicle =>
          vehicle.tripStartTime === undefined ||
          vehicle.tripStartTime === tripStart,
      )
      .filter(
        vehicle =>
          vehicle.tripId === undefined || vehicle.tripId === trip.gtfsId,
      );

    // selected vehicle
    const vehicle =
      matchingVehicles.length > 0 ? matchingVehicles[0] : undefined;
    const nextStop = vehicle && vehicle.next_stop;
    let stopPassed = true;

    return trip.stoptimesForDate.map((stoptime, index) => {
      if (nextStop === stoptime.stop.gtfsId) {
        stopPassed = false;
      } else if (
        stoptime.realtimeDeparture + stoptime.serviceDay > currentTime &&
        (isEmpty(vehicle) || (vehicle && vehicle.next_stop === undefined))
      ) {
        stopPassed = false;
      }
      const nextStoptimeForDate = trip.stoptimesForDate[index + 1];
      const prevStop = trip.stoptimesForDate[index - 1]?.stop;

      return (
        <TripRouteStop
          key={stoptime.stop.gtfsId}
          stoptime={stoptime}
          stop={stoptime.stop}
          nextStop={nextStoptimeForDate ? nextStoptimeForDate.stop : null}
          prevStop={prevStop || null}
          mode={mode}
          color={trip.route && trip.route.color ? `#${trip.route.color}` : null}
          vehicles={vehicles[stoptime.stop.gtfsId]}
          selectedVehicle={vehicle}
          stopPassed={stopPassed}
          realtime={stoptime.realtime}
          currentTime={currentTime}
          realtimeDeparture={stoptime.realtimeDeparture}
          pattern={trip.pattern.code}
          route={trip.route.gtfsId}
          last={index === trip.stoptimesForDate.length - 1}
          first={index === 0}
          className={`bp-${breakpoint}`}
          shortName={trip.route && trip.route.shortName}
          keepTracking={this.props.keepTracking}
          setHumanScrolling={this.props.setHumanScrolling}
        />
      );
    });
  }

  render() {
    return (
      <>
        <div
          className={cx('route-stop-list', this.props.className)}
          role="tabpanel"
          aria-labelledby="route-tab"
        >
          {this.getStops()}
        </div>
        <div className="bottom-whitespace" />
      </>
    );
  }
}

const connectedComponent = createFragmentContainer(
  connectToStores(
    withBreakpoint(TripStopListContainer),
    ['RealTimeInformationStore', 'PositionStore', 'TimeStore'],
    ({ getStore }) => ({
      vehicles: getStore('RealTimeInformationStore').vehicles,
      currentTime: getStore('TimeStore').getCurrentTime(),
    }),
  ),
  {
    trip: graphql`
      fragment TripStopListContainer_trip on Trip {
        route {
          mode
          type
          gtfsId
          color
          shortName
        }
        pattern {
          code
          directionId
        }
        stoptimesForDate {
          stop {
            gtfsId
            name
            desc
            code
            lat
            lon
            zoneId
            alerts {
              alertSeverityLevel
              effectiveEndDate
              effectiveStartDate
            }
          }
          realtimeArrival
          realtimeDeparture
          realtime
          scheduledDeparture
          serviceDay
          realtimeState
        }
        replacementMode
        gtfsId
      }
    `,
  },
);

export { connectedComponent as default, TripStopListContainer as Component };