modules: Move most code in main package to their own modules
parent
4887d55707
commit
825cf44216
GPG Key ID:
B7D77E3E9D102B70
@ -1,89 +0,0 @@
|
|||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"crypto/rand"
|
|
||||||
"fmt"
|
|
||||||
|
|
||||||
"github.com/bwmarrin/discordgo"
|
|
||||||
|
|
||||||
"PermissionGacha/db"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
viewAuditLogRole = "645856347258093568"
|
|
||||||
manageWebhookRole = "645856081750523914"
|
|
||||||
changeNicknameRole = "645977231281356800"
|
|
||||||
embedLinkRole = "645977361254318080"
|
|
||||||
voiceActivityRole = "645977361254318080"
|
|
||||||
)
|
|
||||||
|
|
||||||
func checkForGachaEligibility(dg *discordgo.Session, discordID, intendedLevel string) {
|
|
||||||
member, err := dg.GuildMember(guildID, discordID)
|
|
||||||
if err != nil {
|
|
||||||
logError(dg, "checkForGachaEligibility", "getGuildMember", discordID)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
found := false
|
|
||||||
for _, v := range member.Roles {
|
|
||||||
if v == intendedLevel {
|
|
||||||
found = true
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if found {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if intendedLevel != rolesFromLevel[0] {
|
|
||||||
rollGacha(dg, member)
|
|
||||||
}
|
|
||||||
|
|
||||||
err = dg.GuildMemberRoleAdd(guildID, discordID, intendedLevel)
|
|
||||||
if err != nil {
|
|
||||||
logError(dg, "checkForGachaEligibility", "guildMemberRoleAdd", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func rollGacha(dg *discordgo.Session, member *discordgo.Member) {
|
|
||||||
lookUpAndGive := func(friendlyName, role string) {
|
|
||||||
found := false
|
|
||||||
for _, v := range member.Roles {
|
|
||||||
if v == role {
|
|
||||||
found = true
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
channel, err := dg.UserChannelCreate(member.User.ID)
|
|
||||||
if found {
|
|
||||||
db.RewardShard(member.User.ID, 1)
|
|
||||||
if err == nil {
|
|
||||||
dg.ChannelMessageSend(
|
|
||||||
channel.ID,
|
|
||||||
fmt.Sprintf(
|
|
||||||
"**GACHA > ** You got __%s__! However, it was a duplicate so you got a shard instead.", friendlyName,
|
|
||||||
),
|
|
||||||
)
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
dg.GuildMemberRoleAdd(guildID, member.User.ID, role)
|
|
||||||
dg.ChannelMessageSend(
|
|
||||||
channel.ID,
|
|
||||||
fmt.Sprintf("**GACHA > ** You got __%s__!", friendlyName),
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
gachaRandomness := make([]byte, 1)
|
|
||||||
rand.Read(gachaRandomness)
|
|
||||||
switch gachaRandomness[0] % 5 {
|
|
||||||
case 0:
|
|
||||||
lookUpAndGive("View Audit Log", viewAuditLogRole)
|
|
||||||
case 1:
|
|
||||||
lookUpAndGive("Manage Webhook", manageWebhookRole)
|
|
||||||
case 2:
|
|
||||||
lookUpAndGive("Change Nickname", changeNicknameRole)
|
|
||||||
case 3:
|
|
||||||
lookUpAndGive("Embed Link", embedLinkRole)
|
|
||||||
case 4:
|
|
||||||
lookUpAndGive("Use Voice Activity", voiceActivityRole)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -1,54 +0,0 @@
|
|||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"io/ioutil"
|
|
||||||
"strings"
|
|
||||||
"sync"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/bwmarrin/discordgo"
|
|
||||||
)
|
|
||||||
|
|
||||||
const guildID = "645550569573842945"
|
|
||||||
|
|
||||||
var levelMutex sync.Mutex
|
|
||||||
var levelQueue = make(map[string]bool)
|
|
||||||
var assignedLevels = make(map[string]int)
|
|
||||||
var rolesFromLevel []string
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
levels, _ := ioutil.ReadFile("roles.txt")
|
|
||||||
rolesFromLevel = strings.Split(string(levels), "\n")
|
|
||||||
for k, v := range rolesFromLevel {
|
|
||||||
rolesFromLevel[k] = strings.TrimSpace(v)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func addToLevelQueue(discordID string) {
|
|
||||||
levelMutex.Lock()
|
|
||||||
defer levelMutex.Unlock()
|
|
||||||
levelQueue[discordID] = true
|
|
||||||
}
|
|
||||||
|
|
||||||
func listenToLevelUpdate(dg *discordgo.Session) {
|
|
||||||
for {
|
|
||||||
time.Sleep(5 * time.Second)
|
|
||||||
levelMutex.Lock()
|
|
||||||
for k, v := range levelQueue {
|
|
||||||
if !v {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
levelQueue[k] = false
|
|
||||||
level, _, err := getLevelAndXP(k)
|
|
||||||
if err != nil {
|
|
||||||
logError(dg, err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if assignedLevels[k] != level {
|
|
||||||
assignedLevels[k] = level
|
|
||||||
go checkForGachaEligibility(dg, k, rolesFromLevel[level-1])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
levelMutex.Unlock()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -0,0 +1,3 @@
|
|||||||
|
package config
|
||||||
|
|
||||||
|
const GuildID = "645550569573842945"
|
||||||
@ -0,0 +1,22 @@
|
|||||||
|
package gacha
|
||||||
|
|
||||||
|
import (
|
||||||
|
crand "crypto/rand"
|
||||||
|
"encoding/binary"
|
||||||
|
mrand "math/rand"
|
||||||
|
)
|
||||||
|
|
||||||
|
var rand = mrand.New(cryptoSource{})
|
||||||
|
|
||||||
|
type cryptoSource struct{}
|
||||||
|
|
||||||
|
func (cryptoSource) Int63() int64 {
|
||||||
|
var value uint64
|
||||||
|
err := binary.Read(crand.Reader, binary.BigEndian, &value)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return int64(value ^ uint64(1<<63))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cryptoSource) Seed(seed int64) {}
|
||||||
@ -0,0 +1,50 @@
|
|||||||
|
package gacha
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/bwmarrin/discordgo"
|
||||||
|
|
||||||
|
"PermissionGacha/modules/config"
|
||||||
|
)
|
||||||
|
|
||||||
|
type reward struct {
|
||||||
|
FriendlyName string
|
||||||
|
Reward func(s *discordgo.Session, member *discordgo.Member) (bool, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
var gachaRewards = []reward{
|
||||||
|
reward{
|
||||||
|
FriendlyName: "View Audit Log",
|
||||||
|
Reward: generateRoleReward("645856347258093568"),
|
||||||
|
},
|
||||||
|
reward{
|
||||||
|
FriendlyName: "Manage Webhook",
|
||||||
|
Reward: generateRoleReward("645856081750523914"),
|
||||||
|
},
|
||||||
|
reward{
|
||||||
|
FriendlyName: "Change Nickname",
|
||||||
|
Reward: generateRoleReward("645977231281356800"),
|
||||||
|
},
|
||||||
|
reward{
|
||||||
|
FriendlyName: "Use Voice Activity",
|
||||||
|
Reward: generateRoleReward("645880647474479115"),
|
||||||
|
},
|
||||||
|
reward{
|
||||||
|
FriendlyName: "Embed Link",
|
||||||
|
Reward: generateRoleReward("645977361254318080"),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
func generateRoleReward(roleID string) func(s *discordgo.Session, member *discordgo.Member) (bool, error) {
|
||||||
|
return func(s *discordgo.Session, member *discordgo.Member) (bool, error) {
|
||||||
|
for _, v := range member.Roles {
|
||||||
|
if v == roleID {
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
err := s.GuildMemberRoleAdd(config.GuildID, member.User.ID, roleID)
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
return true, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,87 @@
|
|||||||
|
package gacha
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/bwmarrin/discordgo"
|
||||||
|
|
||||||
|
"PermissionGacha/db"
|
||||||
|
"PermissionGacha/modules/config"
|
||||||
|
)
|
||||||
|
|
||||||
|
func RollReward(s *discordgo.Session, rewardeeID string, timesToReward int) error {
|
||||||
|
if timesToReward <= 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
rewardee, err := s.GuildMember(config.GuildID, rewardeeID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
rewardCount := make(map[int]map[bool]int)
|
||||||
|
shardCount := 0
|
||||||
|
for i := 0; i < timesToReward; i++ {
|
||||||
|
rewardID := rand.Intn(len(gachaRewards))
|
||||||
|
success, err := gachaRewards[rewardID].Reward(s, rewardee)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if _, ok := rewardCount[rewardID]; !ok {
|
||||||
|
rewardCount[rewardID] = make(map[bool]int)
|
||||||
|
}
|
||||||
|
if success {
|
||||||
|
rewardee, err = s.GuildMember(config.GuildID, rewardeeID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
shardCount++
|
||||||
|
}
|
||||||
|
rewardCount[rewardID][success]++
|
||||||
|
}
|
||||||
|
err = db.RewardShard(rewardeeID, shardCount)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
channel, err := s.UserChannelCreate(rewardeeID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if timesToReward == 1 {
|
||||||
|
for k, v := range rewardCount {
|
||||||
|
if v[true] > 0 {
|
||||||
|
//They were rewarded that exact thing.
|
||||||
|
s.ChannelMessageSend(
|
||||||
|
channel.ID,
|
||||||
|
fmt.Sprintf(
|
||||||
|
"**GACHA >** <@%s>, you just levelled up! You got __%s__.",
|
||||||
|
rewardeeID, gachaRewards[k].FriendlyName,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
} else {
|
||||||
|
//They were rewarded that thing but got a shard instead.
|
||||||
|
s.ChannelMessageSend(
|
||||||
|
channel.ID,
|
||||||
|
fmt.Sprintf(
|
||||||
|
"**GACHA >** <@%s>, you just levelled up! You originally got __%s__ but you got a shard instead.",
|
||||||
|
rewardeeID, gachaRewards[k].FriendlyName,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
msg := fmt.Sprintf(
|
||||||
|
"**GACHA >** <@%s>, you just levelled up %d times. Here are your rewards:",
|
||||||
|
rewardeeID, timesToReward,
|
||||||
|
)
|
||||||
|
for k, v := range rewardCount {
|
||||||
|
msg += fmt.Sprintf("\n- __%s__", gachaRewards[k].FriendlyName)
|
||||||
|
if v[false] > 0 {
|
||||||
|
msg += fmt.Sprintf(" (%d converted to shards)", v[false])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
s.ChannelMessageSend(channel.ID, msg)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
@ -0,0 +1,53 @@
|
|||||||
|
package level
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"math"
|
||||||
|
|
||||||
|
"PermissionGacha/db"
|
||||||
|
)
|
||||||
|
|
||||||
|
var levelUpRequirementCache [30]int
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
for i := 1; i <= 30; i++ {
|
||||||
|
//Constructed on Desmos:
|
||||||
|
//y=2^{\frac{x}{2}}+.1x^{2}+1.2x-1
|
||||||
|
fi := float64(i)
|
||||||
|
levelUpRequirementCache[i-1] = int(10000 * (math.Pow(2, fi/2) + 0.1*fi*fi + 1.2*fi - 1))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetLevelAndXP(discordID string) (int, int, error) {
|
||||||
|
rawXP, err := db.GetXP(discordID)
|
||||||
|
if err != nil {
|
||||||
|
return 1, 0, fmt.Errorf("level: GetLevelAndXP: error requesting XP: %w", err)
|
||||||
|
}
|
||||||
|
level, xp := ConvertXPIntoLevel(rawXP)
|
||||||
|
return level, xp, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetLevelUpRequirement(level int) int {
|
||||||
|
if level <= 0 {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
if level == 1 {
|
||||||
|
return levelUpRequirementCache[0]
|
||||||
|
}
|
||||||
|
if level > len(levelUpRequirementCache) {
|
||||||
|
return levelUpRequirementCache[len(levelUpRequirementCache)-1]
|
||||||
|
}
|
||||||
|
return levelUpRequirementCache[level-1] - levelUpRequirementCache[level-2]
|
||||||
|
}
|
||||||
|
|
||||||
|
func ConvertXPIntoLevel(xp int) (int, int) {
|
||||||
|
if xp < levelUpRequirementCache[0] {
|
||||||
|
return 1, xp
|
||||||
|
}
|
||||||
|
for i := 0; i < 30; i++ {
|
||||||
|
if levelUpRequirementCache[i] > xp {
|
||||||
|
return i + 1, xp - levelUpRequirementCache[i-1]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return 30, xp - levelUpRequirementCache[29]
|
||||||
|
}
|
||||||
@ -0,0 +1,57 @@
|
|||||||
|
package level
|
||||||
|
|
||||||
|
import (
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/bwmarrin/discordgo"
|
||||||
|
|
||||||
|
"PermissionGacha/modules/gacha"
|
||||||
|
"PermissionGacha/modules/log"
|
||||||
|
"PermissionGacha/modules/roles"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
levelMutex sync.Mutex
|
||||||
|
levelQueue = make(map[string]bool)
|
||||||
|
lastKnownLevel = make(map[string]int)
|
||||||
|
)
|
||||||
|
|
||||||
|
func QueueCheck(discordID string) {
|
||||||
|
levelMutex.Lock()
|
||||||
|
defer levelMutex.Unlock()
|
||||||
|
levelQueue[discordID] = true
|
||||||
|
}
|
||||||
|
|
||||||
|
func ScheduleUpdates(s *discordgo.Session) {
|
||||||
|
for {
|
||||||
|
time.Sleep(5 * time.Second)
|
||||||
|
levelMutex.Lock()
|
||||||
|
for id, isQueued := range levelQueue {
|
||||||
|
if !isQueued {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
delete(levelQueue, id)
|
||||||
|
level, _, err := GetLevelAndXP(id)
|
||||||
|
if err != nil {
|
||||||
|
log.Error(s, err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if lastKnownLevel[id] != level {
|
||||||
|
lastKnownLevel[id] = level
|
||||||
|
go func() {
|
||||||
|
oldLevel, err := roles.GetOldLevel(s, id)
|
||||||
|
if err != nil {
|
||||||
|
log.Error(s, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if level > oldLevel {
|
||||||
|
gacha.RollReward(s, id, level-oldLevel)
|
||||||
|
roles.GiveLevelRoles(s, id, oldLevel, level)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
levelMutex.Unlock()
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,18 @@
|
|||||||
|
package log
|
||||||
|
|
||||||
|
import (
|
||||||
|
llog "log"
|
||||||
|
|
||||||
|
"github.com/bwmarrin/discordgo"
|
||||||
|
)
|
||||||
|
|
||||||
|
func Error(s *discordgo.Session, a ...interface{}) {
|
||||||
|
s.UpdateStatusComplex(discordgo.UpdateStatusData{
|
||||||
|
Game: &discordgo.Game{
|
||||||
|
Name: "with error",
|
||||||
|
Type: discordgo.GameTypeGame,
|
||||||
|
},
|
||||||
|
Status: string(discordgo.StatusDoNotDisturb),
|
||||||
|
})
|
||||||
|
llog.Println(a...)
|
||||||
|
}
|
||||||
@ -0,0 +1,30 @@
|
|||||||
|
package roles
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/bwmarrin/discordgo"
|
||||||
|
|
||||||
|
"PermissionGacha/modules/config"
|
||||||
|
"PermissionGacha/modules/log"
|
||||||
|
)
|
||||||
|
|
||||||
|
func GiveLevelRoles(s *discordgo.Session, discordID string, oldLevel, newLevel int) {
|
||||||
|
for i := oldLevel; i <= newLevel; i++ {
|
||||||
|
s.GuildMemberRoleAdd(config.GuildID, discordID, rolesFromLevel[i-1])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func RemoveAllLevelRoles(s *discordgo.Session, discordID string) {
|
||||||
|
member, err := s.GuildMember(config.GuildID, discordID)
|
||||||
|
if err != nil {
|
||||||
|
log.Error(s, "RemoveAllLevelRoles", "GuildMember", discordID)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
for _, v := range member.Roles {
|
||||||
|
for _, w := range rolesFromLevel {
|
||||||
|
if v == w {
|
||||||
|
s.GuildMemberRoleRemove(config.GuildID, discordID, v)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,30 @@
|
|||||||
|
package roles
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/bwmarrin/discordgo"
|
||||||
|
|
||||||
|
"PermissionGacha/modules/config"
|
||||||
|
)
|
||||||
|
|
||||||
|
func GetOldLevel(s *discordgo.Session, id string) (int, error) {
|
||||||
|
member, err := s.GuildMember(config.GuildID, id)
|
||||||
|
if err != nil {
|
||||||
|
return 0, fmt.Errorf("roles: EligibleGachaCount: error fetching guild member (%s): %w", id, err)
|
||||||
|
}
|
||||||
|
var foundLevel = 0
|
||||||
|
levelSearchLoop:
|
||||||
|
for i := len(rolesFromLevel); i > 0; i-- {
|
||||||
|
for _, v := range member.Roles {
|
||||||
|
if rolesFromLevel[i-1] == v {
|
||||||
|
foundLevel = i
|
||||||
|
break levelSearchLoop
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if foundLevel == 0 {
|
||||||
|
return 0, nil
|
||||||
|
}
|
||||||
|
return foundLevel, nil
|
||||||
|
}
|
||||||
@ -0,0 +1,16 @@
|
|||||||
|
package roles
|
||||||
|
|
||||||
|
import (
|
||||||
|
"io/ioutil"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
var rolesFromLevel []string
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
listOfRoles, _ := ioutil.ReadFile("roles.txt")
|
||||||
|
rolesFromLevel = strings.Split(string(listOfRoles), "\n")
|
||||||
|
for k, v := range rolesFromLevel {
|
||||||
|
rolesFromLevel[k] = strings.TrimSpace(v)
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -1,21 +0,0 @@
|
|||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"github.com/bwmarrin/discordgo"
|
|
||||||
)
|
|
||||||
|
|
||||||
func cleanUpRole(s *discordgo.Session, discordID string) {
|
|
||||||
member, err := s.GuildMember(guildID, discordID)
|
|
||||||
if err != nil {
|
|
||||||
logError(s, "cleanUpRole", "guildMember", discordID)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
for _, v := range member.Roles {
|
|
||||||
for _, w := range rolesFromLevel {
|
|
||||||
if v == w {
|
|
||||||
s.GuildMemberRoleRemove(guildID, discordID, v)
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Reference in New Issue