BE: Allow customers to extend or reactivate their trial on gitlab.com [RUN ALL RSPEC] [RUN AS-IF-FOSS]

What does this MR do?

Related to issue #290278 (closed)

There are 4 MRs for issue #290278 (closed)

• migration script on Gitlab: !56460 (merged)
• migration script on CustomersDot: https://gitlab.com/gitlab-org/customers-gitlab-com/-/merge_requests/2937
• feature code on Gitlab: !56471 (merged) (this MR)
• feature code on CustomersDot: https://gitlab.com/gitlab-org/customers-gitlab-com/-/merge_requests/2940

Note: the issue's Data requirement has not confirmed yet, so please ignore data requirement during review.

This MR add a put method extend_reactivate in trials controller

• it requires params[:trial_extension_type] as one params
• it validates params[:trial_extension_type] against the namespace
• if the request is valid, it calls GitlabSubscriptions::ExtendReactivateTrialService
• GitlabSubscriptions::ExtendReactivateTrialService will send a put request to CustomersDot

Screenshots (strongly suggested)

Does this MR meet the acceptance criteria?

Conformity

• 📋 Does this MR need a changelog?
  • I have included a changelog entry.
  • I have not included a changelog entry because _____.
• Documentation (if required)
• Code review guidelines
• Merge request performance guidelines
• Style guides
• Database guides
• Separation of EE specific content

Availability and Testing

• Review and add/update tests for this feature/bug. Consider all test levels. See the Test Planning Process.
• Tested in all supported browsers
• Informed Infrastructure department of a default or new setting change, if applicable per definition of done

Security

If this MR contains changes to processing or storing of credentials or tokens, authorization and authentication methods and other items described in the security review guidelines:

• Label as security and @ mention @gitlab-com/gl-security/appsec
• The MR includes necessary changes to maintain consistency between UI, API, email, or other methods
• Security reports checked/validated by a reviewer from the AppSec team

Related to #290278 (closed)

ee/app/controllers/trials_controller.rb

@@ -8,6 +8,8 @@ class TrialsController < ApplicationController
   before_action :check_if_gl_com_or_dev
   before_action :authenticate_user!
   before_action :find_or_create_namespace, only: :apply
+  before_action :find_namespace, only: [:extend_reactivate]
+  before_action :authenticate_namespace_owner!, only: [:extend_reactivate]
 
   feature_category :purchase
 
@@ -49,6 +51,18 @@ def apply
     end
   end
 
+  def extend_reactivate
+    render_404 unless Feature.enabled?(:allow_extend_reactivate_trial)
+
+    @result = GitlabSubscriptions::ExtendReactivateTrialService.new.execute(extend_reactivate_trial_params) if valid_extension?
+
+    if @result&.dig(:success)
+      head 200
+    else
+      render_403
+    end
+  end
+
   protected
 
   # override the ConfirmEmailWarning method in order to skip
@@ -64,6 +78,16 @@ def authenticate_user!
     redirect_to new_trial_registration_path, alert: I18n.t('devise.failure.unauthenticated')
   end
 
+  def authenticate_namespace_owner!
+    user_is_namespace_owner = if @namespace.is_a?(Group)
+                           @namespace.owners.include?(current_user)
+                         else
+                           @namespace.owner == current_user
+                         end
+
+    render_403 unless user_is_namespace_owner
+  end
+
   def company_params
     params.permit(:company_name, :company_size, :first_name, :last_name, :phone_number, :number_of_users, :country)
           .merge(extra_params)
@@ -90,6 +114,15 @@ def apply_trial_params
     }
   end
 
+  def extend_reactivate_trial_params
+    gl_com_params = { gitlab_com_trial: true, sync_to_gl: true }
+
+    {
+      trial_user: params.permit(:namespace_id, :trial_extension_type, :trial_entity, :glm_source, :glm_content).merge(gl_com_params),
+      uid: current_user.id
+    }
+  end
+
   def find_or_create_namespace
     @namespace = if find_namespace?
                    current_user.namespaces.find_by_id(params[:namespace_id])
@@ -100,10 +133,30 @@ def find_or_create_namespace
     render_404 unless @namespace
   end
 
+  def find_namespace
+    @namespace = if find_namespace?
+                   current_user.namespaces.find_by_id(params[:namespace_id])
+                 end
+
+    render_404 unless @namespace
+  end
+
   def find_namespace?
     params[:namespace_id].present? && params[:namespace_id] != '0'
   end
 
+  def valid_extension?
+    trial_extension_type = params[:trial_extension_type].to_i
+
+    return false unless [GitlabSubscription::TRIAL_EXTENDED, GitlabSubscription::TRIAL_REACTIVATED].include?(trial_extension_type)
+
+    return false if trial_extension_type == GitlabSubscription::TRIAL_EXTENDED && !@namespace.can_extend?
+
+    return false if trial_extension_type == GitlabSubscription::TRIAL_REACTIVATED && !@namespace.can_reactivate?
+
+    true
+  end
+
   def can_create_group?
     params[:new_group_name].present? && can?(current_user, :create_group)
   end