Fixed offset checking (broken on Untappd's end

plugins/beers.go

@@ -3,6 +3,7 @@ package plugins
 import (
 	"bitbucket.org/phlyingpenguin/godeepintir/bot"
 	"encoding/json"
+	"errors"
 	"fmt"
 	"io/ioutil"
 	"labix.org/v2/mgo"
@@ -143,7 +144,8 @@ func (p *BeersPlugin) Message(message bot.Message) bool {
 			ChanNick:    message.User.Name,
 		}
 
-		p.Coll.Upsert(bson.M{"untappduser": parts[1]}, u)
+		p.getUntappdCheckins(u)
+
 		p.Bot.SendMessage(channel, "I'll be watching you.")
 		return true
 	}
@@ -254,39 +256,23 @@ type untappdUser struct {
 	UntappdUser   string
 	LastCheckin   int
 	ChanNick      string
+	KnownCheckins [5]int
 }
 
-func (p *BeersPlugin) checkUntappd(channel string) {
-	if p.Bot.Config.UntappdToken == "<Your Token>" {
-		log.Println("No Untappd token, cannot enable plugin.")
-		return
-	}
-
+func (p *BeersPlugin) pullUntappd(user untappdUser) (Beers, error) {
 	access_token := "?access_token=" + p.Bot.Config.UntappdToken
 	baseUrl := "http://api.untappd.com/v4/user/checkins/"
-	frequency := p.Bot.Config.UntappdFreq
 
-	// users := []string{"Tir"}
-
-	for {
-
-		var users []untappdUser
-		p.Coll.Find(bson.M{"untappduser": bson.M{"$exists": true}}).All(&users)
-
-		log.Println("Found ", len(users), " untappd users")
-
-		for _, user := range users {
-
-			url := baseUrl + user.UntappdUser + access_token
-			if user.LastCheckin > 0 {
-				url = url + "?offset=" + strconv.Itoa(user.LastCheckin)
-			}
+	url := baseUrl + user.UntappdUser + access_token + "&limit=5"
+	log.Println("Request:", url)
 
 	resp, err := http.Get(url)
 	if err != nil {
 		panic(err)
 	}
-			fmt.Println("User: ", user)
+	if resp.StatusCode == 500 {
+		return Beers{}, errors.New(resp.Status)
+	}
 	body, err := ioutil.ReadAll(resp.Body)
 	if err != nil {
 		panic(err)
@@ -297,24 +283,57 @@ func (p *BeersPlugin) checkUntappd(channel string) {
 	err = json.Unmarshal(body, &beers)
 	if err != nil {
 		log.Println(user, err)
-				continue
+		return Beers{}, errors.New("Unable to parse JSON")
+	}
+	return beers, nil
+}
+
+func (p *BeersPlugin) getUntappdCheckins(user untappdUser) ([]checkin, error) {
+	beers, err := p.pullUntappd(user)
+	if err != nil {
+		return nil, err
 	}
 
-			log.Printf("%+v\n\n%+v\n", resp, string(body))
+	var output []checkin
+	chks := beers.Response.Checkins.Items
 
-			// fmt.Println(beers.Response)
+	for _, chk := range chks {
+		if chk.Checkin_id > user.LastCheckin {
+			output = append(output, chk)
+		}
+	}
 
-			checkins := beers.Response.Checkins.Items
-			// count := beers.Response.Checkins.Count
-
-			// fmt.Println("Tir has ", count, " checkins:")
-
-			if len(checkins) > 0 {
-				user.LastCheckin = checkins[0].Checkin_id
+	if len(chks) > 0 {
+		user.LastCheckin = chks[0].Checkin_id
 		p.Coll.Upsert(bson.M{"untappduser": user.UntappdUser}, user)
 	}
 
-			for _, checkin := range checkins {
+	return output, nil
+}
+
+func (p *BeersPlugin) checkUntappd(channel string) {
+	if p.Bot.Config.UntappdToken == "<Your Token>" {
+		log.Println("No Untappd token, cannot enable plugin.")
+		return
+	}
+
+	frequency := p.Bot.Config.UntappdFreq
+
+	// users := []string{"Tir"}
+
+	for {
+		var users []untappdUser
+		p.Coll.Find(bson.M{"untappduser": bson.M{"$exists": true}}).All(&users)
+
+		log.Println("Found ", len(users), " untappd users")
+
+		for _, user := range users {
+			chks, err := p.getUntappdCheckins(user)
+			if err != nil {
+				continue
+			}
+
+			for _, checkin := range chks {
 				venue := ""
 				switch v := checkin.Venue.(type) {
 				case map[string]interface{}: