35 Commits
Author SHA1 Message Date
Paul Brinkmeier a175ad22ae fixup! fixup! fixup! Add version to image meta 2025-04-23 17:18:58 +02:00
Paul Brinkmeier c00b8f24f1 fixup! fixup! Add version to image meta
Build image / build-image (push) Failing after 2m33s
2025-04-23 17:09:47 +02:00
Paul Brinkmeier 56d67889b4 fixup! Add version to image meta
Build image / build-image (push) Has been cancelled
2025-04-23 17:05:57 +02:00
Paul Brinkmeier eceec32264 Add version to image meta
Build image / build-image (push) Failing after 2m18s
2025-04-23 17:01:00 +02:00
Paul Brinkmeier bc15cfd21d fixup! Run on tags instead of push
Build image / build-image (push) Successful in 3m55s
2025-04-23 16:45:49 +02:00
Paul Brinkmeier 2f0dd1b59b Run on tags instead of push 2025-04-23 16:44:29 +02:00
Paul Brinkmeier 2c1d1e42bd fixup! fixup! fixup! fixup! fixup! fixup! fixup! Add docker commands 2025-04-23 16:43:19 +02:00
Paul Brinkmeier e125aa5703 fixup! fixup! fixup! fixup! fixup! fixup! Add docker commands
Build image / build-image (push) Successful in 4m17s
2025-04-23 16:35:40 +02:00
Paul Brinkmeier ec87e87fd1 fixup! fixup! fixup! fixup! fixup! Add docker commands
Build image / build-image (push) Has been cancelled
2025-04-23 16:19:31 +02:00
Paul Brinkmeier 76ea336d6f fixup! fixup! fixup! fixup! Add docker commands
Build image / build-image (push) Waiting to run
2025-04-23 15:33:29 +02:00
Paul Brinkmeier 74be0b8a90 fixup! fixup! fixup! Add docker commands
Build image / build-image (push) Waiting to run
2025-04-23 14:50:24 +02:00
Paul Brinkmeier 28bb8be8b2 fixup! fixup! Add docker commands
Build image / build-image (push) Failing after 2m53s
2025-04-23 14:31:51 +02:00
Paul Brinkmeier 225b231262 fixup! Add docker commands
Build image / build-image (push) Failing after 3m12s
2025-04-23 14:27:36 +02:00
Paul Brinkmeier 309266e8e7 Add docker commands
Build image / build-image (push) Failing after 3m35s
2025-04-23 14:23:36 +02:00
Paul Brinkmeier 0b6b3a021b Try using catthehacker/ubuntu container
Build image / build-image (push) Successful in 5m16s
2025-04-23 14:00:56 +02:00
Paul Brinkmeier d3022c010f fixup! fixup! fixup! fixup! Add version file for crane push
Build image / build-image (push) Failing after 4m9s
2025-04-23 13:45:18 +02:00
Paul Brinkmeier 116cc5ce51 fixup! fixup! fixup! Add version file for crane push
Build image / build-image (push) Failing after 3m51s
2025-04-23 13:41:00 +02:00
Paul Brinkmeier 6bab8ac3a9 fixup! fixup! Add version file for crane push
Build image / build-image (push) Failing after 2m48s
2025-04-23 13:36:43 +02:00
Paul Brinkmeier f86c94ae99 fixup! Add version file for crane push
Build image / build-image (push) Failing after 3m2s
2025-04-23 13:32:22 +02:00
Paul Brinkmeier b91ab5d603 Add version file for crane push
Build image / build-image (push) Failing after 3m13s
2025-04-23 13:27:11 +02:00
Paul Brinkmeier c4017843ec fixup! fixup! Add crane tool
Build container / build-container (push) Failing after 3m5s
2025-04-23 13:07:57 +02:00
Paul Brinkmeier b0a989109a fixup! Add crane tool
Build container / build-container (push) Failing after 3m18s
2025-04-23 13:03:10 +02:00
Paul Brinkmeier e8b1556770 Add crane tool
Build container / build-container (push) Failing after 3m27s
2025-04-23 12:57:29 +02:00
Paul Brinkmeier b990bf2b8f Add docker load command
Build container / build-container (push) Failing after 3m48s
2025-04-23 12:42:25 +02:00
Paul Brinkmeier 389b83c72f fixup! fixup! Add sudo installation step
Build container / build-container (push) Successful in 3m28s
2025-04-23 12:35:39 +02:00
Paul Brinkmeier ee00ba9b34 fixup! Add sudo installation step
Build container / build-container (push) Failing after 3m47s
2025-04-23 12:30:47 +02:00
Paul Brinkmeier fa68ab9bcc Add sudo installation step
Build container / build-container (push) Failing after 16s
2025-04-23 12:29:01 +02:00
Paul Brinkmeier 1bcf8e30a0 Try nix-quick-install-action
Build container / build-container (push) Failing after 10s
2025-04-23 12:27:47 +02:00
Paul Brinkmeier 8189157608 fixup! fixup! fixup! Add install nix and build container steps
Build container / build-container (push) Failing after 9s
2025-04-23 12:25:37 +02:00
Paul Brinkmeier 3f9cee98f4 fixup! fixup! Add install nix and build container steps 2025-04-23 12:20:48 +02:00
Paul Brinkmeier 0c8a48cd41 fixup! Add install nix and build container steps
Build container / build-container (push) Failing after 1m18s
2025-04-23 12:20:34 +02:00
Paul Brinkmeier 09bdd39fd8 Add install nix and build container steps
Build container / build-container (push) Failing after 20s
2025-04-23 12:19:13 +02:00
Paul Brinkmeier 9506864c2d Set runs-on
Gitea Actions Demo / build-container (push) Successful in 8s
2025-04-23 12:12:19 +02:00
Paul Brinkmeier aca227bdc6 fixup! Init CI
Gitea Actions Demo / build-container (push) Has been cancelled
2025-04-23 12:10:21 +02:00
Paul Brinkmeier 1020564f8f Init CI 2025-04-23 12:08:46 +02:00
9 changed files with 183 additions and 470 deletions
-1
View File
@@ -1,4 +1,3 @@
result
vrnp
main
*.swp
-1
View File
@@ -11,4 +11,3 @@
- [x] Transfer using JSON
- [x] Correctly implement basic auth
- [ ] Use unidecode to replace non-ascii stuff in the backend
- [ ] Add query parameter for selecting EFAClient
Binary file not shown.

