generate_emu_config: Fix credentials

use an updated fork

tools/generate_emu_config/generate_emu_config.py

@@ -6,8 +6,7 @@ from external_components import (
 )
 from controller_config_generator import parse_controller_vdf
 from steam.client import SteamClient
-from steam.client.cdn import CDNClient
-from steam.enums import common
+from steam.webauth import WebAuth
 from steam.enums.common import EResult
 from steam.enums.emsg import EMsg
 from steam.core.msg import MsgProto
@@ -691,11 +690,14 @@ def main():
         sys.exit(1)
 
     client = SteamClient()
-    login_tmp_folder = os.path.join(get_exe_dir(RELATIVE_DIR), "login_temp")
-    if not os.path.exists(login_tmp_folder):
-        os.makedirs(login_tmp_folder)
-    client.set_credential_location(login_tmp_folder)
-
+    if ANON_LOGIN:
+        result = client.anonymous_login()
+        trials = 5
+        while result != EResult.OK and trials > 0:
+            time.sleep(1000)
+            result = client.anonymous_login()
+            trials -= 1
+    else:
         # first read the 'my_login.txt' file
         my_login_file = os.path.join(get_exe_dir(RELATIVE_DIR), "my_login.txt")
         if not ANON_LOGIN and os.path.isfile(my_login_file):
@@ -716,41 +718,28 @@ def main():
         if env_password:
             PASSWORD = env_password
         
+        # still no username? ask user
+        if not USERNAME:
+            USERNAME = input("Steam user: ")
 
-    if ANON_LOGIN:
-        result = client.anonymous_login()
-        trials = 5
-        while result != EResult.OK and trials > 0:
-            time.sleep(1000)
-            result = client.anonymous_login()
-            trials -= 1
-    elif (len(USERNAME) == 0 or len(PASSWORD) == 0):
-        client.cli_login()
-    else:
-        result = client.login(USERNAME, password=PASSWORD)
-        auth_code, two_factor_code = None, None
+        # the file to save/load credentials
+        REFRESH_TOKENS = os.path.join(get_exe_dir(RELATIVE_DIR), "refresh_tokens.json")
+        refresh_tokens = {}
+        if os.path.isfile(REFRESH_TOKENS):
+            with open(REFRESH_TOKENS) as f:
+                try:
+                    lf = json.load(f)
+                    refresh_tokens = lf if isinstance(lf, dict) else {}
+                except:
+                    pass
+        REFRESH_TOKEN = refresh_tokens.get(USERNAME)
+
+        webauth, result = WebAuth(), None
         while result in (
-            EResult.AccountLogonDenied, EResult.InvalidLoginAuthCode,
-            EResult.AccountLoginDeniedNeedTwoFactor, EResult.TwoFactorCodeMismatch,
             EResult.TryAnotherCM, EResult.ServiceUnavailable,
-            EResult.InvalidPassword,
-            ):
+            EResult.InvalidPassword, None):
 
-            if result == EResult.InvalidPassword:
-                print("invalid password, the password you set is wrong.")
-                exit(1)
-
-            elif result in (EResult.AccountLogonDenied, EResult.InvalidLoginAuthCode):
-                prompt = ("Enter email code: " if result == EResult.AccountLogonDenied else
-                            "Incorrect code. Enter email code: ")
-                auth_code, two_factor_code = input(prompt), None
-
-            elif result in (EResult.AccountLoginDeniedNeedTwoFactor, EResult.TwoFactorCodeMismatch):
-                prompt = ("Enter 2FA code: " if result == EResult.AccountLoginDeniedNeedTwoFactor else
-                            "Incorrect code. Enter 2FA code: ")
-                auth_code, two_factor_code = None, input(prompt)
-
-            elif result in (EResult.TryAnotherCM, EResult.ServiceUnavailable):
+            if result in (EResult.TryAnotherCM, EResult.ServiceUnavailable):
                 if prompt_for_unavailable and result == EResult.ServiceUnavailable:
                     while True:
                         answer = input("Steam is down. Keep retrying? [y/n]: ").lower()
@@ -760,8 +749,25 @@ def main():
                     if answer == 'n': break
 
                 client.reconnect(maxdelay=15)
+            elif result == EResult.InvalidPassword:
+                print("invalid password or refresh_token,")
+                print(f"correct the password or/and delete '{REFRESH_TOKENS}' and try again.")
+                exit(1)
 
-            result = client.login(USERNAME, PASSWORD, None, auth_code, two_factor_code)
+            if not REFRESH_TOKEN:
+                try:
+                    webauth.cli_login(USERNAME, PASSWORD)
+                except Exception as e:
+                    print(f'Unknown exception: {e}, maybe some account info is wrong?')
+                    exit(1)
+                USERNAME, PASSWORD = webauth.username, webauth.password
+                REFRESH_TOKEN = webauth.refresh_token
+
+            result = client.login(USERNAME, PASSWORD, REFRESH_TOKEN)
+
+        with open(REFRESH_TOKENS, 'w') as f:
+            refresh_tokens.update({USERNAME: REFRESH_TOKEN})
+            json.dump(refresh_tokens,f,indent=4)
 
     # read and prepend top_owners_ids.txt
     top_owners_file = os.path.join(get_exe_dir(RELATIVE_DIR), "top_owners_ids.txt")

tools/generate_emu_config/requirements.txt

@@ -1,4 +1,4 @@
-steam[client]
+steam[client] @ git+https://github.com/detiam/steam_websocket@b8239912e6a190f490aede529c08b5049096bdc8
 pyinstaller
 requests
 certifi