initial commit

handlers/carpool.go

@@ -0,0 +1,103 @@
+package handlers
+
+import (
+	"errors"
+	"fmt"
+	"sync"
+	"time"
+
+	"git.coopgo.io/coopgo-platform/multimodal-routing/libs/carpool"
+	"github.com/paulmach/orb"
+	"github.com/paulmach/orb/geojson"
+	"github.com/rs/zerolog/log"
+	"github.com/spf13/viper"
+)
+
+type CarpoolRoutingHandler struct {
+	OperatorAPIs []carpool.CarpoolOperatorAPI
+}
+
+func NewCarpoolRoutingHandler(cfg *viper.Viper) (*CarpoolRoutingHandler, error) {
+	operatorApis := []carpool.CarpoolOperatorAPI{}
+	operators := cfg.Get("operators").([]any)
+	for _, operator := range operators {
+		o := operator.(map[string]any)
+		operatorType, ok := o["type"].(string)
+		if !ok {
+			return nil, errors.New("missing operator type")
+		}
+		if operatorType == "blablacardaily" {
+			operatorId, ok := o["operator_id"].(string)
+			if !ok {
+				return nil, errors.New("missing operator_id")
+			}
+			apiKey, ok := o["api_key"].(string)
+			if !ok {
+				return nil, errors.New("missing api_key")
+			}
+			baseUrl, ok := o["base_url"].(string)
+			if !ok {
+				return nil, errors.New("missing base_url")
+			}
+			operatorapi, err := carpool.NewBBCDailyCarpoolAPI(operatorId, apiKey, baseUrl)
+			if err != nil {
+				return nil, fmt.Errorf("could not create Blablacar Daily API: %w", err)
+			}
+			operatorApis = append(operatorApis, operatorapi)
+		}
+	}
+	return &CarpoolRoutingHandler{
+		OperatorAPIs: operatorApis,
+	}, nil
+}
+
+// TODO add options (WithTimeDelta, etc...)
+func (h CarpoolRoutingHandler) Search(results chan *geojson.FeatureCollection, wg *sync.WaitGroup, departure geojson.Feature, destination geojson.Feature, departureDate time.Time) error {
+	defer wg.Done()
+
+	defaultTimeDelta := 3600 * time.Second
+	defaultDepartureRadius := int64(10)
+	defaultDestinationRadius := int64(10)
+	defaultCount := int64(10)
+
+	var wg2 sync.WaitGroup
+
+	for _, api := range h.OperatorAPIs {
+		wg2.Add(1)
+
+		go func(results chan *geojson.FeatureCollection, wg2 *sync.WaitGroup) {
+			defer wg2.Done()
+			r, err := api.GetDriverJourneys(departure.Point().Lat(), departure.Point().Lon(), destination.Point().Lat(), destination.Point().Lon(), departureDate, &defaultTimeDelta, defaultDepartureRadius, defaultDestinationRadius, defaultCount)
+			if err != nil {
+				log.Error().Err(err).Str("operator", api.GetOperatorId()).Msg("error in carpool api request")
+			}
+			for _, journey := range r {
+				// Departure
+				geo := geojson.NewFeatureCollection()
+				dep := geojson.NewFeature(orb.Point{journey.PassengerPickupLng, journey.PassengerPickupLat})
+				dep.Properties["type"] = "departure"
+				dep.Properties["label"] = journey.PassengerPickupAddress
+
+				// TODO Polyline to GeoJSON LineString
+
+				// Destination
+				dest := geojson.NewFeature(orb.Point{journey.PassengerDropLng, journey.PassengerDropLat})
+				dest.Properties["type"] = "destination"
+				dest.Properties["label"] = journey.PassengerDropAddress
+
+				// Carpool GeoJSON definition
+				geo.Features = append(geo.Features, dep)
+				geo.Features = append(geo.Features, dest)
+				geo.ExtraMembers = geojson.Properties{
+					"journey_type": "carpool",
+					"ocss":         journey,
+				}
+				results <- geo
+			}
+		}(results, &wg2)
+	}
+
+	wg2.Wait()
+
+	return nil
+}