From 9cecccfcdddf0a4c39288f907c48dd0b842c09c9 Mon Sep 17 00:00:00 2001
From: Chris Sexton <3216719+chrissexton@users.noreply.github.com>
Date: Tue, 3 Oct 2023 10:49:25 -0400
Subject: [PATCH] counter: consolidate multiple counter code

---
 plugins/counter/api.go | 90 +++---------------------------------------
 1 file changed, 6 insertions(+), 84 deletions(-)

diff --git a/plugins/counter/api.go b/plugins/counter/api.go
index 1f6d515..35d3a90 100644
--- a/plugins/counter/api.go
+++ b/plugins/counter/api.go
@@ -30,8 +30,8 @@ func (p *CounterPlugin) registerWeb() {
 	subrouter.Use(httprate.LimitByIP(requests, dur))
 	subrouter.HandleFunc("/api/users/{user}/items/{item}/increment/{delta}", p.mkIncrementByNAPI(1))
 	subrouter.HandleFunc("/api/users/{user}/items/{item}/decrement/{delta}", p.mkIncrementByNAPI(-1))
-	subrouter.HandleFunc("/api/users/{user}/items/{item}/increment", p.mkIncrementAPI(1))
-	subrouter.HandleFunc("/api/users/{user}/items/{item}/decrement", p.mkIncrementAPI(-1))
+	subrouter.HandleFunc("/api/users/{user}/items/{item}/increment", p.mkIncrementByNAPI(1))
+	subrouter.HandleFunc("/api/users/{user}/items/{item}/decrement", p.mkIncrementByNAPI(-1))
 	r.Mount("/", subrouter)
 	r.HandleFunc("/api", p.handleCounterAPI)
 	r.HandleFunc("/", p.handleCounter)
@@ -42,7 +42,10 @@ func (p *CounterPlugin) mkIncrementByNAPI(direction int) func(w http.ResponseWri
 	return func(w http.ResponseWriter, r *http.Request) {
 		userName, _ := url.QueryUnescape(chi.URLParam(r, "user"))
 		itemName, _ := url.QueryUnescape(chi.URLParam(r, "item"))
-		delta, _ := strconv.Atoi(chi.URLParam(r, "delta"))
+		delta, err := strconv.Atoi(chi.URLParam(r, "delta"))
+		if err != nil || delta == 0 {
+			delta = direction
+		}
 
 		secret, pass, ok := r.BasicAuth()
 		if !ok || !p.b.CheckPassword(secret, pass) {
@@ -117,87 +120,6 @@ func (p *CounterPlugin) mkIncrementByNAPI(direction int) func(w http.ResponseWri
 	}
 }
 
-func (p *CounterPlugin) mkIncrementAPI(delta int) func(w http.ResponseWriter, r *http.Request) {
-	return func(w http.ResponseWriter, r *http.Request) {
-		userName, _ := url.QueryUnescape(chi.URLParam(r, "user"))
-		itemName, _ := url.QueryUnescape(chi.URLParam(r, "item"))
-
-		secret, pass, ok := r.BasicAuth()
-		if !ok || !p.b.CheckPassword(secret, pass) {
-			err := fmt.Errorf("unauthorized access")
-			log.Error().
-				Err(err).
-				Msg("error authenticating user")
-			w.WriteHeader(401)
-			j, _ := json.Marshal(struct {
-				Status bool
-				Error  string
-			}{false, err.Error()})
-			fmt.Fprint(w, string(j))
-			return
-		}
-
-		// Try to find an ID if possible
-		id := ""
-		u, err := p.b.DefaultConnector().Profile(userName)
-		if err == nil {
-			id = u.ID
-		}
-
-		item, err := GetUserItem(p.db, userName, id, itemName)
-		if err != nil {
-			log.Error().Err(err).Msg("error finding item")
-			w.WriteHeader(400)
-			j, _ := json.Marshal(struct {
-				Status bool
-				Error  error
-			}{false, err})
-			fmt.Fprint(w, string(j))
-			return
-		}
-
-		body, _ := ioutil.ReadAll(r.Body)
-		postData := map[string]string{}
-		err = json.Unmarshal(body, &postData)
-		personalMsg := ""
-		if inputMsg, ok := postData["message"]; ok {
-			personalMsg = fmt.Sprintf("\nMessage: %s", inputMsg)
-		}
-
-		genericChs := p.cfg.GetArray("counter.channels", []string{})
-		specificChs := p.cfg.GetMap("counter.channelItems", map[string]string{})
-		ch, ok := specificChs[itemName]
-		req := &bot.Request{
-			Conn: p.b.DefaultConnector(),
-			Kind: bot.Message,
-			Msg: msg.Message{
-				User: &u,
-				// Noting here that we're only going to do goals in a "default"
-				// channel even if it should send updates to others.
-				Channel: ch,
-				Body:    fmt.Sprintf("%s += %d", itemName, delta),
-				Time:    time.Now(),
-			},
-			Values: nil,
-			Args:   nil,
-		}
-		msg := fmt.Sprintf("%s changed their %s counter by %d for a total of %d via the amazing %s API. %s",
-			userName, itemName, delta, item.Count+delta, p.cfg.Get("nick", "catbase"), personalMsg)
-		if !ok {
-			for _, ch := range genericChs {
-				p.b.Send(p.b.DefaultConnector(), bot.Message, ch, msg)
-				req.Msg.Channel = ch
-			}
-		} else {
-			p.b.Send(p.b.DefaultConnector(), bot.Message, ch, msg)
-			req.Msg.Channel = ch
-		}
-		item.UpdateDelta(req, delta)
-		j, _ := json.Marshal(struct{ Status bool }{true})
-		fmt.Fprint(w, string(j))
-	}
-}
-
 func (p *CounterPlugin) handleCounter(w http.ResponseWriter, r *http.Request) {
 	index, _ := embeddedFS.ReadFile("index.html")
 	w.Write(index)