gitlab-cloud-connector 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 1c5bb789b8f217dd5b85dea32aa05101ece88b73ec439d68a5ef42b5bc362c5f
+  data.tar.gz: 6121096669714066e395e6e7e820f55a81f2787cfabb21d895c110a4ddc5aa9d
+SHA512:
+  metadata.gz: 1a26a433e6339d320825260377887ce161803403f7dee582abef232ac86ce164145672c91b539a944731f7152e168afa15ba5c8f39f189ae954f0bca937e2ef3
+  data.tar.gz: 9406ef59e2f255065f2f84160e316e7e34f715bada69dd943e40af35eaea0eae642481499135bd52e62213636798800689dc0d8d00214a5da008327e2d853174

data/.idea/.gitignore

@@ -0,0 +1,8 @@
+# Default ignored files
+/shelf/
+/workspace.xml
+# Editor-based HTTP Client requests
+/httpRequests/
+# Datasource local storage ignored files
+/dataSources/
+/dataSources.local.xml

data/.idea/gitlab-cloud-connector.iml

@@ -0,0 +1,127 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="RUBY_MODULE" version="4">
+  <component name="FacetManager">
+    <facet type="RailsFacetType" name="Ruby on Rails">
+      <configuration>
+        <RAILS_FACET_CONFIG_ID NAME="RAILS_FACET_SUPPORT_REMOVED" VALUE="false" />
+        <RAILS_FACET_CONFIG_ID NAME="RAILS_TESTS_SOURCES_PATCHED" VALUE="true" />
+        <RAILS_FACET_CONFIG_ID NAME="RAILS_FACET_APPLICATION_ROOT" VALUE="$MODULE_DIR$" />
+      </configuration>
+    </facet>
+  </component>
+  <component name="ModuleRunConfigurationManager">
+    <shared />
+  </component>
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$">
+      <sourceFolder url="file://$MODULE_DIR$/features" isTestSource="true" />
+      <sourceFolder url="file://$MODULE_DIR$/spec" isTestSource="true" />
+      <sourceFolder url="file://$MODULE_DIR$/test" isTestSource="true" />
+      <excludeFolder url="file://$MODULE_DIR$/.bundle" />
+      <excludeFolder url="file://$MODULE_DIR$/public/packs" />
+      <excludeFolder url="file://$MODULE_DIR$/vendor/bundle" />
+      <excludeFolder url="file://$MODULE_DIR$/vendor/cache" />
+      <excludeFolder url="file://$MODULE_DIR$/log" />
+      <excludeFolder url="file://$MODULE_DIR$/tmp" />
+      <excludeFolder url="file://$MODULE_DIR$/public/system" />
+      <excludeFolder url="file://$MODULE_DIR$/components" />
+    </content>
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+    <orderEntry type="library" scope="PROVIDED" name="ast (v2.4.2, asdf: 3.2.1) [gem]" level="application" />
+    <orderEntry type="library" scope="PROVIDED" name="crass (v1.0.6, asdf: 3.2.1) [gem]" level="application" />
+    <orderEntry type="library" scope="PROVIDED" name="mail (v2.8.1, asdf: 3.2.1) [gem]" level="application" />
+    <orderEntry type="library" scope="PROVIDED" name="net-pop (v0.1.2, asdf: 3.2.1) [gem]" level="application" />
+    <orderEntry type="library" scope="PROVIDED" name="rack-test (v2.1.0, asdf: 3.2.1) [gem]" level="application" />
+    <orderEntry type="library" scope="PROVIDED" name="rails-html-sanitizer (v1.6.0, asdf: 3.2.1) [gem]" level="application" />
+    <orderEntry type="library" scope="PROVIDED" name="rainbow (v3.1.1, asdf: 3.2.1) [gem]" level="application" />
+    <orderEntry type="library" scope="PROVIDED" name="rubocop-rails (v2.18.0, asdf: 3.2.1) [gem]" level="application" />
+    <orderEntry type="library" scope="PROVIDED" name="tzinfo (v2.0.6, asdf: 3.2.1) [gem]" level="application" />
+    <orderEntry type="library" scope="PROVIDED" name="websocket-extensions (v0.1.5, asdf: 3.2.1) [gem]" level="application" />
+  </component>
+  <component name="RModuleSettingsStorage">
+    <LOAD_PATH number="0" />
+    <I18N_FOLDERS number="1" string0="$MODULE_DIR$/config/locales" />
+  </component>
+  <component name="RailsGeneratorsCache">
+    <option name="generators">
+      <list>
+        <option value="active_record:migration" />
+        <option value="active_record:model" />
+        <option value="active_record:observer" />
+        <option value="active_record:session_migration" />
+        <option value="controller" />
+        <option value="erb:controller" />
+        <option value="erb:mailer" />
+        <option value="erb:scaffold" />
+        <option value="generator" />
+        <option value="helper" />
+        <option value="integration_test" />
+        <option value="mailer" />
+        <option value="metal" />
+        <option value="migration" />
+        <option value="model" />
+        <option value="model_subclass" />
+        <option value="observer" />
+        <option value="performance_test" />
+        <option value="plugin" />
+        <option value="resource" />
+        <option value="scaffold" />
+        <option value="scaffold_controller" />
+        <option value="session_migration" />
+        <option value="stylesheets" />
+        <option value="test_unit:controller" />
+        <option value="test_unit:helper" />
+        <option value="test_unit:integration" />
+        <option value="test_unit:mailer" />
+        <option value="test_unit:model" />
+        <option value="test_unit:observer" />
+        <option value="test_unit:performance" />
+        <option value="test_unit:plugin" />
+        <option value="test_unit:scaffold" />
+      </list>
+    </option>
+    <option name="myGenerators">
+      <list>
+        <option value="active_record:migration" />
+        <option value="active_record:model" />
+        <option value="active_record:observer" />
+        <option value="active_record:session_migration" />
+        <option value="controller" />
+        <option value="erb:controller" />
+        <option value="erb:mailer" />
+        <option value="erb:scaffold" />
+        <option value="generator" />
+        <option value="helper" />
+        <option value="integration_test" />
+        <option value="mailer" />
+        <option value="metal" />
+        <option value="migration" />
+        <option value="model" />
+        <option value="model_subclass" />
+        <option value="observer" />
+        <option value="performance_test" />
+        <option value="plugin" />
+        <option value="resource" />
+        <option value="scaffold" />
+        <option value="scaffold_controller" />
+        <option value="session_migration" />
+        <option value="stylesheets" />
+        <option value="test_unit:controller" />
+        <option value="test_unit:helper" />
+        <option value="test_unit:integration" />
+        <option value="test_unit:mailer" />
+        <option value="test_unit:model" />
+        <option value="test_unit:observer" />
+        <option value="test_unit:performance" />
+        <option value="test_unit:plugin" />
+        <option value="test_unit:scaffold" />
+      </list>
+    </option>
+  </component>
+  <component name="RakeTasksCache">
+    <option name="myRootTask">
+      <RakeTaskImpl id="rake" />
+    </option>
+  </component>
+</module>

