introducing the restore revision functionality

2 files changed, 18 insertions, 2 deletions

diff --git a/app/controllers/revisions_controller.rb b/app/controllers/revisions_controller.rb
index 7ae22f8..e2c0432 100644
--- a/app/controllers/revisions_controller.rb
+++ b/app/controllers/revisions_controller.rb
@@ -32,4 +32,10 @@ class RevisionsController < ApplicationController
     @node = Node.find(params[:id])
   end
 
+  def restore
+    page = Page.find(params[:id])
+    page.node.restore_revision! page.revision
+    flash[:notice] = "Revision #{page.revision} restored"
+    redirect_to :back
+  end
 end
diff --git a/app/views/revisions/show.html.erb b/app/views/revisions/show.html.erb
index 2a4a9a1..ebda18d 100644
--- a/app/views/revisions/show.html.erb
+++ b/app/views/revisions/show.html.erb
@@ -23,11 +23,21 @@
     <td class="title"><%= page.title %></td>
     <td class="user"><%= page.user.try(:login)  %></td>
     <td class="date"><%= page.updated_at %></td>
-    <td class="actions"><%= link_to 'show', node_path(page.node) %></td>
+    <td>
+      <%= link_to 'show', node_path(page.node) %>
+    </td>
+    <td>
+      <%= link_to(
+            'restore', 
+            restore_revision_path(page), 
+            :method => :put, 
+            :confirm => "Restore this revision?"
+          ) %>
+    </td>
   </tr>
 <% end %>
   <tr class="no_hover">
-    <td colspan="7" class="right"><%= submit_tag 'Diff revisions' %></td>
+    <td colspan="8" class="right"><%= submit_tag 'Diff revisions' %></td>
   </tr>
 </table>