katalyst-google-apis 1.1.1 → 1.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5794e6da69ca70b6edb9a8e7bc30e75b8536d9d436422f3db7e311997aa44fee
-  data.tar.gz: 47e4870067a6c655821415606a2446f80535b76623900290c16fe35441865966
+  metadata.gz: 134e4c5aacc319e53670df643c6252452ab1e865c77132a3de28427b6f762a70
+  data.tar.gz: f6e3756bbc7a329c4920340d4a9e446a81fe47f999afbf8437e6cbced7ad824c
 SHA512:
-  metadata.gz: 14b904f1003fe047a112aea03619c29a9bdca4ec9c486e4143b1c2f02d4995837d0033a1319335d4036beb53e12c6e49483d6a23c9cfdc0303711fc8bbab3d3d
-  data.tar.gz: 5d7693c3922f2573b711a033a5313b35dc0fdfb0156ee4b44f00d1c5df1316df611a9062ad1a7b1750e319b7242d34f9500d4ec4c0cac226ad841ba9de2dc521
+  metadata.gz: 6e5ea2e4950182c34aeacce541c724e33781ff220f71140260aec7fabf103188a667099d644cc49c87df53d76752c9c25423a3114ea22595c94ee5a00c10dfac
+  data.tar.gz: 57940e6e4e33caccb19179975047d1380d6d7a76162e373df7dd15403aa787835f1d6b23d2ad3d2f0b38321bebf1b8d6307c972e7dc0c93606add6b4d907200a

data/app/assets/javascripts/controllers/recaptcha_controller.js

