Merge branch 'main' of github.com:masterElmar/htwkalender into 10-roomfinder

backend/service/db/dbEvents.go

@@ -3,78 +3,81 @@ package db
 import (
 	"github.com/pocketbase/dbx"
 	"github.com/pocketbase/pocketbase"
-	"github.com/pocketbase/pocketbase/models"
 	"htwkalender/model"
-	"log"
 )
 
-func SaveEvents(seminarGroup []model.SeminarGroup, collection *models.Collection, app *pocketbase.PocketBase) ([]*models.Record, error) {
-	var toBeSavedEvents []struct {
-		model.Event
-		string
-	}
-	var savedRecords []*models.Record
-	var insertRecords []*models.Record
+func SaveEvents(seminarGroup []model.SeminarGroup, app *pocketbase.PocketBase) ([]model.Event, error) {
+	var toBeSavedEvents model.Events
+	var savedRecords model.Events
 
 	// check if event is already in database and add to toBeSavedEvents if not
 	for _, seminarGroup := range seminarGroup {
 		for _, event := range seminarGroup.Events {
-			dbGroup, err := findEventByDayWeekStartEndNameCourse(event, seminarGroup.Course, app)
+			event = event.SetCourse(seminarGroup.Course)
+			existsInDatabase, err := findEventByDayWeekStartEndNameCourse(event, seminarGroup.Course, app)
+			alreadyAddedToSave := toBeSavedEvents.Contains(event)
 
-			if dbGroup == nil && err.Error() == "sql: no rows in result set" {
-				toBeSavedEvents = append(toBeSavedEvents, struct {
-					model.Event
-					string
-				}{event, seminarGroup.Course})
-			} else if err != nil {
+			if err != nil {
 				return nil, err
 			}
+
+			if !existsInDatabase && !alreadyAddedToSave {
+				toBeSavedEvents = append(toBeSavedEvents, event)
+			}
 		}
 	}
 
 	// create record for each event that's not already in the database
 	for _, event := range toBeSavedEvents {
-		record := models.NewRecord(collection)
-		record.Set("Day", event.Day)
-		record.Set("Week", event.Week)
-		record.Set("Start", event.Start)
-		record.Set("End", event.End)
-		record.Set("Name", event.Name)
-		record.Set("EventType", event.EventType)
-		record.Set("Prof", event.Prof)
-		record.Set("Rooms", event.Rooms)
-		record.Set("Notes", event.Notes)
-		record.Set("BookedAt", event.BookedAt)
-		record.Set("course", event.string)
-		record.Set("semester", event.Semester)
-		insertRecords = append(insertRecords, record)
-	}
-
-	// save all records
-	for _, record := range insertRecords {
-		if record != nil {
-			err := app.Dao().SaveRecord(record)
-			if err == nil {
-				savedRecords = append(savedRecords, record)
-			} else {
-				log.Println("Error while saving record: ", err)
-				return nil, err
-			}
+		event.MarkAsNew()
+		// auto mapping for event fields to record fields
+		err := app.Dao().Save(&event)
+		if err != nil {
+			return nil, err
+		} else {
+			savedRecords = append(savedRecords, event)
 		}
 	}
 
 	return savedRecords, nil
 }
 
-func findEventByDayWeekStartEndNameCourse(event model.Event, course string, app *pocketbase.PocketBase) (*model.Event, error) {
-	err := app.Dao().DB().Select("*").From("events").Where(
-		dbx.NewExp("Day = {:day} AND Week = {:week} AND Start = {:start} AND End = {:end} AND Name = {:name} AND course = {:course} AND Prof = {:prof} AND Rooms = {:rooms} AND EventType = {:eventType}",
-			dbx.Params{"day": event.Day, "week": event.Week, "start": event.Start, "end": event.End, "name": event.Name, "course": course, "prof": event.Prof, "rooms": event.Rooms, "eventType": event.EventType}),
-	).One(&event)
+// check if event is already in database and return true if it is and false if it's not
+func findEventByDayWeekStartEndNameCourse(event model.Event, course string, app *pocketbase.PocketBase) (bool, error) {
+
+	var dbEvent model.Event
+
+	err := app.Dao().DB().Select("*").From("events").
+		Where(dbx.NewExp(
+			"Day = {:day} AND "+
+				"Week = {:week} AND "+
+				"Start = {:start} AND "+
+				"End = {:end} AND "+
+				"Name = {:name} AND "+
+				"course = {:course} AND "+
+				"Prof = {:prof} AND "+
+				"Rooms = {:rooms} AND "+
+				"EventType = {:eventType}",
+			dbx.Params{
+				"day":       event.Day,
+				"week":      event.Week,
+				"start":     event.Start,
+				"end":       event.End,
+				"name":      event.Name,
+				"course":    course,
+				"prof":      event.Prof,
+				"rooms":     event.Rooms,
+				"eventType": event.EventType}),
+		).One(&dbEvent)
+
 	if err != nil {
-		return nil, err
+		if err.Error() == "sql: no rows in result set" {
+			return false, nil
+		}
+		return false, err
+	} else {
+		return true, nil
 	}
-	return &event, err
 }
 
 func buildIcalQueryForModules(modules []model.FeedCollection) dbx.Expression {
@@ -176,10 +179,21 @@ func DeleteAllEventsForCourse(app *pocketbase.PocketBase, course string, semeste
 	return nil
 }
 
-func FindAllEventsByModule(app *pocketbase.PocketBase, moduleName string) (model.Events, error) {
+func DeleteAllEvents(app *pocketbase.PocketBase) error {
+
+	_, err := app.Dao().DB().Delete("events", dbx.NewExp("1=1")).Execute()
+
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func FindAllEventsByModule(app *pocketbase.PocketBase, module model.Module) (model.Events, error) {
 	var events model.Events
 
-	err := app.Dao().DB().Select("*").From("events").Where(dbx.NewExp("Name = {:moduleName}", dbx.Params{"moduleName": moduleName})).All(&events)
+	err := app.Dao().DB().Select("*").From("events").Where(dbx.NewExp("Name = {:moduleName} AND course = {:course}", dbx.Params{"moduleName": module.Name, "course": module.Course})).All(&events)
 	if err != nil {
 		print("Error while getting events from database: ", err)
 		return nil, err