api

type Api interface {
	// Fetches arrival information for list of location unique identifiers
	FetchPredictions(ids []string) ([]Predictions, error)
	// Fetch train incidents for a location
	FetchTrainIncidents() ([]Incident, error)
	// Given user input for a location, returns the unique identifier (location could have multiple)
	GetCodeFromArg(arg string) []string
	// Given a line name or abbreviation, return colors that represents it.
	// (fg, bg) tuple returned
	GetColorFromLine(arg string) (string, string)
	// Determines if a train isn't for passengers
	IsGhostTrain(line, destination string) bool
}

type DmvApi struct {
	// contains filtered or unexported fields
}

type Incident struct {
	// Message from the transit authority describing the situation
	Description string
	// Last date & time update from the transit authority
	DateUpdated time.Time
	// Lines, stations or stops affected by the incident
	Affected []string
	// Type of incident
	Type string
}

type IncidentsResponse struct {
	Incidents []WMATA_Incident
}

type Predictions struct {
	// Minutes until a train arrives
	Min string
	// The short name for a train station
	LocationName string
	// The short name for the train's destination
	Destination string
	// The full name for the train's destination
	DestinationName string
	// The train's line
	Line string
}

type PredictionsResponse struct {
	Trains []Predictions
}

type WMATA_Incident struct {
	Description   string
	IncidentType  string
	LinesAffected string
	DateUpdated   string
}