From 6b6ceafe1aae23d1973c8b7a6169c05f0a400d28 Mon Sep 17 00:00:00 2001
From: iabdalkader <i.abdalkader@gmail.com>
Date: Sat, 14 May 2022 15:00:20 +0200
Subject: [PATCH] extmod/webrepl: Fix setting password in foreground mode and
 factor code.

The password was not being set when running in foreground mode.  Duplicate
code has been removed.
---
 extmod/webrepl/webrepl.py | 26 +++++++++++++++-----------
 1 file changed, 15 insertions(+), 11 deletions(-)

diff --git a/extmod/webrepl/webrepl.py b/extmod/webrepl/webrepl.py
index 8ddf56143f3a..0de813cda18c 100644
--- a/extmod/webrepl/webrepl.py
+++ b/extmod/webrepl/webrepl.py
@@ -59,24 +59,28 @@ def stop():
         listen_s.close()
 
 
-def start(port=8266, password=None):
+def start(port=8266, password=None, accept_handler=accept_conn):
     stop()
-    if password is None:
+    webrepl_pass = password
+    if webrepl_pass is None:
         try:
             import webrepl_cfg
 
-            _webrepl.password(webrepl_cfg.PASS)
-            setup_conn(port, accept_conn)
-            print("Started webrepl in normal mode")
+            webrepl_pass = webrepl_cfg.PASS
         except:
             print("WebREPL is not configured, run 'import webrepl_setup'")
+
+    _webrepl.password(webrepl_pass)
+    s = setup_conn(port, accept_handler)
+
+    if accept_handler is None:
+        print("Starting webrepl in foreground mode")
+        accept_conn(s)
+    elif password is None:
+        print("Started webrepl in normal mode")
     else:
-        _webrepl.password(password)
-        setup_conn(port, accept_conn)
         print("Started webrepl in manual override mode")
 
 
-def start_foreground(port=8266):
-    stop()
-    s = setup_conn(port, None)
-    accept_conn(s)
+def start_foreground(port=8266, password=None):
+    start(port, password, None)