Continued updating the code to work with entities better

src/pages/training/index.tsx

@@ -20,9 +20,15 @@ import TrainingScore from "@/training/TrainingScore";
 import ModuleBadge from "@/components/ModuleBadge";
 import RecordFilter from "@/components/Medium/RecordFilter";
 import useFilterRecordsByUser from "@/hooks/useFilterRecordsByUser";
+import { mapBy, serialize } from "@/utils";
+import { getEntitiesWithRoles } from "@/utils/entities.be";
+import { getAssignmentsByAssignee } from "@/utils/assignments.be";
+import { getEntitiesUsers } from "@/utils/users.be";
+import { EntityWithRoles } from "@/interfaces/entity";
+import { Assignment } from "@/interfaces/results";
 
-export const getServerSideProps = withIronSessionSsr(({req, res}) => {
-	const user = req.session.user;
+export const getServerSideProps = withIronSessionSsr(async ({req, res}) => {
+	const user = req.session.user as User;
 
 	if (!user) {
 		return {
@@ -42,12 +48,16 @@ export const getServerSideProps = withIronSessionSsr(({req, res}) => {
 		};
 	}
 
+	const entityIDs = mapBy(user.entities, 'id')
+	const entities = await getEntitiesWithRoles(entityIDs)
+	const users = await getEntitiesUsers(entityIDs)
+
 	return {
-		props: {user: req.session.user},
+		props: serialize({user, users, entities}),
 	};
 }, sessionOptions);
 
-const Training: React.FC<{user: User}> = ({user}) => {
+const Training: React.FC<{user: User, entities: EntityWithRoles[], users: User[] }> = ({user, entities, users}) => {
 	const [recordUserId, setRecordTraining] = useRecordStore((state) => [state.selectedUser, state.setTraining]);
 	const [filter, setFilter] = useState<"months" | "weeks" | "days" | "assignments">();
 
@@ -193,7 +203,7 @@ const Training: React.FC<{user: User}> = ({user}) => {
 					</div>
 				) : (
 					<>
-						<RecordFilter user={user} filterState={{filter: filter, setFilter: setFilter}} assignments={false}>
+						<RecordFilter users={users} entities={entities} user={user} filterState={{filter: filter, setFilter: setFilter}} assignments={false}>
 							{user.type === "student" && (
 								<>
 									<div className="flex items-center">