accept codeshares in flightnumber endpoint

explore-flights · Aug 25, 2024 · 4c47edd · 4c47edd
1 parent c226f1d
commit 4c47edd
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/go/api/data/handler.go b/go/api/data/handler.go
@@ -255,7 +255,7 @@ func (h *Handler) Aircraft(ctx context.Context) ([]Aircraft, error) {
 	return result, err
 }
 
-func (h *Handler) FlightNumber(ctx context.Context, fn, airport string, d common.LocalDate) (*common.Flight, error) {
+func (h *Handler) FlightNumber(ctx context.Context, fnRaw, airport string, d common.LocalDate) (*common.Flight, error) {
 	flights, err := loadJson[[]*common.Flight](
 		ctx,
 		h,
@@ -270,8 +270,13 @@ func (h *Handler) FlightNumber(ctx context.Context, fn, airport string, d common
 		}
 	}
 
+	fn, err := common.ParseFlightNumber(fnRaw)
+	if err != nil {
+		return nil, err
+	}
+
 	for _, f := range flights {
-		if f.Number().String() == fn && f.DepartureAirport == airport {
+		if _, ok := f.CodeShares[fn]; (f.Number() == fn || ok) && f.DepartureAirport == airport {
 			return f, nil
 		}
 	}