feat: added courses table

vitty-backend-api/cli/commands/timetableCommands.go

@@ -22,6 +22,13 @@ var TimetableCommands = []*cli.Command{
 		Usage:   "Fix slot times",
 		Action:  fixSlotTimes,
 	},
+	{
+
+		Name:    "seed-course-table",
+		Aliases: []string{"sct"},
+		Usage:   "populate course table",
+		Action:  seedCourseTable,
+	},
 }
 
 func parseTimetable(c *cli.Context) error {
@@ -73,3 +80,31 @@ func fixSlotTimes(c *cli.Context) error {
 	}
 	return nil
 }
+
+func seedCourseTable(c *cli.Context) error {
+	reset := "\033[0m"
+	red := "\033[31m"
+	green := "\033[32m"
+	cyan := "\033[36m "
+
+	fmt.Print(cyan, "Seeding.. ", reset)
+	err := database.DB.Exec(`
+		INSERT INTO courses (course_id, course_name)
+		SELECT
+		DISTINCT  ON(elems.data->>'code')
+			elems.data->>'code' AS CourseCode,
+			elems.data->>'name' AS CourseName
+		FROM
+			timetables,
+			jsonb_array_elements(timetables.slots::jsonb) AS elems(data)	
+	`).Error
+
+	if err != nil {
+		fmt.Println(red, "Failed")
+		fmt.Println("Error: ", err, reset)
+	}
+
+	fmt.Println(green, "Done", reset)
+
+	return nil
+}

vitty-backend-api/internal/models/courses.go

@@ -0,0 +1,6 @@
+package models
+
+type Courses struct {
+	CourseId   string `gorm:"primaryKey"`
+	CourseName string
+}

vitty-backend-api/internal/models/initialize.go

@@ -11,6 +11,7 @@ func InitializeModels() {
 		"User":            &User{},
 		"Timetable":       &Timetable{},
 		"Friend Requests": &FriendRequest{},
+		"Courses":         &Courses{},
 	}
 
 	for name, model := range MODELS {