forgeplus/app/controllers/ci/base_controller.rb

class Ci::BaseController < ApplicationController
  include Ci::DbConnectable

  before_action :require_login
  before_action :connect_to_ci_db

  def load_repo
    namespace = params[:owner]
    id = params[:repo] || params[:id]

    @ci_user, @repo = Ci::Repo.find_with_namespace(namespace, id, current_user.login)
  end

  def load_all_repo
    namespace = current_user.login
    @repos = Ci::Repo.find_all_with_namespace(namespace)
  end

  def load_repo_by_repo_slug(slug)
    @repo_slug = Ci::Repo.load_repo_by_repo_slug(slug)
  end

  private
    def authorize_access_project!
      unless @project.manager?(current_user)
        return render_forbidden
      end
    end

    def authenticate_manager!
      unless @project.manager?(current_user)
        return render_forbidden
      end
    end

    def authenticate_admin!
      return render_forbidden unless current_user.admin?
    end

    def authorize_owner!
      unless @project.owner?(current_user)
        return render_forbidden
      end
    end

    def find_cloud_account
      @cloud_account ||= current_user.ci_cloud_account
      @cloud_account.blank? ? nil : @cloud_account
    end

    def load_ci_user
      @ci_user ||= Ci::User.find_by(user_login: params[:owner])
      @ci_user.blank? ? raise("未找到相关的记录") : @ci_user
    end

  def connect_to_ci_db(options={})
    current = current_user
    owner = params[:owner]
    if owner
      current = User.find_by(login: owner)
    end

    if !(current && !current.is_a?(AnonymousUser) && !current.devops_uninit?)
      return
    end

    if current.ci_cloud_account.server_type == Ci::CloudAccount::SERVER_TYPE_TRUSTIE
      connect_to_trustie_ci_database(options)
    else
      options = options.merge(db_name: current.login)
      connect_to_ci_database(options)
    end

  end

end