fix api link

This commit is contained in:
Gabe Farrell 2023-04-17 20:19:19 -04:00
parent b9e45987e4
commit a5e620a266
5 changed files with 87 additions and 20 deletions

View file

@ -47,10 +47,10 @@ func GetBudget(w http.ResponseWriter, r *http.Request) {
var response BudgetResponse
response.Budget = user.Budget
response.BudgetCategories = user.Categories
cats := make([]string, len(user.Categories))
response.BudgetCategories = user.BudgetCategories
cats := make([]string, len(user.BudgetCategories))
i := 0
for k := range user.Categories {
for k := range user.BudgetCategories {
cats[i] = k
i++
}
@ -120,15 +120,16 @@ func SetCategoryBudget(w http.ResponseWriter, r *http.Request) {
Decimal: newDecimal,
}
if user.Categories == nil {
user.Categories = make(map[string]money.Money)
if user.BudgetCategories == nil {
user.BudgetCategories = make(map[string]money.Money)
user.Categories = append(user.Categories, cat)
}
user.Categories[cat] = newBudget
user.BudgetCategories[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}}}}
update := bson.D{primitive.E{Key: "$set", Value: bson.D{primitive.E{Key: "categories", Value: user.BudgetCategories}}}}
_, err = userCollection.UpdateOne(context.TODO(), filter, update, opts)
if err != nil {
w.WriteHeader(http.StatusUnauthorized)