Update getters for Chrome and VLC media player software

desktop_env/evaluators/metrics/chrome.py

@@ -3,6 +3,8 @@ import os
 import platform
 import sqlite3
 
+from playwright.sync_api import sync_playwright
+
 """
 WARNING: 
 1. Functions from this script assume that no account is registered on Chrome, otherwise the default file path needs to be changed.
@@ -12,6 +14,7 @@ WARNING:
 
 # todo: move to getter module
 
+# The following ones just need to load info from the files of software, no need to connect to the software
 def get_default_search_engine():
     if platform.system() == 'Windows':
         preference_file_path = os.path.join(os.getenv('LOCALAPPDATA'),
@@ -19,8 +22,10 @@ def get_default_search_engine():
     elif platform.system() == 'Darwin':
         preference_file_path = os.path.join(os.getenv('HOME'),
                                             'Library/Application Support/Google/Chrome/Default/Preferences')
-    else:
+    elif platform.system() == 'Linux':
         preference_file_path = os.path.join(os.getenv('HOME'), '.config/google-chrome/Default/Preferences')
+    else:
+        raise Exception('Unsupported operating system')
 
     try:
         with open(preference_file_path, 'r', encoding='utf-8') as file:
@@ -41,8 +46,10 @@ def get_cookie_data():
     elif platform.system() == 'Darwin':
         chrome_cookie_file_path = os.path.join(os.getenv('HOME'),
                                                'Library/Application Support/Google/Chrome/Default/Cookies')
-    else:
+    elif platform.system() == 'Linux':
         chrome_cookie_file_path = os.path.join(os.getenv('HOME'), '.config/google-chrome/Default/Cookies')
+    else:
+        raise Exception('Unsupported operating system')
 
     try:
         conn = sqlite3.connect(chrome_cookie_file_path)
@@ -65,8 +72,10 @@ def get_bookmarks():
     elif platform.system() == 'Darwin':
         preference_file_path = os.path.join(os.getenv('HOME'),
                                             'Library/Application Support/Google/Chrome/Default/Bookmarks')
-    else:
+    elif platform.system() == 'Linux':
         preference_file_path = os.path.join(os.getenv('HOME'), '.config/google-chrome/Default/Bookmarks')
+    else:
+        raise Exception('Unsupported operating system')
 
     try:
         with open(preference_file_path, 'r', encoding='utf-8') as file:
@@ -78,3 +87,75 @@ def get_bookmarks():
     except Exception as e:
         print(f"Error: {e}")
         return None
+
+
+def get_extensions_installed_from_shop():
+    """Find the Chrome extensions directory based on the operating system."""
+    os_name = platform.system()
+    if os_name == 'Windows':
+        chrome_extension_dir = os.path.expanduser(
+            '~') + '\\AppData\\Local\\Google\\Chrome\\User Data\\Default\\Extensions\\'
+    elif os_name == 'Darwin':  # macOS
+        chrome_extension_dir = os.path.expanduser(
+            '~') + '/Library/Application Support/Google/Chrome/Default/Extensions/'
+    elif os_name == 'Linux':
+        chrome_extension_dir = os.path.expanduser('~') + '/.config/google-chrome/Default/Extensions/'
+    else:
+        raise Exception('Unsupported operating system')
+
+    manifests = []
+    for extension_id in os.listdir(chrome_extension_dir):
+        extension_path = os.path.join(chrome_extension_dir, extension_id)
+        if os.path.isdir(extension_path):
+            # Iterate through version-named subdirectories
+            for version_dir in os.listdir(extension_path):
+                version_path = os.path.join(extension_path, version_dir)
+                manifest_path = os.path.join(version_path, 'manifest.json')
+                if os.path.isfile(manifest_path):
+                    with open(manifest_path, 'r') as file:
+                        try:
+                            manifest = json.load(file)
+                            manifests.append(manifest)
+                        except json.JSONDecodeError:
+                            print(f"Error reading {manifest_path}")
+    return manifests
+
+
+# The following ones require Playwright to be installed on the target machine, and the chrome needs to be pre-config on port info to allow remote debugging, see README.md for details
+
+def get_open_tabs_info(remote_debugging_url):
+    with sync_playwright() as p:
+        # connect to remote Chrome instance
+        browser = p.chromium.connect_over_cdp(remote_debugging_url)
+
+        tabs_info = []
+        for context in browser.contexts:
+            for page in context.pages:
+                title = page.title()
+                url = page.url
+                tabs_info.append({'title': title, 'url': url})
+
+        browser.close()
+        return tabs_info
+
+
+def get_active_tab_info(remote_debugging_url):
+    with sync_playwright() as p:
+        # connect to remote Chrome instance
+        browser = p.chromium.connect_over_cdp(remote_debugging_url)
+
+        active_tab_info = {}
+        for context in browser.contexts:
+            for page in context.pages():
+                if page.is_visible("body"):  # check the visibility of the page body to determine the active status
+                    active_tab_info = {
+                        'title': page.title(),
+                        'url': page.url,
+                        'content': page.content()  # get the HTML content of the page
+                    }
+                    break
+            if active_tab_info:
+                break
+
+        browser.close()
+        return active_tab_info