[Enterprise] "Remember me" option in api

2 files changed, 7 insertions, 5 deletions

diff --git a/python/__init__.py b/python/__init__.py
index 1cf00773..12cb7033 100644
--- a/python/__init__.py
+++ b/python/__init__.py
@@ -129,7 +129,7 @@ def _init_plugins():
 
 	if not core_ui_enabled() and core.BNGetProduct() == "Binary Ninja Enterprise Client":
 		# Enterprise client needs to checkout a license reservation or else BNInitPlugins will fail
-		if not core.BNAuthenticateEnterpriseServerWithMethod("Keychain") and (not core.BNIsLicenseValidated() or not enterprise.is_license_still_activated()):
+		if not enterprise.authenticate_with_method("Keychain") and (not core.BNIsLicenseValidated() or not enterprise.is_license_still_activated()):
 			raise RuntimeError(
 				"To use Binary Ninja Enterprise from a headless python script, you must check out a license first.\n"
 				"You can either check out a license for an extended time with the UI, or use the binaryninja.enterprise module.")
diff --git a/python/enterprise.py b/python/enterprise.py
index 5ea57f0c..b94fd7ca 100644
--- a/python/enterprise.py
+++ b/python/enterprise.py
@@ -25,24 +25,26 @@ def is_connected() -> bool:
 	return core.BNIsEnterpriseServerConnected()
 
 
-def authenticate_with_credentials(username: str, password: str):
+def authenticate_with_credentials(username: str, password: str, remember: bool = True):
 	"""
 	Authenticate to the Enterprise Server with username/password credentials.
 	:param str username: Username to use.
 	:param str password: Password to use.
+	:param bool remember: Remember token in keychain
 	"""
-	if not core.BNAuthenticateEnterpriseServerWithCredentials(username, password):
+	if not core.BNAuthenticateEnterpriseServerWithCredentials(username, password, remember):
 		raise RuntimeError(last_error())
 
 
-def authenticate_with_method(method: str):
+def authenticate_with_method(method: str, remember: bool = True):
 	"""
 	Authenticate to the Enterprise Server with a non-password method. Note that many of these will
 	open a URL for a browser-based login prompt, which may not be usable on headless installations.
 	See :func:`authentication_methods` for a list of accepted methods.
 	:param str method: Name of method to use.
+	:param bool remember: Remember token in keychain
 	"""
-	if not core.BNAuthenticateEnterpriseServerWithMethod(method):
+	if not core.BNAuthenticateEnterpriseServerWithMethod(method, remember):
 		raise RuntimeError(last_error())