feat: plannedRoomForStudent

obcode · Jan 7, 2025 · d19cf92 · d19cf92
1 parent 8c799e0
commit d19cf92
Show file tree

Hide file tree

Showing 5 changed files with 199 additions and 11 deletions.
diff --git a/graph/generated/generated.go b/graph/generated/generated.go
diff --git a/graph/nta.resolvers.go b/graph/nta.resolvers.go
diff --git a/graph/room.graphqls b/graph/room.graphqls
@@ -3,6 +3,7 @@ extend type Query {
   plannedRoomNames: [String!]
   plannedRoomNamesInSlot(day: Int!, time: Int!): [String!]
   plannedRoomsInSlot(day: Int!, time: Int!): [PlannedRoom!]
+  plannedRoomForStudent(ancode: Int!, mtknr: String!): PlannedRoom
 }
 
 type Room {

diff --git a/graph/room.resolvers.go b/graph/room.resolvers.go
diff --git a/plexams/rooms.go b/plexams/rooms.go
@@ -524,6 +524,25 @@ func (p *Plexams) PlannedRoomsInSlot(ctx context.Context, day int, time int) ([]
 	return rooms, nil
 }
 
+func (p *Plexams) PlannedRoomForStudent(ctx context.Context, ancode int, mtknr string) (*model.PlannedRoom, error) {
+	plannedRoomsForExam, err := p.dbClient.PlannedRoomsForAncode(ctx, ancode)
+	if err != nil {
+		log.Error().Err(err).Int("ancode", ancode).Msg("cannot get planned rooms for ancode")
+		return nil, err
+	}
+	for _, room := range plannedRoomsForExam {
+		for _, student := range room.StudentsInRoom {
+			if student == mtknr {
+				return room, nil
+			}
+		}
+	}
+
+	err = fmt.Errorf("student %s not found in planned rooms for ancode %d", mtknr, ancode)
+	log.Error().Err(err).Int("ancode", ancode).Str("mtknr", mtknr).Msg("student not found in planned rooms")
+	return nil, err
+}
+
 // func enhancePlannedRooms(plannedRooms []*model.PlannedRoom) []*model.EnhancedPlannedRoom {
 // 	enhancedPlannedRooms := make([]*model.EnhancedPlannedRoom, 0, len(plannedRooms))
 // 	for _, room := range plannedRooms {