traewelling-exporter/go/main.go

package main

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

	"github.com/prometheus/client_golang/prometheus"
	"github.com/prometheus/client_golang/prometheus/promauto"
	"github.com/prometheus/client_golang/prometheus/promhttp"
)

type DashboardResponse struct {
	Data []struct {
		ID    int `json:"id"`
		User  struct {
			DisplayName string `json:"displayName"`
			Username    string `json:"username"`
		} `json:"userDetails"`
		Train struct {
			Trip      int     `json:"trip"`
			Category  string  `json:"category"`
			LineName  string  `json:"lineName"`
			JourneyNumber int `json:"journeyNumber"`
			Distance  float64 `json:"distance"`
			Duration  int     `json:"duration"`
			Points    int     `json:"points"`
			Origin    struct {
				Name             string `json:"name"`
				DeparturePlanned string `json:"departurePlanned"`
				DepartureReal    string `json:"departureReal"`
			} `json:"origin"`
			Destination struct {
				Name           string `json:"name"`
				ArrivalPlanned string `json:"arrivalPlanned"`
				ArrivalReal    string `json:"arrivalReal"`
			} `json:"destination"`
		} `json:"train"`
	} `json:"data"`
}

var (
	totalTripDuration = promauto.NewGaugeVec(
		prometheus.GaugeOpts{
			Name: "traewelling_total_trip_duration_minutes",
			Help: "Gesamtfahrzeit eines Benutzers in Minuten",
		},
		[]string{"username"},
	)
	totalTripPoints = promauto.NewGaugeVec(
		prometheus.GaugeOpts{
			Name: "traewelling_total_trip_points",
			Help: "Gesamtpunkte eines Benutzers",
		},
		[]string{"username"},
	)
	totalTripDistance = promauto.NewGaugeVec(
		prometheus.GaugeOpts{
			Name: "traewelling_total_trip_distance_km",
			Help: "Gesamtentfernung eines Benutzers in Kilometern",
		},
		[]string{"username"},
	)
)

func fetchTraewellingData() (*DashboardResponse, error) {
	url := "https://traewelling.de/api/v1/dashboard"
	client := &http.Client{Timeout: 10 * time.Second}
	req, err := http.NewRequest("GET", url, nil)
	if err != nil {
		return nil, fmt.Errorf("Fehler beim Erstellen der Anfrage: %v", err)
	}

	token := os.Getenv("TRAEWELLING_TOKEN")
	if token == "" {
		return nil, fmt.Errorf("TRAEWELLING_TOKEN ist nicht gesetzt")
	}
	req.Header.Set("Authorization", "Bearer "+token)
	req.Header.Set("Accept", "application/json")

	resp, err := client.Do(req)
	if err != nil {
		return nil, fmt.Errorf("Fehler beim Senden der Anfrage: %v", err)
	}
	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK {
		return nil, fmt.Errorf("Fehlerhafte Antwort: %d", resp.StatusCode)
	}

	var apiResponse DashboardResponse
	err = json.NewDecoder(resp.Body).Decode(&apiResponse)
	if err != nil {
		return nil, fmt.Errorf("Fehler beim Parsen der JSON-Daten: %v", err)
	}

	return &apiResponse, nil
}

func processAndPrintTripDetails(data *DashboardResponse) {
	fmt.Println("Fahrtdetails der Benutzer:")
	fmt.Println("==========================")

	userStats := make(map[string]struct {
		TotalDuration float64
		TotalPoints   int
		TotalDistance float64
	})

	for _, trip := range data.Data {
		fmt.Printf("Benutzer: %s (@%s)\n", trip.User.DisplayName, trip.User.Username)
		fmt.Printf("Fahrt-ID: %d\n", trip.Train.Trip)
		fmt.Printf("Kategorie: %s\n", trip.Train.Category)
		fmt.Printf("Linie: %s (Nummer: %d)\n", trip.Train.LineName, trip.Train.JourneyNumber)
		fmt.Printf("Startbahnhof: %s\n", trip.Train.Origin.Name)
		fmt.Printf("\tGeplante Abfahrt: %s\n", trip.Train.Origin.DeparturePlanned)
		fmt.Printf("\tTatsächliche Abfahrt: %s\n", trip.Train.Origin.DepartureReal)
		fmt.Printf("Zielbahnhof: %s\n", trip.Train.Destination.Name)
		fmt.Printf("\tGeplante Ankunft: %s\n", trip.Train.Destination.ArrivalPlanned)
		fmt.Printf("\tTatsächliche Ankunft: %s\n", trip.Train.Destination.ArrivalReal)
		fmt.Printf("Dauer: %d Minuten\n", trip.Train.Duration)
		fmt.Printf("Entfernung: %.2f km\n", trip.Train.Distance/1000)
		fmt.Printf("Punkte: %d\n", trip.Train.Points)
		fmt.Println("--------------------------")

		stats := userStats[trip.User.Username]
		stats.TotalDuration += float64(trip.Train.Duration)
		stats.TotalPoints += trip.Train.Points
		stats.TotalDistance += trip.Train.Distance / 1000
		userStats[trip.User.Username] = stats
	}

	for username, stats := range userStats {
		totalTripDuration.WithLabelValues(username).Set(stats.TotalDuration)
		totalTripPoints.WithLabelValues(username).Set(float64(stats.TotalPoints))
		totalTripDistance.WithLabelValues(username).Set(stats.TotalDistance)
	}
}

func updateMetrics() {
	data, err := fetchTraewellingData()
	if err != nil {
		fmt.Printf("Fehler beim Abrufen der Daten: %v\n", err)
		return
	}
	processAndPrintTripDetails(data)
}

func main() {
	go func() {
		for {
			updateMetrics()
			time.Sleep(5 * time.Minute)
		}
	}()

	http.Handle("/metrics", promhttp.Handler())
	fmt.Println("Server läuft auf Port 8080...")
	http.ListenAndServe(":8080", nil)
}