feat: claimable gifts

context/context.go

@@ -45,3 +45,21 @@ func (ctx *Context) WishlistDelete(w http.ResponseWriter, r *http.Request) {
 	}
 	http.Redirect(w, r, "/", http.StatusSeeOther)
 }
+
+func (ctx *Context) UpdateClaim(w http.ResponseWriter, r *http.Request) {
+	user := ctx.Auth.ExpectUser(r)
+	if err := r.ParseForm(); err != nil {
+		http.Error(w, err.Error(), http.StatusBadRequest)
+		return
+	}
+	userReference := r.PathValue("userReference")
+	claims := r.Form["claim"]
+	unclaims := r.Form["unclaim"]
+	err := user.ClaimGifts(claims, unclaims)
+	println("err?", err)
+	if err != nil {
+		http.Error(w, "Failed to update claim...", http.StatusInternalServerError)
+		return
+	}
+	http.Redirect(w, r, "/"+userReference, http.StatusSeeOther)
+}

context/foreign_wishlist.go

@@ -8,7 +8,9 @@ import (
 )
 
 type ForeignWishlistProps struct {
+	CurrentUserId string
 	Username      string
+	UserReference string
 	Gifts         []db.Gift
 }
 
@@ -31,9 +33,10 @@ func (ctx *Context) ViewForeignWishlist(w http.ResponseWriter, r *http.Request)
 	}
 	gifts, err := otherUser.GetGifts()
 	if err != nil {
+		log.Printf("An error occurred while fetching %s's wishlist: %s\n", otherUser.Name, err)
 		http.Error(w, "An error occurred while fetching this user's wishlist :(", http.StatusInternalServerError)
 		return
 	}
-	p := ForeignWishlistProps{Username: otherUser.Name, Gifts: gifts}
+	p := ForeignWishlistProps{CurrentUserId: user.Id, Username: otherUser.Name, UserReference: userReference, Gifts: gifts}
 	templates.Execute(w, "foreign_wishlist.gotmpl", p)
 }

db/user.go

@@ -16,6 +16,8 @@ type User struct {
 type Gift struct {
 	Id           string
 	Name         string
+	ClaimantId   string
+	ClaimantName string
 }
 
 func queryForUser(query string, args ...any) (*User, error) {
@@ -78,7 +80,7 @@ func (u *User) GetPassHash() ([]byte, error) {
 }
 
 func (u *User) GetGifts() ([]Gift, error) {
-	stmt := "SELECT gift.id, gift.name FROM gift JOIN user ON gift.recipient_id = user.id WHERE user.id = ?"
+	stmt := "SELECT gift.id, gift.name, claimant.id, claimant.name FROM gift JOIN user ON gift.recipient_id = user.id LEFT JOIN user AS claimant ON gift.claimant_id = claimant.id WHERE user.id = ? ORDER BY gift.name DESC"
 	rows, err := database.Query(stmt, u.Id)
 	if err != nil {
 		return nil, err
@@ -88,10 +90,14 @@ func (u *User) GetGifts() ([]Gift, error) {
 	for rows.Next() {
 		var id string
 		var name string
-		rows.Scan(&id, &name)
+		var claimantId string
+		var claimantName string
+		rows.Scan(&id, &name, &claimantId, &claimantName)
 		gift := Gift{
 			Id:           id,
 			Name:         name,
+			ClaimantId:   claimantId,
+			ClaimantName: claimantName,
 		}
 		gifts = append(gifts, gift)
 	}
@@ -123,3 +129,38 @@ func (u *User) RemoveGift(id string) error {
 	}
 	return nil
 }
+
+func (u *User) executeClaims(tx *sql.Tx, claims, unclaims []string) error {
+	claimStmt := "UPDATE gift SET claimant_id = ? WHERE id = ?"
+	unclaimStmt := "UPDATE gift SET claimant_id = NULL WHERE id = ?"
+	for _, id := range claims {
+		println("exec claim:", claimStmt, u.Id, id)
+		_, err := tx.Exec(claimStmt, u.Id, id)
+		if err != nil {
+			return err
+		}
+	}
+	for _, id := range unclaims {
+		_, err := tx.Exec(unclaimStmt, id)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func (u *User) ClaimGifts(claims, unclaims []string) error {
+	tx, err := database.Begin()
+	if err != nil {
+		return err
+	}
+
+	err = u.executeClaims(tx, claims, unclaims)
+	if err != nil {
+		err = tx.Rollback()
+		return err
+	}
+
+	err = tx.Commit()
+	return err
+}

main.go

@@ -36,6 +36,7 @@ func main() {
 
 	protectedMux.HandleFunc("GET /", ctx.Home)
 	protectedMux.HandleFunc("GET /{userReference}", ctx.ViewForeignWishlist)
+	protectedMux.HandleFunc("POST /{userReference}/update_claim", ctx.UpdateClaim)
 	protectedMux.HandleFunc("POST /wishlist/add", ctx.WishlistAdd)
 	protectedMux.HandleFunc("POST /wishlist/delete", ctx.WishlistDelete)
 	protectedMux.HandleFunc("POST /logout", authMiddleware.LogoutPost)

templates/foreign_wishlist.gotmpl

@@ -8,9 +8,33 @@
   </ul>
 </nav>
 <h2>{{.Username}}'s list</h2>
+<form method="post" action="/{{.UserReference}}/update_claim" autocomplete="off">
+  <input type="submit" value="Update">
   <ul>
     {{range .Gifts}}
-  <li>{{.Name}}</li>
+    <li>
+      {{if eq .ClaimantId $.CurrentUserId}}
+      <label>
+        <input type="checkbox" name="unclaim" value="{{.Id}}">
+        {{.Name}}
+      </label>
+      <span style="color: blue;">Claimed by you</span>
+      {{else}}
+        {{if .ClaimantId}}
+        <label>
+          <input type="checkbox" disabled>
+          {{.Name}}
+        </label>
+        <span style="color: red;">Claimed by {{.ClaimantName}}</span>
+        {{else}}
+        <label>
+          <input type="checkbox" name="claim" value="{{.Id}}">
+          {{.Name}}
+        </label>
+        {{end}}
+      {{end}}
+    </li>
     {{end}}
   </ul>
+</form>
 {{end}}