Add card requests

main.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"os"
 	"strconv"
+	"strings"
 
 	"github.com/hamcha/tg"
 )
@@ -15,6 +16,7 @@ type Config struct {
 	Bind                  string
 	WebhookURL            string
 	WebhookPath           string
+	MaxRequestsPerMessage int
 }
 
 func checkErr(err error, msg string, args ...interface{}) {
@@ -26,6 +28,7 @@ func checkErr(err error, msg string, args ...interface{}) {
 }
 
 var api *tg.Telegram
+var cfg Config
 
 func main() {
 	cfgpath := flag.String("config", "stappa.conf", "Path to config file")
@@ -34,23 +37,24 @@ func main() {
 	cfgfile, err := os.Open(*cfgpath)
 	checkErr(err, "Could not open config file")
 
-	var cfg Config
 	err = json.NewDecoder(cfgfile).Decode(&cfg)
 	checkErr(err, "Could not decode JSON from config file contents")
 
 	cfgfile.Close()
 
+	// Set default maxreq
+	if cfg.MaxRequestsPerMessage < 1 {
+		cfg.MaxRequestsPerMessage = 5
+	}
+
 	api = tg.MakeAPIClient(cfg.Token)
 	api.SetWebhook(cfg.WebhookURL)
 	api.HandleWebhook(cfg.Bind, cfg.WebhookPath, webhook)
 }
 
 func webhook(update tg.APIUpdate) {
-	// Ignore everything that isn't an inline query (for now)
-	if update.Inline == nil {
-		return
-	}
-
+	// Handle inline queries (99% of the usage I hope)
+	if update.Inline != nil {
 		query := update.Inline.Query
 		offset, _ := strconv.Atoi(update.Inline.Offset)
 		results, err := scryfallSearch(query, offset)
@@ -104,3 +108,62 @@ func webhook(update tg.APIUpdate) {
 			return
 		}
 	}
+
+	// Check for card requests
+	if update.Message != nil && update.Message.Text != nil {
+		requests := getCardRequests(*update.Message.Text)
+		if len(requests) > cfg.MaxRequestsPerMessage {
+			api.SendTextMessage(tg.ClientTextMessageData{
+				ChatID:  update.Message.Chat.ChatID,
+				Text:    fmt.Sprintf("You asked for way too many cards (%d!), please only ask me for at most %d cards in a single message.", len(requests), cfg.MaxRequestsPerMessage),
+				ReplyID: &update.Message.MessageID,
+			})
+			return
+		}
+		cardmedia := []tg.APIInputMediaPhoto{}
+		errlist := []string{}
+		for _, cardname := range requests {
+			card, err := scryfallGetCard(cardname)
+			if err != nil {
+				errlist = append(errlist, cardname)
+			} else {
+				cardmedia = append(cardmedia, tg.APIInputMediaPhoto{
+					Type:  "photo",
+					Media: card.ImageUris.Large,
+				})
+			}
+		}
+		if len(cardmedia) > 0 {
+			api.SendAlbum(tg.ClientAlbumData{
+				ChatID:  update.Message.Chat.ChatID,
+				Media:   cardmedia,
+				Silent:  true,
+				ReplyID: &update.Message.MessageID,
+			})
+		}
+		if len(errlist) > 0 {
+			api.SendTextMessage(tg.ClientTextMessageData{
+				ChatID:  update.Message.Chat.ChatID,
+				Text:    "I couldn't find these cards you mentioned: " + strings.Join(errlist, ", "),
+				ReplyID: &update.Message.MessageID,
+			})
+		}
+	}
+}
+
+func getCardRequests(str string) (out []string) {
+	remaining := str
+	for len(remaining) > 1 {
+		nextToken := strings.Index(remaining, "[[")
+		if nextToken < 0 {
+			break
+		}
+		endToken := strings.Index(remaining[nextToken:], "]]")
+		if endToken < 0 {
+			break
+		}
+		out = append(out, remaining[nextToken+2:nextToken+endToken])
+		remaining = remaining[nextToken+2+endToken:]
+	}
+	return
+}

scryfall.go

@@ -73,3 +73,16 @@ func scryfallSearch(query string, offset int) (results CardSearchResults, err er
 	results.Data = results.Data[cardoffset:maxoffset]
 	return
 }
+
+func scryfallGetCard(cardname string) (card CardData, err error) {
+	cardname = url.QueryEscape(cardname)
+	requrl := fmt.Sprintf("https://api.scryfall.com/cards/named?fuzzy=%s")
+	response, err := netClient.Get(requrl)
+	if err != nil {
+		return
+	}
+	defer response.Body.Close()
+
+	err = json.NewDecoder(response.Body).Decode(&card)
+	return
+}