Add Site

internal/handlers/http/auth/auth.go

@@ -52,6 +52,7 @@ func (h *AuthHandler) AuthLogin(c *gin.Context) {
 	}
 
 	var partner *response.Partner
+	var site *response.SiteName
 
 	if authUser.RoleID != role.SuperAdmin {
 		partner = &response.Partner{
@@ -61,6 +62,13 @@ func (h *AuthHandler) AuthLogin(c *gin.Context) {
 		}
 	}
 
+	if authUser.RoleID == role.Casheer {
+		site = &response.SiteName{
+			ID:   authUser.SiteID,
+			Name: authUser.SiteName,
+		}
+	}
+
 	resp := response.LoginResponse{
 		Token:   authUser.Token,
 		Partner: partner,
@@ -69,6 +77,7 @@ func (h *AuthHandler) AuthLogin(c *gin.Context) {
 			ID:   int64(authUser.RoleID),
 			Role: authUser.RoleName,
 		},
+		Site: site,
 	}
 
 	c.JSON(http.StatusOK, response.BaseResponse{

internal/handlers/response/auth.go

@@ -1,10 +1,11 @@
 package response
 
 type LoginResponse struct {
-	Token   string   `json:"token"`
-	Name    string   `json:"name"`
-	Role    Role     `json:"role"`
-	Partner *Partner `json:"partner"`
+	Token   string    `json:"token"`
+	Name    string    `json:"name"`
+	Role    Role      `json:"role"`
+	Partner *Partner  `json:"partner"`
+	Site    *SiteName `json:"site,omitempty"`
 }
 
 type Role struct {

internal/handlers/response/site.go

@@ -19,6 +19,11 @@ type Site struct {
 	UpdatedAt        string `json:"updated_at"`
 }
 
+type SiteName struct {
+	ID   *int64 `json:"id"`
+	Name string `json:"name"`
+}
+
 type SiteList struct {
 	Sites  []Site `json:"sites"`
 	Total  int64  `json:"total"`