cmdTools.go

package main

import (
	"strings"

	log "github.com/Sirupsen/logrus"
	"github.com/bwmarrin/discordgo"
	"github.com/go-ini/ini"
	"strconv"
)

func init() {
	//Owner Commands
	makeCmd("devmode", cmdDevModeToggle).owner().add()
	makeCmd("logmode", cmdLogModeToggle).owner().add()
	makeCmd("cfgreload", cmdCfgRefresh).owner().add()
	makeCmd("listguilds", cmdGuildList).owner().add()
	makeCmd("listchans", cmdChanList).owner().add()
	makeCmd("msgchan", cmdMsgChan).owner().add()
	makeCmd("e6Sample", cmde621SampleToggle).owner().add()
	makeCmd("makeinvite", cmdMakeInvite).owner().add()
	makeCmd("e6filter", cmde621FilterToggle).owner().add()
	makeCmd("e6filterscore", cmde621FilterScore).owner().add()
	makeCmd("status", cmdChangeStatus).helpText("`status message` `type` `url` `reset?`\nStatus Message: message to show as status\nType: 0 - None, 1 - Playing, 2 - Listening to, 3 - Streaming").owner().add()
	//User Commands
	makeCmd("ver", cmdVersion).helpText("gets version infomation of bot").add()
	makeCmd("help", cmdHelp).helpText("i dunno what this does").add()
  makeCmd("vid", cmdGetVideoLink).helpText("Gets video link to the voice channel you are in").add()
}

func cmdDevModeToggle(message []string, s *discordgo.Session, m *discordgo.MessageCreate) {
	if devMode == false {
		devMode = true
		cfg.Section("bot").Key("dev_mode").SetValue("true")
		cfg.SaveTo(cfgFile)
		s.ChannelMessageSend(m.ChannelID, "DEV MODE ENABLED")
		log.Println("DEV MODE ENABLED")
	} else {
		devMode = false
		cfg.Section("bot").Key("dev_mode").SetValue("false")
		cfg.SaveTo(cfgFile)
		s.ChannelMessageSend(m.ChannelID, "DEV MODE DISABLED")
		log.Println("DEV MODE DISABLED")
	}
}

func cmdLogModeToggle(message []string, s *discordgo.Session, m *discordgo.MessageCreate) {
	if logAll == false {
		logAll = true
		cfg.Section("bot").Key("logall").SetValue("true")
		cfg.SaveTo(cfgFile)
		s.ChannelMessageSend(m.ChannelID, "LOG ENABLED")
		log.Println("LOGALL ENABLED")
	} else {
		logAll = false
		cfg.Section("bot").Key("logall").SetValue("false")
		cfg.SaveTo(cfgFile)
		s.ChannelMessageSend(m.ChannelID, "LOG DISABLED")
		log.Println("LOGALL DISABLED")
	}
}

func cmdCfgRefresh(message []string, s *discordgo.Session, m *discordgo.MessageCreate) {
	var err error
	cfg, err = ini.Load(cfgFile)
	if err != nil {
		log.Printf("FAILED TO READ BOT INI FILE: %v\n\n\n\nMAKE SURE ITS IN BOTDIR\\config\\bot.ini", err)
		s.ChannelMessageSend(m.ChannelID, "Failed to refresh config")
		return
	}
	s.ChannelMessageSend(m.ChannelID, "Refreshed config file")
}

func cmdGuildList(message []string, s *discordgo.Session, m *discordgo.MessageCreate) {
	guildIDLen := len(guildID)
	msg := ""
	var guild *discordgo.Guild
	var err error
	if guildID[0] != "" {
		guild, err = s.Guild(guildID[0])
		if err != nil {
			return
		}
		msg = guild.Name + " (" + guildID[0] + ")"
	}
	for i := 1; i < guildIDLen; i++ {
		guild, err = s.Guild(guildID[i])
		if err != nil {
			return
		}
		msg = msg + "\n" + guild.Name + " (" + guildID[i] + ")"
	}
	s.ChannelMessageSend(m.ChannelID, msg)
}