Before

Width:  |  Height:  |  Size: 4.0 KiB

-14
View File
@@ -81,17 +81,3 @@ static const uint8_t PROGMEM spinny[][8] = {
0x00,
},
};
static const uint16_t PROGMEM tiny_train_dims[] = {24, 9, 1};
static const uint8_t PROGMEM tiny_train[][27] = {
{
0x00, 0x04, 0x00,
0x00, 0x02, 0x00,
0x01, 0xff, 0xfe,
0x00, 0xa4, 0x94,
0x01, 0x24, 0x92,
0x01, 0x24, 0x92,
0x03, 0xfc, 0xff,
0x03, 0xff, 0xff,
0x01, 0xb0, 0x36,
},
};
+118 -208
View File
@@ -1,189 +1,39 @@
#include <ArduinoJson.h>
#include <Adafruit_GFX.h>
#include <Adafruit_ST7735.h>
#include <ESP8266WiFi.h>
#include <ESP8266HTTPClient.h>
#include <memory>
#include <SPI.h>
#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include <ESP8266HTTPClient.h>
#include <WiFiClientSecureBearSSL.h>
#include "bitmaps.h"
#include "wifi_credentials.h"
#define TFT_CS D8
#define TFT_DC D4
#define TFT_RESET D3
// Screen stuff
#define SCREEN_WIDTH 128 // OLED display width, in pixels
#define SCREEN_HEIGHT 64 // OLED display height, in pixels
#define OLED_RESET -1 // Reset pin # (or -1 if sharing Arduino reset pin)
#define SCREEN_ADDRESS 0x3C
// Button stuff
#define BUTTON_PIN D6
// Display stuff
#define FRAME_DELAY 80
#define COLOR_BG (rgb(0, 0, 0))
#define COLOR_DIVIDER (rgb(0xff, 0x60, 0x0d))
#define COLOR_TEXT (rgb(0xff, 0x60, 0x0d))
// Layout
#define WIDTH 160
#define HEIGHT 128
#define CX 6
#define CY 8
#define MARGIN_LEFT 5
#define MARGIN_RIGHT 5
#define MAIN_X MARGIN_LEFT
#define MAIN_Y 9
#define MAIN_WIDTH (WIDTH - 2 * MARGIN_LEFT)
#define MAIN_HEIGHT (8 * CY)
#define CLOCK_X MARGIN_LEFT
#define CLOCK_Y 113
#define STATUS_X 83
#define STATUS_Y CLOCK_Y
#define STATUS_WIDTH (12 * CX)
#define STATUS_HEIGHT CY
// Function definitions
void drawLayout();
uint16_t rgb(uint16_t, uint16_t, uint16_t);
String fetchDepartures();
// Peripherals
Adafruit_ST7735 display(TFT_CS, TFT_DC, TFT_RESET);
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);
// App state interface
class State {
public:
virtual void enter() = 0;
virtual void tick() = 0;
virtual ~State() = default;
};
class StateConnecting : public State {
public:
void enter() override;
void tick() override;
};
class StateFetching : public State {
public:
void enter() override;
void tick() override;
};
class StateShowingDepartures : public State {
private:
// Todo: discard JsonDocument asap, parse into better format
JsonDocument departures;
unsigned long entered;
public:
StateShowingDepartures(String&);
void enter() override;
void tick() override;
};
// App state implementation
uint64_t currentTick = 0;
// Initially true so that enter() of the initial state is called.
bool stateChanged = true;
std::unique_ptr<State> state = std::make_unique<StateConnecting>();
template <typename StateType, typename... Args>
void setState(Args&&... args) {
state = std::make_unique<StateType>(std::forward<Args>(args)...);
stateChanged = true;
}
void StateConnecting::enter() {
Serial.println("Entering StateConnecting");
display.fillRect(STATUS_X, STATUS_Y, STATUS_WIDTH, STATUS_HEIGHT, COLOR_BG);
display.setCursor(STATUS_X, STATUS_Y);
display.printf("connecting");
}
void StateConnecting::tick() {
if (WiFi.status() != WL_CONNECTED) {
display.fillRect(STATUS_X + 11 * CX, STATUS_Y, CX, CY, COLOR_BG);
display.setCursor(STATUS_X + 11 * CX, STATUS_Y);
display.print("|/-\\"[currentTick % 4]);
delay(125);
return;
}
setState<StateFetching>();
}
// Fetching is a bit of an ugly duckling; it performs all its logic in enter().
// This is because the request it does is synchronous.
// Sadly this means no animation is possible right now.
// At least the fetch call still does yield() under the hood :)
void StateFetching::enter() {
Serial.println("Entering StateFetching");
if (WiFi.status() != WL_CONNECTED) {
setState<StateConnecting>();
return;
}
display.fillRect(STATUS_X, STATUS_Y, STATUS_WIDTH, STATUS_HEIGHT, COLOR_BG);
display.setCursor(STATUS_X, STATUS_Y);
display.print(" fetching");
String departuresRaw = fetchDepartures();
setState<StateShowingDepartures>(departuresRaw);
};
void StateFetching::tick() {
}
StateShowingDepartures::StateShowingDepartures(String &departuresRaw) {
deserializeJson(departures, departuresRaw);
}
void StateShowingDepartures::enter() {
Serial.println("Entering StateShowingDepartures");
entered = millis();
display.fillRect(STATUS_X, STATUS_Y, STATUS_WIDTH, STATUS_HEIGHT, COLOR_BG);
display.fillRect(MAIN_X, MAIN_Y, MAIN_WIDTH, MAIN_HEIGHT, COLOR_BG);
int line = 0;
for (JsonVariant departure : departures["departures"].as<JsonArray>()) {
display.setCursor(MAIN_X, MAIN_Y + (CY + 3) * line);
display.printf("%2s %-15.15s ",
departure["symbol"].as<const char*>(),
departure["direction"].as<const char *>()
);
if (departure["leaving"].as<String>().equals("sofort")) {
int16_t x = display.getCursorX();
int16_t y = display.getCursorY();
display.drawBitmap(x + 6 * CX - tiny_train_dims[0], y, tiny_train[0], tiny_train_dims[0], tiny_train_dims[1], COLOR_TEXT);
} else {
display.printf("%6s", departure["leaving"].as<const char*>());
}
line++;
}
}
void StateShowingDepartures::tick() {
unsigned long elapsed = millis() - entered;
if (elapsed > 25000) {
setState<StateFetching>();
}
}
// App state
enum {
CONNECTING,
SHOWING_DEPARTURES
} state = CONNECTING;
int frame = 0;
int buttonPushed = false;
JsonDocument departures;
void ICACHE_RAM_ATTR onButtonFalling() {
buttonPushed = true;
@@ -192,52 +42,34 @@ void ICACHE_RAM_ATTR onButtonFalling() {
void setup() {
Serial.begin(9600);
Serial.println("serial init");
display.initR();
display.setSPISpeed(40000000);
display.setRotation(3);
Serial.println("display initialized");
Serial.printf("display dimensions: %" PRId16 "x%" PRId16 "\n", display.width(), display.height());
display.setTextSize(1);
display.setTextColor(COLOR_TEXT);
drawLayout();
// SSD1306_SWITCHCAPVCC = generate display voltage from 3.3V internally
if(!display.begin(SSD1306_SWITCHCAPVCC, SCREEN_ADDRESS)) {
Serial.println("SSD1306 allocation failed");
for(;;); // Don't proceed, loop forever
}
Serial.println("SSD1306 initialized");
display.clearDisplay();
WiFi.begin(WIFI_SSID, WIFI_PASSWORD);
pinMode(BUTTON_PIN, INPUT_PULLUP);
attachInterrupt(BUTTON_PIN, onButtonFalling, FALLING);
state = std::make_unique<StateConnecting>();
display.setTextColor(SSD1306_INVERSE);
display.setTextSize(1);
display.setCursor(0, 0);
display.printf("Connecting to\n%s\n", WIFI_SSID);
display.display();
/*
display.print(" 5 Mannheim sofort\n"
"26 Kirchheim 9 min\n"
" 5 Weinheim 20:13");
*/
}
#define QUERY_STRING "stop_id=de:08221:1225&platform=A"
void loop() {
if (stateChanged) {
stateChanged = false;
// Note: enter() may call setState(). In that case we want to end up right back here.
state->enter();
} else {
state->tick();
currentTick++;
}
// Let background system do its thing
yield();
}
uint16_t rgb(uint16_t r, uint16_t g, uint16_t b) {
// Color layout: RRRRRGGGGGGBBBBB (5, 6, 5)
return ((b >> 3) & 0b11111) << 11 | ((g >> 2) & 0b111111) << 5 | ((r >> 3) & 0b11111);
}
void drawLayout() {
display.fillScreen(COLOR_BG);
display.drawFastHLine(5, 4, 150, COLOR_DIVIDER);
display.drawFastHLine(5, 109, 150, COLOR_DIVIDER);
display.drawFastHLine(5, 123, 150, COLOR_DIVIDER);
display.setCursor(CLOCK_X, CLOCK_Y);
display.print("13:12");
void drawSpinner(int frame) {
display.fillRect(120, 56, spinny_dims[0], spinny_dims[1], SSD1306_BLACK);
display.drawBitmap(120, 56, spinny[frame % spinny_dims[2]], spinny_dims[0], spinny_dims[1], SSD1306_WHITE);
}
// TODO: Error handling
@@ -246,10 +78,11 @@ String fetchDepartures() {
https->setInsecure();
HTTPClient client;
if (!client.begin(*https, "vrnp.beany.club", 443, "/departures?" QUERY_STRING)) {
display.fillScreen(COLOR_BG);
if (!client.begin(*https, "vrnp.beany.club", 443, "/departures?stop_id=de:08221:1225&platform=A")) {
display.clearDisplay();
display.setCursor(0, 0);
display.print("begin failed");
display.display();
for (;;);
}
@@ -257,11 +90,88 @@ String fetchDepartures() {
client.addHeader("Accept", "application/json");
int statusCode = client.GET();
if (statusCode != 200) {
display.fillScreen(COLOR_BG);
display.clearDisplay();
display.setCursor(0, 0);
display.printf("http non-ok: %d\n", statusCode);
display.display();
for (;;);
}
return client.getString();
}
void loop() {
switch (state) {
case CONNECTING: {
if (WiFi.status() != WL_CONNECTED) {
drawSpinner(frame++);
display.display();
delay(FRAME_DELAY);
return;
}
display.clearDisplay();
display.fillRect(0, 56, 128, 8, SSD1306_INVERSE);
display.setCursor(0, 56);
display.print("requesting departures");
display.display();
String departuresRaw = fetchDepartures();
deserializeJson(departures, departuresRaw);
state = SHOWING_DEPARTURES;
frame = 0;
return;
}
case SHOWING_DEPARTURES:
if (WiFi.status() != WL_CONNECTED) {
state = CONNECTING;
return;
}
/*
if (buttonPushed) {
buttonPushed = false;
delay(FRAME_DELAY);
return;
}
*/
// Marquee effect for direction strings longer than 11 characters
// Neither particularly efficient nor legible
// (This could use some love)
int availableWidth = 11;
display.clearDisplay();
display.setCursor(0, 0);
for (JsonVariant departure : departures["departures"].as<JsonArray>()) {
const char *directionStr = departure["direction"].as<const char *>();
char buf[128] = {0};
const char *dirStart;
if (strlen(directionStr) <= availableWidth || (strlen(directionStr) + 3 + availableWidth) > 127) {
dirStart = directionStr;
} else {
memcpy(buf, directionStr, strlen(directionStr));
memcpy(buf + strlen(directionStr), " ", 3);
memcpy(buf + strlen(directionStr) + 3, directionStr, availableWidth);
buf[strlen(directionStr) + 3 + availableWidth] = '\0';
dirStart = buf + frame % (strlen(directionStr) + 3);
}
display.printf("%2s %-11.11s %6s\n",
departure["symbol"].as<const char*>(),
dirStart,
departure["leaving"].as<const char*>()
);
}
display.display();
frame++;
delay(500);
return;
}
}
-1
View File
@@ -43,4 +43,3 @@ def generate_bitmap_from_gif(file: Path, fd):
if __name__ == "__main__":
generate_bitmap_from_gif(Path("./animations/spinny.gif"), sys.stdout)
generate_bitmap_from_gif(Path("./animations/tiny_train.gif"), sys.stdout)
Generated
-34
View File
@@ -1,23 +1,5 @@
{
"nodes": {
"flake-utils": {
"inputs": {
"systems": "systems"
},
"locked": {
"lastModified": 1731533236,
"narHash": "sha256-l0KFg5HjrsfsO/JpG+r7fRrqm12kzFHyUHqHCVpMMbI=",
"owner": "numtide",
"repo": "flake-utils",
"rev": "11707dc2f618dd54ca8739b309ec4fc024de578b",
"type": "github"
},
"original": {
"owner": "numtide",
"repo": "flake-utils",
"type": "github"
}
},
"gitignore": {
"inputs": {
"nixpkgs": [
@@ -56,25 +38,9 @@
},
"root": {
"inputs": {
"flake-utils": "flake-utils",
"gitignore": "gitignore",
"nixpkgs": "nixpkgs"
}
},
"systems": {
"locked": {
"lastModified": 1681028828,
"narHash": "sha256-Vy1rq5AaRuLzOxct8nz4T6wlgyUR7zLU309k9mBC768=",
"owner": "nix-systems",
"repo": "default",
"rev": "da67096a3b9bf56a91d16901293e51ba5b49a27e",
"type": "github"
},
"original": {
"owner": "nix-systems",
"repo": "default",
"type": "github"
}
}
},
"root": "root",
+8 -9
View File
@@ -3,20 +3,20 @@
inputs = {
nixpkgs.url = "github:nixos/nixpkgs?ref=nixos-unstable";
flake-utils.url = "github:numtide/flake-utils";
gitignore = {
url = "github:hercules-ci/gitignore.nix";
inputs.nixpkgs.follows = "nixpkgs";
};
};
outputs = { self, nixpkgs, flake-utils, gitignore }: flake-utils.lib.eachDefaultSystem (system:
outputs = { self, nixpkgs, gitignore }:
let
pkgs = import nixpkgs { inherit system; };
system = "x86_64-linux";
pkgs = nixpkgs.legacyPackages.${system};
vrnp-static = pkgs.buildGoModule {
pname = "vrnp";
version = "0.0.9";
version = "0.0.4";
vendorHash = null;
# For building the package, we use only the files not ignored by Git as inputs.
@@ -31,15 +31,15 @@
CGO_ENABLED = 0;
};
in {
devShell = pkgs.mkShell {
devShell.${system} = pkgs.mkShellNoCC {
packages = [
pkgs.go
(pkgs.python3.withPackages (ps: with ps; [ pillow ]))
];
};
packages = {
packages.${system} = {
default = vrnp-static;
image-meta = pkgs.runCommand "image-meta" {} ''
image-meta = pkgs.runCommand "vrnp-version" {} ''
mkdir -p $out
echo -n ${vrnp-static.version} > $out/version
echo -n git.pbrinkmeier.de/paul/vrnp:${vrnp-static.version} > $out/name
@@ -55,6 +55,5 @@
config.Cmd = [ "${vrnp-static}/bin/vrnp" ];
};
};
}
);
};
}
+22 -167
View File
@@ -1,6 +1,5 @@
package main
import "cmp"
import "encoding/json"
import "errors"
import "fmt"
@@ -9,9 +8,7 @@ import "net/http"
import "net/url"
import "os"
import "slices"
import "strings"
import "strconv"
import "sync/atomic"
import "time"
// JSON unmarshaling types for departure monitor API
@@ -46,150 +43,28 @@ type DMDateTime struct {
Minute string `json:"minute"`
}
type EFAClient interface {
GetName() string
BuildRequest(string) (*http.Request, error)
}
var allEfaClients []EFAClient = []EFAClient{
BwegtEFAClient{},
VRNEFAClient{},
KVVEFAClient{},
VAGEFAClient{},
}
type VRNEFAClient struct {
}
func (c VRNEFAClient) GetName() string {
return "VRN"
}
func (c VRNEFAClient) BuildRequest(stopId string) (*http.Request, error) {
func FetchDepartures(stopId string) (DMResponse, error) {
// Create request object
req, err := http.NewRequest("GET", "https://www.vrn.de/mngvrn/XML_DM_REQUEST", nil)
if err != nil {
return nil, err
}
// Configure our request
query := url.Values{}
query.Set("coordOutputFormat", "EPSG:4326")
query.Set("depType", "stopEvents")
query.Set("includeCompleteStopSeq", "0")
query.Set("limit", "10")
query.Set("locationServerActive", "0")
query.Set("mode", "direct")
query.Set("name_dm", stopId)
query.Set("outputFormat", "json")
query.Set("type_dm", "stop")
query.Set("useOnlyStops", "1")
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("useRealtime", "1")
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
}
type BwegtEFAClient struct {
}
func (c BwegtEFAClient) GetName() string {
return "bwegt"
}
func (c BwegtEFAClient) BuildRequest(stopId string) (*http.Request, error) {
// Create request object
req, err := http.NewRequest("GET", "https://www.bwegt.de/bwegt-efa/XML_DM_REQUEST", nil)
if err != nil {
return nil, err
}
// Configure our request
query := url.Values{}
query.Set("coordOutputFormat", "EPSG:4326")
query.Set("depType", "stopEvents")
query.Set("includeCompleteStopSeq", "0")
query.Set("limit", "10")
query.Set("locationServerActive", "0")
query.Set("mode", "direct")
query.Set("name_dm", stopId)
query.Set("outputFormat", "json")
query.Set("type_dm", "stop")
query.Set("useOnlyStops", "1")
query.Set("useRealtime", "1")
req.URL.RawQuery = query.Encode()
return req, nil
}
type VAGEFAClient struct {
}
func (c VAGEFAClient) GetName() string {
return "VAG"
}
func (c VAGEFAClient) BuildRequest(stopId string) (*http.Request, error) {
// Create request object
req, err := http.NewRequest("GET", "https://efa.vagfr.de/vagfr3/XSLT_DM_REQUEST", nil)
if err != nil {
return nil, err
}
// Configure our request
query := url.Values{}
query.Set("coordOutputFormat", "EPSG:4326")
query.Set("depType", "stopEvents")
query.Set("includeCompleteStopSeq", "0")
query.Set("limit", "10")
query.Set("locationServerActive", "0")
query.Set("mode", "direct")
query.Set("name_dm", stopId)
query.Set("outputFormat", "json")
query.Set("type_dm", "stop")
query.Set("useOnlyStops", "1")
query.Set("useRealtime", "1")
req.URL.RawQuery = query.Encode()
return req, nil
}
func FetchDepartures(c EFAClient, stopId string) (DMResponse, error) {
req, err := c.BuildRequest(stopId)
if err != nil {
return DMResponse{}, err
}
// Configure our request
query := url.Values{}
query.Set("coordOutputFormat", "EPSG:4326")
query.Set("depType", "stopEvents")
query.Set("includeCompleteStopSeq", "0")
query.Set("limit", "10")
query.Set("locationServerActive", "0")
query.Set("mode", "direct")
query.Set("name_dm", stopId)
query.Set("outputFormat", "json")
query.Set("type_dm", "stop")
query.Set("useOnlyStops", "1")
query.Set("useRealtime", "1")
req.URL.RawQuery = query.Encode()
// Send the request, wait max 10 seconds
client := http.Client{
Timeout: 10 * time.Second,
@@ -220,7 +95,6 @@ type Departure struct {
Symbol string `json:"symbol"`
Direction string `json:"direction"`
Leaving string `json:"leaving"`
Countdown int `json:"-"`
}
func ParseDepartures(response DMResponse, allowedPlatform *string) (Departures, error) {
@@ -243,12 +117,6 @@ func ParseDepartures(response DMResponse, allowedPlatform *string) (Departures,
return Departures{}, errors.New("could not parse countdown")
}
matchesPlatform := allowedPlatform == nil || d.Platform == *allowedPlatform
isInThePast := countdown < 0
if !matchesPlatform || isInThePast {
continue
}
if countdown == 0 {
leaving = "sofort"
}
@@ -260,37 +128,25 @@ func ParseDepartures(response DMResponse, allowedPlatform *string) (Departures,
leaving = fmt.Sprintf("%02s:%02s", dt.Hour, dt.Minute)
}
if allowedPlatform != nil && d.Platform != *allowedPlatform {
continue
}
ds = append(ds, Departure{
d.Platform,
d.ServingLine.Symbol,
direction,
leaving,
countdown,
})
}
slices.SortFunc(ds, func(a Departure, b Departure) int {
return cmp.Compare(a.Countdown, b.Countdown)
})
return Departures{ds}, nil
}
func main() {
password := os.Getenv("VRNP_PASSWORD")
if len(password) == 0 {
log.Fatal("Required environment variable VRNP_PASSWORD is not set")
}
clientAllowlist := strings.Split(os.Getenv("VRNP_CLIENT_ALLOWLIST"), ",")
// Use round-robin to send incoming requests to different servers
var efaClient atomic.Uint64
var efaClients []EFAClient
for _, efaClient := range allEfaClients {
if len(clientAllowlist) != 0 && slices.Contains(clientAllowlist, efaClient.GetName()) {
efaClients = append(efaClients, efaClient)
}
panic("Required environment variable VRNP_PASSWORD is not set")
}
http.HandleFunc("/departures", func(w http.ResponseWriter, r *http.Request) {
@@ -314,8 +170,7 @@ func main() {
platform = &query["platform"][0]
}
c := efaClients[(efaClient.Add(1) - 1) % uint64(len(efaClients))]
ds, err := FetchDepartures(c, stopId)
ds, err := FetchDepartures(stopId)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return