logged in status

Author: avoidik

cmd/web/helpers.go

@@ -0,0 +1,12 @@
+package main
+
+import "net/http"
+
+func (app *App) LoggedIn(r *http.Request) (bool, error) {
+	session := app.sessions.Load(r)
+	loggedIn, err := session.Exists("currentUserId")
+	if err != nil {
+		return false, err
+	}
+	return loggedIn, nil
+}

cmd/web/views.go

@@ -14,6 +14,7 @@ type HtmlData struct {
 	Form     interface{}
 	Path     string
 	Flash    string
+	LoggedIn bool
 	Snippet  *models.Snippet
 	Snippets []*models.Snippet
 }
@@ -29,6 +30,13 @@ func (app *App) RenderHtml(w http.ResponseWriter, r *http.Request, page string,
 
 	data.Path = r.URL.Path
 
+	var err error
+	data.LoggedIn, err = app.LoggedIn(r)
+	if err != nil {
+		app.ServerError(w, err)
+		return
+	}
+
 	files := []string{
 		filepath.Join(app.htmlDir, "base.html"),
 		filepath.Join(app.htmlDir, page),

ui/html/base.html

@@ -15,9 +15,21 @@ <h1><a href="/">Snippetbox</a></h1>
             <a href="/" {{if eq .Path "/"}}class="live"{{end}}>
                 Home
             </a>
+            {{if .LoggedIn}}
             <a href="/snippet/new" {{if eq .Path "/snippet/new"}}class="live"{{end}}>
                 New Snippet
             </a>
+            <form action="/user/logout" method="POST">
+                <button>Logout</button>
+            </form>
+            {{else}}
+            <a href="/user/login" {{if eq .Path "/user/login"}}class="live"{{end}}>
+                Login
+            </a>
+            <a href="/user/signup" {{if eq .Path "/user/signup"}}class="live"{{end}}>
+                Signup
+            </a>
+            {{end}}
         </nav>
         <section>
             {{- template "page-body" .}}