package commands import ( "errors" "strconv" "strings" "time" "github.com/bwmarrin/discordgo" "github.com/foxtrot/scuzzy/actions" ) func (c *Commands) handleSetSlowmode(s *discordgo.Session, m *discordgo.MessageCreate) error { slowmodeSplit := strings.Split(m.Content, " ") if len(slowmodeSplit) < 2 { return errors.New("You must supply at least an amount of time") } slowmodeTimeStr := slowmodeSplit[1] slowModeTime, err := strconv.Atoi(slowmodeTimeStr) if err != nil { return err } if len(slowmodeSplit) == 3 { if slowmodeSplit[2] == "all" { channels, err := s.GuildChannels(c.Config.GuildID) if err != nil { return err } for _, channel := range channels { currPos := channel.Position s.ChannelEditComplex(channel.ID, &discordgo.ChannelEdit{ Position: currPos, RateLimitPerUser: slowModeTime, }) } } } else { currChan, err := s.Channel(m.ChannelID) if err != nil { return err } currPos := currChan.Position _, err = s.ChannelEditComplex(m.ChannelID, &discordgo.ChannelEdit{ Position: currPos, RateLimitPerUser: slowModeTime, }) if err != nil { return err } } msg := c.CreateDefinedEmbed("Slow Mode", "Successfully set Slow Mode to `"+slowmodeTimeStr+"`.", "success", m.Author) _, err = s.ChannelMessageSendEmbed(m.ChannelID, msg) if err != nil { return err } return nil } func (c *Commands) handleUnsetSlowmode(s *discordgo.Session, m *discordgo.MessageCreate) error { slowmodeSplit := strings.Split(m.Content, " ") secs := 0 if len(slowmodeSplit) == 2 { if slowmodeSplit[1] == "all" { channels, err := s.GuildChannels(c.Config.GuildID) if err != nil { return err } for _, channel := range channels { currPos := channel.Position s.ChannelEditComplex(channel.ID, &discordgo.ChannelEdit{ Position: currPos, RateLimitPerUser: secs, }) } } } else { currChan, err := s.Channel(m.ChannelID) if err != nil { return err } currPos := currChan.Position _, err = s.ChannelEditComplex(m.ChannelID, &discordgo.ChannelEdit{ Position: currPos, RateLimitPerUser: secs, }) if err != nil { return err } } msg := c.CreateDefinedEmbed("Slow Mode", "Successfully unset Slow Mode", "success", m.Author) _, err := s.ChannelMessageSendEmbed(m.ChannelID, msg) if err != nil { return err } return nil } func (c *Commands) handlePurgeChannel(s *discordgo.Session, m *discordgo.MessageCreate) error { purgeSplit := strings.SplitN(m.Content, " ", 2) if len(purgeSplit) < 2 { return errors.New("No message count supplied") } msgCount, err := strconv.Atoi(purgeSplit[1]) if err != nil { return nil } if msgCount > 100 { return errors.New("You may only purge upto 100 messages at a time.") } chanMsgs, err := s.ChannelMessages(m.ChannelID, msgCount, "", "", "") if err != nil { return err } msg := c.CreateDefinedEmbed("Purge Channel", "Purging `"+purgeSplit[1]+"` messages.", "", m.Author) r, err := s.ChannelMessageSendEmbed(m.ChannelID, msg) if err != nil { return err } var delMsgs []string for _, v := range chanMsgs { delMsgs = append(delMsgs, v.ID) } err = s.ChannelMessagesBulkDelete(m.ChannelID, delMsgs) if err != nil { return err } err = s.ChannelMessageDelete(m.ChannelID, r.ID) msg = c.CreateDefinedEmbed("Purge Channel", "Purged `"+purgeSplit[1]+"` messages!", "success", m.Author) msgS, err := s.ChannelMessageSendEmbed(m.ChannelID, msg) if err != nil { return err } time.Sleep(time.Second * 10) err = s.ChannelMessageDelete(m.ChannelID, msgS.ID) if err != nil { return err } return nil } func (c *Commands) handleKickUser(s *discordgo.Session, m *discordgo.MessageCreate) error { var ( mHandle *discordgo.Member kickReason string err error ) args := strings.Split(m.Content, " ") if len(args) < 2 { return errors.New("You must specify a user to kick.") } if len(args) == 3 { kickReason = strings.Join(args[2:], " ") } member := args[1] idStr := strings.ReplaceAll(member, "<@!", "") idStr = strings.ReplaceAll(idStr, "<@", "") idStr = strings.ReplaceAll(idStr, ">", "") mHandle, err = s.GuildMember(c.Config.GuildID, idStr) if err != nil { return err } err = actions.KickUser(s, c.Config.GuildID, mHandle.User.ID, kickReason) if err != nil { return err } msg := "User `" + mHandle.User.Username + "#" + mHandle.User.Discriminator + "` was kicked.\n" if len(kickReason) > 0 { msg += "Reason: `" + kickReason + "`\n" } embed := c.CreateDefinedEmbed("Kick User", msg, "success", m.Author) _, err = s.ChannelMessageSendEmbed(m.ChannelID, embed) if err != nil { return err } return nil } func (c *Commands) handleBanUser(s *discordgo.Session, m *discordgo.MessageCreate) error { var ( mHandle *discordgo.User banReason string err error ) args := strings.Split(m.Content, " ") if len(args) < 2 { return errors.New("You must specify a user to ban.") } if len(args) == 3 { banReason = strings.Join(args[2:], " ") } member := args[1] idStr := strings.ReplaceAll(member, "<@!", "") idStr = strings.ReplaceAll(idStr, "<@", "") idStr = strings.ReplaceAll(idStr, ">", "") mHandle, err = s.User(idStr) if err != nil { return err } err = actions.BanUser(s, c.Config.GuildID, mHandle.ID, banReason) if err != nil { return err } msg := "User `" + mHandle.Username + "#" + mHandle.Discriminator + "` was banned.\n" if len(banReason) > 0 { msg += "Reason: `" + banReason + "`\n" } embed := c.CreateDefinedEmbed("Ban User", msg, "success", m.Author) _, err = s.ChannelMessageSendEmbed(m.ChannelID, embed) if err != nil { return err } return nil } func (c *Commands) handleIgnoreUser(s *discordgo.Session, m *discordgo.MessageCreate) error { ignArgs := strings.Split(m.Content, " ") if len(ignArgs) < 2 { return errors.New("You did not specify a user.") } member := ignArgs[1] idStr := strings.ReplaceAll(member, "<@!", "") idStr = strings.ReplaceAll(idStr, "<@", "") idStr = strings.ReplaceAll(idStr, ">", "") c.Config.IgnoredUsers = append(c.Config.IgnoredUsers, idStr) eMsg := c.CreateDefinedEmbed("Ignore User", "<@!"+idStr+"> is now being ignored.", "success", m.Author) _, err := s.ChannelMessageSendEmbed(m.ChannelID, eMsg) if err != nil { return err } err = c.handleSaveConfig(s, m) if err != nil { return err } return nil } func (c *Commands) handleUnIgnoreUser(s *discordgo.Session, m *discordgo.MessageCreate) error { ignArgs := strings.Split(m.Content, " ") if len(ignArgs) < 2 { return errors.New("You did not specify a user.") } member := ignArgs[1] idStr := strings.ReplaceAll(member, "<@!", "") idStr = strings.ReplaceAll(idStr, "<@", "") idStr = strings.ReplaceAll(idStr, ">", "") for k, v := range c.Config.IgnoredUsers { if v == idStr { c.Config.IgnoredUsers[k] = c.Config.IgnoredUsers[len(c.Config.IgnoredUsers)-1] c.Config.IgnoredUsers = c.Config.IgnoredUsers[:len(c.Config.IgnoredUsers)-1] } } eMsg := c.CreateDefinedEmbed("Unignore User", "<@!"+idStr+"> is not being ignored.", "success", m.Author) _, err := s.ChannelMessageSendEmbed(m.ChannelID, eMsg) if err != nil { return err } err = c.handleSaveConfig(s, m) if err != nil { return err } return nil }