feat: Add direct connections endpoint to airports controller

Author: teetangh

app/controllers/api/v1/airports_controller.rb

@@ -108,6 +108,25 @@ def index
         end
       end
 
+      # GET /api/v1/airports/direct-connections
+      def direct_connections
+        raise ArgumentError, 'Source airport is missing' unless params[:sourceAirportCode].present?
+
+        source_airport = params[:sourceAirportCode]
+        limit = params[:limit] || 10
+        offset = params[:offset] || 0
+
+        begin
+          destination_airports = Route.direct_connections(key: [source_airport, limit, offset])
+                                      .pluck(:destinationairport)
+          render json: destination_airports, status: :ok
+        rescue ArgumentError => e
+          render json: { error: 'Invalid request', message: e.message }, status: :bad_request
+        rescue StandardError => e
+          render json: { error: 'Internal server error', message: e.message }, status: :internal_server_error
+        end
+      end
+
       private
 
       def set_airport

app/models/airport.rb

@@ -1,3 +1,8 @@
+# frozen_string_literal: true
+
+require 'couchbase-orm'
+
+# GeoCoordinates nested document
 class GeoCoordinates < CouchbaseOrm::NestedDocument
   attribute :lat, :float
   attribute :lon, :float
@@ -8,6 +13,7 @@ class GeoCoordinates < CouchbaseOrm::NestedDocument
   validates :alt, presence: true, numericality: { greater_than_or_equal_to: -1000, less_than_or_equal_to: 10_000 }
 end
 
+# Airport model
 class Airport < CouchbaseOrm::Base
   attribute :airportname, :string
   attribute :city, :string

app/models/route.rb

@@ -26,4 +26,8 @@ class Route < CouchbaseOrm::Base
   validates :equipment, presence: true
   validates :schedule, presence: true
   validates :distance, presence: true, numericality: { greater_than_or_equal_to: 0 }
+
+  n1ql :direct_connections, query_fn: proc { |bucket, values, options|
+    cluster.query("SELECT distinct raw meta(route).id FROM `#{bucket.name}` AS airport JOIN `#{bucket.name}` AS route ON route.sourceairport = airport.faa WHERE airport.faa = #{quote(values[0])} AND route.stops = 0 LIMIT #{values[1]} OFFSET #{values[2]}", options)
+  }
 end

test/integration/airports_spec.rb

@@ -191,30 +191,30 @@
     end
   end
 
-  # describe 'GET /api/v1/airports/direct-connections' do
-  #   let(:destination_airport_code) { 'JFK' }
-  #   let(:limit) { 10 }
-  #   let(:offset) { 0 }
-  #   let(:expected_connections) { %w[DEL LHR EZE ATL CUN MEX LAX SAN SEA SFO] }
-
-  #   context 'when the destination airport code is provided' do
-  #     it 'returns the direct connections' do
-  #       get '/api/v1/airports/direct-connections',
-  #           params: { destinationAirportCode: destination_airport_code, limit: limit, offset: offset }
-
-  #       expect(response).to have_http_status(:ok)
-  #       expect(response.content_type).to eq('application/json; charset=utf-8')
-  #       expect(JSON.parse(response.body)).to eq(expected_connections)
-  #     end
-  #   end
-
-  #   context 'when the destination airport code is not provided' do
-  #     it 'returns a bad request error' do
-  #       get '/api/v1/airports/direct-connections'
-
-  #       expect(response).to have_http_status(:bad_request)
-  #       expect(JSON.parse(response.body)).to eq({ 'message' => 'Destination airport code is required' })
-  #     end
-  #   end
-  # end
+  describe 'GET /api/v1/airports/direct-connections' do
+    let(:destination_airport_code) { 'JFK' }
+    let(:limit) { 10 }
+    let(:offset) { 0 }
+    let(:expected_connections) { %w[DEL LHR EZE ATL CUN MEX LAX SAN SEA SFO] }
+
+    context 'when the destination airport code is provided' do
+      it 'returns the direct connections' do
+        get '/api/v1/airports/direct-connections',
+            params: { destinationAirportCode: destination_airport_code, limit:, offset: }
+
+        expect(response).to have_http_status(:ok)
+        expect(response.content_type).to eq('application/json; charset=utf-8')
+        expect(JSON.parse(response.body)).to eq(expected_connections)
+      end
+    end
+
+    context 'when the destination airport code is not provided' do
+      it 'returns a bad request error' do
+        get '/api/v1/airports/direct-connections'
+
+        expect(response).to have_http_status(:bad_request)
+        expect(JSON.parse(response.body)).to eq({ 'message' => 'Destination airport code is required' })
+      end
+    end
+  end
 end