TerraOceanPlugin/GoBot/trust.go

77 lines
1.4 KiB
Go

2019-10-10 13:59:00 +07:00
package main
import (
"errors"
"github.com/bwmarrin/discordgo"
2019-10-12 08:07:10 +07:00
"TerraOceanBot/db"
)
2019-10-11 12:25:09 +07:00
var trustCache map[string]float64
var errNotMember = errors.New("trust: only members have trust value")
func getTrust(s *discordgo.Session, discordID string) (float64, error) {
2019-10-11 12:25:09 +07:00
if v, ok := trustCache[discordID]; ok {
return v, nil
}
member, err := s.GuildMember(guildID, discordID)
if err != nil {
return 0.0, err
}
isMember := false
for _, role := range member.Roles {
if role == memberRoleID {
isMember = true
}
}
if !isMember {
return 0.0, errNotMember
}
2019-10-12 08:07:10 +07:00
votes, err := db.GetTrustVote(discordID)
2019-10-11 12:25:09 +07:00
if err != nil {
return 0.0, err
}
votes = append(votes, 3)
var total float64
for _, v := range votes {
total += float64(v)
}
return total / float64(len(votes)), nil
}
func getTotalTrust(s *discordgo.Session) (float64, error) {
members, err := getAllMembers(s)
if err != nil {
return 0.0, err
}
var total float64
for _, member := range members {
isMember := false
for _, role := range member.Roles {
if role == memberRoleID {
isMember = true
}
}
if !isMember {
continue
}
trust, err := getTrust(s, member.User.ID)
2019-10-11 12:25:09 +07:00
if err != nil {
return 0.0, err
}
total += trust
}
return total, nil
2019-10-10 13:59:00 +07:00
}
func updateTrust(sourceID, targetID string, voteValue int) error {
2019-10-12 08:07:10 +07:00
err := db.UpdateTrust(sourceID, targetID, voteValue, false)
if err != nil {
return err
}
delete(trustCache, targetID)
return nil
}