Add initial support for karma ranking

Add initial support for karma rankings, posting to sprunge.us as a
pastebin service.

Signed-off-by: Adam Jimerson <[email protected]>

Author: vendion

karma.go

@@ -7,16 +7,29 @@ package main
 import (
 	"encoding/json"
 	"fmt"
+	"sort"
 	"strings"
 
 	"github.com/danryan/hal"
 )
 
+// DevKarma contains fields for user karma information.
 type DevKarma struct {
 	Name  string `json:"name"`
 	Karma int    `json:"karma"`
 }
 
+func (d DevKarma) String() string {
+	return fmt.Sprintf("%s: %d\n", d.Name, d.Karma)
+}
+
+// ByKarma implements sort.Interface for []DevKarma based on the Karma field.
+type ByKarma []DevKarma
+
+func (a ByKarma) Len() int           { return len(a) }
+func (a ByKarma) Swap(i, j int)      { a[i], a[j] = a[j], a[i] }
+func (a ByKarma) Less(i, j int) bool { return a[i].Karma < a[j].Karma }
+
 var karmaHandler = hear(`(.+)(\+\+|\-\-)`, "(nickname)++ or (nickname)--", "Increases or Decreases a persons karma", func(res *hal.Response) error {
 	nick := res.Match[1]
 	sign := res.Match[2]
@@ -49,6 +62,8 @@ var karmaHandler = hear(`(.+)(\+\+|\-\-)`, "(nickname)++ or (nickname)--", "Incr
 		return err
 	}
 
+	updateDevsWithKarma(key, res)
+
 	err = res.Send(fmt.Sprintf("Karma for %s has been updated", nick))
 	return err
 })
@@ -78,3 +93,81 @@ var karmaStatsHandler = hear(`karma stats(.*)`, "karma stats (username)", "Shows
 	err = res.Send(fmt.Sprintf("User %s currently has %d karma", nick, dev.Karma))
 	return err
 })
+
+var karmaRankingHandler = hear(`karma ranking(\s(top|lowest)\s\d*)?`, "karma ranking", "Shows the current karma standings, with optional top/lowest standings", func(res *hal.Response) error {
+	var devs []DevKarma
+	names := getDevKarmaList(res)
+	hal.Logger.Debugf("Value of names: %#v", names)
+
+	for _, v := range names {
+		var dev DevKarma
+		val, err := res.Robot.Store.Get("Karma::" + v)
+		if err != nil {
+			hal.Logger.Errorf("couldn't fetch karma details: %v", err)
+			err = res.Reply("Sorry, I couldn't fetch karma ratings")
+			return err
+		}
+		json.Unmarshal(val, &dev)
+
+		devs = append(devs, dev)
+	}
+	hal.Logger.Debugf("Value of devs: %#v", devs)
+
+	if len(devs) == 0 {
+	}
+
+	sort.Sort(sort.Reverse(ByKarma(devs)))
+	hal.Logger.Debugf("Sorted devs list: %#v", devs)
+
+	ranking := []string{
+		"Current rankings:\n",
+	}
+
+	for _, val := range devs {
+		ranking = append(ranking, val.String())
+	}
+
+	var msg string
+	for _, m := range ranking {
+		msg = msg + m
+	}
+	msg, err := sprungeSend(msg)
+	if err != nil {
+		hal.Logger.Error(err)
+		err = res.Reply("Sorry, I had some unexpected difficulties posting the rankings")
+		return err
+	}
+
+	err = res.Reply(fmt.Sprintf("The current rankings can be found here %s", msg))
+	return err
+})
+
+func updateDevsWithKarma(name string, r *hal.Response) {
+	names := getDevKarmaList(r)
+	if !inSlice(names, name) {
+		if len(names) == 0 || names[0] == "" {
+			names[0] = name
+		} else {
+			names = append(names, name)
+		}
+	}
+	list := strings.Join(names, ",")
+	r.Robot.Store.Set("KarmaList", []byte(list))
+}
+
+func getDevKarmaList(r *hal.Response) []string {
+	val, _ := r.Robot.Store.Get("KarmaList")
+	list := string(val)
+	names := strings.Split(list, ",")
+
+	return names
+}
+
+func inSlice(n []string, name string) bool {
+	for _, val := range n {
+		if val == name {
+			return true
+		}
+	}
+	return false
+}

main.go

@@ -42,8 +42,7 @@ var selfHandler = hear(`who are you`, "self", "", func(res *hal.Response) error
 })
 
 var quitHandler = hear(`(.*)+/quit(.*)+`, "quit", "", func(res *hal.Response) error {
-	name := res.UserName()
-	return res.Send(fmt.Sprintf("No!  Bad %s!", name))
+	return res.Send(fmt.Sprintf("No!  Bad %s!", res.UserName()))
 })
 
 var helpHandler = hear(`help`, "help", "Displays this message", func(res *hal.Response) error {
@@ -127,6 +126,7 @@ func run() int {
 		devlunchRSVPHandler,
 		karmaHandler,
 		karmaStatsHandler,
+		karmaRankingHandler,
 	)
 
 	if err := robot.Run(); err != nil {

sprunge.go

@@ -0,0 +1,27 @@
+package main
+
+import (
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"net/url"
+)
+
+func sprungeSend(t string) (string, error) {
+	resp, err := http.PostForm("http://sprunge.us",
+		url.Values{"sprunge": {t}})
+	if err != nil {
+		return "", fmt.Errorf("unable to POST to sprunge.us server: %v", err)
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != 200 {
+		return "", fmt.Errorf("response from the server: %s", resp.Status)
+	}
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		return "", fmt.Errorf("unable to parse response from the server: %v", err)
+	}
+
+	return string(body), nil
+}