package com.eleith.calchoochoo.utils;

import android.database.Cursor;
import android.database.sqlite.SQLiteDatabase;
import android.support.annotation.Nullable;

import com.eleith.calchoochoo.data.PossibleTrip;

import org.joda.time.LocalDateTime;
import org.joda.time.LocalTime;

import java.util.ArrayList;

public class PossibleTripUtils {
  @Nullable
  public static PossibleTrip getPossibleTripFromCursor(Cursor cursor) {

    if (cursor.moveToNext()) {
      PossibleTrip possibleTrip = new PossibleTrip();
      Float price = cursor.getFloat(cursor.getColumnIndex("price"));
      Integer direction = cursor.getInt(cursor.getColumnIndex("direction"));
      String tripShortName = cursor.getString(cursor.getColumnIndex("trip_short_name"));
      String routeLongName = cursor.getString(cursor.getColumnIndex("route_long_name"));

      String routeId = cursor.getString(cursor.getColumnIndex("route_id"));
      String stop1Id = cursor.getString(cursor.getColumnIndex("st1__stop_id"));
      String stop2Id = cursor.getString(cursor.getColumnIndex("st2__stop_id"));
      String stop1ParentId = cursor.getString(cursor.getColumnIndex("st1__parent_station"));
      String stop2ParentId = cursor.getString(cursor.getColumnIndex("st2__parent_station"));
      String stop1Name = cursor.getString(cursor.getColumnIndex("st1__stop_name"));
      String stop2Name = cursor.getString(cursor.getColumnIndex("st2__stop_name"));
      String tripId = cursor.getString(cursor.getColumnIndex("st1__trip_id"));
      Integer stopOneSequence = cursor.getInt(cursor.getColumnIndex("st1__stop_sequence"));
      Integer stopTwoSequence = cursor.getInt(cursor.getColumnIndex("st2__stop_sequence"));

      LocalTime stopOneDepartureTime = new LocalTime(DataStringUtils.adjustLateTimes(cursor.getString(cursor.getColumnIndex("st1__departure_time"))));
      LocalTime stopOneArrivalTime = new LocalTime(DataStringUtils.adjustLateTimes(cursor.getString(cursor.getColumnIndex("st1__arrival_time"))));
      LocalTime stopTwoDepartureTime = new LocalTime(DataStringUtils.adjustLateTimes(cursor.getString(cursor.getColumnIndex("st2__departure_time"))));
      LocalTime stopTwoArrivalTime = new LocalTime(DataStringUtils.adjustLateTimes(cursor.getString(cursor.getColumnIndex("st2__arrival_time"))));

      if (stopOneSequence < stopTwoSequence) {
        possibleTrip.setArrivalTime(stopTwoArrivalTime);
        possibleTrip.setDepartureTime(stopOneDepartureTime);
        possibleTrip.setFirstStopSequence(stopOneSequence);
        possibleTrip.setLastStopSequence(stopTwoSequence);
        possibleTrip.setFirstStopId(stop1Id);
        possibleTrip.setFirstParentStopId(stop1ParentId);
        possibleTrip.setLastStopId(stop2Id);
        possibleTrip.setLastParentStopId(stop2ParentId);
        possibleTrip.setFirstStopName(stop1Name);
        possibleTrip.setLastStopName(stop2Name);
      } else {
        possibleTrip.setArrivalTime(stopOneArrivalTime);
        possibleTrip.setDepartureTime(stopTwoDepartureTime);
        possibleTrip.setFirstStopSequence(stopTwoSequence);
        possibleTrip.setLastStopSequence(stopOneSequence);
        possibleTrip.setFirstStopId(stop2Id);
        possibleTrip.setFirstParentStopId(stop2ParentId);
        possibleTrip.setLastStopId(stop1Id);
        possibleTrip.setLastParentStopId(stop1ParentId);
        possibleTrip.setFirstStopName(stop2Name);
        possibleTrip.setLastStopName(stop1Name);
      }

      possibleTrip.setRouteLongName(routeLongName);
      possibleTrip.setPrice(price);
      possibleTrip.setTripId(tripId);
      possibleTrip.setTripShortName(tripShortName);
      possibleTrip.setTripDirection(direction);
      possibleTrip.setRouteId(routeId);

      return possibleTrip;
    }

    return null;
  }

  public static ArrayList<PossibleTrip> getPossibleTripsFromCursor(Cursor cursor) {
    ArrayList<PossibleTrip> possibleTrips = new ArrayList<>();

    while (cursor.getPosition() < cursor.getCount()) {
      PossibleTrip possibleTrip = getPossibleTripFromCursor(cursor);
      if (possibleTrip != null) {
        possibleTrips.add(possibleTrip);
      }
    }

    return possibleTrips;
  }

