Save responses in Store

back/store.go

@@ -54,7 +54,7 @@ func NewMemoryStore(genString GenString) (*Store, error) {
 		pool:      pool,
 		genString: genString,
 	}
-	log.Println("creating schema")
+
 	err = store.createSchema()
 	if err != nil {
 		defer pool.Close()
@@ -84,13 +84,14 @@ const schema = `
 		id INTEGER PRIMARY KEY,
 		alpha_id TEXT NOT NULL,
 		event_id INTEGER NOT NULL,
+		guest_name TEXT NOT NULL,
 
 		UNIQUE (alpha_id),
 		FOREIGN KEY (event_id) REFERENCES event(id)
 	);
 
 	CREATE TABLE response_time (
-		response_id INTEGER PRIMARY KEY,
+		response_id INTEGER NOT NULL,
 		date DATE NOT NULL,
 		time INTEGER NOT NULL,
 
@@ -103,7 +104,7 @@ const schema = `
 	BEGIN
 		SELECT
 			CASE
-				WHEN NEW.date BETWEEN event.earliest_date AND event.latest_date
+				WHEN NEW.date NOT BETWEEN event.earliest_date AND event.latest_date
 					THEN RAISE(ABORT, 'response date is out of range')
 			END
 		FROM response
@@ -213,36 +214,58 @@ func (s *Store) GetEventMetadata(ctx context.Context, query GetEventMetadataQuer
 	return result, nil
 }
 
-type GetEventResponsesQuery struct {
+type GetEventResponseSummaryQuery struct {
 	AlphaID string
 }
 
-type GetEventResponsesResult struct {
+type GetEventResponseSummaryResult struct {
 	TotalResponses int
 }
 
-func (s *Store) GetEventResponses(ctx context.Context, query GetEventResponsesQuery) (GetEventResponsesResult, error) {
+func (s *Store) GetEventResponseSummary(ctx context.Context, query GetEventResponseSummaryQuery) (GetEventResponseSummaryResult, error) {
 	conn := s.pool.Get(ctx)
 	defer s.pool.Put(conn)
 
-	var result GetEventResponsesResult
-	const dbQuery = `
+	var found bool
+	var eventID int
+
+	const eventIDQuery = `
+		SELECT id
+		FROM event
+		WHERE alpha_id = ?`
+	var err = sqlitex.Exec(conn, eventIDQuery,
+		func(stmt *sqlite.Stmt) error {
+			found = true
+			eventID = stmt.ColumnInt(0)
+			return nil
+		}, query.AlphaID)
+	if err != nil {
+		return GetEventResponseSummaryResult{}, err
+	}
+	if !found {
+		return GetEventResponseSummaryResult{}, errors.New("event not found")
+	}
+
+	var result GetEventResponseSummaryResult
+	const responseCountQuery = `
 		SELECT COUNT(*)
 		FROM response
-		JOIN event ON response.event_id = event.id
-		WHERE event.alpha_id = ?;`
-	err := sqlitex.Exec(conn, dbQuery,
+		WHERE event_id = ?;`
+	err = sqlitex.Exec(conn, responseCountQuery,
 		func(stmt *sqlite.Stmt) error {
 			result.TotalResponses = stmt.ColumnInt(0)
 			return nil
-		}, query.AlphaID)
-	// TODO return an error if the event does not exist?
-	return result, err
+		}, eventID)
+	if err != nil {
+		return GetEventResponseSummaryResult{}, err
+	}
+	return result, nil
 }
 
 type CreateEventResponseCommand struct {
 	EventAlphaID string
-	DateHours    map[date.Date]int
+	GuestName    string
+	DateHours    map[date.Date]map[int]struct{}
 }
 
 type CreateEventResponseResult struct {
@@ -258,22 +281,95 @@ func (s *Store) CreateEventResponse(ctx context.Context, cmd CreateEventResponse
 	defer sqlitex.Save(conn)(&err)
 	responseAlphaID, err := s.genString(responseAlphaIDLength)
 	if err != nil {
-		return CreateEventResponseResult{}, err
+		return
 	}
 
-	// Create response
 	const responseQuery = `
-		INSERT INTO response(event_id, alpha_id)
-			SELECT event.id AS event_id, ? AS alpha_id
+		INSERT INTO response(event_id, alpha_id, guest_name)
+			SELECT event.id AS event_id, ? AS alpha_id, ? as guest_name
 			FROM event
 			WHERE event.alpha_id = ?;`
-	err = sqlitex.Exec(conn, responseQuery, nil, responseAlphaID, cmd.EventAlphaID)
+	err = sqlitex.Exec(conn, responseQuery, nil, responseAlphaID, cmd.GuestName, cmd.EventAlphaID)
 	if err != nil {
-		return CreateEventResponseResult{}, err
+		return
+	}
+
+	const responseTimeQuery = `
+		INSERT INTO response_time(response_id, date, time)
+			SELECT response.id AS response_id, ? AS date, ? AS time
+			FROM response
+			WHERE response.alpha_id = ?;`
+	for d, hs := range cmd.DateHours {
+		for h := range hs {
+			err = sqlitex.Exec(conn, responseTimeQuery, nil, d.Format(dbDateLayout), h, responseAlphaID)
+			if err != nil {
+				return
+			}
+		}
 	}
-	// TODO Create response times
 
 	return CreateEventResponseResult{
 		ResponseAlphaID: responseAlphaID,
 	}, nil
 }
+
+type GetEventResponseQuery struct {
+	EventAlphaID    string
+	ResponseAlphaID string
+}
+
+type GetEventResponseResult struct {
+	GuestName string
+	DateHours map[date.Date]map[int]struct{}
+}
+
+func (s *Store) GetEventResponse(ctx context.Context, query GetEventResponseQuery) (GetEventResponseResult, error) {
+	conn := s.pool.Get(ctx)
+	defer s.pool.Put(conn)
+
+	var found bool
+	var result GetEventResponseResult
+	var responseID int
+
+	const responseQuery = `
+		SELECT guest_name, response.id
+		FROM response
+		JOIN event ON response.event_id = event.id
+		WHERE response.alpha_id = ? AND event.alpha_id = ?;`
+	var err = sqlitex.Exec(conn, responseQuery,
+		func(stmt *sqlite.Stmt) error {
+			found = true
+			result.GuestName = stmt.ColumnText(0)
+			responseID = stmt.ColumnInt(1)
+			return nil
+		}, query.ResponseAlphaID, query.EventAlphaID)
+	if err != nil {
+		return GetEventResponseResult{}, err
+	}
+	if !found {
+		// TODO return a constant or typed error
+		return GetEventResponseResult{}, errors.New("not found")
+	}
+
+	result.DateHours = make(map[date.Date]map[int]struct{})
+	const responseTimeQuery = `
+		SELECT date, time
+		FROM response_time
+		WHERE response_id = ?`
+	err = sqlitex.Exec(conn, responseTimeQuery,
+		func(stmt *sqlite.Stmt) error {
+			d, err := date.Parse(dbDateLayout, stmt.ColumnText(0))
+			if err != nil {
+				return err
+			}
+			if result.DateHours[d] == nil {
+				result.DateHours[d] = make(map[int]struct{})
+			}
+			result.DateHours[d][stmt.ColumnInt(1)] = struct{}{}
+			return nil
+		}, responseID)
+	if err != nil {
+		return GetEventResponseResult{}, err
+	}
+	return result, nil
+}