Adding legacy API paths

2024-08-06 22:36:58 -05:00
parent bb44e26dd4
commit a9ca12395e
4 changed files with 23 additions and 5 deletions
@@ -0,0 +1,136 @@
+package backend
+
+import (
+	"encoding/json"
+	"log"
+	"net/http"
+	"time"
+
+	"github.com/google/uuid"
+)
+
+type CountLog struct {
+	ID       string `json:"id" bson:"_id"`
+	Platform string `json:"platform" bson:"platform"`
+	Date     int    `json:"date" bson:"date"`
+}
+
+func (c CountLog) GetID() string {
+	return c.ID
+}
+
+type countLogReq struct {
+	ID       string
+	Platform string
+}
+
+func (b *Backend) countLog(w http.ResponseWriter, r *http.Request) {
+	hdr, err := b.VerifyHeader(w, r, "count", false)
+	if hdr == nil {
+		if err != nil {
+			log.Println("request key parsing error:", err)
+		}
+		return
+	}
+	defer r.Body.Close()
+	var req countLogReq
+	err = json.NewDecoder(r.Body).Decode(&req)
+	if err != nil || req.Platform == "" {
+		if r.URL.Query().Get("platform") != "" {
+			//TODO: remove legacy code
+			req.Platform = r.URL.Query().Get("platform")
+		} else {
+			ReturnError(w, http.StatusBadRequest, "invalidBody", "Bad request")
+			return
+		}
+	}
+	ap := b.GetApp(hdr.Key)
+	count := ap.CountTable()
+	if count == nil {
+		ReturnError(w, http.StatusInternalServerError, "misconfigured", "Server Misconfigured")
+		return
+	}
+	curDate := getDate(time.Now())
+	if req.ID == "" {
+		err = addToCountTable(w, count, req.Platform, curDate)
+		if err != nil {
+			log.Println("error adding to count table:", err)
+		}
+		return
+	}
+	l, err := count.Get(req.ID)
+	if err == ErrNotFound {
+		err = addToCountTable(w, count, req.Platform, curDate)
+		if err != nil {
+			log.Println("error adding to count table:", err)
+		}
+		return
+	}
+	if l.Date >= curDate {
+		json.NewEncoder(w).Encode(map[string]string{"id": req.ID})
+		w.WriteHeader(http.StatusCreated)
+		return
+	}
+	err = count.PartUpdate(req.ID, map[string]any{"date": curDate})
+	if err != nil {
+		log.Println("error updating count log:", err)
+		ReturnError(w, http.StatusInternalServerError, "internal", "Server error")
+		return
+	}
+	w.WriteHeader(http.StatusCreated)
+	json.NewEncoder(w).Encode(map[string]string{"id": req.ID})
+}
+
+func addToCountTable(w http.ResponseWriter, c CountTable, platform string, curDate int) error {
+	id, err := uuid.NewV7()
+	if err != nil {
+		log.Println("error generating new log UUID:", err)
+		ReturnError(w, http.StatusInternalServerError, "internal", "Server error")
+		return err
+	}
+	err = c.Insert(CountLog{
+		ID:       id.String(),
+		Platform: platform,
+		Date:     curDate,
+	})
+	if err != nil {
+		log.Println("error inserting new count log:", err)
+		ReturnError(w, http.StatusInternalServerError, "internal", "Server error")
+		return err
+	}
+	w.WriteHeader(http.StatusCreated)
+	json.NewEncoder(w).Encode(map[string]string{"id": id.String()})
+	return nil
+}
+
+func (b *Backend) getCount(w http.ResponseWriter, r *http.Request) {
+	hdr, err := b.VerifyHeader(w, r, "management", true)
+	if hdr == nil {
+		if err != nil {
+			log.Println("request key parsing error:", err)
+		}
+		return
+	}
+	var ap App
+	if hdr.Key.AppID == b.managementKeyID {
+		ap = b.apps[r.PathValue("appID")]
+		if ap == nil {
+			ReturnError(w, http.StatusBadRequest, "badRequest", "Bad request")
+			return
+		}
+	} else {
+		ap = b.GetApp(hdr.Key)
+	}
+	count := ap.CountTable()
+	if count == nil {
+		ReturnError(w, http.StatusBadRequest, "badRequest", "Trying to get user count on app that doesn't have a count table")
+		return
+	}
+	out, err := count.Count(r.URL.Query().Get("platform"))
+	if err != nil {
+		log.Println("error getting count:", err)
+		ReturnError(w, http.StatusInternalServerError, "internal", "Server error")
+		return
+	}
+	json.NewEncoder(w).Encode(map[string]int{"count": out})
+}