api for initial widgets

widgets/budget.go

@@ -0,0 +1,173 @@
+package widgets
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"strconv"
+
+	"github.com/jacobmveber-01839764/BudgetBuddy/db"
+	"github.com/jacobmveber-01839764/BudgetBuddy/money"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+type BudgetResponse struct {
+	// total amount allowed to spend in a month
+	Budget money.Money `json:"budget"`
+	// total amount allowed to spend by category
+	BudgetCategories map[string]money.Money `json:"budget_categories"`
+	// categories in the budget
+	Categories []string `json:"categories"`
+	// transactions mapped to a category
+	Expenses map[string][]db.Transaction `json:"expenses"`
+}
+
+func GetBudget(w http.ResponseWriter, r *http.Request) {
+	// get session key from request
+	session := r.Header.Get("x-session-key")
+
+	// get collection handle from db
+	var userCollection = db.Client.Database("budgetbuddy").Collection("users")
+
+	var user = db.UserSchema{}
+
+	err := userCollection.FindOne(context.Background(), bson.D{primitive.E{Key: "session", Value: session}}).Decode(&user)
+	if err != nil {
+		w.WriteHeader(http.StatusUnauthorized)
+		fmt.Fprintf(w, "{\"error\":\"invalid session key\"}")
+		return
+	}
+	var response BudgetResponse
+
+	response.Budget = user.Budget
+	response.BudgetCategories = user.Categories
+	cats := make([]string, len(user.Categories))
+	i := 0
+	for k := range user.Categories {
+		cats[i] = k
+		i++
+	}
+	response.Categories = cats
+	response.Expenses = make(map[string][]db.Transaction)
+	for _, e := range user.Expenses {
+		if response.Expenses[e.Category] == nil {
+			response.Expenses[e.Category] = make([]db.Transaction, 0)
+		}
+		response.Expenses[e.Category] = append(response.Expenses[e.Category], e)
+	}
+
+	ret, err := json.Marshal(response)
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		fmt.Fprintf(w, "{\"error\":\"problem marshalling response\"}")
+		return
+	}
+
+	w.Write(ret)
+
+}
+func SetCategoryBudget(w http.ResponseWriter, r *http.Request) {
+	// get session key from request
+	session := r.Header.Get("x-session-key")
+
+	// get collection handle from db
+	var userCollection = db.Client.Database("budgetbuddy").Collection("users")
+
+	var user = db.UserSchema{}
+
+	err := userCollection.FindOne(context.Background(), bson.D{primitive.E{Key: "session", Value: session}}).Decode(&user)
+	if err != nil {
+		w.WriteHeader(http.StatusUnauthorized)
+		fmt.Fprintf(w, "{\"error\":\"invalid session key\"}")
+		return
+	}
+
+	// get form values
+	r.ParseForm()
+	cat := r.FormValue("category")
+	if cat == "" {
+		w.WriteHeader(http.StatusBadRequest)
+		fmt.Fprintf(w, "{\"error\":\"category must be specified\"}")
+		return
+	}
+	newWhole, err := strconv.Atoi(r.Form.Get("whole"))
+	if err != nil {
+		w.WriteHeader(http.StatusBadRequest)
+		fmt.Fprintf(w, "{\"error\":\"incorrect whole value\"}")
+		return
+	}
+	newDecimal, err := strconv.Atoi(r.Form.Get("decimal"))
+	if err != nil {
+		w.WriteHeader(http.StatusBadRequest)
+		fmt.Fprintf(w, "{\"error\":\"incorrect decimal value\"}")
+		return
+	}
+	// TODO: figure out how to efficiently determing currency
+	newBudget := money.Money{
+		Currency: money.Currency(r.FormValue("currency")),
+		Whole:    newWhole,
+		Decimal:  newDecimal,
+	}
+
+	if user.Categories == nil {
+		user.Categories = make(map[string]money.Money)
+	}
+
+	user.Categories[cat] = newBudget
+
+	filter := bson.D{primitive.E{Key: "session", Value: session}}
+	opts := options.Update().SetUpsert(true)
+	update := bson.D{primitive.E{Key: "$set", Value: bson.D{primitive.E{Key: "categories", Value: user.Categories}}}}
+	_, err = userCollection.UpdateOne(context.TODO(), filter, update, opts)
+	if err != nil {
+		w.WriteHeader(http.StatusUnauthorized)
+		fmt.Fprintf(w, "{\"error\":\"invalid session key\"}")
+		return
+	}
+
+	w.Write([]byte("{\"status\":200}"))
+}
+
+func SetBudget(w http.ResponseWriter, r *http.Request) {
+	// get session key from request
+	session := r.Header.Get("x-session-key")
+
+	// get form values
+	r.ParseForm()
+	newWhole, err := strconv.Atoi(r.Form.Get("whole"))
+	if err != nil {
+		w.WriteHeader(http.StatusBadRequest)
+		fmt.Fprintf(w, "{\"error\":\"incorrect whole value\"}")
+		return
+	}
+	newDecimal, err := strconv.Atoi(r.Form.Get("decimal"))
+	if err != nil || newDecimal < 0 || newDecimal > 99 {
+		w.WriteHeader(http.StatusBadRequest)
+		fmt.Fprintf(w, "{\"error\":\"incorrect decimal value\"}")
+		return
+	}
+	// TODO: figure out how to efficiently determing currency
+	newBudget := money.Money{
+		Currency: money.Currency(r.FormValue("currency")),
+		Whole:    newWhole,
+		Decimal:  newDecimal,
+	}
+
+	// get collection handle from db
+	var userCollection = db.Client.Database("budgetbuddy").Collection("users")
+
+	filter := bson.D{primitive.E{Key: "session", Value: session}}
+	opts := options.Update().SetUpsert(true)
+	update := bson.D{primitive.E{Key: "$set", Value: bson.D{primitive.E{Key: "budget", Value: newBudget}}}}
+	_, err = userCollection.UpdateOne(context.TODO(), filter, update, opts)
+	if err != nil {
+		w.WriteHeader(http.StatusUnauthorized)
+		fmt.Fprintf(w, "{\"error\":\"invalid session key\"}")
+		return
+	}
+
+	w.Write([]byte("{\"status\":200}"))
+}