From b345506e7fa6757a1be73ac16118376bb0bd361f Mon Sep 17 00:00:00 2001
From: Chris Sexton <chrissexton@users.noreply.github.com>
Date: Fri, 15 Jan 2016 09:44:22 -0500
Subject: [PATCH] Fix beer plugin bugs

---
 plugins/beers.go | 37 ++++++++++++++++++++++++++++---------
 1 file changed, 28 insertions(+), 9 deletions(-)

diff --git a/plugins/beers.go b/plugins/beers.go
index 4df02cf..dec7d96 100644
--- a/plugins/beers.go
+++ b/plugins/beers.go
@@ -48,7 +48,7 @@ func NewBeersPlugin(bot *bot.Bot) *BeersPlugin {
 			id integer primary key,
 			nick string,
 			count integer,
-			lastDrunk integer
+			lastDrunk datetime
 		);`); err != nil {
 			log.Fatal(err)
 		}
@@ -57,7 +57,7 @@ func NewBeersPlugin(bot *bot.Bot) *BeersPlugin {
 			id integer primary key,
 			untappdUser string
 			channel string
-			lastCheckin integer
+			lastCheckin datetime
 			chanNick string
 		);`); err != nil {
 			log.Fatal(err)
@@ -77,9 +77,9 @@ func NewBeersPlugin(bot *bot.Bot) *BeersPlugin {
 func (u *userBeers) Save(db *sql.DB) error {
 	if !u.saved {
 		res, err := db.Exec(`insert into beers (
-			nick string,
-			count integer,
-			lastDrunk integer
+			nick,
+			count,
+			lastDrunk
 		) values (?, ?, ?)`, u.nick, u.count, u.lastDrunk)
 		if err != nil {
 			return err
@@ -89,12 +89,21 @@ func (u *userBeers) Save(db *sql.DB) error {
 			return err
 		}
 		u.id = id
+	} else {
+		_, err := db.Exec(`update beers set
+			count = ?,
+			lastDrunk = ?
+		where id = ?;`, u.count, time.Now(), u.id)
+		if err != nil {
+			log.Println("Error updating beers: ", err)
+			return err
+		}
 	}
 	return nil
 }
 
 func getUserBeers(db *sql.DB, nick string) *userBeers {
-	var ub userBeers
+	ub := userBeers{saved: true}
 	err := db.QueryRow(`select id, nick, count, lastDrunk from beers
 		where nick = ?`, nick).Scan(
 		&ub.id,
@@ -102,8 +111,15 @@ func getUserBeers(db *sql.DB, nick string) *userBeers {
 		&ub.count,
 		&ub.lastDrunk,
 	)
+	if err == sql.ErrNoRows {
+		log.Println("DIDN'T FIND THAT USER: ", err)
+		return &userBeers{
+			nick:  nick,
+			count: 0,
+		}
+	}
 	if err != nil && err != sql.ErrNoRows {
-		log.Println(err)
+		log.Println("Error finding beers: ", err)
 		return nil
 	}
 
@@ -140,6 +156,7 @@ func (p *BeersPlugin) Message(message bot.Message) bool {
 				// you can't be negative
 				msg := fmt.Sprintf("Sorry %s, you can't have negative beers!", nick)
 				p.Bot.SendMessage(channel, msg)
+				return true
 			}
 			if parts[1] == "+=" {
 				p.setBeers(nick, p.getBeers(nick)+count)
@@ -212,7 +229,7 @@ func (p *BeersPlugin) Message(message bot.Message) bool {
 		log.Println("Creating Untappd user:", u.untappdUser, "nick:", u.chanNick)
 
 		var count int
-		err := p.db.QueryRow(`select count(*) from untappd 
+		err := p.db.QueryRow(`select count(*) from untappd
 			where untappdUser = ?`, u.untappdUser).Scan(&count)
 		if err != nil {
 			log.Println("Error registering untappd: ", err)
@@ -272,7 +289,9 @@ func (p *BeersPlugin) setBeers(user string, amount int) {
 	ub := getUserBeers(p.db, user)
 	ub.count = amount
 	ub.lastDrunk = time.Now()
-	ub.Save(p.db)
+	if err := ub.Save(p.db); err != nil {
+		log.Println("Error saving beers: ", err)
+	}
 }
 
 func (p *BeersPlugin) addBeers(user string) {