func cmdChanList(message []string, s *discordgo.Session, m *discordgo.MessageCreate) {
	if len(message) < 2 {
		s.ChannelMessageSend(m.ChannelID, "Not enough params")
		return
	}
	gid := message[1]
	chans, err := s.GuildChannels(gid)
	var msg string
	var guild *discordgo.Guild
	if err != nil {
		s.ChannelMessageSend(m.ChannelID, "Not vaild")
		return
	}
	var c *discordgo.Channel
	for _, c = range chans {
		var chanType string
		if c.Type == discordgo.ChannelTypeGuildText {
			chanType = "text"
		} else {
			chanType = "voice"
		}
		msg = msg + "\n" + c.Name + " (" + c.ID + ") Type: " + chanType

	}
	guild, _ = s.Guild(gid)
	s.ChannelMessageSend(m.ChannelID, guild.Name+" has the channels\n-------------"+msg)
}
func cmdMsgChan(message []string, s *discordgo.Session, m *discordgo.MessageCreate) {
	if len(message) <= 2 {
		s.ChannelMessageSend(m.ChannelID, "Not enough params")
		return
	}
	chanid := message[1]
	msg := strings.TrimPrefix(m.Content, message[0]+" "+chanid)
	s.ChannelMessageSend(chanid, msg)
	s.ChannelMessageSend(m.ChannelID, chanid+" was sent the message: "+msg)
}

func cmde621SampleToggle(message []string, s *discordgo.Session, m *discordgo.MessageCreate) {
	if e6Sample == false {
		e6Sample = true
		cfg.Section("bot").Key("e621Sample").SetValue("true")
		cfg.SaveTo(cfgFile)
		s.ChannelMessageSend(m.ChannelID, "e621/e926 SAMPLE ENABLED")
		log.Println("e621 SAMPLE ENABLED")
	} else {
		e6Sample = false
		cfg.Section("bot").Key("e621Sample").SetValue("false")
		cfg.SaveTo(cfgFile)
		s.ChannelMessageSend(m.ChannelID, "e621/e926 SAMPLE DISABLED")
		log.Println("e621 SAMPLE DISABLED")
	}
}

func cmde621FilterToggle(message []string, s *discordgo.Session, m *discordgo.MessageCreate) {
	if e6Filter == false {
		e6Filter = true
		cfg.Section("bot").Key("e621Filter").SetValue("true")
		cfg.SaveTo(cfgFile)
		s.ChannelMessageSend(m.ChannelID, "e621/e926 FILTER ENABLED")
		log.Println("e621 FILTER ENABLED")
	} else {
		e6Filter = false
		cfg.Section("bot").Key("e621Filter").SetValue("false")
		cfg.SaveTo(cfgFile)
		s.ChannelMessageSend(m.ChannelID, "e621/e926 FILTER DISABLED")
		log.Println("e621 FILTER DISABLED")
	}
}

func cmde621FilterScore(message []string, s *discordgo.Session, m *discordgo.MessageCreate) {
	score := strings.TrimPrefix(m.Content, message[0] + " ")
	if e6Filter == false {
		s.ChannelMessageSend(m.ChannelID, "FILTER DISABLED, PLEASE ENABLE FILTER WITH ``" +globalPrefix + "e6filter`")
		log.Println("e621 FILTER ENABLED")
	} else {
		var extraMessage string
		extraMessage = ""
		if score == "<e6filterscore"	{
			score = "2"
			extraMessage = "NOTHING SET, SETTING "
		}
		e6FilterScore = score
		cfg.Section("bot").Key("e621FilterScore").SetValue(e6FilterScore)
		cfg.SaveTo(cfgFile)
		s.ChannelMessageSend(m.ChannelID, extraMessage +"FILTER SCORE TO " + e6FilterScore)
		log.Println("e621 FILTER SCORE SET TO " + e6FilterScore)
	}
}

