rootspring / getguildbase.go

package get_user_guild_base_info import ( "errors" "net/http" "time" "github.com/anti-raid/splashtail/animusmagic" "github.com/anti-raid/splashtail/types" "github.com/anti-raid/splashtail/utils" "github.com/anti-raid/splashtail/utils/mewext" "github.com/anti-raid/splashtail/webserver/state" "github.com/go-chi/chi/v5" "go.uber.org/zap" docs "github.com/infinitybotlist/eureka/doclib" "github.com/infinitybotlist/eureka/ratelimit" "github.com/infinitybotlist/eureka/uapi" ) func Docs() *docs.Doc { return &docs.Doc{ Summary: "Get User Guild Base Info", Description: "This endpoint will return basic user and guild information given their IDs", Resp: types.UserGuildBaseData{}, Params: []docs.Parameter{ { Name: "user_id", Description: "The ID of the user to get information about", In: "path", Required: true, Schema: docs.IdSchema, }, { Name: "guild_id", Description: "Whether to refresh the user's guilds from discord", In: "path", Required: true, Schema: docs.IdSchema, }, }, } } func Route(d uapi.RouteData, r *http.Request) uapi.HttpResponse { limit, err := ratelimit.Ratelimit{ Expiry: 5 * time.Minute, MaxRequests: 5, Bucket: "get_user_guild_base_info", }.Limit(d.Context, r) if err != nil { state.Logger.Error("Error while ratelimiting", zap.Error(err), zap.String("bucket", "get_user_guild_base_info")) return uapi.DefaultResponse(http.StatusInternalServerError) } if limit.Exceeded { return uapi.HttpResponse{ Json: types.ApiError{ Message: "You are being ratelimited. Please try again in " + limit.TimeToReset.String(), }, Headers: limit.Headers(), Status: http.StatusTooManyRequests, } } guildId := chi.URLParam(r, "guild_id") userId := chi.URLParam(r, "user_id") if guildId == "" || userId == "" { return uapi.DefaultResponse(http.StatusBadRequest) } // Fetch from animus magic clusterId, err := mewext.GetClusterIDFromGuildID(guildId, state.MewldInstanceList.Map, int(state.MewldInstanceList.ShardCount)) if err != nil { state.Logger.Error("Error getting cluster ID", zap.Error(err)) return uapi.HttpResponse{ Status: http.StatusInternalServerError, Json: types.ApiError{ Message: "Error getting cluster ID: " + err.Error(), }, } } for _, v := range state.MewldInstanceList.Instances { if v.ClusterID == clusterId { if !v.Active || v.CurrentlyKilling || len(v.ClusterHealth) == 0 { return uapi.HttpResponse{ Status: http.StatusInternalServerError, Json: types.ApiError{ Message: "Cluster is not healthy", }, } } } } resps, err := state.AnimusMagicClient.Request(d.Context, state.Rueidis, &animusmagic.RequestData{ ClusterID: utils.Pointer(uint16(clusterId)), Message: &animusmagic.AnimusMessage{ GetBaseGuildAndUserInfo: &struct { GuildID string `json:"guild_id"` UserID string `json:"user_id"` }{ GuildID: guildId, UserID: userId, }, }, }) if err != nil { state.Logger.Error("Error sending request to animus magic", zap.Error(err)) if errors.Is(err, animusmagic.ErrOpError) && len(resps) > 0 { return uapi.HttpResponse{ Status: http.StatusInternalServerError, Json: resps[0].Error, } } return uapi.HttpResponse{ Status: http.StatusInternalServerError, Json: types.ApiError{ Message: "Error sending request to animus magic: " + err.Error(), }, } } if len(resps) != 1 { return uapi.HttpResponse{ Status: http.StatusInternalServerError, Json: types.ApiError{ Message: "Unexpected number of responses", }, } } resp := resps[0] return uapi.HttpResponse{ Json: resp.Resp.GetBaseGuildAndUserInfo, } }