extracted import code from rake to ImportManager

app/import_manager.rb

···
+require_relative 'importers/likes_importer'
+require_relative 'importers/posts_importer'
+require_relative 'importers/reposts_importer'
+require_relative 'item_queue'
+require_relative 'post_downloader'
+
+class ImportManager
+  attr_accessor :report, :time_limit
+
+  def initialize(user)
+    @user = user
+  end
+
+  def start(sets, include_pending)
+    queued_items = []
+    importers = []
+    sets = [sets] unless sets.is_a?(Array)
+
+    sets.each do |set|
+      case set
+      when 'likes'
+        queued_items += @user.likes.pending.to_a if include_pending
+        importers << LikesImporter.new(@user)
+      when 'reposts'
+        queued_items += @user.reposts.pending.to_a if include_pending
+        importers << RepostsImporter.new(@user)
+      when 'posts'
+        queued_items += @user.quotes.pending.to_a + @user.pins.pending.to_a if include_pending
+        importers << PostsImporter.new(@user)
+      when 'all'
+        queued_items += @user.all_pending_items if include_pending
+        importers += [
+          LikesImporter.new(@user),
+          RepostsImporter.new(@user),
+          PostsImporter.new(@user)
+        ]
+      else
+        raise "Invalid collection: #{set}"
+      end
+    end
+
+    queue = ItemQueue.new(queued_items)
+
+    downloader = PostDownloader.new
+    downloader.report = @report
+
+    download_thread = Thread.new { downloader.import_from_queue(queue) }
+
+    import_threads = importers.map do |import|
+      import.item_queue = queue
+      import.report = @report
+
+      Thread.new { import.run_import(@time_limit) }
+    end
+
+    import_threads.each { |i| i.join }
+
+    downloader.stop_when_empty = true
+    download_thread.join
+  end
+end

app/importers/base_importer.rb

···
 class BaseImporter
   attr_accessor :item_queue, :report
 
-  def initialize(user_did)
-    @did = DID.new(user_did)
-    @user = User.find_or_create_by!(did: user_did)
+  def initialize(user)
+    @did = DID.new(user.did)
+    @user = user
     @imported_count = 0
   end
 

lib/tasks/import.rake

···
-require_relative '../../app/importers/likes_importer'
-require_relative '../../app/importers/posts_importer'
-require_relative '../../app/importers/reposts_importer'
-
 require_relative '../../app/console_report'
+require_relative '../../app/import_manager'
 require_relative '../../app/item_queue'
+require_relative '../../app/models/user'
 require_relative '../../app/post_downloader'
 
 task :import_user do
···
   user = User.find_or_create_by!(did: ENV['USER'])
   pending = !ENV['SKIP_PENDING']
 
-  case ENV['COLLECTION']
-  when 'likes'
-    queue = ItemQueue.new(pending ? user.likes.pending.to_a : [])
-    importers = [LikesImporter.new(ENV['USER'])]
-  when 'reposts'
-    queue = ItemQueue.new(pending ? user.reposts.pending.to_a : [])
-    importers = [RepostsImporter.new(ENV['USER'])]
-  when 'posts'
-    queue = ItemQueue.new(pending ? user.quotes.pending.to_a + user.pins.pending.to_a : [])
-    importers = [PostsImporter.new(ENV['USER'])]
-  when 'all'
-    queue = ItemQueue.new(pending ? user.all_pending_items : [])
-    importers = [
-      LikesImporter.new(ENV['USER']),
-      RepostsImporter.new(ENV['USER']),
-      PostsImporter.new(ENV['USER'])
-    ]
-  when nil
+  unless ENV['COLLECTION']
     raise "Required COLLECTION parameter missing"
-  else
-    raise "Invalid collection: #{ENV['COLLECTION']}"
   end
 
-  report = ConsoleReport.new
-
-  downloader = PostDownloader.new
-  downloader.report = report
-
-  download_thread = Thread.new { downloader.import_from_queue(queue) }
+  import = ImportManager.new(user)
+  import.report = ConsoleReport.new
+  import.time_limit = ENV['UNTIL']
 
   trap("SIGINT") {
     puts "\n\n\n\n\n"
     exit
   }
 
-  import_threads = importers.map do |import|
-    import.item_queue = queue
-    import.report = report
-
-    Thread.new { import.run_import(ENV['UNTIL']) }
-  end
-
-  import_threads.each { |i| i.join }
-
-  downloader.stop_when_empty = true
-  download_thread.join
+  import.start(ENV['COLLECTION'], pending)
 
   puts "\n\n\n\n\n"
 end