From 8f470ae322a2cd9d4e0842766ec69646001f8ae2 Mon Sep 17 00:00:00 2001
From: Simon Quigley <simon@tsimonq2.net>
Date: Wed, 27 Sep 2023 06:31:33 -0500
Subject: [PATCH] Complete Lesson 31

---
 app/controllers/users_controller.rb | 1 +
 app/helpers/application_helper.rb   | 3 +++
 app/views/layouts/_header.html.erb  | 6 ++++++
 3 files changed, 10 insertions(+)

diff --git a/app/controllers/users_controller.rb b/app/controllers/users_controller.rb
index b5ffab5..b107d28 100644
--- a/app/controllers/users_controller.rb
+++ b/app/controllers/users_controller.rb
@@ -11,6 +11,7 @@ class UsersController < ApplicationController
   def create
     @user = User.new(user_params)
     if @user.save
+      session[:user_id] = @user.id
       redirect_to @user, notice: "Thanks for signing up!" 
     else
       render :new, status: :unprocessable_entity
diff --git a/app/helpers/application_helper.rb b/app/helpers/application_helper.rb
index de6be79..3778320 100644
--- a/app/helpers/application_helper.rb
+++ b/app/helpers/application_helper.rb
@@ -1,2 +1,5 @@
 module ApplicationHelper
+  def current_user
+    @current_user ||= User.find(session[:user_id]) if session[:user_id]
+  end
 end
diff --git a/app/views/layouts/_header.html.erb b/app/views/layouts/_header.html.erb
index 6b9ca8a..f6bdbcb 100644
--- a/app/views/layouts/_header.html.erb
+++ b/app/views/layouts/_header.html.erb
@@ -13,12 +13,18 @@
         </li>
       </ul>
       <ul class="navbar-nav">
+        <% if current_user %>
+        <li class="nav-item">
+          <%= link_to current_user.name, current_user %>
+        </li>
+        <% else %>
         <li class="nav-item">
           <%= link_to "Sign In", new_session_path, class: "nav-link" %>
         </li>
         <li class="nav-item">
           <%= link_to "Sign Up", signup_path, class: "nav-link" %>
         </li>
+        <% end %>
       </ul>
     </div>
   </nav>