Merge pull request #3 from ockibagusp/session-3-Is-types

User tables

Author: ockibagusp

README.md

@@ -26,11 +26,11 @@ file: golang-website-example.sql -> new database: golang-website-example_test
 
 ## User Tables
 
-| Username | Password |
-| --- | --- |
-| ockibagusp | user123 |
-| sugriwa | user123 |
-| subali | user123 |
+| Username | Password | is Admin |
+| --- | --- | --- |
+| admin | admin123 | yes |
+| sugriwa | user123 | no |
+| subali | user123 | no |
 
 
 ## Router
@@ -108,7 +108,6 @@ $ go test github.com/ockibagusp/golang-website-example/test -v
 
 ## TODO List
 - mock unit test
-- session: IsAdmin, IsUser and IsAuth
 - list pagination with next, previous, first and last
 - Mutex: BankAccount
 - too much

controllers/home_controller.go

@@ -1,10 +1,12 @@
 package controllers
 
 import (
+	"fmt"
 	"net/http"
 
 	"github.com/labstack/echo/v4"
 	"github.com/ockibagusp/golang-website-example/middleware"
+	"github.com/ockibagusp/golang-website-example/models"
 	log "github.com/sirupsen/logrus"
 )
 
@@ -15,7 +17,7 @@ import (
  * @method: GET
  * @route: /
  */
-func (Controller) Home(c echo.Context) error {
+func (controller Controller) Home(c echo.Context) error {
 	// Please note the the second parameter "home.html" is the template name and should
 	// be equal to one of the keys in the TemplateRegistry array defined in main.go
 	session, _ := middleware.GetAuth(c)
@@ -24,12 +26,27 @@ func (Controller) Home(c echo.Context) error {
 		"route":    c.Path(),
 	})
 	log.Info("START request method GET for home")
-	// ---
+
+	var message string
+	if session.Values["username"] != "" {
+		var user models.User
+		if err := controller.DB.Select("name").Where(
+			"username = ?", session.Values["username"],
+		).First(&user); err.Error != nil { // TODO: why?
+			log.Warnf(`session values "username" error: %v`, err.Error)
+		}
+
+		message = fmt.Sprintf("%v!", user.Name)
+		// or,
+		// session.Values["username"].(string) + "!"
+	}
+
 	log.Info("END request method GET for home: [+]success")
 	return c.Render(http.StatusOK, "home.html", echo.Map{
-		"name":    "Home",
-		"nav":     "home", // (?)
-		"session": session,
-		"msg":     "Ocki Bagus Pratama!",
+		"name":          "Home",
+		"nav":           "home", // (?)
+		"session":       session,
+		"flash_success": middleware.GetFlashSuccess(c),
+		"msg":           message,
 	})
 }

controllers/session_controller.go

@@ -52,7 +52,7 @@ func (controller *Controller) Login(c echo.Context) error {
 
 		var user models.User
 		// err := controller.DB.Select(...).Where(...).First(...).Error
-		if err := controller.DB.Select("username", "password").Where(
+		if err := controller.DB.Select("username", "password", "is_admin").Where(
 			"username = ?", passwordForm.Username,
 		).First(&user).Error; err != nil {
 			middleware.SetFlashError(c, err.Error())