arbeitszeitmessung/Backend/endpoints/time.go

package endpoints

import (
	"arbeitszeitmessung/helper"
	"arbeitszeitmessung/models"
	"arbeitszeitmessung/templates"
	"context"
	"encoding/json"
	"log"
	"net/http"
	"strconv"
	"time"
)

// Frontend relevant backend functionality -> not used by the arduino devices
func TimeHandler(w http.ResponseWriter, r *http.Request) {
	helper.SetCors(w)
	switch r.Method {
	case "GET":
		getBookings(w, r)
	case "POST":
		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 parseTimestamp(r *http.Request , get_key string, fallback string) (time.Time, error) {
	_timestamp_get := r.URL.Query().Get(get_key)
	if(_timestamp_get == "") {
		_timestamp_get = fallback
	}
	timestamp_get, err := time.Parse("2006-01-02", _timestamp_get)
	if(err != nil){
		return time.Now(), err
	}
	return timestamp_get, nil
}


// Returns bookings from DB with similar card uid -> checks for card uid in http query params
func getBookings(w http.ResponseWriter, r *http.Request) {
	var user models.User
	var err error
	if(helper.GetEnv("GO_ENV", "production") == "debug"){
		user, err = (*models.User).GetByPersonalNummer(nil, 123)
	}else{
		if(!Session.Exists(r.Context(), "user")){
			log.Println("No user in session storage!")
			http.Error(w, "Not logged in!", http.StatusForbidden)
			return
		}

		user, err = (*models.User).GetByPersonalNummer(nil, Session.GetInt(r.Context(), "user"))
	}
	if(err != nil){
		log.Println("No user found with the given personal number!")
		http.Error(w, "No user found", http.StatusNotFound)
		return
	}


	tsFrom, err := parseTimestamp(r, "time_from", "2000-01-01")
	if(err != nil ){
		log.Println("Error parsing 'from' time", err)
		http.Error(w, "Timestamp 'from' cannot be parsed!", http.StatusBadRequest)
		return
	}
	tsTo, err := parseTimestamp(r, "time_to", time.Now().Format("2006-01-02"))
	if(err != nil ){
		log.Println("Error parsing 'to' time", err)
		http.Error(w, "Timestamp 'to' cannot be parsed!", http.StatusBadRequest)
		return
	}
	tsTo = tsTo.AddDate(0,0,1) // so that today is inside

	bookings, err := (*models.Booking).GetBookingsGrouped(nil, user.CardUID, tsFrom, tsTo)
	if err != nil {
		log.Println("Error getting bookings: ", err)
		http.Error(w, "Internal Server Error", http.StatusInternalServerError)
		return
	}
	ctx := context.WithValue(r.Context(), "user", user)
	templates.TimeDashboard(bookings).Render(ctx, w)
	// w.Header().Set("Content-Type", "application/json")
	// json.NewEncoder(w).Encode(bookings)
}

func updateBooking(w http.ResponseWriter, r *http.Request){
	r.ParseForm()
	for index, possibleBooking := range r.PostForm{
		if(index[:7] == "booking"){
			booking_id, err := strconv.Atoi(index[8:])
			if(err != nil){
				log.Println("Error parsing bookingId", err)
				continue
			}
			booking, err := (*models.Booking).GetBookingById(nil, booking_id)
			if(err != nil){
				log.Println("Error getting booking!", err)
				continue
			}
			parsedTime, err := time.ParseInLocation("15:04", possibleBooking[0], time.Local)
			if(err != nil){
				log.Println("Error parsing time!", err)
				continue
			}
			log.Println("Parsing time", parsedTime)
			booking.UpdateTime(parsedTime)
		}
	}
	getBookings(w, r)
}

// Updates a booking form the given json body
func updateBookingAPI(w http.ResponseWriter, r *http.Request) {
	_booking_id := r.URL.Query().Get("counter_id")
	if _booking_id == "" {
		http.Error(w, "Missing bookingID query parameter", http.StatusBadRequest)
		return
	}
	booking_id, err := strconv.Atoi(_booking_id)
	if err != nil {
		http.Error(w, "Invalid bookingID query parameter", http.StatusBadRequest)
		return
	}
	bookingDB, err := (*models.Booking).GetBookingById(nil, booking_id)
	if err != nil {
		log.Println("Error getting booking: ", err)
		http.Error(w, "Internal Server Error", http.StatusInternalServerError)
		return
	}
	var booking models.Booking
	dec := json.NewDecoder(r.Body)
	dec.DisallowUnknownFields()
	err = dec.Decode(&booking)
	if err != nil {
		log.Println("Error parsing booking: ", err)
		http.Error(w, "Internal Server Error", http.StatusInternalServerError)
		return
	}
	if booking.CounterId != 0 && booking.CounterId != bookingDB.CounterId {
		log.Println("Booking Ids do not match")
		http.Error(w, "Booking Ids do not match", http.StatusBadRequest)
		return
	}
	bookingDB.Update(booking)
	bookingDB.Save()
	w.Header().Set("Content-Type", "application/json")
	json.NewEncoder(w).Encode(bookingDB)
}