gofio

commit 44a7b24af5654585091f2a020fa7b102c6f4452f
parent 51f293c4d0d2ed8ab9f03004901c1ca31e80e228
Author: tms <nemi@skaut.cz>
Date:   Fri,  9 Apr 2021 14:38:08 +0200

token as file

Diffstat:
M
main.go
 | 
18
+++++++++++++-----

1 file changed, 13 insertions(+), 5 deletions(-)
diff --git a/main.go b/main.go
@@ -16,22 +16,30 @@ import (
 
 const base_url = "https://www.fio.cz/ib_api/rest/"
 
-var token = flag.String("t", "", "fio api token")
+var tokenFile = flag.String("t", "", "fio api token")
+var token string
 
 var page string
 
 func main() {
 	flag.Parse()
 
-	if *token == "" {
-		log.Fatal("Token was not provided")
+	if *tokenFile == "" {
+		log.Fatal("Token file was not provided")
+		os.Exit(1)
+	}
+
+	data, err := os.ReadFile(*tokenFile)
+	token = strings.TrimSuffix(string(data), "\n")
+	if err != nil {
+		log.Fatal("Cannot read token file")
 		os.Exit(1)
 	}
 
 	http.HandleFunc("/", serveFiles)
 
 	log.Println("Listening on :3000...")
-	err := http.ListenAndServe(":3000", nil)
+	err = http.ListenAndServe(":3000", nil)
 	if err != nil {
 		log.Fatal(err)
 		os.Exit(1)
@@ -39,7 +47,7 @@ func main() {
 }
 func serveFiles(w http.ResponseWriter, r *http.Request) {
 	index, _ := os.ReadFile("./web/index.html")
-	fio := &fio.Fio{Base_url: base_url, Token: *token}
+	fio := &fio.Fio{Base_url: base_url, Token: token}
 
 	if page != "" && !hasNew(fio) {
 		log.Println("No new updates")