wovnrb 3.11.0 → 3.13.0

data/lib/wovnrb/api_translator.rb

@@ -1,183 +1,186 @@
-require 'addressable'
-require 'digest'
-require 'json'
-require 'zlib'
-
-module Wovnrb
-  class ApiTranslator
-    def initialize(store, headers, uuid)
-      @store = store
-      @headers = headers
-      @uuid = uuid
-    end
-
-    def translate(body)
-      connection = prepare_connection
-      request = prepare_request(body)
-
-      begin
-        response = connection.request(request)
-      rescue => e
-        WovnLogger.error("\"#{e.message}\" error occurred when contacting WOVNio translation API")
-        return body
-      end
-
-      case response
-      when Net::HTTPSuccess
-        begin
-          raw_response_body = @store.dev_mode? ? response.body : Zlib::GzipReader.new(StringIO.new(response.body)).read
-        rescue Zlib::GzipFile::Error
-          raw_response_body = response.body
-        end
-
-        begin
-          JSON.parse(raw_response_body)['body'] || body
-        rescue JSON::JSONError
-          body
-        end
-      else
-        WovnLogger.error("HTML-swapper call failed. Received \"#{response.message}\" from WOVNio translation API.")
-        body
-      end
-    end
-
-    private
-
-    def prepare_connection
-      connection = Net::HTTP.new(api_uri.host, api_uri.port)
-
-      connection.open_timeout = api_timeout
-      connection.read_timeout = api_timeout
-
-      connection
-    end
-
-    def prepare_request(body)
-      if @store.compress_api_requests?
-        gzip_request(body)
-      else
-        json_request(body)
-      end
-    end
-
-    def gzip_request(html_body)
-      api_params = build_api_params(html_body)
-      compressed_body = compress_request_data(api_params.to_json)
-      request = Net::HTTP::Post.new(request_path(html_body), {
-                                      'Accept-Encoding' => 'gzip',
-                                      'Content-Type' => 'application/json',
-                                      'Content-Encoding' => 'gzip',
-                                      'Content-Length' => compressed_body.bytesize.to_s,
-                                      'X-Request-Id' => @uuid
-                                    })
-      request.body = compressed_body
-
-      request
-    end
-
-    def json_request(html_body)
-      api_params = build_api_params(html_body)
-      request = Net::HTTP::Post.new(request_path(html_body), {
-                                      'Accept-Encoding' => 'gzip',
-                                      'Content-Type' => 'application/json',
-                                      'X-Request-Id' => @uuid
-                                    })
-      request.body = api_params.to_json
-
-      request
-    end
-
-    def request_path(body)
-      "#{api_uri.path}/translation?cache_key=#{cache_key(body)}"
-    end
-
-    def cache_key(body)
-      cache_key_components = {
-        'token' => token,
-        'settings_hash' => settings_hash,
-        'body_hash' => Digest::MD5.hexdigest(body),
-        'path' => page_pathname,
-        'lang' => lang_code,
-        'version' => "wovnrb_#{VERSION}"
-      }.map { |k, v| "#{k}=#{v}" }.join('&')
-
-      CGI.escape("(#{cache_key_components})")
-    end
-
-    def build_api_params(body)
-      result = {
-        'url' => page_url,
-        'token' => token,
-        'lang_code' => lang_code,
-        'url_pattern' => url_pattern,
-        'lang_param_name' => lang_param_name,
-        'translate_canonical_tag' => translate_canonical_tag,
-        'insert_hreflangs' => insert_hreflangs,
-        'product' => 'WOVN.rb',
-        'version' => VERSION,
-        'body' => body
-      }
-
-      result['custom_lang_aliases'] = JSON.dump(custom_lang_aliases) unless custom_lang_aliases.empty?
-      result['custom_domain_langs'] = JSON.dump(custom_domain_langs) unless custom_domain_langs.empty?
-
-      result
-    end
-
-    def compress_request_data(data_hash)
-      ActiveSupport::Gzip.compress(data_hash)
-    end
-
-    def api_uri
-      Addressable::URI.parse("#{@store.settings['api_url']}/v0")
-    end
-
-    def api_timeout
-      @headers.search_engine_bot? ? @store.settings['api_timeout_search_engine_bots'] : @store.settings['api_timeout_seconds']
-    end
-
-    def settings_hash
-      Digest::MD5.hexdigest(JSON.dump(@store.settings))
-    end
-
-    def token
-      @store.settings['project_token']
-    end
-
-    def lang_code
-      @headers.lang_code
-    end
-
-    def url_pattern
-      @store.settings['url_pattern']
-    end
-
-    def lang_param_name
-      @store.settings['lang_param_name']
-    end
-
-    def custom_lang_aliases
-      @store.settings['custom_lang_aliases']
-    end
-
-    def translate_canonical_tag
-      @store.settings['translate_canonical_tag']
-    end
-
-    def insert_hreflangs
-      @store.settings['insert_hreflangs']
-    end
-
-    def custom_domain_langs
-      @store.custom_domain_langs.to_html_swapper_hash
-    end
-
-    def page_url
-      "#{@headers.protocol}://#{@headers.url}"
-    end
-
-    def page_pathname
-      @headers.pathname_with_trailing_slash_if_present
-    end
-  end
-end
+require 'addressable'
+require 'digest'
+require 'json'
+require 'zlib'
+
+module Wovnrb
+  class ApiTranslator
+    def initialize(store, headers, uuid, response_status_code)
+      @store = store
+      @headers = headers
+      @uuid = uuid
+      @response_status_code = response_status_code
+    end
+
+    def translate(body)
+      connection = prepare_connection
+      request = prepare_request(body)
+
+      begin
+        response = connection.request(request)
+      rescue => e
+        WovnLogger.error("\"#{e.message}\" error occurred when contacting WOVNio translation API")
+        return body
+      end
+
+      case response
+      when Net::HTTPSuccess
+        begin
+          raw_response_body = @store.dev_mode? ? response.body : Zlib::GzipReader.new(StringIO.new(response.body)).read
+        rescue Zlib::GzipFile::Error
+          raw_response_body = response.body
+        end
+
+        begin
+          JSON.parse(raw_response_body)['body'] || body
+        rescue JSON::JSONError
+          body
+        end
+      else
+        WovnLogger.error("HTML-swapper call failed. Received \"#{response.message}\" from WOVNio translation API.")
+        body
+      end
+    end
+
+    private
+
+    def prepare_connection
+      connection = Net::HTTP.new(api_uri.host, api_uri.port)
+
+      connection.open_timeout = api_timeout
+      connection.read_timeout = api_timeout
+
+      connection
+    end
+
+    def prepare_request(body)
+      if @store.compress_api_requests?
+        gzip_request(body)
+      else
+        json_request(body)
+      end
+    end
+
+    def gzip_request(html_body)
+      api_params = build_api_params(html_body)
+      compressed_body = compress_request_data(api_params.to_json)
+      request = Net::HTTP::Post.new(request_path(html_body), {
+                                      'Accept-Encoding' => 'gzip',
+                                      'Content-Type' => 'application/json',
+                                      'Content-Encoding' => 'gzip',
+                                      'Content-Length' => compressed_body.bytesize.to_s,
+                                      'X-Request-Id' => @uuid
+                                    })
+      request.body = compressed_body
+
+      request
+    end
+
+    def json_request(html_body)
+      api_params = build_api_params(html_body)
+      request = Net::HTTP::Post.new(request_path(html_body), {
+                                      'Accept-Encoding' => 'gzip',
+                                      'Content-Type' => 'application/json',
+                                      'X-Request-Id' => @uuid
+                                    })
+      request.body = api_params.to_json
+
+      request
+    end
+
+    def request_path(body)
+      "#{api_uri.path}/translation?cache_key=#{cache_key(body)}"
+    end
+
+    def cache_key(body)
+      cache_key_components = {
+        'token' => token,
+        'settings_hash' => settings_hash,
+        'body_hash' => Digest::MD5.hexdigest(body),
+        'path' => page_pathname,
+        'lang' => lang_code,
+        'version' => "wovnrb_#{VERSION}"
+      }.map { |k, v| "#{k}=#{v}" }.join('&')
+
+      CGI.escape("(#{cache_key_components})")
+    end
+
+    def build_api_params(body)
+      result = {
+        'url' => page_url,
+        'token' => token,
+        'lang_code' => lang_code,
+        'url_pattern' => url_pattern,
+        'lang_param_name' => lang_param_name,
+        'translate_canonical_tag' => translate_canonical_tag,
+        'insert_hreflangs' => insert_hreflangs,
+        'hreflang_x_default_lang' => @store.settings['hreflang_x_default_lang'], # should not use fallback, so html swapper can calculate the default lang from js_data
+        'product' => 'WOVN.rb',
+        'version' => VERSION,
+        'page_status_code' => @response_status_code,
+        'body' => body
+      }
+
+      result['custom_lang_aliases'] = JSON.dump(custom_lang_aliases) unless custom_lang_aliases.empty?
+      result['custom_domain_langs'] = JSON.dump(custom_domain_langs) unless custom_domain_langs.empty?
+
+      result
+    end
+
+    def compress_request_data(data_hash)
+      ActiveSupport::Gzip.compress(data_hash)
+    end
+
+    def api_uri
+      Addressable::URI.parse("#{@store.settings['api_url']}/v0")
+    end
+
+    def api_timeout
+      @headers.search_engine_bot? ? @store.settings['api_timeout_search_engine_bots'] : @store.settings['api_timeout_seconds']
+    end
+
+    def settings_hash
+      Digest::MD5.hexdigest(JSON.dump(@store.settings))
+    end
+
+    def token
+      @store.settings['project_token']
+    end
+
+    def lang_code
+      @headers.lang_code
+    end
+
+    def url_pattern
+      @store.settings['url_pattern']
+    end
+
+    def lang_param_name
+      @store.settings['lang_param_name']
+    end
+
+    def custom_lang_aliases
+      @store.settings['custom_lang_aliases']
+    end
+
+    def translate_canonical_tag
+      @store.settings['translate_canonical_tag']
+    end
+
+    def insert_hreflangs
+      @store.settings['insert_hreflangs']
+    end
+
+    def custom_domain_langs
+      @store.custom_domain_langs.to_html_swapper_hash
+    end
+
+    def page_url
+      "#{@headers.protocol}://#{@headers.url}"
+    end
+
+    def page_pathname
+      @headers.pathname_with_trailing_slash_if_present
+    end
+  end
+end