1 files changed, 43 insertions, 0 deletions
diff --git a/lib/tasks/referential.rake b/lib/tasks/referential.rake
index 7bab6e040..d53157312 100644
--- a/lib/tasks/referential.rake
+++ b/lib/tasks/referential.rake
@@ -67,4 +67,47 @@ namespace :referential do
       referential.update(ready: true)
     end
   end
+
+  def update_checksums_for_referential referential
+    thing = %w(\\ | / —)
+    Referential.force_register_models_with_checksum if Rails.env.development? && Referential.models_with_checksum.empty?
+    puts "\n \e[33m***\e[0m Referential #{referential.name}"
+    referential.switch do
+      Referential.models_with_checksum.each do |klass|
+        i = 0
+        j = 0
+        prev_size = 1
+        head =  "Updating checksums for #{klass.name}: "
+        print head
+        print "⎯"*(80-head.size)
+        print "  "
+        count = klass.count
+        klass.find_each do |o|
+          o.update_checksum!
+          if j%10 == 0
+            out = "#{"\b"*prev_size}\e[33m#{thing[i]}\e[0m (#{j}/#{count})"
+            prev_size = out.size - prev_size - 9
+            print out
+            i = (i+1) % thing.size
+          end
+          j += 1
+        end
+        print "#{"\b"*prev_size}\e[32m✓\e[0m (#{count}/#{count})\n"
+      end
+    end
+  end
+
+  desc 'Update all the checksums in the given referential'
+  task :update_checksums_in_referential, [:slug] => :environment do |t, args|
+    referential = Referential.find_by_slug(args[:slug])
+    update_checksums_for_referential referential
+  end
+
+  desc 'Update all the checksums in the given organisation'
+  task :update_checksums_in_organisation, [:organisation_id] => :environment do |t, args|
+    thing = %w(\\ | / —)
+    Organisation.find(args[:organisation_id]).referentials.find_each do |referential|
+      update_checksums_for_referential referential
+    end
+  end
 end