summaryrefslogtreecommitdiff
path: root/lib/gitlab/import_export/avatar_restorer.rb
blob: 352539eb594ead70aec67accc7d348ba31776f15 (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
module Gitlab
  module ImportExport
    class AvatarRestorer

      def initialize(project:, shared:)
        @project = project
        @shared = shared
      end

      def restore
        return true unless avatar_export_file

        @project.avatar = File.open(avatar_export_file)
        @project.save!
      rescue => e
        @shared.error(e)
        false
      end

      private

      def avatar_export_file
        @avatar_export_file ||= Dir["#{avatar_export_path}/*"].first
      end

      def avatar_export_path
        File.join(@shared.export_path, 'avatar')
      end
    end
  end
end