From 41bf03682ca992cf147f1e807ad211f351d52235 Mon Sep 17 00:00:00 2001
From: soukainna <soukaina1lafdili@gmail.com>
Date: Fri, 19 May 2023 11:26:36 +0200
Subject: [PATCH] logic to update vehicles

---
 grpcapi/grpcapi.go   | 19 ++++++++++++++++---
 handlers/vehicles.go | 13 +++++++++++++
 storage/mongodb.go   | 11 +++++++++++
 storage/storage.go   |  1 +
 4 files changed, 41 insertions(+), 3 deletions(-)

diff --git a/grpcapi/grpcapi.go b/grpcapi/grpcapi.go
index 6938c09..ee4e861 100644
--- a/grpcapi/grpcapi.go
+++ b/grpcapi/grpcapi.go
@@ -79,9 +79,6 @@ func (s FleetsServerImpl) GetVehicles(ctx context.Context, req *GetVehiclesReque
 	return &GetVehiclesResponse{Vehicles: vehicles}, nil
 }
 
-func (s FleetsServerImpl) UpdateVehicle(context.Context, *UpdateVehicleRequest) (*UpdateVehicleResponse, error) {
-	return nil, status.Errorf(codes.Unimplemented, "method UpdateVehicle not implemented")
-}
 func (s FleetsServerImpl) CreateBooking(ctx context.Context, req *CreateBookingRequest) (*CreateBookingResponse, error) {
 	b := req.Booking.ToStorageType()
 	booking, err := s.handler.AddBooking(b)
@@ -195,3 +192,19 @@ func Run(done chan error, cfg *viper.Viper, handler handlers.FleetsHandler) {
 		done <- err
 	}
 }
+
+/////////////////////update////////////////////////
+func (s FleetsServerImpl) UpdateVehicle(ctx context.Context, req *UpdateVehicleRequest) (*UpdateVehicleResponse, error) {
+	b := req.Vehicle.ToStorageType()
+	vehicle, err := s.handler.UpdateVehicle(b)
+	if err != nil {
+		fmt.Println(err)
+		return nil, status.Errorf(codes.AlreadyExists, "vehicle update failed : %v", err)
+	}
+	response, err := VehicleFromStorageType(vehicle)
+	if err != nil {
+		fmt.Println(err)
+		return nil, status.Errorf(codes.Internal, "issue while retrieving vehicle : %v", err)
+	}
+	return &UpdateVehicleResponse{Vehicle: response}, nil
+}
diff --git a/handlers/vehicles.go b/handlers/vehicles.go
index eba2647..f063432 100644
--- a/handlers/vehicles.go
+++ b/handlers/vehicles.go
@@ -3,6 +3,8 @@ package handlers
 import (
 	"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"
 )
@@ -56,3 +58,14 @@ func (h FleetsHandler) GetVehicles(namespaces []string, filters storage.VehicleF
 	}
 	return newvehicles, err
 }
+
+/////////////////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
+}
diff --git a/storage/mongodb.go b/storage/mongodb.go
index e6a9a4d..9468ecf 100644
--- a/storage/mongodb.go
+++ b/storage/mongodb.go
@@ -236,3 +236,14 @@ func (s MongoDBStorage) DeleteBooking(bookingid string) error {
 	return err
 
 }
+
+/////////////////////update///////////////
+func (s MongoDBStorage) UpdateVehicle(vehicle Vehicle) error {
+	collection := s.Client.Database(s.DbName).Collection(s.Collections["vehicles"])
+	if _, err := collection.ReplaceOne(context.TODO(), bson.M{"_id": vehicle.ID}, vehicle); err != nil {
+		fmt.Println(err)
+		return err
+	}
+
+	return nil
+}
diff --git a/storage/storage.go b/storage/storage.go
index 399ca39..3d9e902 100644
--- a/storage/storage.go
+++ b/storage/storage.go
@@ -11,6 +11,7 @@ type Storage interface {
 	CreateVehicle(Vehicle) error
 	GetVehicle(id string) (*Vehicle, error)
 	GetVehicles(namespaces []string) ([]Vehicle, error)
+	UpdateVehicle(Vehicle) error
 
 	//Bookings management
 	CreateBooking(Booking) error