From 865d3fcb6780e209d209e4e7735c318ef32733dc Mon Sep 17 00:00:00 2001
From: Michael Hunteman <michael@huntm.net>
Date: Sat, 6 Jul 2024 16:27:15 -0700
Subject: Populate RSVP form

---
 server/guest/store.go | 28 ++++++++++++++++++++++++++--
 1 file changed, 26 insertions(+), 2 deletions(-)

(limited to 'server')

diff --git a/server/guest/store.go b/server/guest/store.go
index 525bb71..db9fc3d 100644
--- a/server/guest/store.go
+++ b/server/guest/store.go
@@ -25,16 +25,40 @@ func (m MemStore) FindGuest(creds Credentials) (Guest, error) {
 	}
 	defer rows.Close()
 
+	found := false
 	for rows.Next() {
 		err := rows.Scan(&guest.Id, &guest.FirstName, &guest.LastName, &guest.Attendance, &guest.Email, &guest.Message, &guest.PartySize)
 		if err != nil {
 			return guest, err
 		}
 		if guest.FirstName == creds.FirstName && guest.LastName == creds.LastName {
-			return guest, nil
+			found = true
+			break
 		}
 	}
-	return guest, errors.New("Guest does not exist")
+
+	rows, err = m.db.Query(context.Background(), "select * from party")
+	if err != nil {
+		return guest, err
+	}
+	defer rows.Close()
+
+	for rows.Next() {
+		var guestId int
+		var partyGuest PartyGuest
+		err := rows.Scan(&guestId, &partyGuest.FirstName, &partyGuest.LastName)
+		if err != nil {
+			return guest, err
+		}
+		if guestId == guest.Id {
+			guest.PartyList = append(guest.PartyList, partyGuest)
+		}
+	}
+
+	if found {
+		return guest, nil
+	}
+	return guest, errors.New("Guest not found")
 }
 
 func (m MemStore) Get() ([]Guest, error) {
-- 
cgit v1.2.3