TerraOceanPlugin/GoBot/discord/backend/trust.go

package backend

import (
	"errors"

	"github.com/bwmarrin/discordgo"

	"TerraOceanBot/db"
	"TerraOceanBot/discord/config"
)

var trustCache map[string]float64

var ErrNotMember = errors.New("trust: only members have trust value")

func GetTrust(s *discordgo.Session, discordID string) (float64, error) {
	if v, ok := trustCache[discordID]; ok {
		return v, nil
	}
	member, err := s.GuildMember(config.GuildID, discordID)
	if err != nil {
		return 0.0, err
	}
	isMember := false
	for _, role := range member.Roles {
		if role == config.MemberRoleID {
			isMember = true
		}
	}
	if !isMember {
		return 0.0, ErrNotMember
	}
	votes, err := db.GetTrustVote(discordID)
	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 == config.MemberRoleID {
				isMember = true
			}
		}
		if !isMember {
			continue
		}
		trust, err := GetTrust(s, member.User.ID)
		if err != nil {
			return 0.0, err
		}
		total += trust
	}
	return total, nil
}

func UpdateTrust(sourceID, targetID string, voteValue int) error {
	err := db.UpdateTrust(sourceID, targetID, voteValue, false)
	if err != nil {
		return err
	}
	delete(trustCache, targetID)
	return nil
}