statbot/processor_general.go

package main

import (
	"fmt"
	"net/http"
	"strings"
)

/*
 * General Message Processor
 */
type generalProcessor struct{}

func (p *generalProcessor) GetName() string {
	return "stat_bot General Processor"
}

func (p *generalProcessor) GetHelp() string {
	return ""
}

func (p *generalProcessor) ProcessMessage(slack *Slack, m *Message)      {}
func (p *generalProcessor) ProcessBotMessage(slack *Slack, m *Message)   {}
func (p *generalProcessor) ProcessAdminMessage(slack *Slack, m *Message) {}
func (p *generalProcessor) ProcessUserMessage(slack *Slack, m *Message)  {}

func (p *generalProcessor) ProcessAdminUserMessage(slack *Slack, m *Message) {
	// Check if we were mentioned
	if strings.HasPrefix(m.Text, "<@"+slack.id+">") {
		parts := strings.Fields(m.Text)
		var action, target string

		if len(parts) >= 2 {
			action = parts[1]
			if len(parts) >= 3 {
				target = parts[2]
			}
		}

		if action == "mkadmin" && target != "" {
			// Make a user an admin
			if strings.HasPrefix(target, "<@") && strings.HasSuffix(target, ">") {
				target = strings.Trim(target, "<@>")
				if e := addAdmin(target); e == nil {
					m.Text = "User <@" + target + "> has been made an admin"
				} else {
					m.Text = fmt.Sprintf("%s", e)
				}
			} else {
				m.Text = "Please specify an existing user starting with a '@'"
			}
			slack.postMessage(*m)
		} else if action == "rmadmin" && target != "" {
			// Revoke a user as an admin
			if strings.HasPrefix(target, "<@") && strings.HasSuffix(target, ">") {
				target = strings.Trim(target, "<@>")
				if e := removeAdmin(target); e == nil {
					m.Text = "Admin privileges revoked from <@" + target + ">"
				} else {
					m.Text = fmt.Sprintf("%s", e)
				}
			} else {
				m.Text = "Please specify an existing user starting with a '@'"
			}
			slack.postMessage(*m)
		} else {
			// huh?
			m.Text = fmt.Sprintf("Beep boop beep, that does not compute\n")
			slack.postMessage(*m)
		}
	}
}
func (p *generalProcessor) ProcessBotUserMessage(slack *Slack, m *Message) {}

func (p *generalProcessor) ProcessChannelMessage(slack *Slack, m *Message)      {}
func (p *generalProcessor) ProcessAdminChannelMessage(slack *Slack, m *Message) {}
func (p *generalProcessor) ProcessBotChannelMessage(slack *Slack, m *Message)   {}

/*
 *General Statistics Processor
 */
type generalStatProcessor struct{}

func (p *generalStatProcessor) GetName() string {
	return "General Statistics"
}

func (p *generalStatProcessor) GetStatKeys() []string {
	return []string{
		"bot-message",
		//"bot-reaction-*",
		"channel-message",
		//"channel-reaction-*",
		"message-hour-*",
		"message-dow-*",
		"message-dom-*",
		//"reaction-*-hour-*",
		//"reaction-*-dow-*",
		//"reaction-*-dom-*",
	}
}

func (p *generalStatProcessor) ProcessMessage(m *Message) {
	if m.Type == "message" {
		incrementUserStat(m.User, "message-hour-"+m.Time.Format("15"))
		incrementUserStat(m.User, "message-dow-"+m.Time.Format("Mon"))
		incrementUserStat(m.User, "message-dom-"+m.Time.Format("02"))
		//} else if m.Type == "reaction_added" {
		//incrementUserStat(m.User, "reaction-"+m.Name+"-hour-"+m.Time.Format("15"))
		//incrementUserStat(m.User, "reaction-"+m.Name+"-dow-"+m.Time.Format("Mon"))
		//incrementUserStat(m.User, "reaction-"+m.Name+"-dom-"+m.Time.Format("02"))
	}
}
func (p *generalStatProcessor) ProcessBotMessage(m *Message) {}

