From d1b167c00295d46d520c9f7dd8319dbb7b734917 Mon Sep 17 00:00:00 2001
From: Hykilpikonna <admin@moecraft.cc>
Date: Sun, 19 Apr 2020 19:27:55 -0400
Subject: [PATCH] [+] Map classinfo to courseinfo

---
 src/logic/course-info.ts                       |  1 +
 src/pages/course-selection/course-selection.ts | 10 +++++++++-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/logic/course-info.ts b/src/logic/course-info.ts
index 49769b4..f82f472 100644
--- a/src/logic/course-info.ts
+++ b/src/logic/course-info.ts
@@ -12,6 +12,7 @@ export default class CourseInfo
     courseCount: number
     gradeLevels: number[]
     enrollments: number
+    classes: ClassInfo[]
 
     /**
      * Construct with a json object
diff --git a/src/pages/course-selection/course-selection.ts b/src/pages/course-selection/course-selection.ts
index d8b9a43..f2390e2 100644
--- a/src/pages/course-selection/course-selection.ts
+++ b/src/pages/course-selection/course-selection.ts
@@ -38,7 +38,15 @@ export default class CourseSelection extends Vue
                     let info = new CourseInfo(json);
 
                     // Index
-                    info.courseIds.forEach(id => this.courseIdIndex[id] = index);
+                    info.courseIds.forEach(id =>
+                    {
+                        this.courseIdIndex[id] = index;
+
+                        // Add class info into course
+                        let classInfo = this.classes.find(c => c.id == id)
+                        if (classInfo == null) return;
+                        info.classes.push(classInfo);
+                    });
                     return info;
                 });
                 this.directory = result.data.studentInfos;