Welcome to mirror list, hosted at ThFree Co, Russian Federation.

import_csv_service.rb « issues « services « app - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 7fa2ecc3afd6ad385e7209e6ce963a7064baa7af (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
# frozen_string_literal: true

module Issues
  class ImportCsvService
    def initialize(user, project, upload)
      @user = user
      @project = project
      @uploader = upload.build_uploader
      @results = { success: 0, error_lines: [], parse_error: false }
    end

    def execute
      # Cache remote file locally for processing
      @uploader.cache_stored_file! unless @uploader.file_storage?

      process_csv
      email_results_to_user

      cleanup_cache unless @uploader.file_storage?

      @results
    end

    private

    def process_csv
      CSV.foreach(@uploader.file.path, col_sep: detect_col_sep, headers: true).with_index(2) do |row, line_no|
        issue = Issues::CreateService.new(@project, @user, title: row[0], description: row[1]).execute

        if issue.persisted?
          @results[:success] += 1
        else
          @results[:error_lines].push(line_no)
        end
      end
    rescue ArgumentError, CSV::MalformedCSVError
      @results[:parse_error] = true
    end

    def email_results_to_user
      Notify.import_issues_csv_email(@user.id, @project.id, @results).deliver_now
    end

    def detect_col_sep
      header = File.open(@uploader.file.path, &:readline)

      if header.include?(",")
        ","
      elsif header.include?(";")
        ";"
      elsif header.include?("\t")
        "\t"
      else
        raise CSV::MalformedCSVError
      end
    end

    def cleanup_cache
      cached_file_path = @uploader.file.cache_path

      File.delete(cached_file_path)
      Dir.delete(File.dirname(cached_file_path))
    end
  end
end