func awardBits(cmd *modulebase.ModuleCommand) (string, error) {
if len(cmd.Args) < 2 {
return awardBitsHelpString, nil
}
amount, err := strconv.Atoi(cmd.Args[0])
if err != nil {
return "Invalid amount", nil
}
if amount <= 0 {
return "Amount must be a positive nonzero integer", nil
}
userName := strings.Join(cmd.Args[1:], " ")
user, err := utils.FindUser(cmd.Guild, userName)
if err != nil {
return "Unable to find user", nil
}
bits.AddBits(cmd.Session, cmd.Guild.ID, user.ID, amount, "Awarded bits", false)
message := fmt.Sprintf("Awarded %d bits to %v",
amount,
utils.GetPreferredName(cmd.Guild, user.ID))
return message, nil
}
func giveBits(cmd *modulebase.ModuleCommand) (string, error) {
if len(cmd.Args) < 2 {
return giveBitsHelpString, nil
}
amount, err := strconv.Atoi(cmd.Args[0])
if err != nil {
return "Invalid amount", nil
}
if amount <= 0 {
return "Amount must be a positive nonzero integer", nil
}
userName := strings.Join(cmd.Args[1:], " ")
user, err := utils.FindUser(cmd.Guild, userName)
if err != nil {
return "Unable to find user", nil
}
gifter := cmd.Message.Author
if bits.GetBits(cmd.Guild.ID, gifter.ID) < amount {
return "Not enough bits", nil
}
bits.RemoveBits(cmd.Session, cmd.Guild.ID, gifter.ID, amount, "Gift")
bits.AddBits(cmd.Session, cmd.Guild.ID, user.ID, amount, "Gift", false)
message := fmt.Sprintf("Transferred %d bits from %v to %v",
amount,
utils.GetPreferredName(cmd.Guild, gifter.ID),
utils.GetPreferredName(cmd.Guild, user.ID))
return message, nil
}
func voiceStateUpdateCallback(s *discordgo.Session, v *discordgo.VoiceStateUpdate) {
log.Debugf("Voice bonus On voice state update: %v", v.VoiceState)
// Check if it was a part
if v.ChannelID == "" {
return
}
guild, _ := s.State.Guild(v.GuildID)
if guild == nil {
log.WithFields(log.Fields{
"guild": v.GuildID,
}).Warning("Failed to grab guild")
return
}
member, _ := s.State.Member(v.GuildID, v.UserID)
if member == nil {
log.WithFields(log.Fields{
"member": member,
}).Warning("Failed to grab member")
return
}
if member.User.Bot {
return
}
c := voicebonusCollection{ramendb.GetCollection(v.GuildID, ConfigName)}
if !c.Enabled() {
return
}
span := c.Timespan()
if span == nil {
log.Error("Timespan was nil for %v", guild.ID)
return
}
log.Debugf("Timespan is: %v", span)
currentTime := time.Now().UTC()
year, month, day := currentTime.Date()
weekday := currentTime.Weekday()
nextDay := day + utils.GetDaysTillWeekday(int(weekday), span.Weekday)
startDate := time.Date(year, month, nextDay, span.Hour, span.Minute, 0, 0, time.UTC)
previousDate := time.Date(year, month, nextDay-7, span.Hour, span.Minute, 0, 0, time.UTC)
log.Debugf("Start Date is: %v", startDate)
log.Debugf("Previous Date is: %v", previousDate)
lastJoinTime := time.Unix(getUserLastJoin(v.GuildID, v.UserID), 0)
log.Debugf("Last join date: %v, now: %v", lastJoinTime, currentTime)
if currentTime.Before(startDate) {
// Check for previous week parts. Needed if timespan crosses UTC midnight
if time.Since(previousDate).Hours() > float64(span.Duration) ||
lastJoinTime.After(previousDate) {
log.Debugf("Previous time status: %v %v",
time.Since(previousDate).Hours() > float64(span.Duration),
lastJoinTime.After(previousDate))
return
}
} else if time.Since(startDate).Hours() > float64(span.Duration) ||
lastJoinTime.After(startDate) {
log.Debugf("Current time status: %v %v",
time.Since(startDate).Hours() > float64(span.Duration),
lastJoinTime.After(startDate))
return
}
log.Debug("Giving bits for join")
bits.AddBits(s, v.GuildID, v.UserID, c.Amount(), "Voice join bonus", true)
username := utils.GetPreferredName(guild, v.UserID)
message := fmt.Sprintf(joinMessage, username, c.Amount(),
bits.GetBits(v.GuildID, v.UserID))
channel, _ := s.UserChannelCreate(v.UserID)
s.ChannelMessageSend(channel.ID, message)
updateUserLastJoin(v.GuildID, v.UserID, currentTime.Unix())
}