CHANGE: added personaldaten db and auto logout function

Backend/main.go

@@ -21,27 +21,35 @@ func main() {
 	}
 	defer models.DB.Close()
 
+	// handles the different http routes
 	http.HandleFunc("/time/new", timeCreateHandler)
 	http.HandleFunc("/time", timeHandler)
+	http.HandleFunc("/logout", logoutHandler)
 
+	// starting the http server
 	fmt.Printf("Server is running at http://localhost:8000 exposed to port %s\n", getEnv("EXPOSED_PORT", "8000"))
 	log.Fatal(http.ListenAndServe(":8080", nil))
 }
 
+// Relevant for arduino inputs -> creates new Booking from get and put method
+// GET only for demo purpose
 func timeCreateHandler(w http.ResponseWriter, r *http.Request) {
 	setCors(w)
+	// switch with request methods
 	switch r.Method {
 	case "PUT":
 		createBooking(w, r)
 	case "GET":
 		createBooking(w, r)
 	case "OPTIONS":
+		// just support options header for non GET Requests from SWAGGER
 		w.WriteHeader(http.StatusOK)
 	default:
 		http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
 	}
 }
 
+// Frontendrelevant backend functionality -> not used by the arduino devices
 func timeHandler(w http.ResponseWriter, r *http.Request) {
 	setCors(w)
 	switch r.Method {
@@ -50,14 +58,49 @@ func timeHandler(w http.ResponseWriter, r *http.Request) {
 	case "PUT":
 		updateBooking(w, r)
 	case "OPTIONS":
+		// just support options header for non GET Requests from SWAGGER
 		w.WriteHeader(http.StatusOK)
 	default:
 		http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
 	}
+}
+
+func logoutHandler(w http.ResponseWriter, r *http.Request) {
+	setCors(w)
+	switch r.Method {
+	case "GET":
+		autoLogout(w, r)
+	default:
+		http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
+	}
+}
+
+func autoLogout(w http.ResponseWriter, r *http.Request) {
+	users, err := (*models.User).GetAll(nil)
+	var logged_out_users []models.User
+	if err != nil {
+		fmt.Printf("Error getting user list %v\n", err)
+	}
+	for _, user := range users {
+		if user.CheckAnwesenheit() {
+			err = user.Logout()
+			if err != nil {
+				fmt.Printf("Error logging out user %v\n", err)
+			}
+			logged_out_users = append(logged_out_users, user)
+		}
+
+	}
+	w.Header().Set("Content-Type", "application/json")
+	w.WriteHeader(http.StatusOK)
+	json.NewEncoder(w).Encode(logged_out_users)
 
 }
 
+// Creates a booking from the http query params -> no body needed
+// after that entry wi'll be written to database and the booking is returned as json
 func createBooking(w http.ResponseWriter, r *http.Request) {
+
 	booking := (*models.Booking).FromUrlParams(nil, r.URL.Query())
 	if booking.Verify() {
 		err := booking.Insert()
@@ -77,13 +120,14 @@ func createBooking(w http.ResponseWriter, r *http.Request) {
 	w.WriteHeader(http.StatusBadRequest)
 }
 
+// Returns bookings from DB with similar card uid -> checks for card uid in http query params
 func getBookings(w http.ResponseWriter, r *http.Request) {
-	card_id := r.URL.Query().Get("card_uid")
-	if card_id == "" {
+	card_uid := r.URL.Query().Get("card_uid")
+	if card_uid == "" {
 		http.Error(w, "Missing cardID query parameter", http.StatusBadRequest)
 		return
 	}
-	bookings, err := (*models.Booking).GetBookingsByCardID(nil, card_id)
+	bookings, err := (*models.Booking).GetBookingsByCardID(nil, card_uid)
 	if err != nil {
 		log.Println("Error getting bookings: ", err)
 		http.Error(w, "Internal Server Error", http.StatusInternalServerError)
@@ -93,6 +137,7 @@ func getBookings(w http.ResponseWriter, r *http.Request) {
 	json.NewEncoder(w).Encode(bookings)
 }
 
+// Updates a booking form the given json body
 func updateBooking(w http.ResponseWriter, r *http.Request) {
 	_booking_id := r.URL.Query().Get("counter_id")
 	if _booking_id == "" {
@@ -130,8 +175,12 @@ func updateBooking(w http.ResponseWriter, r *http.Request) {
 	json.NewEncoder(w).Encode(_booking)
 }
 
-// func getBooking(w http.ResponseWriter, r *http.Request)
-
+// Returns env with default fallback value.
+//
+// Params:
+//
+//	key - enviroment var name
+//	fallback - default value
 func getEnv(key, fallback string) string {
 	if value, ok := os.LookupEnv(key); ok {
 		return value
@@ -139,6 +188,8 @@ func getEnv(key, fallback string) string {
 	return fallback
 }
 
+// setting cors, important for later frontend use
+// in DEBUG == "true" everything is set to "*" so that no cors errors will be happen
 func setCors(w http.ResponseWriter) {
 	if os.Getenv("DEBUG") == "true" {
 		w.Header().Set("Access-Control-Allow-Origin", "*")

Backend/models/booking.go

@@ -25,9 +25,9 @@ type Booking struct {
 
 var DB *sql.DB
 
-func (b Booking) New(card_id string, geraet_id int16, check_in_out int16) Booking {
+func (b *Booking) New(card_uid string, geraet_id int16, check_in_out int16) Booking {
 	return Booking{
-		CardUID:    card_id,
+		CardUID:    card_uid,
 		GeraetID:   geraet_id,
 		CheckInOut: check_in_out,
 	}
@@ -89,6 +89,7 @@ func (b *Booking) GetBookingsByCardID(card_id string) ([]Booking, error) {
 	if err != nil {
 		return nil, err
 	}
+	defer qStr.Close()
 	var bookings []Booking
 	rows, err := qStr.Query(card_id)
 	if err == sql.ErrNoRows {

Backend/models/user.go

@@ -0,0 +1,66 @@
+package models
+
+import (
+	"fmt"
+)
+
+type User struct {
+	CardUID                 string
+	Name                    string
+	Vorname                 string
+	HauptbeschaeftigungsOrt int8
+}
+
+func (u *User) GetAll() ([]User, error) {
+	qStr, err := DB.Prepare((`SELECT card_uid, vorname, nachname, hauptbeschaeftigung_ort FROM personal_daten;`))
+	var users []User
+	if err != nil {
+		fmt.Printf("Error preparing query statement %v\n", err)
+		return users, err
+	}
+	defer qStr.Close()
+	rows, err := qStr.Query()
+	if err != nil {
+		return users, err
+	}
+	defer rows.Close()
+	for rows.Next() {
+		var user User
+		if err := rows.Scan(&user.CardUID, &user.Vorname, &user.Name, &user.HauptbeschaeftigungsOrt); err != nil {
+			return users, nil
+		}
+		users = append(users, user)
+	}
+	if err = rows.Err(); err != nil {
+		return users, nil
+	}
+	return users, nil
+}
+
+// Returns true if there is a booking 1 for today -> meaning the user is at work
+// Returns false if there is no booking today or the user is already booked out of the system
+func (u *User) CheckAnwesenheit() bool {
+	qStr, err := DB.Prepare((`SELECT check_in_out FROM anwesenheit WHERE card_uid = $1 AND "timestamp" >= now()::date + interval '1h' ORDER BY "timestamp" DESC`))
+	if err != nil {
+		fmt.Printf("Error preparing query statement %v\n", err)
+		return false
+	}
+	defer qStr.Close()
+	var check_in_out int
+	err = qStr.QueryRow(u.CardUID).Scan(&check_in_out)
+	if err != nil {
+		return false
+	}
+	return check_in_out == 1
+}
+
+// Creates a new booking for the user -> check_in_out will be 255 for automatic check out
+func (u *User) Logout() error {
+	booking := (*Booking).New(nil, u.CardUID, 0, 255)
+	err := booking.Insert()
+	if err != nil {
+		fmt.Printf("Error inserting booking %v\n", err)
+		return err
+	}
+	return nil
+}