func (p *generalStatProcessor) ProcessUserMessage(m *Message) {
	if m.Type == "message" {
		incrementUserStat(m.User, "bot-message")
		//} else if m.Type == "reaction_added" {
		//	incrementUserStat(m.User, "bot-reaction-"+m.Name)
	}
}
func (p *generalStatProcessor) ProcessBotUserMessage(m *Message) {}

func (p *generalStatProcessor) ProcessChannelMessage(m *Message) {
	if m.Type == "message" {
		incrementUserStat(m.User, "channel-message")

		incrementChannelStat(m.Channel, "message-hour-"+m.Time.Format("15"))
		incrementChannelStat(m.Channel, "message-dow-"+m.Time.Format("Mon"))
		incrementChannelStat(m.Channel, "message-dom-"+m.Time.Format("02"))
		//} else if m.Type == "reaction_added" {
		//	incrementUserStat(m.User, "channel-reaction-"+m.Name)

		//	incrementChannelStat(m.Channel, "reaction-"+m.Name+"-hour-"+m.Time.Format("15"))
		//	incrementChannelStat(m.Channel, "reaction-"+m.Name+"-dow-"+m.Time.Format("Mon"))
		//	incrementChannelStat(m.Channel, "reaction-"+m.Name+"-dom-"+m.Time.Format("02"))
	}
}
func (p *generalStatProcessor) ProcessBotChannelMessage(m *Message) {}

/*
 * Web Site Module
 */
type generalWebModule struct{}

func (wm *generalWebModule) GetName() string {
	return "General Web Module"
}
func (wm *generalWebModule) GetRoutes() map[string]func(http.ResponseWriter, *http.Request) {
	ret := make(map[string]func(http.ResponseWriter, *http.Request))
	ret["/"] = wm.handleStats
	return ret
}
func (wm *generalWebModule) Register() {
	for k, v := range wm.GetRoutes() {
		r.HandleFunc(k, v)
	}
}

func (wm *generalWebModule) GetMenuEntries() []menuItem {
	var ret []menuItem
	ret = append(ret, menuItem{Text: "Stats", Link: "/"})
	return ret
}

func (wm *generalWebModule) GetBottomMenuEntries() []menuItem {
	var ret []menuItem
	return ret
}

