91 lines
2.0 KiB
Go
91 lines
2.0 KiB
Go
package handlers
|
|
|
|
import (
|
|
"slices"
|
|
"time"
|
|
|
|
// "Documents/go_projet/go/pkg/mod/git.coopgo.io/coopgo-platform/fleets@v0.0.0-20230310144446-feb935f8bf4e/storage"
|
|
|
|
"git.coopgo.io/coopgo-platform/fleets/storage"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
func (h FleetsHandler) AddVehicle(vehicle storage.Vehicle) (*storage.Vehicle, error) {
|
|
if vehicle.ID == "" {
|
|
vehicle.ID = uuid.NewString()
|
|
}
|
|
|
|
vehicle.Metadata = map[string]any{
|
|
"created": time.Now(),
|
|
}
|
|
|
|
// Store the account
|
|
if err := h.storage.CreateVehicle(vehicle); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &vehicle, nil
|
|
}
|
|
|
|
func (h FleetsHandler) GetVehicle(id string) (vehicle *storage.Vehicle, err error) {
|
|
vehicle, err = h.storage.GetVehicle(id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
bookings, err := h.storage.GetBookingsForVehicle(id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
vehicle.Bookings = bookings
|
|
return vehicle, nil
|
|
}
|
|
|
|
func (h FleetsHandler) GetVehicles(namespaces []string, filters storage.VehicleFilters) ([]storage.Vehicle, error) {
|
|
vehicles, err := h.storage.GetVehicles(namespaces)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Filter vehicles and bookings using slices.Collect
|
|
return slices.Collect(func(yield func(storage.Vehicle) bool) {
|
|
for _, v := range vehicles {
|
|
bookings, err := h.storage.GetBookingsForVehicle(v.ID)
|
|
if err != nil {
|
|
continue // Skip vehicles with booking errors
|
|
}
|
|
|
|
// Filter deleted bookings unless specifically requested
|
|
if !filters.IncludeDeleted {
|
|
v.Bookings = slices.Collect(func(yieldBooking func(storage.Booking) bool) {
|
|
for _, booking := range bookings {
|
|
if !booking.Deleted {
|
|
if !yieldBooking(booking) {
|
|
return
|
|
}
|
|
}
|
|
}
|
|
})
|
|
} else {
|
|
v.Bookings = bookings
|
|
}
|
|
|
|
if v.MatchesFilters(filters) {
|
|
if !yield(v) {
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}), nil
|
|
}
|
|
|
|
/////////////////update///////////////
|
|
func (h FleetsHandler) UpdateVehicle(vehicle storage.Vehicle) (*storage.Vehicle, error) {
|
|
|
|
// Store the account
|
|
if err := h.storage.UpdateVehicle(vehicle); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &vehicle, nil
|
|
}
|