From aee0a117a889461ce8ced6fcf73207fe017f1d99 Mon Sep 17 00:00:00 2001
From: GitLab Bot <gitlab-bot@gitlab.com>
Date: Mon, 20 Dec 2021 13:37:47 +0000
Subject: Add latest changes from gitlab-org/gitlab@14-6-stable-ee

---
 lib/version_check.rb | 44 +++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 41 insertions(+), 3 deletions(-)

(limited to 'lib/version_check.rb')

diff --git a/lib/version_check.rb b/lib/version_check.rb
index a8b7c7371ca..e5a4c244c7a 100644
--- a/lib/version_check.rb
+++ b/lib/version_check.rb
@@ -2,22 +2,60 @@
 
 require "base64"
 
-# This class is used to build image URL to
-# check if it is a new version for update
 class VersionCheck
+  include ReactiveCaching
+
+  self.reactive_cache_work_type = :external_dependency
+  self.reactive_cache_worker_finder = ->(_id, *args) { from_cache }
+
   def self.data
     { version: Gitlab::VERSION }
   end
 
-  def self.url
+  def self.headers
+    { "REFERER": Gitlab.config.gitlab.url }
+  end
+
+  # This is temporary and will be removed when the new UI is hooked up
+  # to the version_check.json endpoint.
+  def self.image_url
     encoded_data = Base64.urlsafe_encode64(data.to_json)
 
     "#{host}/check.svg?gitlab_info=#{encoded_data}"
   end
 
+  def self.url
+    encoded_data = Base64.urlsafe_encode64(data.to_json)
+
+    "#{host}/check.json?gitlab_info=#{encoded_data}"
+  end
+
   def self.host
     'https://version.gitlab.com'
   end
+
+  def self.from_cache(*)
+    new
+  end
+
+  def id
+    Gitlab::VERSION
+  end
+
+  def calculate_reactive_cache(*)
+    response = Gitlab::HTTP.try_get(self.class.url, headers: self.class.headers)
+
+    case response&.code
+    when 200
+      response.body
+    end
+  end
+
+  def response
+    with_reactive_cache do |data|
+      Gitlab::Json.parse(data) if data
+    end
+  end
 end
 
 VersionCheck.prepend_mod
-- 
cgit v1.2.1