af83/chouette-core

# frozen_string_literal: true

module Control
  class PassingTimesInTimeRange < Control::Base
    module Options
      extend ActiveSupport::Concern

      included do
        option :passing_time_scope
        option :before, serialize: TimeOfDay::Type::SecondOffset
        option :after,  serialize: TimeOfDay::Type::SecondOffset

        enumerize :passing_time_scope, in: %w[all first last]
        validates :passing_time_scope, presence: true

        %w[before after].each do |option|
          define_method "#{option}=" do |time_of_day|
            if time_of_day.is_a?(Hash)
              time_of_day = TimeOfDay.from_input_hash(time_of_day).without_utc_offset.second_offset
              time_of_day = nil if time_of_day.zero?
            end

            super time_of_day
          end
        end
      end
    end
    include Options

    validate :before_or_after_present
    validate :time_of_day_for_before_and_after

    private

    def before_or_after_present
      return if before || after

      errors.add(:before, :invalid)
      errors.add(:after, :invalid)
    end

    def time_of_day_for_before_and_after
      errors.add(:after, :invalid) if before && after && before <= after
    end

    class Run < Control::Base::Run
      include Options

      def run
        return if vehicle_journey_at_stops.blank?

        faulty_models.each_instance do |model|
          control_messages.create(
            message_attributes: {
              name: model.try(:published_journey_name) || model.id
            },
            criticity: criticity,
            source: model,
            message_key: :passing_times_in_time_range
          )
        end
      end

      def faulty_models
        context.vehicle_journeys.where(id: vehicle_journey_ids)
      end

      def vehicle_journey_ids
        VehicleJourneys.new(
          context,
          vehicle_journey_at_stops,
          after,
          before
        ).vehicle_journey_ids
      end

      class VehicleJourneys
        def initialize(context, vehicle_journey_at_stops, after, before)
          @context = context
          @vehicle_journey_at_stops = vehicle_journey_at_stops
          @after = after
          @before = before
        end
        attr_reader :context, :vehicle_journey_at_stops, :after, :before

        def vehicle_journey_ids
          context
            .vehicle_journey_at_stops
            .joins(time_zones, stop_point: :stop_area)
            .where.not("#{second_offset_range} @> #{departure_second_offset} AND #{second_offset_range} @> #{arrival_second_offset}")
            .select(:vehicle_journey_id)
            .distinct
            .from(base_query)
        end

        def base_query
          "(#{vehicle_journey_at_stops.to_sql}) AS vehicle_journey_at_stops"
        end

        def second_offset_range
          "'[#{lower},#{upper}]'::int4range"
        end

        def lower
          after ? after.second_offset : ''
        end

        def upper
          before ? before.second_offset : ''
        end

        def departure_second_offset
          second_offset_expression :departure
        end

        def arrival_second_offset
          second_offset_expression :arrival
        end

        def second_offset_expression(state)
          <<~SQL
            (EXTRACT(EPOCH FROM #{state}_time) + time_zones.utc_offset + #{state}_day_offset * 86400)::integer
          SQL
        end

        def time_zones
          <<~SQL
            INNER JOIN public.time_zones ON time_zones.name = COALESCE(stop_areas.time_zone, 'UTC')
          SQL
        end
      end

      def vehicle_journey_at_stops
        VehicleJourneyAtStops.for(context, passing_time_scope)
      end

      class VehicleJourneyAtStops
        def self.for(context, passing_time_scope)
          const_get(passing_time_scope.classify)
            .new(context)
            .vehicle_journey_at_stops
        end

        class Base
          def initialize(context)
            @context = context
          end
          attr_reader :context

          delegate :vehicle_journey_at_stops, to: :context
        end

        class All < Base
        end

        class First < Base
          def vehicle_journey_at_stops
            super.departures
          end
        end

        class Last < Base
          def vehicle_journey_at_stops
            super.arrivals
          end
        end
      end
    end
  end
end