data/.idea/inspectionProfiles/Project_Default.xml

@@ -0,0 +1,9 @@
+<component name="InspectionProjectProfileManager">
+  <profile version="1.0">
+    <option name="myName" value="Project Default" />
+    <inspection_tool class="Rubocop" enabled="true" level="WARNING" enabled_by_default="true">
+      <option name="myUseStandardGemIfPossible" value="true" />
+      <option name="myRunRubocopOnSave" value="true" />
+    </inspection_tool>
+  </profile>
+</component>

data/.idea/misc.xml

@@ -0,0 +1,4 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectRootManager" version="2" project-jdk-name="asdf: 3.2.1" project-jdk-type="RUBY_SDK" />
+</project>

data/.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/gitlab-cloud-connector.iml" filepath="$PROJECT_DIR$/.idea/gitlab-cloud-connector.iml" />
+    </modules>
+  </component>
+</project>

data/.idea/vcs.xml

@@ -0,0 +1,7 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="Git" />
+    <mapping directory="$PROJECT_DIR$/gitlab-cloud-connector" vcs="Git" />
+  </component>
+</project>

data/.rubocop.yml

@@ -0,0 +1,3 @@
+inherit_gem:
+  gitlab-styles:
+    - rubocop-default.yml

data/CHANGELOG.md

