FEATURE: use random user avatar image from 'avatars' directory.

lib/discourse_dev/user.rb

@@ -6,9 +6,11 @@
 
 module DiscourseDev
   class User < Record
+    attr_reader :images
 
     def initialize(count = DEFAULT_COUNT)
       super(::User, count)
+      @images = Dir[File.join(__dir__, '..', '..', 'avatars', '*.*')]
     end
 
     def data
@@ -31,6 +33,7 @@ def data
     def create!
       super do |user|
         user.activate
+        set_random_avatar(user)
         Faker::Number.between(from: 0, to: 2).times do
           group = Group.random
 
@@ -42,5 +45,48 @@ def create!
     def self.random
       super(::User)
     end
+
+    def set_random_avatar(user)
+      return if images.blank?
+      return unless Faker::Boolean.boolean
+
+      avatar_index = Faker::Number.between(from: 0, to: images.count - 1)
+      avatar_path = images[avatar_index]
+      create_avatar(user, avatar_path)
+      @images.delete_at(avatar_index)
+    end
+
+    def create_avatar(user, avatar_path)
+      tempfile = copy_to_tempfile(avatar_path)
+      filename = "avatar#{File.extname(avatar_path)}"
+      upload = UploadCreator.new(tempfile, filename, type: "avatar").create_for(user.id)
+
+      if upload.present? && upload.persisted?
+        user.create_user_avatar
+        user.user_avatar.update(custom_upload_id: upload.id)
+        user.update(uploaded_avatar_id: upload.id)
+      else
+        STDERR.puts "Failed to upload avatar for user #{user.username}: #{avatar_path}"
+        STDERR.puts upload.errors.inspect if upload
+      end
+    rescue
+      STDERR.puts "Failed to create avatar for user #{user.username}: #{avatar_path}"
+    ensure
+      tempfile.close! if tempfile
+    end
+
+    private
+
+    def copy_to_tempfile(source_path)
+      extension = File.extname(source_path)
+      tmp = Tempfile.new(['discourse-upload', extension])
+
+      File.open(source_path) do |source_stream|
+        IO.copy_stream(source_stream, tmp)
+      end
+
+      tmp.rewind
+      tmp
+    end
   end
 end

lib/discourse_dev/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module DiscourseDev
-  VERSION = "0.0.8"
+  VERSION = "0.0.9"
 end