147 lines
3.5 KiB
Go
147 lines
3.5 KiB
Go
package character
|
|
|
|
import (
|
|
"bamort/database"
|
|
"bamort/models"
|
|
"bamort/user"
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// GetCharacterShares returns the list of users a character is shared with
|
|
func GetCharacterShares(c *gin.Context) {
|
|
charID := c.Param("id")
|
|
|
|
var character models.Char
|
|
if err := character.FirstID(charID); err != nil {
|
|
respondWithError(c, http.StatusNotFound, "Character not found")
|
|
return
|
|
}
|
|
|
|
// Check ownership
|
|
if !checkCharacterOwnership(c, &character) {
|
|
return
|
|
}
|
|
|
|
var shares []models.CharShare
|
|
if err := database.DB.Where("character_id = ?", character.ID).Find(&shares).Error; err != nil {
|
|
respondWithError(c, http.StatusInternalServerError, "Failed to retrieve shares")
|
|
return
|
|
}
|
|
|
|
// Get user details for each share
|
|
type ShareWithUser struct {
|
|
models.CharShare
|
|
Username string `json:"username"`
|
|
Email string `json:"email"`
|
|
}
|
|
|
|
var sharesWithUsers []ShareWithUser
|
|
for _, share := range shares {
|
|
var u user.User
|
|
if err := u.FirstId(share.UserID); err == nil {
|
|
sharesWithUsers = append(sharesWithUsers, ShareWithUser{
|
|
CharShare: share,
|
|
Username: u.Username,
|
|
Email: u.Email,
|
|
})
|
|
}
|
|
}
|
|
|
|
c.JSON(http.StatusOK, sharesWithUsers)
|
|
}
|
|
|
|
// UpdateCharacterShares updates the list of users a character is shared with
|
|
func UpdateCharacterShares(c *gin.Context) {
|
|
charID := c.Param("id")
|
|
|
|
var character models.Char
|
|
if err := character.FirstID(charID); err != nil {
|
|
respondWithError(c, http.StatusNotFound, "Character not found")
|
|
return
|
|
}
|
|
|
|
// Check ownership
|
|
if !checkCharacterOwnership(c, &character) {
|
|
return
|
|
}
|
|
|
|
type UpdateSharesRequest struct {
|
|
UserIDs []uint `json:"user_ids" binding:"required"`
|
|
}
|
|
|
|
var request UpdateSharesRequest
|
|
if err := c.ShouldBindJSON(&request); err != nil {
|
|
respondWithError(c, http.StatusBadRequest, err.Error())
|
|
return
|
|
}
|
|
|
|
// Delete existing shares
|
|
if err := database.DB.Where("character_id = ?", character.ID).Delete(&models.CharShare{}).Error; err != nil {
|
|
respondWithError(c, http.StatusInternalServerError, "Failed to delete existing shares")
|
|
return
|
|
}
|
|
|
|
// Create new shares
|
|
for _, userID := range request.UserIDs {
|
|
// Don't share with yourself
|
|
if userID == character.UserID {
|
|
continue
|
|
}
|
|
|
|
share := models.CharShare{
|
|
CharacterID: character.ID,
|
|
UserID: userID,
|
|
Permission: "read",
|
|
}
|
|
|
|
if err := database.DB.Create(&share).Error; err != nil {
|
|
respondWithError(c, http.StatusInternalServerError, "Failed to create share")
|
|
return
|
|
}
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"message": "Shares updated successfully"})
|
|
}
|
|
|
|
// GetAvailableUsersForSharing returns a list of users (excluding the owner)
|
|
func GetAvailableUsersForSharing(c *gin.Context) {
|
|
charID := c.Param("id")
|
|
|
|
var character models.Char
|
|
if err := character.FirstID(charID); err != nil {
|
|
respondWithError(c, http.StatusNotFound, "Character not found")
|
|
return
|
|
}
|
|
|
|
// Check ownership
|
|
if !checkCharacterOwnership(c, &character) {
|
|
return
|
|
}
|
|
|
|
var users []user.User
|
|
if err := database.DB.Where("user_id != ?", character.UserID).Find(&users).Error; err != nil {
|
|
respondWithError(c, http.StatusInternalServerError, "Failed to retrieve users")
|
|
return
|
|
}
|
|
|
|
// Remove sensitive data
|
|
type UserInfo struct {
|
|
UserID uint `json:"user_id"`
|
|
Username string `json:"username"`
|
|
Email string `json:"email"`
|
|
}
|
|
|
|
var userInfos []UserInfo
|
|
for _, u := range users {
|
|
userInfos = append(userInfos, UserInfo{
|
|
UserID: u.UserID,
|
|
Username: u.Username,
|
|
Email: u.Email,
|
|
})
|
|
}
|
|
|
|
c.JSON(http.StatusOK, userInfos)
|
|
}
|