Compare commits
8 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
51d20dba37 | ||
|
|
2a400f4ee4 | ||
|
|
d50691776b | ||
|
|
14968bfd4c | ||
|
|
f8e7b01a28 | ||
|
|
332a84aa57 | ||
|
|
e9fdea664f | ||
|
|
7704fa9ecb |
@@ -16,6 +16,11 @@ import (
|
||||
)
|
||||
|
||||
func TestDatabase(t *testing.T) {
|
||||
// set enviroment variables
|
||||
os.Setenv("ACCESS_SECRET", "12345678910111213141516171819202")
|
||||
os.Setenv("REFRESH_SECRET", "9998979695949392919089888786858")
|
||||
os.Setenv("DOMAIN", "localhost")
|
||||
|
||||
dbName := "user.db"
|
||||
if _, err := os.Stat(dbName); err == nil {
|
||||
t.Log("remove user.db to start test with empty database")
|
||||
@@ -154,6 +159,10 @@ func TestDatabase(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestLoginAndAuthorization(t *testing.T) {
|
||||
os.Setenv("ACCESS_SECRET", "12345678910111213141516171819202")
|
||||
os.Setenv("REFRESH_SECRET", "9998979695949392919089888786858")
|
||||
os.Setenv("DOMAIN", "localhost")
|
||||
|
||||
gin.SetMode(gin.TestMode)
|
||||
|
||||
// Setup your AccessHandler and router
|
||||
|
||||
@@ -3,6 +3,7 @@ package handlers
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"gitea.tecamino.com/paadi/access-handler/internal/utils"
|
||||
@@ -16,12 +17,12 @@ import (
|
||||
// 🔐 AUTHENTICATION CONSTANTS
|
||||
// -----------------------------
|
||||
|
||||
// JWT secrets (replace "*" with strong random values in production!)
|
||||
var ACCESS_SECRET = []byte("ShFRprALcXjlosJ2hFCnGYGG3Ce2uRx6")
|
||||
var REFRESH_SECRET = []byte("pQIjuX6g6Tzf0FEfdScxttT3hlL9NFaa")
|
||||
// JWT secrets
|
||||
var ACCESS_SECRET = []byte(os.Getenv("ACCESS_SECRET"))
|
||||
var REFRESH_SECRET = []byte(os.Getenv("REFRESH_SECRET"))
|
||||
|
||||
// DOMAIN defines where cookies are valid. Change this in production.
|
||||
var DOMAIN = "localhost"
|
||||
var DOMAIN = os.Getenv("DOMAIN")
|
||||
|
||||
// ACCESS_TOKEN_TIME defines how long access tokens are valid.
|
||||
var ACCESS_TOKEN_TIME = 15 * time.Minute
|
||||
@@ -52,39 +53,18 @@ func (aH *AccessHandler) Login(c *gin.Context) {
|
||||
}
|
||||
|
||||
// Fetch user record from DB
|
||||
var dbRecord []models.User
|
||||
err := aH.dbHandler.GetByKey(&dbRecord, "user_name", user.Name, false)
|
||||
if err != nil {
|
||||
aH.logger.Error("Login", err)
|
||||
c.JSON(http.StatusBadRequest, models.NewJsonErrorResponse(err))
|
||||
return
|
||||
}
|
||||
|
||||
if len(dbRecord) == 0 {
|
||||
aH.logger.Error("Login", "no user "+user.Name+" found")
|
||||
c.JSON(http.StatusUnauthorized, models.NewJsonMessageResponse("invalid credentials"))
|
||||
return
|
||||
}
|
||||
|
||||
if len(dbRecord) > 1 {
|
||||
aH.logger.Error("Login", "more than one record found")
|
||||
c.JSON(http.StatusInternalServerError, models.NewJsonMessageResponse("internal error"))
|
||||
return
|
||||
}
|
||||
|
||||
if !dbRecord[0].ExpirationIsValid() {
|
||||
aH.logger.Error("Login", fmt.Sprintf("user %s is expired", user.Name))
|
||||
c.JSON(http.StatusUnauthorized, models.NewJsonMessageResponse("user "+user.Name+" is expired"))
|
||||
dbUser, hasError := aH.getUserFromDB(c, user.Name)
|
||||
if hasError {
|
||||
return
|
||||
}
|
||||
|
||||
// Check password
|
||||
if !utils.CheckPassword(user.Password, dbRecord[0].Password) {
|
||||
if !utils.CheckPassword(user.Password, dbUser.Password) {
|
||||
aH.logger.Error("Login", "invalid password")
|
||||
c.JSON(http.StatusUnauthorized, models.NewJsonMessageResponse("invalid credentials"))
|
||||
return
|
||||
}
|
||||
user = dbRecord[0]
|
||||
user = dbUser
|
||||
|
||||
// -----------------------------
|
||||
// 🔑 TOKEN CREATION
|
||||
@@ -101,7 +81,6 @@ func (aH *AccessHandler) Login(c *gin.Context) {
|
||||
"role": user.Role,
|
||||
"type": "access",
|
||||
"exp": accessTokenExp.Unix(),
|
||||
"userExpiration": user.Expiration,
|
||||
})
|
||||
|
||||
// Create refresh token
|
||||
@@ -111,7 +90,6 @@ func (aH *AccessHandler) Login(c *gin.Context) {
|
||||
"role": user.Role,
|
||||
"type": "refresh",
|
||||
"exp": refreshTokenExp.Unix(),
|
||||
"userExpiration": user.GetExpiration(),
|
||||
})
|
||||
|
||||
// Sign tokens
|
||||
@@ -187,12 +165,9 @@ func (aH *AccessHandler) Refresh(c *gin.Context) {
|
||||
|
||||
username := claims["username"].(string)
|
||||
id := int(claims["id"].(float64))
|
||||
role := claims["role"].(string)
|
||||
|
||||
if !expirationDateValid(claims["userExpiration"].(string)) {
|
||||
aH.Logout(c)
|
||||
aH.logger.Error("Refresh", fmt.Sprintf("user %s is expired", username))
|
||||
c.JSON(http.StatusUnauthorized, models.NewJsonMessageResponse("user "+username+" is expired"))
|
||||
user, hasError := aH.getUserFromDB(c, username)
|
||||
if hasError {
|
||||
return
|
||||
}
|
||||
|
||||
@@ -202,7 +177,7 @@ func (aH *AccessHandler) Refresh(c *gin.Context) {
|
||||
newAccess := jwt.NewWithClaims(jwt.SigningMethodHS256, jwt.MapClaims{
|
||||
"id": id,
|
||||
"username": username,
|
||||
"role": role,
|
||||
"role": user.Role,
|
||||
"exp": accessExp.Unix(),
|
||||
})
|
||||
accessString, _ := newAccess.SignedString(ACCESS_SECRET)
|
||||
@@ -262,15 +237,33 @@ func (aH *AccessHandler) Logout(c *gin.Context) {
|
||||
c.JSON(http.StatusOK, gin.H{"message": "logged out"})
|
||||
}
|
||||
|
||||
func expirationDateValid(expiration string) bool {
|
||||
if expiration == "" {
|
||||
return true // No expiration = always valid
|
||||
}
|
||||
|
||||
t, err := time.Parse(time.RFC3339, expiration)
|
||||
func (aH *AccessHandler) getUserFromDB(c *gin.Context, userName string) (user models.User, hasError bool) {
|
||||
hasError = true
|
||||
// Fetch user record from DB
|
||||
var dbRecord []models.User
|
||||
err := aH.dbHandler.GetByKey(&dbRecord, "user_name", userName, false)
|
||||
if err != nil {
|
||||
return false // Invalid date format
|
||||
aH.logger.Error("Login", err)
|
||||
c.JSON(http.StatusBadRequest, models.NewJsonErrorResponse(err))
|
||||
return
|
||||
}
|
||||
|
||||
return time.Now().Before(t)
|
||||
if len(dbRecord) == 0 {
|
||||
aH.logger.Error("Login", "no user "+userName+" found")
|
||||
c.JSON(http.StatusUnauthorized, models.NewJsonMessageResponse("invalid credentials"))
|
||||
return
|
||||
}
|
||||
|
||||
if len(dbRecord) > 1 {
|
||||
aH.logger.Error("Login", "more than one record found")
|
||||
c.JSON(http.StatusInternalServerError, models.NewJsonMessageResponse("internal error"))
|
||||
return
|
||||
}
|
||||
|
||||
if !dbRecord[0].ExpirationIsValid() {
|
||||
aH.logger.Error("Login", fmt.Sprintf("user %s is expired", userName))
|
||||
c.JSON(http.StatusUnauthorized, models.NewJsonMessageResponse("user "+userName+" is expired"))
|
||||
return
|
||||
}
|
||||
return dbRecord[0], false
|
||||
}
|
||||
|
||||
@@ -12,7 +12,7 @@ func (r *Permissions) DefaultPermissions() {
|
||||
*r = append(*r,
|
||||
Permission{Name: "settings", Permission: 7},
|
||||
Permission{Name: "userSettings", Permission: 7},
|
||||
Permission{Name: "members", Permission: 7},
|
||||
Permission{Name: "members", Permission: 31},
|
||||
Permission{Name: "events", Permission: 7},
|
||||
Permission{Name: "responsible", Permission: 7},
|
||||
)
|
||||
|
||||
@@ -12,6 +12,7 @@ type Settings struct {
|
||||
SecondaryColor string `json:"secondaryColor,omitempty"`
|
||||
SecondaryColorText string `json:"secondaryColorText,omitempty"`
|
||||
Icon string `json:"icon,omitempty"`
|
||||
AppName string `json:"appName,omitempty"`
|
||||
DatabaseName string `json:"databaseName,omitempty"`
|
||||
DatabaseToken string `json:"databaseToken,omitempty"`
|
||||
}
|
||||
|
||||
@@ -1,6 +1,8 @@
|
||||
package models
|
||||
|
||||
import "time"
|
||||
import (
|
||||
"time"
|
||||
)
|
||||
|
||||
type User struct {
|
||||
Id uint `gorm:"primaryKey" json:"id"`
|
||||
@@ -8,7 +10,7 @@ type User struct {
|
||||
Email string `gorm:"column:email" json:"email"`
|
||||
Role string `gorm:"column:role" json:"role,omitempty"`
|
||||
Password string `gorm:"column:password" json:"password"`
|
||||
Expiration *time.Time `gorm:"column:expiration" json:"expiration,omitempty"`
|
||||
Expiration string `gorm:"column:expiration" json:"expiration,omitempty"`
|
||||
Settings Settings `gorm:"type:json" json:"settings"`
|
||||
}
|
||||
|
||||
@@ -17,15 +19,10 @@ func (u *User) IsValid() bool {
|
||||
}
|
||||
|
||||
func (u *User) ExpirationIsValid() bool {
|
||||
if u.Expiration == nil {
|
||||
if u.Expiration == "" || u.Expiration == "never" {
|
||||
return true
|
||||
}
|
||||
return time.Now().Before(*u.Expiration)
|
||||
}
|
||||
|
||||
func (u *User) GetExpiration() string {
|
||||
if u.Expiration == nil {
|
||||
return ""
|
||||
}
|
||||
return u.Expiration.Format(time.RFC3339)
|
||||
loc := time.Now().Location()
|
||||
parsedTime, _ := time.ParseInLocation("2006-01-02 15:04:05", u.Expiration, loc)
|
||||
return parsedTime.After(time.Now())
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user