From c36ca15055439e98a0900b92175a9b92edeca0ef Mon Sep 17 00:00:00 2001
From: Kyle Drake <kyle@kyledrake.net>
Date: Mon, 27 Oct 2014 15:43:29 -0700
Subject: [PATCH] add space used migration

---
 Rakefile                                  | 16 ++++++++++++++++
 environment.rb                            |  2 +-
 migrations/049_add_space_used_to_sites.rb |  9 +++++++++
 views/_news.erb                           |  1 +
 4 files changed, 27 insertions(+), 1 deletion(-)
 create mode 100644 migrations/049_add_space_used_to_sites.rb

diff --git a/Rakefile b/Rakefile
index f3535212..fb4371b1 100644
--- a/Rakefile
+++ b/Rakefile
@@ -172,6 +172,22 @@ task :update_screenshots => [:environment] do
   sleep
 end
 
+desc 'prime_space_used'
+task :prime_space_used => [:environment] do
+  Site.select(:id,:username,:space_used).all.each do |s|
+    s.space_used += s.actual_space_used
+    s.save_changes validate: false
+  end
+end
+
+desc 'prime site_updated_at'
+task :prime_site_updated_at => [:environment] do
+  Site.select(:id,:username,:site_updated_at, :updated_at).all.each do |s|
+    s.site_updated_at = s.updated_at
+    s.save_changes validate: false
+  end
+end
+
 =begin
 desc 'Update screenshots'
 task :update_screenshots => [:environment] do
diff --git a/environment.rb b/environment.rb
index 41aca134..ab9f9611 100644
--- a/environment.rb
+++ b/environment.rb
@@ -89,7 +89,7 @@ Sequel.extension :migration
 Sequel::Model.plugin :validation_helpers
 Sequel::Model.plugin :force_encoding, 'UTF-8'
 Sequel::Model.plugin :defaults_setter
-Sequel::Model.plugin :timestamps, create: :created_at, update: :updated_at
+Sequel::Model.plugin :timestamps, create: :created_at, update: :DONT_UPDATE
 Sequel.default_timezone = 'UTC'
 Sequel::Migrator.apply DB, './migrations'
 
diff --git a/migrations/049_add_space_used_to_sites.rb b/migrations/049_add_space_used_to_sites.rb
new file mode 100644
index 00000000..fc3bdb0c
--- /dev/null
+++ b/migrations/049_add_space_used_to_sites.rb
@@ -0,0 +1,9 @@
+Sequel.migration do
+  up {
+    DB.add_column :sites, :space_used, :bigint, default: 0, index: true
+  }
+
+  down {
+    DB.drop_column :sites, :space_used
+  }
+end
\ No newline at end of file
diff --git a/views/_news.erb b/views/_news.erb
index ab64dbb9..391e63d7 100644
--- a/views/_news.erb
+++ b/views/_news.erb
@@ -50,6 +50,7 @@
           <a href="/site/<%= event_site.username %>" class="user" title="<%= event_site.title %>"><%= event_site.title.shorten(45) %></a> has been updated.
         <% end %>
         <span class="date">
+          <% binding.pry %>
           <a href="?event_id=<%= event.id %>"><%= event.created_at.ago %></a>
         </span>
       </div>