Move the pending address from the node onto the draft

1 files changed, 33 insertions, 0 deletions

diff --git a/lib/tasks/pages.rake b/lib/tasks/pages.rake
new file mode 100644
index 00000000..0cfbc5dc
--- /dev/null
+++ b/lib/tasks/pages.rake
@@ -0,0 +1,33 @@
+namespace :pages do
+  desc "Backfill pages.slug and pages.parent_node_id from each page's " \
+       "node. Historical accuracy is not attempted. Every revision gets " \
+       "the node's current address, which is right for head and draft and " \
+       "harmless for older revisions, and avoids nil checks everywhere. " \
+       "Dry run unless WRITE=1."
+  task :backfill_address => :environment do
+    write = ENV["WRITE"] == "1"
+    puts "DRY RUN -- nothing written. Re-run with WRITE=1." unless write
+
+    touched = 0
+    Node.find_each do |node|
+      scope = node.pages.where("slug IS DISTINCT FROM :s OR parent_node_id IS DISTINCT FROM :p",
+                               :s => node.slug, :p => node.parent_id)
+      count = scope.count
+      next if count.zero?
+
+      scope.update_all(:slug => node.slug, :parent_node_id => node.parent_id) if write
+      touched += count
+    end
+
+    # Autosaves carry no node_id -- has_many :pages does not cover them.
+    Node.where.not(:autosave_id => nil).includes(:autosave).find_each do |node|
+      a = node.autosave
+      next if a.slug == node.slug && a.parent_node_id == node.parent_id
+
+      a.update_columns(:slug => node.slug, :parent_node_id => node.parent_id) if write
+      touched += 1
+    end
+
+    puts "#{write ? "updated" : "would update"} #{touched} pages"
+  end
+end