@@ -0,0 +1,5 @@
+## [Unreleased]
+
+## [0.1.0] - 2024-06-11
+
+- Initial release

data/Dangerfile

@@ -0,0 +1,5 @@
+ # frozen_string_literal: true
+
+ require 'gitlab-dangerfiles'
+
+ Gitlab::Dangerfiles.for_project(self, 'gitlab-cloud-connector', &:import_defaults)

data/README.md

@@ -0,0 +1,231 @@
+# Gitlab::Cloud::Connector
+
+TODO: Delete this and the text below, and describe your gem
+
+Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/gitlab/cloud/connector`. To experiment with that code, run `bin/console` for an interactive prompt.
+
+## Installation
+
+TODO: Replace `UPDATE_WITH_YOUR_GEM_NAME_IMMEDIATELY_AFTER_RELEASE_TO_RUBYGEMS_ORG` with your gem name right after releasing it to RubyGems.org. Please do not do it earlier due to security reasons. Alternatively, replace this section with instructions to install your gem from git if you don't plan to release to RubyGems.org.
+
+Install the gem and add to the application's Gemfile by executing:
+
+    $ bundle add UPDATE_WITH_YOUR_GEM_NAME_IMMEDIATELY_AFTER_RELEASE_TO_RUBYGEMS_ORG
+
+If bundler is not being used to manage dependencies, install the gem by executing:
+
+    $ gem install UPDATE_WITH_YOUR_GEM_NAME_IMMEDIATELY_AFTER_RELEASE_TO_RUBYGEMS_ORG
+
+## Usage
+
+You need to configure the Gitlab Cloud Connector gem first.
+Create a class that inherits `Gitlab::CloudConnector::PurchasableDecorator` class and overrides the `available_for?` method.
+
+Example:
+
+```ruby
+require 'gitlab/cloud_connector/unit_primitives'
+
+
+class CustomPurchasableDecorator < Gitlab::CloudConnector::PurchasableDecorator
+  extend ::Gitlab::Utils::Override
+  
+  override :available_for?
+  def available_for?(namespace)
+    add_on_purchases(namespace).any?
+  end
+  
+  private
+
+  def add_on_purchases(namespace = nil) 
+    results = GitlabSubscriptions::AddOnPurchase
+      .by_add_on_name(add_ons)
+      .active
+    results = results.by_namespace_id(namespace.self_and_ancestor_ids) if namespace
+    
+    results
+  end
+end
+```
+
+Create `config/initializers/cloud_connector.rb` initializer 
+
+```ruby
+# config/initializers/cloud_connector.rb
+
+require 'gitlab/cloud_connector'
+require 'custom_purchasable_decorator'
+
+Gitlab::CloudConnector.configure do |config|
+  config.purchasable_decorator_class = CustomPurchasableDecorator
+  config.token_issuer = -> { Doorkeeper::OpenidConnect.configuration.issuer }
+  config.token_subject = Gitlab::CurrentSettings.uuid
+  config.gitlab_version = Gitlab::VERSION
+  config.gitlab_realm = 'self-managed'
+end
+```
+
+The CloudConnector unit primitives are grouped by `backend_services` and by the way how they are delivered: `delivered_by` (trough common interface i.e Duo Chat).
+
+```ruby
+  Gitlab::CloudConnector.unit_primitives
+ ```
+
+To access specific unit primitive:
+
+```ruby
+  code_suggestions = Gitlab::CloudConnector.unit_primitives[:code_suggestions]
+  code_suggestions.min_gitlab_version
+  code_suggestions.cut_off_date
+  code_suggestions.add_ons
+  code_suggestions.feature_categories
+  code_suggestions.free_access?
+  code_suggestions.supported_gitlab_version?('16.8')
+ ```
+
+You can group unit primitives by backend service that is hosting the unit primitive.
+
+```ruby
+  ai_gateway = Gitlab::CloudConnector.backend_services[:ai_gateway]
+  ai_gateway.min_gitlab_version # returns the smallest required gitlab version from all grouped unit primitives
+  ai_gateway.cut_off_date # returns the latest cut-off-date from all grouped unit primitives
+  ai_gateway.add_ons # returns uniq union of all add-on names bundled with grouped unit primitives
+  ai_gateway.feature_categories # returns uniq union of all feature-categories that introduced grouped unit primitives 
+  ai_gateway.delivered_by # return groups of unit primitives that are delivered together and hosted by this backend service
+  ai_gateway.unit_primitives # return list of unit primitives that are hosted by this backend service
+```
+
+You can group unit primitives that are delivered together. The unit primitives can share common user interface. (i.e. Duo Chat)
+
+```ruby
+  delivered_by = Gitlab::CloudConnector.delivered_by[:duo_chat]
+  delivered_by.min_gitlab_version # returns the smallest required gitlab version from all grouped unit primitives
+  delivered_by.cut_off_date # returns the latest cut-off-date from all grouped unit primitives
+  delivered_by.add_ons # returns uniq union of all add-on names bundled with grouped unit primitives
+  delivered_by.feature_categories # returns uniq union of all feature-categories that introduced grouped unit primitives 
+  delivered_by.free_access? # checks if at least one UP in the group has free access
+  delivered_by.supported_gitlab_version?('16.8') # checks if at least one UP in the group is supported by provided gitlab version
+  delivered_by.unit_primitives # return list of unit primitives that are delivered together via duo_chat
+```
+
+To list all available add-ons for the instance:
+
+```ruby 
+  Gitlab::CloudConnector.add_ons
+```
+
+### AccessTokens
+
+You can issue an isntance access token that will contain all allowed scopes across all backend services.
+Note that audience header will contain the list of audiences, containing multiple supported backends, 
+that can authenticate this JWT token.
+
+```ruby
+   Gitlab::CloudConnector.instance_access_token(License.last, gitlab_version: '16.8')
+```
+
+You can issue a token per backend service, that will contain all allowed scopes for unit primitives that are hosted by specified backend service.
+
+```ruby
+   Gitlab::CloudConnector.backend_services[:ai_gateway].access_token(License.last, gitlab_version: '16.8')
+```
+
+You can issue a token per delivered_by group, that will contain all allowed scopes for unit primitives that are delivered by a common user interface (i.e. Duo Chat) 
+
+```ruby
+   Gitlab::CloudConnector.delivered_by[:duo_chat].access_token(License.last, gitlab_version: '16.8')
+```
+
+### Permission checks
+
+You can call all public methods implemented in PurchasableDecorator on delivered_by group or on unit primitive level 
+
+If you want to check if the access to the group has free access
+
+```ruby 
+  Gitlab::CloudConnector.delivered_by[:duo_chat].free_access?
+```
+
+You can also call any other public method that is implemented by `CloudConnector::PurchasableInterface`
+
+
+Or you can check for specific unit primitive:
+
+```ruby 
+  Gitlab::CloudConnector.unit_primitives[:documentation_search].free_access?
+```
+
+
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitLab: []() # TODO add link
+
+### Introduce new unit primitive
+
+The GitLab codebase provides bin/unit-primitive, a dedicated tool to create new unit primitive definitions. 
+The tool asks various questions about the new unit primitive, then creates a YAML definition in config/backend_services/[backend_service]/unit_primitives/[unit_primitive_name].yml file.
+
+```sh
+bin/unit_primitive new_feature
+
+>> Description of the unit primitive (enter to skip):
+?>
+You picked the bakcend 'ai_gateway'
+
+>> Cut off date when the service is no longer available for free (enter to skip):
+?> 2024-12-22 00:00
+
+>>Minimum gitlab version (enter to skip):
+?> 16.7
+
+>>Minimum gitlab version when feature has free access (enter to skip):
+?> 16.8
+
+>>Minimum backend service version (enter to skip):
+?> 2.3
+
+You picked the add_ons '["duo_enterprise", "duo_pro"]'
+You picked the delivered_by '["duo_chat"]'
+You picked the group 'group::cloud connector'
+You picked the feature_category 'cloud_connector'
+
+>> URL of the original unit primitive issue (enter to skip):
+?>
+
+>> URL of the MR introducing the unit primitive (enter to skip):
+?>
+
+You picked the gitlab_realms '["saas", "self-managed"]'
+
+create config/cloud_connector/backend_services/ai_gateway/unit_primitives/new_feature.yml
+---
+name: new_feature
+description:
+cut_off_date: 2024-12-22 00:00
+minumum_gitlab_version: '16.7'
+min_gitlab_version_for_free_access: '16.8'
+min_backend_version: '2.3'
+bundled_with:
+  - duo_enterprise
+  - duo_pro
+delivered_by:
+  - duo_chat
+unit_primitive_issue_url:
+introduced_by_url:
+group: group::cloud connector
+feature_category: cloud_connector
+milestone: 16.7
+gitlab_realms:
+  - saas
+  - self-managed 
+```
+
+
+

data/Rakefile

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+require "rubocop/rake_task"
+
+RuboCop::RakeTask.new
+
+task default: %i[spec rubocop]

data/config/cloud_connector/backend_services/ai_gateway/unit_primitives/code_suggestions.yml

@@ -0,0 +1,19 @@
+---
+name: code_suggestions
+description:
+backend: 'gitlab-ai-gateway'
+cut_off_date: 2024-02-15 00:00:00 UTC
+min_gitlab_version_for_free_access:
+min_gitlab_version: '16.8'
+min_backend_version:
+delivered_by:
+  - code_suggestions
+bundled_with:
+  - duo_pro
+  - duo_enterprise
+introduced_by_url:
+group: group::code suggestions
+feature_category: code_suggestions
+gitlab_realms:
+  - saas
+  - self-managed

data/config/cloud_connector/backend_services/ai_gateway/unit_primitives/documentation_search.yml

@@ -0,0 +1,19 @@
+---
+name: documentation_search
+description:
+backend: 'gitlab-ai-gateway'
+cut_off_date:
+min_gitlab_version_for_free_access: '16.8'
+min_gitlab_version: '16.9'
+min_backend_version:
+delivered_by:
+  - duo_chat
+bundled_with:
+  - duo_pro
+  - duo_enterprise
+introduced_by_url:
+group: group::duo_chat
+feature_category: duo_chat
+gitlab_realms:
+  - saas
+  - self-managed

data/config/cloud_connector/backend_services/ai_gateway/unit_primitives/duo_chat.yml

@@ -0,0 +1,19 @@
+---
+name: duo_chat
+description:
+backend: 'gitlab-ai-gateway'
+cut_off_date:
+min_gitlab_version_for_free_access: '16.8'
+min_gitlab_version: '16.9'
+min_backend_version:
+delivered_by:
+  - duo_chat
+bundled_with:
+  - duo_pro
+  - duo_enterprise
+introduced_by_url:
+group: group::duo_chat
+feature_category: duo_chat
+gitlab_realms:
+  - saas
+  - self-managed

data/lib/gitlab/cloud_connector/auth/json_web_token.rb

@@ -0,0 +1,82 @@
+# frozen_string_literal: true
+
+require "active_support"
+require "active_support/all"
+
+module Gitlab
+  module CloudConnector
+    module Auth
+      class JsonWebToken
+        NOT_BEFORE_TIME = 5.seconds.to_i.freeze
+        DEFAULT_EXPIRES_IN = 3.days.to_i.freeze
+        NoSigningKeyError = Class.new(StandardError)
+
+        attr_reader :expires_at, :scopes
+
+        def initialize(audience:, subject:, scopes:, expires_at: nil, extra_claims: nil, gitlab_realm: nil)
+          @id = SecureRandom.uuid
+          @audience = audience
+          @subject = subject
+          @scopes = Array.wrap(scopes)
+          @issuer = issuer
+          @issued_at = Time.current.to_i
+          @not_before = @issued_at - NOT_BEFORE_TIME
+          @expires_at = expires_at || (@issued_at + expires_in)
+          @extra_claims = extra_claims
+          @gitlab_realm = gitlab_realm || ::Gitlab::CloudConnector.gitlab_realm
+        end
+
+        def encode
+          headers = { typ: 'JWT' }
+
+          JWT.encode(payload, key, 'RS256', headers)
+        end
+
+        def payload
+          reserved_claims = {
+            jti: @id,
+            aud: @audience,
+            sub: @subject,
+            iss: @issuer,
+            iat: @issued_at,
+            nbf: @not_before,
+            exp: @expires_at
+          }
+
+          custom_claims = {
+            gitlab_realm: @gitlab_realm,
+            scopes: @scopes,
+            extra_claims: @extra_claims
+          }
+
+          reserved_claims.merge(custom_claims)
+        end
+
+        private
+
+        def expires_in
+          ::Gitlab::CloudConnector.token_expires_in || DEFAULT_EXPIRES_IN
+        end
+
+        def issuer
+          issuer = Gitlab::CloudConnector.token_issuer
+
+          if issuer.is_a?(Proc)
+            p "lambda?"
+            issuer.call
+          else
+            issuer
+          end
+        end
+
+        def key
+          key_data = Settings.value(:jwt_signing_key)
+
+          raise NoSigningKeyError unless key_data
+
+          OpenSSL::PKey::RSA.new(key_data)
+        end
+      end
+    end
+  end
+end

data/lib/gitlab/cloud_connector/backend_service.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module CloudConnector
+    class  BackendService < BaseGroup
+      include Deliverable
+    end
+  end
+end

data/lib/gitlab/cloud_connector/base_group.rb

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+require_relative 'token_issuer'
+
+module Gitlab
+  module CloudConnector
+    class BaseGroup
+      include TokenIssuer
+
+      attr_reader :name, :unit_primitives
+
+      def initialize(name, unit_primitives)
+        @name = name
+        @unit_primitives = unit_primitives
+      end
+
+      def cut_off_date
+        unit_primitives.values.map(&:cut_off_date).compact.max
+      end
+
+      def bundled_with
+        unit_primitives.values.flat_map(&:bundled_with).uniq
+      end
+      alias :add_ons :bundled_with
+
+      def feature_categories
+        unit_primitives.values.map(&:feature_category).uniq
+      end
+
+      def min_gitlab_version
+        unit_primitives.values.flat_map(&:min_gitlab_version).compact.min
+      end
+
+      def min_gitlab_version_for_free_access
+        unit_primitives.values.flat_map(&:min_gitlab_version_for_free_access).compact.min
+      end
+    end
+  end
+end

data/lib/gitlab/cloud_connector/configuration.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+require_relative 'unit_primitives'
+require_relative 'purchasable_decorator'
+
+module Gitlab
+  module CloudConnector
+    class Configuration
+      attr_reader :purchasable_decorator_class
+      attr_accessor :token_issuer, :token_subject, :gitlab_version, :token_expires_in, :gitlab_realm
+
+      def purchasable_decorator_class=(purchasable_decorator_class)
+        raise ArgumentError, 'Must be a class inheriting from Gitlab::CloudConnector::PurchasableDecorator' unless
+          purchasable_decorator_class < Gitlab::CloudConnector::PurchasableDecorator
+
+        @purchasable_decorator_class = purchasable_decorator_class
+      end
+
+      def purchasable_decorator_class
+        @purchasable_decorator_class ||= Gitlab::CloudConnector::PurchasableDecorator
+      end
+    end
+  end
+end