From ae7b2f170aa0e9a8414acd0899ccfc08e69fd74f Mon Sep 17 00:00:00 2001
From: Paul Brinkmeier <hallo@pbrinkmeier.de>
Date: Fri, 25 Apr 2025 12:04:47 +0200
Subject: [PATCH] Add EFAClient interface

---
 .gitignore |  1 +
 main.go    | 56 ++++++++++++++++++++++++++++++++++++++++++++++++++++--
 2 files changed, 55 insertions(+), 2 deletions(-)

diff --git a/.gitignore b/.gitignore
index 5e7cfbf..de8e544 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,3 +1,4 @@
 result
 vrnp
+main
 *.swp
diff --git a/main.go b/main.go
index 24e6c8d..44a2b9b 100644
--- a/main.go
+++ b/main.go
@@ -8,6 +8,7 @@ import "net/http"
 import "net/url"
 import "os"
 import "slices"
+import "strings"
 import "strconv"
 import "time"
 
@@ -43,11 +44,23 @@ type DMDateTime struct {
 	Minute string `json:"minute"`
 }
 
-func FetchDepartures(stopId string) (DMResponse, error) {
+type EFAClient interface {
+	GetName() string
+	BuildRequest(string) (*http.Request, error)
+}
+
+type VRNEFAClient struct {
+}
+
+func (c VRNEFAClient) GetName() string {
+	return "VRN"
+}
+
+func (c VRNEFAClient) BuildRequest(stopId string) (*http.Request, error) {
 	// Create request object
 	req, err := http.NewRequest("GET", "https://www.vrn.de/mngvrn/XML_DM_REQUEST", nil)
 	if err != nil {
-		return DMResponse{}, err
+		return nil, err
 	}
 
 	// Configure our request
@@ -65,6 +78,45 @@ func FetchDepartures(stopId string) (DMResponse, error) {
 	query.Set("useRealtime", "1")
 	req.URL.RawQuery = query.Encode()
 
+	return req, nil
+}
+
+type KVVEFAClient struct {
+}
+
+func (c KVVEFAClient) GetName() string {
+	return "KVV"
+}
+
+func (c KVVEFAClient) BuildRequest(stopId string) (*http.Request, error) {
+	form := url.Values{}
+	form.Set("action", "XSLT_DM_REQUEST")
+	form.Set("name_dm", stopId)
+	form.Set("type_dm", "stop")
+	form.Set("useRealtime", "1")
+	form.Set("limit", "10")
+	form.Set("mode", "direct")
+	form.Set("outputFormat", "json")
+	body := strings.NewReader(form.Encode())
+
+	req, err := http.NewRequest("POST", "https://www.kvv.de/tunnelEfaDirect.php", body)
+	if err != nil {
+		return nil, err
+	}
+
+	req.Header.Set("User-Agent", "coolio/1.0")
+	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
+
+	return req, nil
+}
+
+func FetchDepartures(stopId string) (DMResponse, error) {
+	var c EFAClient = KVVEFAClient{}
+	req, err := c.BuildRequest(stopId)
+	if err != nil {
+		return DMResponse{}, err
+	}
+
 	// Send the request, wait max 10 seconds
 	client := http.Client{
 		Timeout: 10 * time.Second,