Complete Lesson 31

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

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

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>