  public static Cursor getPossibleTripQuery(SQLiteDatabase db, String trip_id, String stop1_id, String stop2_id) {
    String query = "SELECT " +
        "routes.route_id as route_id, " +
        "routes.route_long_name as route_long_name, " +
        "fare_attributes.price as price, " +
        "trips.direction_id as direction, " +
        "trips.trip_short_name as trip_short_name, " +
        "st1.trip_id as st1__trip_id, st1.arrival_time as st1__arrival_time, st1.departure_time as st1__departure_time, " +
        "st1.stop_name as st1__stop_name, st1.stop_id as st1__stop_id, st1.stop_sequence as st1__stop_sequence, st1.parent_station as st1__parent_station, " +
        "st2.trip_id as st2__trip_id, st2.arrival_time as st2__arrival_time, st2.departure_time as st2__departure_time, " +
        "st2.stop_name as st2__stop_name, st2.stop_id as st2__stop_id, st2.stop_sequence as st2__stop_sequence, st2.parent_station as st2__parent_station " +
        "FROM " +
        "    (SELECT * " +
        "    FROM stops, stop_times " +
        "    WHERE " +
        "     stops.stop_id = stop_times.stop_id " +
        "     AND stop_times.trip_id = ? " +
        "     AND stops.stop_id = ?) AS st1, " +
        "    (Select * " +
        "    FROM stops, stop_times " +
        "    WHERE " +
        "     stops.stop_id = stop_times.stop_id " +
        "     AND stop_times.trip_id = ? " +
        "     AND stops.stop_id = ?) AS st2, " +
        "  trips, " +
        "  routes, " +
        "  calendar, " +
        "  fare_rules, " +
        "  fare_attributes " +
        "WHERE st1.trip_id = st2.trip_id " +
        "  AND trips.route_id = routes.route_id " +
        "  AND trips.trip_id = st1.trip_id " +
        "  AND calendar.service_id = trips.service_id " +
        "  AND fare_rules.origin_id = st1.zone_id " +
        "  AND fare_rules.destination_id = st2.zone_id " +
        "  AND fare_rules.route_id = routes.route_id " +
        "  AND fare_rules.fare_id = fare_attributes.fare_id " +
        "ORDER BY st1.departure_time";
    String[] args = {trip_id, stop1_id, trip_id, stop2_id};
    return db.rawQuery(query, args);
  }

  public static Cursor getPossibleTripsByParentStopQuery(SQLiteDatabase db, Long dateTime, String stop1_id, String stop2_id) {
    String calendarFilter = CalendarDateUtils.getFilterForDate(db, dateTime);
    String query = "SELECT " +
        "routes.route_id as route_id, " +
        "routes.route_long_name as route_long_name, " +
        "fare_attributes.price as price, " +
        "trips.direction_id as direction, " +
        "trips.trip_short_name as trip_short_name, " +
        "st1.stop_name as st1__stop_name, st1.platform_code as st1__platform_code, st1.trip_id as st1__trip_id, st1.arrival_time as st1__arrival_time, st1.departure_time as st1__departure_time, " +
        "st1.stop_id as st1__stop_id, st1.stop_sequence as st1__stop_sequence, st1.drop_off_type as st1__drop_off_type, st1.parent_station as st1__parent_station, " +
        "st2.stop_name as st2__stop_name, st2.platform_code as st2__platform_code, st2.trip_id as st2__trip_id, st2.arrival_time as st2__arrival_time, st2.departure_time as st2__departure_time, " +
        "st2.stop_id as st2__stop_id, st2.stop_sequence as st2__stop_sequence, st2.drop_off_type as st2__drop_off_type, st2.parent_station as st2__parent_station " +
        "FROM " +
        "    (SELECT * " +
        "    FROM stops, stop_times " +
        "    WHERE " +
        "     stop_times.stop_id = stops.stop_id " +
        "     AND stops.parent_station = ?) AS st1, " +
        "    (Select * " +
        "    FROM stops, stop_times " +
        "    WHERE " +
        "     stop_times.stop_id = stops.stop_id " +
        "     AND stops.parent_station = ?) AS st2, " +
        "  trips, " +
        "  routes, " +
        "  calendar, " +
        "  fare_rules, " +
        "  fare_attributes " +
        "WHERE st1.trip_id = st2.trip_id " +
        "  AND st1.platform_code = st2.platform_code " +
        "  AND cast(st1.stop_sequence as integer) < cast(st2.stop_sequence as integer) " +
        "  AND trips.trip_id = st1.trip_id " +
        "  AND trips.route_id = routes.route_id " +
        "  AND calendar.service_id = trips.service_id " +
        "  AND fare_rules.origin_id = st1.zone_id " +
        "  AND fare_rules.destination_id = st2.zone_id " +
        "  AND fare_rules.route_id = routes.route_id " +
        "  AND fare_rules.fare_id = fare_attributes.fare_id " +
        calendarFilter +
        "ORDER BY st1.departure_time ";
    String[] args = {stop1_id, stop2_id};
    return db.rawQuery(query, args);
  }

  public static ArrayList<PossibleTrip> filterByDateTimeAndDirection(ArrayList<PossibleTrip> possibleTrips, LocalDateTime dateTime, Boolean arriving) {
    ArrayList<PossibleTrip> possibleTripsFiltered = new ArrayList<>();

    for (PossibleTrip possibleTrip : possibleTrips) {
      LocalTime departureTime = possibleTrip.getDepartureTime();
      LocalTime arrivalTime = possibleTrip.getArrivalTime();

      if (!arriving) {
        if (departureTime.isAfter(dateTime.toLocalTime())) {
          possibleTripsFiltered.add(possibleTrip);
        }
      } else {
        if (arrivalTime.isBefore(dateTime.toLocalTime()) && arrivalTime.isAfter(dateTime.minusHours(3).toLocalTime())) {
          possibleTripsFiltered.add(possibleTrip);
        }
      }
    }

    return possibleTripsFiltered;
  }
}