55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
#!/usr/bin/env python3
|
|
import sys
|
|
import json
|
|
import base64
|
|
from playwright.sync_api import sync_playwright
|
|
|
|
def run_bridge():
|
|
# Read command from stdin
|
|
try:
|
|
raw_input = sys.stdin.read()
|
|
if not raw_input:
|
|
print(json.dumps({"status": "error", "message": "No input provided"}))
|
|
return
|
|
|
|
args = json.loads(raw_input)
|
|
except Exception as e:
|
|
print(json.dumps({"status": "error", "message": f"Invalid JSON input: {str(e)}"}))
|
|
return
|
|
|
|
url = args.get("url")
|
|
action = args.get("action", "extract_text")
|
|
selector = args.get("selector", "body")
|
|
|
|
if not url:
|
|
print(json.dumps({"status": "error", "message": "No URL provided"}))
|
|
return
|
|
|
|
try:
|
|
with sync_playwright() as p:
|
|
browser = p.chromium.launch(headless=True)
|
|
page = browser.new_page()
|
|
|
|
# Navigate and wait for network to be idle
|
|
page.goto(url, wait_until="networkidle")
|
|
|
|
result = {"status": "success", "url": url}
|
|
|
|
if action == "extract_text":
|
|
result["content"] = page.inner_text(selector)
|
|
elif action == "screenshot":
|
|
screenshot_bytes = page.screenshot()
|
|
result["screenshot_base64"] = base64.b64encode(screenshot_bytes).decode("utf-8")
|
|
else:
|
|
result["status"] = "error"
|
|
result["message"] = f"Unknown action: {action}"
|
|
|
|
browser.close()
|
|
print(json.dumps(result))
|
|
|
|
except Exception as e:
|
|
print(json.dumps({"status": "error", "message": f"Playwright Error: {str(e)}"}))
|
|
|
|
if __name__ == "__main__":
|
|
run_bridge()
|