Switch back and forth on same page (not thoroughly tested)

refinery · May 10, 2010 · bc8a42f · bc8a42f
1 parent d6a644a
commit bc8a42f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/vendor/plugins/refinery/app/views/shared/_site_bar.html.erb b/vendor/plugins/refinery/app/views/shared/_site_bar.html.erb
@@ -2,9 +2,9 @@
   <div id='site_bar_content' class='clearfix'>
     <div id='editor_switch'>
       <% if admin? %>
-        <%= link_to "Switch to live view", "/" %>
+        <%= link_to "Switch to live view", begin request.request_uri.to_s.match(/\/admin(.*)\/edit/)[1] rescue "/" end %>
       <% else %>
-        <%= link_to "Switch to editor view", admin_root_url %>
+        <%= link_to "Switch to editor view", (request.request_uri.to_s == "/") ? admin_root_url : "/admin" + request.request_uri.to_s + "/edit" %>
       <% end %>
     </div>
     <div id='site_bar_branding'>
@@ -15,4 +15,4 @@
     </div>
   </div>
 </div>
-<%= stylesheet_link_tag('refinery/site_bar') %>
+<%= stylesheet_link_tag('refinery/site_bar') %>