@@ -111,20 +111,21 @@ function getMetaElement(name) {
 function getCspNonce() {
   const element = getMetaElement("csp-nonce");
   if (element) {
-    const {nonce: nonce, content: content} = element;
+    const { nonce: nonce, content: content } = element;
     return nonce === "" ? content : nonce;
   }
 }
 
 function createRecaptchaScriptElement() {
   const element = document.createElement("script");
-  element.src = "https://www.google.com/recaptcha/enterprise.js?render=explicit";
+  element.src =
+    "https://www.google.com/recaptcha/enterprise.js?render=explicit";
   const cspNonce = getCspNonce();
   if (cspNonce) {
     element.setAttribute("nonce", cspNonce);
   }
-  element.async = true
-  element.defer = true
+  element.async = true;
+  element.defer = true;
 
   return element;
 }
@@ -133,7 +134,9 @@ function createRecaptchaScriptElement() {
  * Inject recaptcha enterprise script into the head, unless it is already there.
  */
 function injectRecaptchaScripts() {
-  let element = document.head.querySelector("script[src*='recaptcha/enterprise']");
+  let element = document.head.querySelector(
+    "script[src*='recaptcha/enterprise']",
+  );
   if (!element) {
     element = createRecaptchaScriptElement();
     document.head.appendChild(element);

data/app/services/katalyst/google_apis/error.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module Katalyst
+  module GoogleApis
+    class Error < StandardError
+      attr_reader :code, :status
+
+      def initialize(code:, status:, message:)
+        super(message)
+
+        @code = code
+        @status = status
+      end
+
+      def inspect
+        "#<#{self.class.name} code=#{code.inspect} status=#{status.inspect} message=#{message.inspect}>"
+      end
+    end
+  end
+end

data/app/services/katalyst/google_apis/gemini/generate_content_service.rb

@@ -0,0 +1,95 @@
+# frozen_string_literal: true
+
+module Katalyst
+  module GoogleApis
+    module Gemini
+      # Use Gemini GPT to generate a response to a prompt.
+      class GenerateContentService
+        attr_reader :response, :result, :error, :content_text
+
+        def self.call(parent:, model:, payload:, credentials: Katalyst::GoogleApis.credentials)
+          new(parent:, model:, credentials:).call(payload:)
+        end
+
+        def initialize(credentials:, model:, parent:)
+          @credentials = credentials
+          @model = model
+          @parent = parent
+        end
+
+        def call(payload:)
+          @response = Curl.post(url, payload.to_json) do |http|
+            http.headers["Content-Type"] = "application/json; UTF-8"
+            @credentials.apply!(http.headers)
+          end
+
+          if %r{^application\/json}.match?(@response.content_type)
+            @result = JSON.parse(response.body, symbolize_names: true)
+          else
+            raise GoogleApis::Error.new(
+              code:    response.response_code,
+              status:  Rack::Utils::HTTP_STATUS_CODES[response.response_code],
+              message: "Unexpected HTTP response received (#{response.response_code}, #{@response.content_type})",
+            )
+          end
+
+          if result[:error].present?
+            @error = GoogleApis::Error.new(**result[:error])
+          else
+            @content_text = result.dig(:candidates, 0, :content, :parts, 0, :text)
+          end
+
+          self
+        rescue StandardError => e
+          @error = e
+          raise e
+        ensure
+          report_error
+        end
+
+        def success?
+          result.present? && content_text.present?
+        end
+
+        def content_json
+          return @content_json if instance_variable_defined?(:@content_json)
+
+          @content_json = JSON.parse(content_text, symbolize_names: true)
+        end
+
+        def inspect
+          "#<#{self.class.name} result: #{@result.inspect} error: #{@error.inspect}>"
+        end
+
+        private
+
+        def url
+          "https://aiplatform.googleapis.com/v1#{@parent}#{@model}:generateContent"
+        end
+
+        def report_error
+          return if error.nil?
+
+          if defined?(Sentry)
+            Sentry.add_breadcrumb(sentry_breadcrumb(error))
+          else
+            Rails.logger.error(error)
+          end
+        end
+
+        def sentry_breadcrumb(error)
+          Sentry::Breadcrumb.new(
+            type:     "http",
+            category: "gemini",
+            data:     {
+              url:,
+              method:      "POST",
+              status_code: error.code,
+              reason:      error.message,
+            },
+          )
+        end
+      end
+    end
+  end
+end

data/app/services/katalyst/google_apis/recaptcha/assessment_service.rb

@@ -8,8 +8,8 @@ module Katalyst
       class AssessmentService
         attr_accessor :response, :result, :error
 
-        def self.call(parent:, credentials: GoogleApis.credentials, **)
-          new(credentials:, parent:).call(**)
+        def self.call(assessment:, parent:, credentials: GoogleApis.credentials)
+          new(credentials:, parent:).call(assessment:)
         end
 
         def initialize(credentials:, parent:)
@@ -23,18 +23,22 @@ module Katalyst
             @credentials.apply!(http.headers)
           end
 
-          @result = JSON.parse(@response.body, symbolize_names: true)
-
-          self
-        rescue Curl::Easy::Error => e
-          if defined?(Sentry)
-            Sentry.add_breadcrumb(sentry_breadcrumb(e))
+          if %r{^application\/json}.match?(@response.content_type)
+            @result = JSON.parse(response.body, symbolize_names: true)
           else
-            Rails.logger.error(e)
+            raise GoogleApis::Error.new(
+              code:    response.response_code,
+              status:  Rack::Utils::HTTP_STATUS_CODES[response.response_code],
+              message: "Unexpected HTTP response received (#{response.response_code}, #{@response.content_type})",
+            )
           end
 
-          @error = e
           self
+        rescue StandardError => e
+          @error = e
+          raise e
+        ensure
+          report_error
         end
 
         def valid?
@@ -63,14 +67,26 @@ module Katalyst
           "https://recaptchaenterprise.googleapis.com/v1/#{@parent}/assessments"
         end
 
+        def report_error
+          return if error.nil?
+
+          if defined?(Sentry)
+            Sentry.add_breadcrumb(sentry_breadcrumb(error))
+          else
+            Rails.logger.error(error)
+          end
+        end
+
         def sentry_breadcrumb(error)
           Sentry::Breadcrumb.new(
-            type:        "http",
-            category:    "recaptcha",
-            url:,
-            method:      "POST",
-            status_code: error.code,
-            reason:      error.message,
+            type:     "http",
+            category: "recaptcha",
+            data:     {
+              url:,
+              method:      "POST",
+              status_code: error.code,
+              reason:      error.message,
+            },
           )
         end
       end

data/config/ci.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+# Run using bin/ci
+
+CI.run do
+  step "Setup", "bin/setup"
+
+  step "Style: Ruby", "bundle exec rubocop --no-server"
+
+  step "Style: JS/CSS", "bundle exec rake prettier:lint"
+
+  step "Security: Brakeman vulnerability audit", "bundle exec brakeman -q -w2"
+
+  step "Tests: rspec", "bundle exec rspec"
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: katalyst-google-apis
 version: !ruby/object:Gem::Version
-  version: 1.1.1
+  version: 1.2.1
 platform: ruby
 authors:
 - Katalyst Interactive
@@ -77,8 +77,11 @@ files:
 - app/helpers/katalyst/google_apis/form_builder.rb
 - app/helpers/katalyst/google_apis/govuk_form_builder.rb
 - app/services/katalyst/google_apis/credentials.rb
+- app/services/katalyst/google_apis/error.rb
+- app/services/katalyst/google_apis/gemini/generate_content_service.rb
 - app/services/katalyst/google_apis/recaptcha/assessment_service.rb
 - app/validators/recaptcha_validator.rb
+- config/ci.rb
 - config/importmap.rb
 - config/locales/en.yml
 - lib/katalyst-google-apis.rb