Add Site

internal/entity/auth.go

@@ -26,6 +26,7 @@ type UserDB struct {
 	RoleName      string                `gorm:"column:role_name" json:"role_name"`
 	PartnerID     *int64                `gorm:"column:partner_id" json:"partner_id"`
 	SiteID        *int64                `gorm:"column:site_id" json:"site_id"`
+	SiteName      string                `gorm:"column:name" json:"site_name"`
 	PartnerName   string                `gorm:"column:partner_name" json:"partner_name"`
 	PartnerStatus string                `gorm:"column:partner_status" json:"partner_status"`
 	CreatedAt     time.Time             `gorm:"column:created_at" json:"created_at"`
@@ -52,6 +53,7 @@ func (u *UserDB) ToUser() *User {
 		PartnerID:   u.PartnerID,
 		PartnerName: u.PartnerName,
 		SiteID:      u.SiteID,
+		SiteName:    u.SiteName,
 	}
 
 	return userEntity
@@ -88,6 +90,8 @@ func (u *UserDB) ToUserAuthenticate(signedToken string) *AuthenticateUser {
 		PartnerID:     u.PartnerID,
 		PartnerName:   u.PartnerName,
 		PartnerStatus: u.PartnerStatus,
+		SiteID:        u.SiteID,
+		SiteName:      u.SiteName,
 	}
 }
 

internal/entity/jwt.go

@@ -9,6 +9,7 @@ type JWTAuthClaims struct {
 	Role      int    `json:"role"`
 	PartnerID int64  `json:"partner_id"`
 	SiteID    int64  `json:"site_id"`
+	SiteName  string `json:"site_name"`
 	jwt.StandardClaims
 }
 

internal/entity/user.go

@@ -22,6 +22,7 @@ type User struct {
 	RoleName    string
 	PartnerID   *int64
 	SiteID      *int64
+	SiteName    string
 	PartnerName string
 }
 
@@ -33,6 +34,8 @@ type AuthenticateUser struct {
 	PartnerID     *int64
 	PartnerName   string
 	PartnerStatus string
+	SiteID        *int64
+	SiteName      string
 }
 
 type UserRoleDB struct {