dimitri/sncf-bot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Move command-line argument parsing to another function

Browse source
This commit is contained in:
Dimitri Lozeve 2022-09-12 20:41:30 +02:00
parent aacd35d766
commit 0329f4d696
1 changed files with 20 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

38
sncf.ss
View file

@ -2,16 +2,32 @@
(export main) (export main)
(import :std/format (import :std/getopt
:std/getopt
:std/srfi/19 :std/srfi/19
:std/sugar :std/sugar
:dlozeve/fancy/format
:dlozeve/sncf/api :dlozeve/sncf/api
:dlozeve/sncf/display :dlozeve/sncf/display
:dlozeve/sncf/mattermost) :dlozeve/sncf/mattermost)
(def (main . args) (def (main . args)
(define-values (station datetime mattermost-url mattermost-channel) (parse-arguments args))
(def sncf-key (getenv "SNCF_AUTH_KEY" #f))
(unless sncf-key
(display "No SNCF API authentication key found. Set the SNCF_AUTH_KEY environment variable.\n"
(current-error-port))
(exit 1))
(define-values (station-name station-id)
(if station
(get-station-id sncf-key station)
(values "Vernon - Giverny (Vernon)" "stop_area:SNCF:87415604")))
(define-values (departures disruptions) (get-departures sncf-key station-id datetime))
(display-all departures disruptions station-name datetime)
(when mattermost-url
(let ((tab-str-md
(with-output-to-string (lambda () (display-all departures disruptions station-name datetime)))))
(post-to-mattermost mattermost-url tab-str-md channel: mattermost-channel))))
(def (parse-arguments args)
(def gopt (getopt (optional-argument 'station help: "Name of the station (default Vernon-Giverny).") (def gopt (getopt (optional-argument 'station help: "Name of the station (default Vernon-Giverny).")
(optional-argument 'datetime help: "Date and time (ISO 8601 format).") (optional-argument 'datetime help: "Date and time (ISO 8601 format).")
(flag 'help "-h" "--help" help: "Display this help.") (flag 'help "-h" "--help" help: "Display this help.")
@ -28,21 +44,7 @@
(when help (when help
(getopt-display-help gopt "sncf") (getopt-display-help gopt "sncf")
(exit 0)) (exit 0))
(def sncf-key (getenv "SNCF_AUTH_KEY" #f)) (values station datetime mattermost-url mattermost-channel))
(unless sncf-key
(display "No SNCF API authentication key found. Set the SNCF_AUTH_KEY environment variable.\n"
(current-error-port))
(exit 1))
(define-values (station-name station-id)
(if station
(get-station-id sncf-key station)
(values "Vernon - Giverny (Vernon)" "stop_area:SNCF:87415604")))
(let-values (((departures disruptions) (get-departures sncf-key station-id datetime)))
(display-all departures disruptions station-name datetime)
(when mattermost-url
(let ((tab-str-md
(with-output-to-string (lambda () (display-all departures disruptions station-name datetime)))))
(post-to-mattermost mattermost-url tab-str-md channel: mattermost-channel)))))
(catch (getopt-error? exn) (catch (getopt-error? exn)
(getopt-display-help exn "sncf" (current-error-port)) (getopt-display-help exn "sncf" (current-error-port))
(exit 1)) (exit 1))