Check error code of Write

martinbjeldbak · Aug 27, 2022 · 443828c · 443828c
1 parent ae43fe3
commit 443828c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/commands/auth.go b/commands/auth.go
@@ -52,7 +52,12 @@ var authCmd = &cobra.Command{
 					w.WriteHeader(http.StatusNotFound)
 					return
 				}
-				w.Write(file)
+				_, err = w.Write(file)
+				if err != nil {
+					fmt.Printf("Error writing to response: %v", err)
+					w.WriteHeader(http.StatusNotFound)
+					return
+				}
 			})
 
 			if err := http.ListenAndServe(fmt.Sprintf(":%v", localCallbackPort), nil); err != nil {