idfm-api/api/schedules.go

package api

import (
	"encoding/json"
	"flag"
	"fmt"
	"math"
	"net/http"
	"net/url"
	"time"

	"github.com/gin-gonic/gin"
)

const IDFM_BASEURL = "https://prim.iledefrance-mobilites.fr/marketplace"

var IDFM_TOKEN = ""

func init() {
	flag.StringVar(&IDFM_TOKEN, "token-IDFM", IDFM_TOKEN, "Token to access IDFM API")
}

type IDFMSchedule struct {
	Siri struct {
		ServiceDelivery struct {
			ResponseTimestamp         time.Time `json:"ResponseTimestamp"`
			ProducerRef               string    `json:"ProducerRef"`
			ResponseMessageIdentifier string    `json:"ResponseMessageIdentifier"`
			StopMonitoringDelivery    []struct {
				ResponseTimestamp  time.Time `json:"ResponseTimestamp"`
				Version            string    `json:"Version"`
				Status             string    `json:"Status"`
				MonitoredStopVisit []struct {
					RecordedAtTime time.Time `json:"RecordedAtTime"`
					ItemIdentifier string    `json:"ItemIdentifier"`
					MonitoringRef  struct {
						Value string `json:"value"`
					} `json:"MonitoringRef"`
					MonitoredVehicleJourney struct {
						LineRef struct {
							Value string `json:"value"`
						} `json:"LineRef"`
						OperatorRef struct {
							Value string `json:"value"`
						} `json:"OperatorRef"`
						FramedVehicleJourneyRef struct {
							DataFrameRef struct {
								Value string `json:"value"`
							} `json:"DataFrameRef"`
							DatedVehicleJourneyRef string `json:"DatedVehicleJourneyRef"`
						} `json:"FramedVehicleJourneyRef"`
						DirectionName []struct {
							Value string `json:"value"`
						} `json:"DirectionName"`
						DestinationRef struct {
							Value string `json:"value"`
						} `json:"DestinationRef"`
						DestinationName []struct {
							Value string `json:"value"`
						} `json:"DestinationName"`
						JourneyNote []struct {
							Value string `json:"value"`
						} `json:"JourneyNote"`
						MonitoredCall struct {
							StopPointName []struct {
								Value string `json:"value"`
							} `json:"StopPointName"`
							VehicleAtStop      bool `json:"VehicleAtStop"`
							DestinationDisplay []struct {
								Value string `json:"value"`
							} `json:"DestinationDisplay"`
							ExpectedArrivalTime   time.Time `json:"ExpectedArrivalTime"`
							ExpectedDepartureTime time.Time `json:"ExpectedDepartureTime"`
							DepartureStatus       string    `json:"DepartureStatus"`
						} `json:"MonitoredCall"`
					} `json:"MonitoredVehicleJourney"`
				} `json:"MonitoredStopVisit"`
			} `json:"StopMonitoringDelivery"`
		} `json:"ServiceDelivery"`
	} `json:"siri"`
}

type PGSchedule struct {
	Destination string `json:"destionation"`
	Message     string `json:"message"`
}

func declareSchedulesRoutes(router *gin.RouterGroup) {
	router.GET("/schedules/:type/:code/:station/:way", func(c *gin.Context) {
		t := string(c.Param("type"))
		station := convertLineType(string(c.Param("station")))

		rurl, err := url.JoinPath(IDFM_BASEURL, "stop-monitoring")
		if err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": err.Error()})
			return
		}

		requrl, err := url.Parse(rurl)
		if err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": err.Error()})
			return
		}

		reqquery := url.Values{}
		reqquery.Add("MonitoringRef", station)
		requrl.RawQuery = reqquery.Encode()

		req, err := http.NewRequest("GET", requrl.String(), nil)
		if err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": err.Error()})
			return
		}

		req.Header.Add("Accept", "application/json")
		req.Header.Add("apikey", IDFM_TOKEN)

		res, err := http.DefaultClient.Do(req)
		if err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": err.Error()})
			return
		}
		defer res.Body.Close()

		var schedules IDFMSchedule

		dec := json.NewDecoder(res.Body)
		if err = dec.Decode(&schedules); err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": err.Error()})
			return
		}

		pgs := []PGSchedule{}
		for _, vehicule := range schedules.Siri.ServiceDelivery.StopMonitoringDelivery[0].MonitoredStopVisit {
			msg := vehicule.MonitoredVehicleJourney.MonitoredCall.ExpectedDepartureTime.String()

			if t == "metros" || t == "buses" || t == "tramways" {
				if vehicule.MonitoredVehicleJourney.MonitoredCall.VehicleAtStop {
					msg = "À quai"
				} else {
					msg = fmt.Sprintf("%d mn", int(math.Floor(time.Until(vehicule.MonitoredVehicleJourney.MonitoredCall.ExpectedDepartureTime).Minutes())))
				}
			}

			pgs = append(pgs, PGSchedule{
				Destination: vehicule.MonitoredVehicleJourney.MonitoredCall.DestinationDisplay[0].Value,
				Message:     msg,
			})
		}

		c.JSON(http.StatusOK, pgs)
	})
}