func (wm *generalWebModule) handleStats(w http.ResponseWriter, req *http.Request) {
	initRequest(w, req)

	type ChannelStats struct {
		Name         string
		MemberCount  int
		MessageCount int
	}
	type MessageStats struct {
		Hours map[int]int
		Dow   map[string]int
	}
	type UserStats struct {
		Name     string
		Hours    map[int]int
		Dow      map[string]int
		Messages int
	}
	type StatData struct {
		TotalChannelMessages int
		TotalChannels        int
		ChannelStats         []ChannelStats
		UserStats            []UserStats
		MessageStats         MessageStats
		Error                string
	}

	// Get the global stats
	var s StatData
	var err error

	openDatabase()
	chanlst := getChannelList()
	userlst := getUserList()
	var chanstats []ChannelStats
	var userstats []UserStats
	s.MessageStats.Hours = make(map[int]int)
	for i := 0; i < 24; i++ {
		s.MessageStats.Hours[i] = 0
	}
	s.MessageStats.Dow = make(map[string]int)
	for _, k := range []string{"Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"} {
		s.MessageStats.Dow[k] = 0
	}

	for _, k := range chanlst {
		chanstats = append(chanstats, ChannelStats{
			Name:         getChannelName(k),
			MemberCount:  getChannelMemberCount(k),
			MessageCount: getChannelMessageCount(k),
		})
		for i := 0; i < 24; i++ {
			if hrI, err := getChannelStat(k, fmt.Sprintf("message-hour-%0d", i)); err == nil {
				s.MessageStats.Hours[i] = s.MessageStats.Hours[i] + hrI
			}
		}
		for _, dow := range []string{"Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"} {
			if dowI, err := getChannelStat(k, fmt.Sprintf("message-dow-%s", dow)); err == nil {
				s.MessageStats.Dow[dow] = s.MessageStats.Dow[dow] + dowI
			}
		}
	}
	for _, k := range userlst {
		usrName := getUserName(k)
		if usrName == "stat_bot" || usrName == "bot" {
			continue
		}
		usrHourStats := make(map[int]int)
		usrDowStats := make(map[string]int)
		var chanMsg int
		var err error
		if chanMsg, err = getUserStat(k, "channel-message"); err != nil {
			chanMsg = 0
		}
		for i := 0; i < 24; i++ {
			if hrI, err := getUserStat(k, fmt.Sprintf("message-hour-%0d", i)); err == nil {
				usrHourStats[i] = hrI
			} else {
				usrHourStats[i] = 0
			}
		}
		for _, dow := range []string{"Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"} {
			if dowI, err := getUserStat(k, fmt.Sprintf("message-dow-%s", dow)); err == nil {
				usrDowStats[dow] = dowI
			} else {
				usrDowStats[dow] = 0
			}
		}
		userstats = append(userstats, UserStats{
			Name:     usrName,
			Messages: chanMsg,
			Hours:    usrHourStats,
			Dow:      usrDowStats,
		})
	}

	s.ChannelStats = chanstats
	s.UserStats = userstats
	s.TotalChannelMessages = getTotalChannelMsgCount()
	s.TotalChannels = len(chanlst)

	closeDatabase()

	sc := "var stats = {totalchannelmessages:"
	sc = fmt.Sprintf("%s%d,", sc, s.TotalChannelMessages)
	sc = sc + "channels:["
	if len(s.ChannelStats) > 0 {
		for _, k := range s.ChannelStats {
			sc = fmt.Sprintf("%s{name:\"%s\",member_count:%d,message_count:%d},",
				sc,
				k.Name,
				k.MemberCount,
				k.MessageCount,
			)
		}
		// Trim the last ,
		sc = sc[:len(sc)-1]
	}
	sc = sc + "],"
	sc = sc + "users:["
	if len(s.UserStats) > 0 {
		for _, usr := range s.UserStats {
			sc = fmt.Sprintf("%s{name:\"%s\",message_count:%d,",
				sc,
				usr.Name,
				usr.Messages,
			)
			sc = sc + "messages:{"
			sc = sc + "hours:{"
			for i, k := range usr.Hours {
				sc = sc + fmt.Sprintf("%d:%d,", i, k)
			}
			sc = sc[:len(sc)-1]
			sc = sc + "},"
			sc = sc + "dow:{"
			for _, k := range []string{"Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"} {
				sc = sc + fmt.Sprintf("%s:%d,", k, usr.Dow[k])
			}
			sc = sc[:len(sc)-1]
			sc = sc + "}}"
			sc = sc + "},"
		}
		// Trim the last ,
		sc = sc[:len(sc)-1]
	}
	sc = sc + "],"
	// Get all of the message stats
	sc = sc + "messages:{"
	sc = sc + "hours:{"
	for i, k := range s.MessageStats.Hours {
		sc = sc + fmt.Sprintf("%d:%d,", i, k)
	}
	sc = sc[:len(sc)-1]
	sc = sc + "},"
	sc = sc + "dow:{"
	for _, k := range []string{"Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"} {
		sc = sc + fmt.Sprintf("%s:%d,", k, s.MessageStats.Dow[k])
	}
	sc = sc[:len(sc)-1]
	sc = sc + "}}"
	sc = sc + "};"

	addToInlineScript(sc)

	site.Scripts = append(site.Scripts, "/assets/js/main_stats.js")

	if err != nil {
		setFlashMessage("Error Counting Channel Messages", "error")
	}
	site.TemplateData = s

	setMenuItemActive("Stats")
	showPage("stats.html", site, w)
}