Populate RSVP form

1 files changed, 26 insertions, 2 deletions

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) {