implemented assigning users to events

backend/pkg/db/availabilities/userAvailabilities.go

@@ -9,7 +9,9 @@ type eventAvailabilities struct {
 	AvailabilityID int    `db:"availabilityID"`
 }
 
-func Event(eventID int) (map[string]string, error) {
+type AvailabilityMap map[int][]string
+
+func Event(eventID int) (AvailabilityMap, error) {
 	// get the availabilities for the event
 	var availabilitiesRows []eventAvailabilities
 
@@ -17,17 +19,18 @@ func Event(eventID int) (map[string]string, error) {
 		return nil, err
 	} else {
 		// transform the result into a map
-		eventAvailabilities := map[string]string{}
+		eventAvailabilities := AvailabilityMap{}
 
 		// get the availabilities
-		if availabilitiesMap, err := Keys(); err != nil {
-			return nil, err
-		} else {
-			for _, a := range availabilitiesRows {
-				eventAvailabilities[a.UserName] = availabilitiesMap[a.AvailabilityID].AvailabilityName
+		for _, a := range availabilitiesRows {
+			// if there is no slice for this availability, create it
+			if _, exists := eventAvailabilities[a.AvailabilityID]; !exists {
+				eventAvailabilities[a.AvailabilityID] = make([]string, 0)
 			}
 
-			return eventAvailabilities, nil
+			eventAvailabilities[a.AvailabilityID] = append(eventAvailabilities[a.AvailabilityID], a.UserName)
 		}
+
+		return eventAvailabilities, nil
 	}
 }

backend/pkg/db/events/events.go

@@ -32,7 +32,7 @@ type EventWithAssignment struct {
 
 type EventWithAvailabilities struct {
 	EventWithAssignment
-	Availabilities map[string]string `json:"availabilities"`
+	Availabilities availabilities.AvailabilityMap `json:"availabilities"`
 }
 
 type EventCreate struct {
@@ -268,8 +268,22 @@ func User(userName string) ([]EventWithAssignment, error) {
 }
 
 // set the availability of an user for a specific event
-func UserAvailability(userName string, eventID, availabilityID int) error {
+func UserAvailability(eventID, availabilityID int, userName string) error {
 	_, err := db.DB.Exec("INSERT INTO USER_AVAILABILITIES (userName, eventID, availabilityID) VALUES ($1, $2, $3) ON CONFLICT (userName, eventID) DO UPDATE SET availabilityID = $3", userName, eventID, availabilityID)
 
 	return err
 }
+
+// set the assignment of an user to a task for a specific event
+func SetAssignment(eventID, taskID int, userName string) error {
+	_, err := db.DB.Exec("UPDATE USER_ASSIGNMENTS SET userName = $1 WHERE eventID = $2 AND taskID = $3", userName, eventID, taskID)
+
+	return err
+}
+
+// remove the assignment of an user
+func DeleteAssignment(eventID, taskID int) error {
+	_, err := db.DB.Exec("UPDATE USER_ASSIGNMENTS SET userName = null WHERE eventID = $1 AND taskID = $2", eventID, taskID)
+
+	return err
+}