aboutsummaryrefslogtreecommitdiff
path: root/pkg/navitia_api_client/departures.go
blob: 1932767ac19f7a2a32106fed91df52e1feebf844 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
package navitia_api_client

import (
	"encoding/json"
	"fmt"
	"net/http"
	"time"

	"git.adyxax.org/adyxax/trains/pkg/model"
)

type DeparturesResponse struct {
	Disruptions []interface{} `json:"disruptions"`
	Notes       []interface{} `json:"notes"`
	Departures  []struct {
		DisplayInformations struct {
			Direction      string        `json:"direction"`
			Code           string        `json:"code"`
			Network        string        `json:"network"`
			Links          []interface{} `json:"links"`
			Color          string        `json:"color"`
			Name           string        `json:"name"`
			PhysicalMode   string        `json:"physical_mode"`
			Headsign       string        `json:"headsign"`
			Label          string        `json:"label"`
			Equipments     []interface{} `json:"equipments"`
			TextColor      string        `json:"text_color"`
			TripShortName  string        `json:"trip_short_name"`
			CommercialMode string        `json:"commercial_mode"`
			Description    string        `json:"description"`
		} `json:"display_informations"`
		StopDateTime struct {
			Links                  []interface{} `json:"links"`
			ArrivalDateTime        string        `json:"arrival_date_time"`
			AdditionalInformations []interface{} `json:"additional_informations"`
			DepartureDateTime      string        `json:"departure_date_time"`
			BaseArrivalDateTime    string        `json:"base_arrival_date_time"`
			BaseDepartureDateTime  string        `json:"base_departure_date_time"`
			DataFreshness          string        `json:"data_freshness"`
		} `json:"stop_date_time"`
	} `json:"departures"`
	Context struct {
		Timezone        string `json:"timezone"`
		CurrentDatetime string `json:"current_datetime"`
	} `json:"context"`
}

func (c *NavitiaClient) GetDepartures(trainStop string) (departures []model.Departure, err error) {
	request := fmt.Sprintf("%s/coverage/sncf/stop_areas/%s/departures", c.baseURL, trainStop)
	start := time.Now()
	c.mutex.Lock()
	defer c.mutex.Unlock()
	if cachedResult, ok := c.cache[request]; ok {
		if start.Sub(cachedResult.ts) < 60*1000*1000*1000 {
			return cachedResult.result.([]model.Departure), nil
		}
	}
	req, err := http.NewRequest("GET", request, nil)
	if err != nil {
		return nil, newHttpClientError("http.NewRequest error", err)
	}
	resp, err := c.httpClient.Do(req)
	if err != nil {
		return nil, newHttpClientError("httpClient.Do error", err)
	}
	defer resp.Body.Close()
	if resp.StatusCode == http.StatusOK {
		var data DeparturesResponse
		if err = json.NewDecoder(resp.Body).Decode(&data); err != nil {
			return nil, newJsonDecodeError("GetDepartures "+trainStop, err)
		}
		// TODO test for no json error
		// TODO handle pagination
		for i := 0; i < len(data.Departures); i++ {
			t, err := time.Parse("20060102T150405", data.Departures[i].StopDateTime.ArrivalDateTime)
			if err != nil {
				return nil, newDateParsingError(data.Departures[i].StopDateTime.ArrivalDateTime, err)
			}
			departures = append(departures, model.Departure{data.Departures[i].DisplayInformations.Direction, t.Format("Mon, 02 Jan 2006 15:04:05")})
		}
		c.cache[request] = cachedResult{
			ts:     start,
			result: departures,
		}
	} else {
		err = newApiError(resp.StatusCode, "GetDepartures "+trainStop)
	}
	return
}