package main

import (
	"fmt"
	"os"
	"os/signal"
	"syscall"
	"strings"
	"time"
	"math/rand"

	"github.com/bwmarrin/discordgo"
)

var (
	auth Auth
	redditbooru_sources []string
)

func init() {
	// set the randomized seed
	rand.Seed(time.Now().UnixNano())

	// build the authentication struct
	build_auth()

	// build the redditbooru sources slice
	build_redditbooru_sources()
}

func main() {
	// create Discord session
	dg, err := discordgo.New("Bot " + auth.discord)
	if err != nil {
		fmt.Println("Error creating Discord session, ", err)
		return
	}

	// register the message_create func as a callback for MessageCreate events
	dg.AddHandler(message_create)

	// open a websocket connection to Discord and begin listening
	err = dg.Open()
	if err != nil {
		fmt.Println("Error opening connection, ", err)
		return
	}

	// wait here until term signal
	fmt.Println("Bot is now running. Press Ctrl+C to exit.")
	sc := make(chan os.Signal, 1)
	signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt, os.Kill)
	<-sc

	// cleanly close
	dg.Close()

}

func message_create (s *discordgo.Session, m *discordgo.MessageCreate) {
	// ignore messages from the bot
	if m.Author.ID == s.State.User.ID {
		return
	}

	// make sure we match our bot's message token
	if len(m.Content) > 2 {
		input_token := m.Content[0:2]
		if strings.ToLower(string(input_token)) != "t." {
			return
		}
	}

	// get our message without our bot's token
	message := strings.ToLower(m.Content[2:])

	// determine our actions
	if message == "coinflip" || message == "coin" { // flip a coin
		s.ChannelMessageSend(m.ChannelID, coinflip(m.Author.ID))
	} else if message == "roll" {  // roll a number
		s.ChannelMessageSend(m.ChannelID, roll(m.Author.ID))
	} else if len(message) > 0 { // as long as there is a message, try to find a picture
		// get url
		url := <-get_image(message)

		// make sure we have a url returned
		if len(url) > 0 {
			s.ChannelMessageSend(m.ChannelID, url)
		} else {
			s.ChannelMessageSend(m.ChannelID, "I couldn't find that, sauce?")
		}
	}
}