| import os | |
| import subprocess | |
| import sys | |
| import shutil | |
| from typing import Optional | |
| PLATFORM = sys.platform | |
| NODE = 'node' if PLATFORM in ['linux', 'darwin'] else 'node.exe' | |
| def _find_node_path() -> Optional[str]: | |
| """Try multiple ways to find Node.js path.""" | |
| local_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), f'binaries/{NODE}') | |
| if os.path.isfile(local_path): | |
| return local_path | |
| system_path = shutil.which(NODE) | |
| if system_path: | |
| return system_path | |
| return NODE | |
| NODE_PATH = _find_node_path() | |
| VM_PATH = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'vm/botGuard.js') | |
| def generate_po_token(visitor_data: str) -> str: | |
| """ | |
| Run nodejs to generate poToken through botGuard. | |
| Raises: | |
| RuntimeError: If Node.js is not available | |
| """ | |
| try: | |
| result = subprocess.check_output( | |
| [NODE_PATH, VM_PATH, visitor_data], | |
| stderr=subprocess.PIPE | |
| ).decode() | |
| return result.replace("\n", "") | |
| except FileNotFoundError as e: | |
| raise RuntimeError( | |
| f"Node.js is required but not found. Tried path: {NODE_PATH}\n" | |
| "Please install Node.js or ensure it's in your PATH." | |
| ) from e | |
| except subprocess.CalledProcessError as e: | |
| raise RuntimeError( | |
| f"Failed to execute botGuard.js: {e.stderr.decode().strip()}" | |
| ) from e | |