73 lines
1.9 KiB
Go
73 lines
1.9 KiB
Go
package api
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/spf13/viper"
|
|
"gorm.io/gorm"
|
|
"strings"
|
|
|
|
"git.solsynth.dev/hydrogen/passport/pkg/internal/database"
|
|
"git.solsynth.dev/hydrogen/passport/pkg/internal/models"
|
|
"git.solsynth.dev/hydrogen/passport/pkg/internal/services"
|
|
"github.com/gofiber/fiber/v2"
|
|
)
|
|
|
|
func getOtherUserinfo(c *fiber.Ctx) error {
|
|
alias := c.Params("alias")
|
|
|
|
var account models.Account
|
|
if err := database.C.
|
|
Where(&models.Account{Name: alias}).
|
|
Preload("Profile").
|
|
Preload("Badges", func(db *gorm.DB) *gorm.DB {
|
|
prefix := viper.GetString("database.prefix")
|
|
return db.Order(fmt.Sprintf("%sbadges.type DESC", prefix))
|
|
}).
|
|
First(&account).Error; err != nil {
|
|
return fiber.NewError(fiber.StatusBadRequest, err.Error())
|
|
}
|
|
|
|
groups, err := services.GetUserAccountGroup(account)
|
|
if err != nil {
|
|
return fiber.NewError(fiber.StatusInternalServerError, fmt.Sprintf("unable to get account groups: %v", err))
|
|
}
|
|
for _, group := range groups {
|
|
for k, v := range group.PermNodes {
|
|
if _, ok := account.PermNodes[k]; !ok {
|
|
account.PermNodes[k] = v
|
|
}
|
|
}
|
|
}
|
|
|
|
return c.JSON(account)
|
|
}
|
|
|
|
func getOtherUserinfoBatch(c *fiber.Ctx) error {
|
|
idFilter := c.Query("id")
|
|
nameFilter := c.Query("name")
|
|
idSet := strings.Split(idFilter, ",")
|
|
nameSet := strings.Split(nameFilter, ",")
|
|
if len(idSet) == 0 && len(nameSet) == 0 {
|
|
return fiber.NewError(fiber.StatusBadRequest, "query filter is required")
|
|
}
|
|
|
|
if len(idSet)+len(nameSet) > 100 {
|
|
return fiber.NewError(fiber.StatusBadRequest, "only support 100 users in a single batch")
|
|
}
|
|
|
|
tx := database.C.Model(&models.Account{}).Limit(100)
|
|
if len(idFilter) > 0 {
|
|
tx = tx.Where("id IN ?", idSet)
|
|
}
|
|
if len(nameFilter) > 0 {
|
|
tx = tx.Where("name IN ?", nameSet)
|
|
}
|
|
|
|
var accounts []models.Account
|
|
if err := tx.Find(&accounts).Error; err != nil {
|
|
return fiber.NewError(fiber.StatusBadRequest, err.Error())
|
|
}
|
|
|
|
return c.JSON(accounts)
|
|
}
|