gitlab_controller.rb 1.88 KB
Newer Older
Valery Sizov's avatar
Valery Sizov committed
1
class Import::GitlabController < Import::BaseController
2
3
  before_action :verify_gitlab_import_enabled
  before_action :gitlab_auth, except: :callback
Valery Sizov's avatar
Valery Sizov committed
4
5
6
7

  rescue_from OAuth2::Error, with: :gitlab_unauthorized

  def callback
8
    session[:gitlab_access_token] = client.get_token(params[:code], callback_import_gitlab_url)
9
    redirect_to status_import_gitlab_url
Valery Sizov's avatar
Valery Sizov committed
10
11
12
13
  end

  def status
    @repos = client.projects
14

Valery Sizov's avatar
Valery Sizov committed
15
16
17
    @already_added_projects = current_user.created_projects.where(import_type: "gitlab")
    already_added_projects_names = @already_added_projects.pluck(:import_source)

18
    @repos = @repos.to_a.reject{ |repo| already_added_projects_names.include? repo["path_with_namespace"] }
Valery Sizov's avatar
Valery Sizov committed
19
20
21
  end

  def jobs
22
    jobs = current_user.created_projects.where(import_type: "gitlab").to_json(only: [:id, :import_status])
Valery Sizov's avatar
Valery Sizov committed
23
24
25
26
27
28
    render json: jobs
  end

  def create
    @repo_id = params[:repo_id].to_i
    repo = client.project(@repo_id)
Valery Sizov's avatar
Valery Sizov committed
29
    @project_name = repo["name"]
30
31
32
33

    repo_owner = repo["namespace"]["path"]
    repo_owner = current_user.username if repo_owner == client.user["username"]
    @target_namespace = params[:new_namespace].presence || repo_owner
34

Valery Sizov's avatar
Valery Sizov committed
35
    namespace = get_or_create_namespace || (render and return)
Valery Sizov's avatar
Valery Sizov committed
36

37
    @project = Gitlab::GitlabImport::ProjectCreator.new(repo, namespace, current_user, access_params).execute
Valery Sizov's avatar
Valery Sizov committed
38
39
40
41
42
  end

  private

  def client
43
    @client ||= Gitlab::GitlabImport::Client.new(session[:gitlab_access_token])
Valery Sizov's avatar
Valery Sizov committed
44
45
  end

Douwe Maan's avatar
Douwe Maan committed
46
47
48
49
  def verify_gitlab_import_enabled
    not_found! unless gitlab_import_enabled?
  end

Valery Sizov's avatar
Valery Sizov committed
50
  def gitlab_auth
51
    if session[:gitlab_access_token].blank?
Valery Sizov's avatar
Valery Sizov committed
52
53
54
55
56
      go_to_gitlab_for_permissions
    end
  end

  def go_to_gitlab_for_permissions
57
    redirect_to client.authorize_url(callback_import_gitlab_url)
Valery Sizov's avatar
Valery Sizov committed
58
59
60
61
62
  end

  def gitlab_unauthorized
    go_to_gitlab_for_permissions
  end
63
64
65
66
67
68

  private

  def access_params
    { gitlab_access_token: session[:gitlab_access_token] }
  end
Valery Sizov's avatar
Valery Sizov committed
69
end