debian-mirror-gitlab/app/controllers/projects/google_cloud/base_controller.rb

# frozen_string_literal: true

class Projects::GoogleCloud::BaseController < Projects::ApplicationController
  feature_category :five_minute_production_app
  urgency :low

  before_action :admin_project_google_cloud!
  before_action :google_oauth2_enabled!
  before_action :feature_flag_enabled!

  private

  def admin_project_google_cloud!
    unless can?(current_user, :admin_project_google_cloud, project)
      track_event('admin_project_google_cloud!', 'access_denied', 'invalid_user')
      access_denied!
    end
  end

  def google_oauth2_enabled!
    config = Gitlab::Auth::OAuth::Provider.config_for('google_oauth2')
    if config.app_id.blank? || config.app_secret.blank?
      track_event('google_oauth2_enabled!', 'access_denied', { reason: 'google_oauth2_not_configured', config: config })
      access_denied! 'This GitLab instance not configured for Google Oauth2.'
    end
  end

  def feature_flag_enabled!
    enabled_for_user = Feature.enabled?(:incubation_5mp_google_cloud, current_user)
    enabled_for_group = Feature.enabled?(:incubation_5mp_google_cloud, project.group)
    enabled_for_project = Feature.enabled?(:incubation_5mp_google_cloud, project)
    feature_is_enabled = enabled_for_user || enabled_for_group || enabled_for_project
    unless feature_is_enabled
      track_event('feature_flag_enabled!', 'access_denied', 'feature_flag_not_enabled')
      access_denied!
    end
  end

  def validate_gcp_token!
    is_token_valid = GoogleApi::CloudPlatform::Client.new(token_in_session, nil)
                                                     .validate_token(expires_at_in_session)

    return if is_token_valid

    return_url = project_google_cloud_index_path(project)
    state = generate_session_key_redirect(request.url, return_url)
    @authorize_url = GoogleApi::CloudPlatform::Client.new(nil,
                                                          callback_google_api_auth_url,
                                                          state: state).authorize_url
    redirect_to @authorize_url
  end

  def generate_session_key_redirect(uri, error_uri)
    GoogleApi::CloudPlatform::Client.new_session_key_for_redirect_uri do |key|
      session[key] = uri
      session[:error_uri] = error_uri
    end
  end

  def token_in_session
    session[GoogleApi::CloudPlatform::Client.session_key_for_token]
  end

  def expires_at_in_session
    session[GoogleApi::CloudPlatform::Client.session_key_for_expires_at]
  end

  def handle_gcp_error(action, error)
    track_event(action, 'gcp_error', error)
    @js_data = { screen: 'gcp_error', error: error.to_s }.to_json
    render status: :unauthorized, template: 'projects/google_cloud/errors/gcp_error'
  end

  def track_event(action, label, property)
    options = { label: label, project: project, user: current_user }

    if property.is_a?(String)
      options[:property] = property
    else
      options[:extra] = property
    end

    Gitlab::Tracking.event('Projects::GoogleCloud', action, **options)
  end
end