Add reddit grabber function & techgore command

cmd/aerial/aerial.go

@@ -65,6 +65,7 @@ func main() {
 	a.cs.AddCmd("time", "shows the current bot time", bot.NoPermissions, curTime)
 	a.cs.AddCmd("streams", "shows the different Ponyville FM stream links", bot.NoPermissions, streams)
 	a.cs.AddCmd("derpi", "grabs a random (safe) image from Derpibooru with the given search results", bot.NoPermissions, derpi)
+	a.cs.AddCmd("techgore", "grabs a random post from /r/techsupportgore for your viewing pleasure", bot.NoPermissions, techgore)
 
 	dg.AddHandler(a.Handle)
 	dg.AddHandler(pesterLink)

cmd/aerial/pvfm.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"log"
 	"math/rand"
+	"os"
 	"strings"
 	"time"
 
@@ -224,6 +225,10 @@ func streams(s *discordgo.Session, m *discordgo.Message, parv []string) error {
 }
 
 func derpi(s *discordgo.Session, m *discordgo.Message, parv []string) error {
+	if m.ChannelID != string(os.Getenv("DIABEETUSID")) {
+		s.ChannelMessageSend(m.ChannelID, "Please use this command only in #diabeetus")
+		return nil
+	}
 	searchResults, err := derpiSearch.SearchDerpi(m.Content[7:len(m.Content)])
 	if err != nil {
 		s.ChannelMessageSend(m.ChannelID, "An error occured.")
@@ -236,3 +241,16 @@ func derpi(s *discordgo.Session, m *discordgo.Message, parv []string) error {
 	s.ChannelMessageSend(m.ChannelID, "http:"+searchResults.Search[randomRange(0, len(searchResults.Search))].Image)
 	return nil
 }
+
+func techgore(s *discordgo.Session, m *discordgo.Message, parv []string) error {
+	if m.ChannelID != string(os.Getenv("NERDERYID")) {
+		s.ChannelMessageSend(m.ChannelID, "Please use this command only in #nerdery")
+		return nil // only works in #nerdery
+	}
+	results, err := redditSearch("techsupportgore")
+	if err != nil {
+		return err
+	}
+	s.ChannelMessageSend(m.ChannelID, results[randomRange(0, len(results))].URL)
+	return nil
+}

cmd/aerial/reddit.go

@@ -0,0 +1,15 @@
+package main
+
+import "github.com/jzelinskie/geddit"
+
+func redditSearch(subreddit string) ([]*geddit.Submission, error) {
+	listOptions := geddit.ListingOptions{
+		Limit: 50,
+	}
+	reddit := geddit.NewSession("discordbot")
+	results, err := reddit.SubredditSubmissions(subreddit, geddit.NewSubmissions, listOptions)
+	if err != nil {
+		return nil, err
+	}
+	return results, nil
+}