FEATURE: Bulk Invite

app/controllers/admin/backups_controller.rb

@@ -2,7 +2,7 @@ require_dependency "backup_restore"
 
 class Admin::BackupsController < Admin::AdminController
 
-  skip_before_filter :check_xhr, only: [:index, :show, :logs, :check_chunk, :upload_chunk]
+  skip_before_filter :check_xhr, only: [:index, :show, :logs, :check_backup_chunk, :upload_backup_chunk]
 
   def index
     respond_to do |format|
@@ -87,7 +87,7 @@ class Admin::BackupsController < Admin::AdminController
     render nothing: true
   end
 
-  def check_chunk
+  def check_backup_chunk
     identifier         = params.fetch(:resumableIdentifier)
     filename           = params.fetch(:resumableFilename)
     chunk_number       = params.fetch(:resumableChunkNumber)
@@ -95,15 +95,13 @@ class Admin::BackupsController < Admin::AdminController
 
     # path to chunk file
     chunk = Backup.chunk_path(identifier, filename, chunk_number)
-    # check whether the chunk has already been uploaded
-    has_chunk_been_uploaded = File.exists?(chunk) && File.size(chunk) == current_chunk_size
-    # 200 = exists, 404 = not uploaded yet
-    status = has_chunk_been_uploaded ? 200 : 404
+    # check chunk upload status
+    status = HandleChunkUpload.check_chunk(chunk, current_chunk_size: current_chunk_size)
 
     render nothing: true, status: status
   end
 
-  def upload_chunk
+  def upload_backup_chunk
     filename   = params.fetch(:resumableFilename)
     total_size = params.fetch(:resumableTotalSize).to_i
 
@@ -118,15 +116,10 @@ class Admin::BackupsController < Admin::AdminController
 
     # path to chunk file
     chunk = Backup.chunk_path(identifier, filename, chunk_number)
-    dir = File.dirname(chunk)
-
-    # ensure directory exists
-    FileUtils.mkdir_p(dir) unless Dir.exists?(dir)
-    # save chunk to the directory
-    File.open(chunk, "wb") { |f| f.write(file.tempfile.read) }
+    # upload chunk
+    HandleChunkUpload.upload_chunk(chunk, file: file)
 
     uploaded_file_size = chunk_number * chunk_size
-
     # when all chunks are uploaded
     if uploaded_file_size + current_chunk_size >= total_size
       # merge all the chunks in a background thread

app/controllers/invites_controller.rb

@@ -3,7 +3,7 @@ class InvitesController < ApplicationController
   skip_before_filter :check_xhr
   skip_before_filter :redirect_to_login_if_required
 
-  before_filter :ensure_logged_in, only: [:destroy, :create]
+  before_filter :ensure_logged_in, only: [:destroy, :create, :check_csv_chunk, :upload_csv_chunk]
 
   def show
     invite = Invite.find_by(invite_key: params[:id])
@@ -51,4 +51,48 @@ class InvitesController < ApplicationController
     render nothing: true
   end
 
+  def check_csv_chunk
+    guardian.ensure_can_bulk_invite_to_forum!(current_user)
+
+    filename           = params.fetch(:resumableFilename)
+    identifier         = params.fetch(:resumableIdentifier)
+    chunk_number       = params.fetch(:resumableChunkNumber)
+    current_chunk_size = params.fetch(:resumableCurrentChunkSize).to_i
+
+    # path to chunk file
+    chunk = Invite.chunk_path(identifier, filename, chunk_number)
+    # check chunk upload status
+    status = HandleChunkUpload.check_chunk(chunk, current_chunk_size: current_chunk_size)
+
+    render nothing: true, status: status
+  end
+
+  def upload_csv_chunk
+    guardian.ensure_can_bulk_invite_to_forum!(current_user)
+
+    filename = params.fetch(:resumableFilename)
+    return render status: 415, text: I18n.t("bulk_invite.file_should_be_csv") unless filename.to_s.end_with?(".csv")
+
+    file               = params.fetch(:file)
+    identifier         = params.fetch(:resumableIdentifier)
+    chunk_number       = params.fetch(:resumableChunkNumber).to_i
+    chunk_size         = params.fetch(:resumableChunkSize).to_i
+    total_size         = params.fetch(:resumableTotalSize).to_i
+    current_chunk_size = params.fetch(:resumableCurrentChunkSize).to_i
+
+    # path to chunk file
+    chunk = Invite.chunk_path(identifier, filename, chunk_number)
+    # upload chunk
+    HandleChunkUpload.upload_chunk(chunk, file: file)
+
+    uploaded_file_size = chunk_number * chunk_size
+    # when all chunks are uploaded
+    if uploaded_file_size + current_chunk_size >= total_size
+      # handle bulk_invite processing in a background thread
+      Jobs.enqueue(:bulk_invite, filename: filename, identifier: identifier, chunks: chunk_number, current_user_id: current_user.id)
+    end
+
+    render nothing: true
+  end
+
 end