inigorb 0.27.4 → 0.27.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 61225748fde7b180e704d9e45ee2f831eaad0374a2b2164e614f921a48aef653
-  data.tar.gz: 4d077ed1e60e36916bb104d6c7a0b3e7a6a5134ff39ac0ce52209cdefcd694bc
+  metadata.gz: 788d97815fbd001b6eebffed2c9f7c9ef5ea8eddb4228c060ce8dfc155b71aa0
+  data.tar.gz: b21a0572e9fabb0c6c55fe1a112c2e09f0fc5891e9a9353a05599e43a843a1ed
 SHA512:
-  metadata.gz: ee3e3bd4854fb47afb922b985b6ec77638a14dd778618d503d74b846e24f63b6d5c7aeeda123e03ca5ed8148d42ed556d847f28a988da0d80fa80f46665acc38
-  data.tar.gz: e71610eb47bbc0416792de36948cf8a9e27b999f4d65695951a3c5f9ff7717f39c8c5646557f753b1ef375dde48f9f8f2ca42624c6fae381259c4f49b1ea6a9b
+  metadata.gz: 4120ce3939645d632e5f72e17bf5342e2e7c869368e64cb05f6d9fb411f6eaf73afb3088828c081db4a6d3ea15341403c25396d021303b8387a2fc49a1214669
+  data.tar.gz: 9a97a310ced373289fccb74631452212da345acb5e7d2ea1c45ec46b1e7f95ec016fccf7d0d15ec6994758416d1301ddc7eee0cbd111a63989a1e6582c4c7364

data/lib/inigorb.rb

@@ -9,6 +9,7 @@ module Inigo
     @@initialized = false
     @@path = ''
     @@schema = ''
+    @@operation_store = nil
 
     def self.instance
       @@instance
@@ -42,6 +43,14 @@ module Inigo
       @@schema = value
     end
 
+    def self.operation_store
+      @@operation_store
+    end
+
+    def self.operation_store=(value)
+      @@operation_store = value
+    end
+
     def initialize(app)
       @app = app
     end
@@ -54,13 +63,15 @@ module Inigo
 
       request = Rack::Request.new(env)
 
-      # 'path' guard -> /graphql
-      if request.path != self.class.path
+      path = self.class.path
+      path += "/" unless path.end_with?("/")
+      # 'path' guard -> /graphql, /graphql/whatever
+      if request.path != self.class.path && !request.path.start_with?(path)
         return @app.call(env)
       end
 
       # GraphiQL request
-      if request.get? && request.accept.include?('text/html')
+      if request.get? && env['HTTP_ACCEPT'].include?('text/html')
         return @app.call(env)
       end
 
@@ -74,10 +85,71 @@ module Inigo
       if request.post?
         # Read request from body
         request.body.each { |chunk| gReq += chunk }
+
+        if self.class.operation_store.present? && has_operation_id?(gReq)
+          parsed = JSON.parse(gReq)
+          operationId = get_operation_id(parsed)
+
+          if !operationId
+            request.body.rewind
+            return @app.call(env)
+          end
+
+          parts = operationId.split('/')
+          # Query can't be resolved
+          if parts.length != 2
+            request.body.rewind
+            return @app.call(env)
+          end
+          data = self.class.operation_store.get(client_name: parts[0], operation_alias: parts[1])
+          # Query can't be resolved
+          if !data
+            request.body.rewind
+            return @app.call(env)
+          end
+          if data.name
+            parsed.merge!('operationName' => data.name)
+          end
+  
+          if data.body
+            parsed.merge!('query' => data.body)
+          end
+
+          gReq = ''
+          gReq = JSON.dump(parsed)
+          env['rack.input'] = StringIO.new(gReq)
+          env['CONTENT_LENGTH'] = gReq.length.to_s
+          env['CONTENT_TYPE'] = 'application/json'
+        end
         request.body.rewind
       elsif request.get?
+        operation_id = get_operation_id(request.params)
+        if operation_id && self.class.operation_store
+            parts = operation_id.split('/')
+            # Query can't be resolved
+            if parts.length != 2
+              return @app.call(env)
+            end
+            data = self.class.operation_store.get(client_name: parts[0], operation_alias: parts[1])
+            # Query can't be resolved
+            if !data
+              return @app.call(env)
+            end
+
+            if data.name
+              request.params['operationName'] = data.name
+            end
+    
+            if data.body
+              request.params['query'] = data.body
+            end
+
+            # Update the env with the modified query string
+            env['QUERY_STRING'] = Rack::Utils.build_nested_query(request.params)
+        end
+
         # Read request from query param
-        gReq = JSON.dump({ 'query' => request.params['query'] })
+        gReq = JSON.dump(request.params)
       end
 
       q = Query.new(self.class.instance, gReq)
@@ -87,7 +159,7 @@ module Inigo
 
       # Introspection query
       if resp.any?
-        return respond(resp)
+        return respond(200, { 'Content-Type' => 'application/json'}, resp)
       end
 
       # Modify query if required
@@ -108,12 +180,12 @@ module Inigo
       end
 
       # Forward to request handler
-      response = @app.call(env)
+      status, headers, response = @app.call(env)
 
       # Inigo: process response
-      processed_response = q.process_response(response[2].body.to_s)
+      processed_response = q.process_response(response.body.to_s)
       if processed_response
-        return respond(processed_response)
+        return respond(status, headers, processed_response)
       end
 
       response
@@ -121,12 +193,17 @@ module Inigo
 
     private
 
-    def self.initialize_middleware(schema = nil)
+    def self.initialize_middleware(schema = nil, operation_store = nil)
       return if @@initialized
 
       if schema
         @@schema = schema
       end
+
+      if operation_store
+        @@operation_store = operation_store
+      end
+
       # get all the inigo settings from env
       settings = ENV.select { |k, v| k.start_with?('INIGO') }
 
@@ -190,14 +267,32 @@ module Inigo
       JSON.dump(headers)
     end
 
-    def respond(data)
-      response = Rack::Response.new
+    def respond(status, headers, data)      
       response_hash = {}
       response_hash['data'] = data['data'] if data['data']
       response_hash['errors'] = data['errors'] if data['errors']
       response_hash['extensions'] = data['extensions'] if data['extensions']
-      response.write(JSON.dump(response_hash))
-      response.finish
+
+      [status, headers, [JSON.dump(response_hash)]]
+    end
+
+    # operates with string data not to parse the request body unnecessary
+    def has_operation_id?(str_data)
+      # Relay / Apollo 1.x and Apollo Link have the same field just in different places.
+      str_data.include?('operationId')
+    end
+
+    # extracts operation id from parsed body hash
+    def get_operation_id(json_data)
+      # Relay / Apollo 1.x
+      if json_data.include?('operationId')
+        json_data['operationId']
+      # Apollo Link
+      elsif json_data.include?('extensions') && json_data['extensions'].include?('operationId')
+        json_data['extensions']['operationId']
+      else
+        nil
+      end
     end
 
   end  

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: inigorb
 version: !ruby/object:Gem::Version
-  version: 0.27.4
+  version: 0.27.6
 platform: ruby
 authors:
 - Inigo
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-08-02 00:00:00.000000000 Z
+date: 2023-08-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: jwt