func cmdMakeInvite(message []string, s *discordgo.Session, m *discordgo.MessageCreate) {
	i := &discordgo.Invite{
		MaxAge: 0,
		Uses:   0,
	}
	var err error
	if len(message) >= 2 {

		i, err = s.ChannelInviteCreate(message[1], *i)
		if err != nil {
			s.ChannelMessageSend(m.ChannelID, "Invite creation failed")
			return
		}

	} else {
		i, _ = s.ChannelInviteCreate(m.ChannelID, *i)
	}

	s.ChannelMessageSend(m.ChannelID, "discord.gg/"+i.Code)
}

func cmdVersion(message []string, s *discordgo.Session, m *discordgo.MessageCreate) {
	s.ChannelMessageSend(m.ChannelID, "Version: "+Version+"-"+BinaryOS+"-"+BinaryArch+"\nBuild Time: "+BuildTime+"\nGitHash: "+GitHash+"\nDiscordGo: "+discordgo.VERSION+"\nHostName: "+HostName)
}

func cmdHelp(message []string, s *discordgo.Session, m *discordgo.MessageCreate) {
	if len(message) == 2 {
		if cmd, ok := commands[strings.ToLower(message[1])]; ok {
			//log.Println("cmd = " + cmd.Name + " help = " + cmd.Help)
			s.ChannelMessageSendEmbed(m.ChannelID, &discordgo.MessageEmbed{
				Color: 0xffffff,
				Author: &discordgo.MessageEmbedAuthor{
					Name: "Help",
				},
				Description: cmd.Help,
				Title:      globalPrefix + cmd.Name,
				//Timestamp:   time.Now().Format(time.RFC3339),
			})
		}
		return
	}
	var cmds []string
	for _, cmd := range commands {
		if m.Author.ID == ownerID || !cmd.Owner {
			cmds = append(cmds,globalPrefix+cmd.Name)
		}
	}
	_, err := s.ChannelMessageSendEmbed(m.ChannelID, &discordgo.MessageEmbed{
		Color:       0xffffff,
		Title:       "List of Commands",
		Description: strings.Join(cmds, ", "),
	})
	if err != nil {
		log.Println("Help Embed Error: " + err.Error())
	}

}

func cmdGetVideoLink(message []string, s *discordgo.Session, m *discordgo.MessageCreate){
	g, err := s.State.Guild(m.GuildID)
	if err != nil {
		s.ChannelMessage(m.ChannelID, "ERROR: " + err.Error())
		for _, vs := range g.VoiceStates {
			if vs.UserID == m.Author.ID {
				vidembed := &discordgo.MessageEmbed{
					Color:       0x202053,
					Author: &discordgo.MessageEmbedAuthor{
						Name:    "Discord Video",
					},
					Title:     "Join " + getNameFromCID(vs.ChannelID, s) + "'s video",
					Description: "https://www.discordapp.com/channels/"+m.GuildID+"/"+vs.ChannelID+"/",
				}
				s.ChannelMessageSendEmbed(m.ChannelID, vidembed)
			}
		}
	}
}

func cmdChangeStatus(message []string, s *discordgo.Session, m *discordgo.MessageCreate) {
	if len(message) == 1 {
		s.ChannelMessageSend(m.ChannelID, "Not enough params\nRun help on this command to see details")
		return
	}
	_message := ""
	_messageType := statusType
	_reset := false
	var err error
	if len(message) >= 2 {
		if len(message) > 2 {
			_message = strings.TrimPrefix(m.Content, message[0]+" ")
			_messageType, err = strconv.Atoi(message[len(message)-1])
			if err != nil {
				_messageType = statusType
			} else {
				_message = strings.TrimSuffix(_message, message[len(message)-1])
			}
		} else if len(message) == 2 {
			_messageType, err = strconv.Atoi(message[1])
			if err != nil {
				_message = strings.TrimPrefix(m.Content, message[0]+" ")
			}
		}
	}
	changeStatus(s, _message, _messageType, statusURL, _reset)
	cfg.Section("bot").Key("statusMessage").SetValue(_message)
	cfg.Section("bot").Key("statusType").SetValue(strconv.Itoa(_messageType))
	cfg.SaveTo(cfgFile)
	s.ChannelMessageSend(m.ChannelID, "Changed status to " + _message )
}