From 8c34d937b3d542d584bd20fc5d4964b3e9922ca9 Mon Sep 17 00:00:00 2001
From: Leonardo Mosquera <ldmosquera@gmail.com>
Date: Fri, 18 Nov 2022 19:04:09 +0000
Subject: [PATCH] Add support for suspension fields

---
 script/bulk_import/base.rb         |  2 ++
 script/bulk_import/generic_bulk.rb | 10 +++++++++-
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/script/bulk_import/base.rb b/script/bulk_import/base.rb
index 7537c87ec5..f92a6861c9 100644
--- a/script/bulk_import/base.rb
+++ b/script/bulk_import/base.rb
@@ -453,6 +453,8 @@ class BulkImport::Base
     user[:last_emailed_at] ||= NOW
     user[:created_at] ||= NOW
     user[:updated_at] ||= user[:created_at]
+    user[:suspended_at] ||= user[:suspended_at]
+    user[:suspended_till] ||= user[:suspended_till] || (200.years.from_now if user[:suspended_at].present?)
     user
   end
 
diff --git a/script/bulk_import/generic_bulk.rb b/script/bulk_import/generic_bulk.rb
index d81985e517..8265ac604f 100644
--- a/script/bulk_import/generic_bulk.rb
+++ b/script/bulk_import/generic_bulk.rb
@@ -83,6 +83,12 @@ class BulkImport::Generic < BulkImport::Base
     create_users(users) do |row|
       sso_record = JSON.parse(row["sso_record"]) if row["sso_record"].present?
 
+      if row["suspension"].present?
+        suspension = JSON.parse(row["suspension"])
+        suspended_at = suspension['suspended_at']
+        suspended_till = suspension['suspended_till']
+      end
+
       {
         imported_id: row["id"],
         username: row["username"],
@@ -91,7 +97,9 @@ class BulkImport::Generic < BulkImport::Base
         external_id: sso_record&.fetch("external_id"),
         created_at: to_datetime(row["created_at"]),
         admin: row["admin"],
-        moderator: row["moderator"]
+        moderator: row["moderator"],
+        suspended_at: suspended_at,
+        suspended_till: suspended_till,
       }
     end
   end