mirror-nourybot/pkg/api/ivr/whois.go
2021-10-29 22:26:37 +02:00

78 lines
1.8 KiB
Go

package ivr
import (
"encoding/json"
"fmt"
"io/ioutil"
"net/http"
"strings"
log "github.com/sirupsen/logrus"
)
// https://api.ivr.fi
type whoisApiResponse struct {
Id string `json:"id"`
DisplayName string `json:"displayName"`
Login string `json:"login"`
Bio string `json:"bio"`
ChatColor string `json:"chatColor"`
Partner bool `json:"partner"`
// Affiliate bool `json:"affiliate"`
Bot bool `json:"bot"`
CreatedAt string `json:"createdAt"`
Roles roles
Error string `json:"error"`
}
type roles struct {
IsAffiliate bool `json:"isAffiliate"`
IsPartner bool `json:"isPartner"`
IsSiteAdmin bool `json:"isSiteAdmin"`
IsStaff bool `json:"isStaff"`
}
// Userid returns the userID of a given user
func Whois(username string) string {
baseUrl := "https://api.ivr.fi/twitch/resolve"
resp, err := http.Get(fmt.Sprintf("%s/%s", baseUrl, username))
if err != nil {
log.Error(err)
}
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
log.Error(err)
}
var responseObject whoisApiResponse
json.Unmarshal(body, &responseObject)
// time string format 2011-05-19T00:28:28.310449Z
// discard everything after T
created := strings.Split(responseObject.CreatedAt, "T")
reply := fmt.Sprintf("User: %s, ID: %s, Created on: %s, Color: %s, Affiliate: %v, Partner: %v, Staff: %v, Admin: %v, Bot: %v, Bio: %v",
responseObject.DisplayName,
responseObject.Id,
created[0],
responseObject.ChatColor,
responseObject.Roles.IsAffiliate,
responseObject.Roles.IsPartner,
responseObject.Roles.IsStaff,
responseObject.Roles.IsSiteAdmin,
responseObject.Bot,
responseObject.Bio,
)
// User not found
if responseObject.Error != "" {
return fmt.Sprintf(responseObject.Error + " FeelsBadMan")
} else {
return reply
}
}