asking to delte client id on logout now. useful if user wants to login agian

1 files changed, 10 insertions, 18 deletions

diff --git a/auth/server.go b/auth/server.go
index 5a18a36..02b9382 100644
--- a/auth/server.go
+++ b/auth/server.go
@@ -29,25 +29,17 @@ import (
 
 func listen(clientId, verifier string) {
   http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
-    query := r.URL.Query()
+	  code, codeExists := r.URL.Query()["code"]
 
-	code, codePresent := query["code"]
-	if !codePresent {
-	  // TODO: check if error message present
-	  fmt.Println("Error: response from MyAnimeList doesn't contain required code.")
-	  os.Exit(1)
-	}
-
-	accessToken, refreshToken, expiresIn := requestToken(clientId, verifier, code[0])
-
-	if accessToken != "" {
-	  w.WriteHeader(200)
-	  w.Write([]byte("<h1>You have successfully logged into macli.</h1>"))
-	}
-
-	setToken(accessToken)
-	setRefreshToken(refreshToken)
-	setExpiresIn(expiresIn)
+		if codeExists {
+	    accessToken, refreshToken, expiresIn := requestToken(clientId, verifier, code[0])
+		  setToken(accessToken)
+		  setRefreshToken(refreshToken)
+      setExpiresIn(expiresIn)
+			fmt.Println("\x1b[32mYou have successfully logged into macli.\x1b[0m")
+			fmt.Println("\x1b[32mYou can close the web browser tab now.\x1b[0m")
+			os.Exit(0)
+		}
   })
 
   err := http.ListenAndServe(":8000", nil)