complete signup, login, logout workflow

2 files changed, 18 insertions, 0 deletions

diff --git a/app/controllers/application_controller.rb b/app/controllers/application_controller.rb
index e8065d9..693bd86 100644
--- a/app/controllers/application_controller.rb
+++ b/app/controllers/application_controller.rb
@@ -1,3 +1,10 @@
 class ApplicationController < ActionController::Base
   protect_from_forgery
+
+  helper_method :current_user
+
+  private
+  def current_user
+    @current_user ||= User.find(session[:user_id]) if session[:user_id]
+  end
 end
diff --git a/app/views/home/index.html.haml b/app/views/home/index.html.haml
index 11f18de..68e39bb 100644
--- a/app/views/home/index.html.haml
+++ b/app/views/home/index.html.haml
@@ -2,3 +2,14 @@
 
 Try to fetch a 
 = link_to "cert", cert_path
+
+-if current_user
+  %p 
+    logged in as
+    = current_user.login
+    = link_to "logout", logout_path
+-else
+  %p
+    = link_to "login", login_path
+    or 
+    = link_to "signup", signup_path