Author SHA1 Message Date
paul 740b654be6 Add bwegt efa client 2025-04-25 13:37:01 +02:00
paul dcc3cfa44f Send requests to different servers using round-robin 2025-04-25 12:30:19 +02:00
paul ae7b2f170a Add EFAClient interface 2025-04-25 12:04:47 +02:00
paul fad4da3f6b Use flake-utils, use mkShell
This fixes the devshell on darwin
2025-04-25 03:29:24 +02:00
paul 76cca9c1c0 Add tiny_train sprite 2025-04-25 03:18:49 +02:00
8 changed files with 193 additions and 52 deletions
+1
View File
@@ -1,3 +1,4 @@
result
vrnp
main
*.swp
Binary file not shown.

After

Width:  |  Height:  |  Size: 4.0 KiB

+14
View File
@@ -81,3 +81,17 @@ 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,
},
};
+3 -6
View File
@@ -102,20 +102,17 @@ void setup() {
pinMode(BUTTON_PIN, INPUT_PULLUP);
attachInterrupt(BUTTON_PIN, onButtonFalling, FALLING);
/*
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"
// TODO: Error handling
String fetchDepartures() {
std::unique_ptr<BearSSL::WiFiClientSecure> https(new BearSSL::WiFiClientSecure);
https->setInsecure();
HTTPClient client;
if (!client.begin(*https, "vrnp.beany.club", 443, "/departures?stop_id=de:08221:1225&platform=A")) {
if (!client.begin(*https, "vrnp.beany.club", 443, "/departures?" QUERY_STRING)) {
display.fillScreen(COLOR_BG);
display.setCursor(0, 0);
display.print("begin failed");
+1
View File
@@ -43,3 +43,4 @@ 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,5 +1,23 @@
{
"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": [
@@ -38,9 +56,25 @@
},
"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 -7
View File
@@ -3,16 +3,16 @@
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, gitignore }:
outputs = { self, nixpkgs, flake-utils, gitignore }: flake-utils.lib.eachDefaultSystem (system:
let
system = "x86_64-linux";
pkgs = nixpkgs.legacyPackages.${system};
pkgs = import nixpkgs { inherit system; };
vrnp-static = pkgs.buildGoModule {
pname = "vrnp";
@@ -31,15 +31,15 @@
CGO_ENABLED = 0;
};
in {
devShell.${system} = pkgs.mkShellNoCC {
devShell = pkgs.mkShell {
packages = [
pkgs.go
(pkgs.python3.withPackages (ps: with ps; [ pillow ]))
];
};
packages.${system} = {
packages = {
default = vrnp-static;
image-meta = pkgs.runCommand "vrnp-version" {} ''
image-meta = pkgs.runCommand "image-meta" {} ''
mkdir -p $out
echo -n ${vrnp-static.version} > $out/version
echo -n git.pbrinkmeier.de/paul/vrnp:${vrnp-static.version} > $out/name
@@ -55,5 +55,6 @@
config.Cmd = [ "${vrnp-static}/bin/vrnp" ];
};
};
};
}
);
}
+96 -3
View File
@@ -8,7 +8,9 @@ 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
@@ -43,11 +45,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 +79,76 @@ 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
}
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
}
func FetchDepartures(c EFAClient, stopId string) (DMResponse, error) {
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,
@@ -151,6 +235,14 @@ func main() {
panic("Required environment variable VRNP_PASSWORD is not set")
}
// Use round-robin to send incoming requests to different servers
var efaClient atomic.Uint64
efaClients := []EFAClient{
BwegtEFAClient{},
VRNEFAClient{},
KVVEFAClient{},
}
http.HandleFunc("/departures", func(w http.ResponseWriter, r *http.Request) {
user, pass, ok := r.BasicAuth()
if !(ok && user == "admin" && pass == password) {
@@ -172,7 +264,8 @@ func main() {
platform = &query["platform"][0]
}
ds, err := FetchDepartures(stopId)
c := efaClients[(efaClient.Add(1) - 1) % uint64(len(efaClients))]
ds, err := FetchDepartures(c, stopId)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return