main(): Handle errors from HTTP responses

Add some simple error handling for known responses from our TimeTask
HTTP requests. Check a couple of known strings to determine whether
there was an error. If so, exit with a failing error code.

Remove our old `log` statements. These were used during debugging to see
some output and check responses from TimeTask. We now have an idea of
what those responses are, and are handling some of the error cases. Thus
the log statements are no longer needed.

1 files changed, 15 insertions, 5 deletions

diff --git a/main.go b/main.go
index 3b7ace5..861dc8b 100644
--- a/main.go
+++ b/main.go
@@ -3,8 +3,8 @@ package main
 import (
 	"fmt"
 	"io/ioutil"
-	"log"
 	"os"
+	"strings"
 	"time"
 
 	"github.com/teddywing/timetasker/timetask"
@@ -97,17 +97,27 @@ func main() {
 		password,
 	)
 	kingpin.FatalIfError(err, "login request failed")
-	log.Printf("%+v\n", resp)
 
 	defer resp.Body.Close()
 	body, err := ioutil.ReadAll(resp.Body)
-	log.Println(string(body))
+	if strings.Contains(
+		string(body),
+		"The username and password don't appear to be valid.",
+	) {
+		kingpin.Errorf("TimeTask authentication failed")
+		os.Exit(1)
+	}
 
 	resp, err = timetask.SubmitTimeEntry(*client, time_entry)
 	kingpin.FatalIfError(err, "time entry submission request failed")
-	log.Printf("%+v\n", resp)
 
 	defer resp.Body.Close()
 	body, err = ioutil.ReadAll(resp.Body)
-	log.Println(string(body))
+	if strings.Contains(
+		string(body),
+		"No time entries were created.",
+	) {
+		kingpin.Errorf("time entry creation failed")
+		os.Exit(1)
+	}
 }