fix issue with missing token/login detection

nitram509 · May 12, 2024 · a979388 · a979388
1 parent a951306
commit a979388
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 11 deletions.
diff --git a/login.go b/login.go
@@ -119,7 +119,10 @@ func doLogin(args *GlobalOptions, host string, encryptedPwd string) error {
 }
 
 func checkIsLoginRequired(httpResponseBody string) bool {
-	return len(httpResponseBody) < 10 || strings.Contains(httpResponseBody, "/login.cgi") || strings.Contains(httpResponseBody, "/redirect.html")
+	return len(httpResponseBody) < 10 ||
+		strings.Contains(httpResponseBody, "/login.cgi") ||
+		strings.Contains(httpResponseBody, "/wmi/login") ||
+		strings.Contains(httpResponseBody, "/redirect.html")
 }
 
 func getSessionToken(resp *http.Response) string {

diff --git a/poe_status.go b/poe_status.go
@@ -42,16 +42,6 @@ func (poe *PoeStatusCommand) Run(args *GlobalOptions) error {
 	}
 	var statuses []PoePortStatus
 	statuses, err = findPortStatusInHtml(args.model, strings.NewReader(statusPage))
-
-	// FIXME: remove
-	if len(statuses) == 0 {
-		println("DEBUG: status are zero")
-		println("--------")
-		println(string(statusPage))
-		println("--------")
-
-	}
-
 	if err != nil {
 		return err
 	}