fix commit

storage/postgresql.go

@@ -61,6 +61,46 @@ func NewPostgresqlStorage(cfg *viper.Viper) (PostgresqlStorage, error) {
 	}, nil
 }
 
+func (s PostgresqlStorage) CreateFirebaseToken(user_id string, fcm_token string, device_platform string) (err error) {
+	_, err = uuid.Parse(user_id)
+	if err != nil {
+		log.Error().Err(err).Msg("Postgresql Storage CreateFirebaseToken invalid User ID")
+		return err
+	}
+	_, err = s.DbConnection.Exec(fmt.Sprintf("INSERT INTO %s (user_id , fcm_token , device_platform) VALUES($1,$2,$3)", s.Tables["users_firebase"]),
+		user_id,
+		fcm_token,
+		device_platform)
+	if err != nil {
+		if strings.Contains(err.Error(), "duplicate key") {
+			_ = s.UpdateFirebaseToken(user_id, device_platform, fcm_token)
+		}
+	}
+
+	return nil
+}
+
+func (s PostgresqlStorage) GetFirebaseToken(user_id string) (fcm string, device_platform string, err error) {
+	err = s.DbConnection.QueryRow(fmt.Sprintf("SELECT fcm_token , device_platform FROM %s WHERE user_id = $1", s.Tables["users_firebase"]), user_id).
+		Scan(
+			&fcm,
+			&device_platform,
+		)
+	if err != nil {
+		return "", "", err
+	}
+	return fcm, device_platform, nil
+}
+
+func (s PostgresqlStorage) UpdateFirebaseToken(user_id string, fcm_token string, device_platform string) error {
+	query := fmt.Sprintf("UPDATE %s SET fcm_token = $1 device_platform = $2 WHERE user_id = $3", s.Tables["users_firebase"])
+	_, err := s.DbConnection.Exec(query, fcm_token, device_platform, user_id)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
 func (s PostgresqlStorage) CreatePassenger(passenger internal.Passenger) (err error) {
 	_, err = uuid.Parse(passenger.Passenger.ID)
 	if err != nil {

storage/storage.go

@@ -19,6 +19,9 @@ type Storage interface {
 	FilterUserBookingsByStatus(user_type string, status internal.BookingStatus, user_id string) (bookings []internal.Booking, err error)
 	DriverJourneys(departure_route *geojson.Feature, departure_date int64) (drivers []internal.Driver, err error)
 	GetAllDrivers(date int64) (drivers []internal.Driver, err error)
+	CreateFirebaseToken(user_id string, fcm_token string, device_platform string) (err error)
+	UpdateFirebaseToken(user_id string, fcm_token string, device_platform string) error
+	GetFirebaseToken(user_id string) (fcm string, device_platform string, err error)
 }
 
 func NewStorage(cfg *viper.Viper) (Storage, error) {