From cf884263c8d6fc4a02d977c71e10b47d310d3bf5 Mon Sep 17 00:00:00 2001 From: Jon Banafato Date: Mon, 6 Apr 2020 00:21:17 -0400 Subject: [PATCH] Add api token flag Currently the `demo` API token is hardcoded, which is not usable for real API calls. Add a new CLI flag to allow users to supply their own registered API token. --- main.go | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/main.go b/main.go index 45b4e00..dad6e2b 100644 --- a/main.go +++ b/main.go @@ -13,6 +13,7 @@ import ( ) func main() { + apiToken := flag.String("a", "demo", "World Trading Data API Token") ledgerBinary := flag.String("b", "ledger", "Ledger Binary") ledgerFile := flag.String("f", "ledger.ledger", "Ledger File") priceDbFile := flag.String("p", "prices.db", "Price Database File") @@ -27,7 +28,7 @@ func main() { defer pricedb.Close() for _, c := range commodities { - priceString, err := GetPriceString(c) + priceString, err := GetPriceString(c, *apiToken) if err != nil { continue } @@ -35,8 +36,8 @@ func main() { } } -func GetPriceString(ticker string) (string, error) { - resp, err := http.Get("https://api.worldtradingdata.com/api/v1/stock?symbol=" + ticker + "&api_token=demo") +func GetPriceString(ticker string, apiToken string) (string, error) { + resp, err := http.Get("https://api.worldtradingdata.com/api/v1/stock?symbol=" + ticker + "&api_token=" + apiToken) if err != nil { return "", err }