mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-04-28 11:40:47 +00:00
5193ef7501
- Change default mode to self-chat across all modules - Update README to reflect self-chat as primary mode with security warning - Move session/media storage from usr/whatsapp to tmp/whatsapp - Reorder config UI: move Mode above Allowed Numbers - Add warning banner when allowed_numbers is empty in self-chat mode - Move Bridge Port and Poll Interval to bottom of settings - Update mode descriptions to clarify self-chat handles both self
83 lines
2.6 KiB
Python
83 lines
2.6 KiB
Python
"""WhatsApp QR code pairing endpoint."""
|
|
|
|
from helpers.api import ApiHandler, Request
|
|
from helpers.errors import format_error
|
|
from helpers import plugins
|
|
|
|
|
|
PLUGIN_NAME = "_whatsapp_integration"
|
|
|
|
|
|
class QrCode(ApiHandler):
|
|
|
|
async def process(self, input: dict, request: Request) -> dict:
|
|
config = plugins.get_plugin_config(PLUGIN_NAME) or {}
|
|
port = int(config.get("bridge_port", 3100))
|
|
mode = config.get("mode", "self-chat")
|
|
|
|
from plugins._whatsapp_integration.helpers.bridge_manager import (
|
|
ensure_bridge_http_up,
|
|
get_bridge_url,
|
|
is_process_alive,
|
|
)
|
|
from plugins._whatsapp_integration.helpers.storage_paths import (
|
|
get_bridge_media_dir,
|
|
get_bridge_session_dir,
|
|
)
|
|
from plugins._whatsapp_integration.helpers.wa_client import get_qr
|
|
|
|
session_dir = get_bridge_session_dir()
|
|
cache_dir = get_bridge_media_dir()
|
|
base_url = get_bridge_url(port)
|
|
|
|
# Start bridge if not running
|
|
if not is_process_alive():
|
|
try:
|
|
ok = await ensure_bridge_http_up(
|
|
port, session_dir, cache_dir, mode=mode,
|
|
)
|
|
if not ok:
|
|
return {
|
|
"status": "error",
|
|
"message": "Failed to start bridge",
|
|
"qr": None,
|
|
}
|
|
except Exception as e:
|
|
return {
|
|
"status": "error",
|
|
"message": format_error(e),
|
|
"qr": None,
|
|
}
|
|
|
|
# Fetch QR status from bridge
|
|
try:
|
|
data = await get_qr(base_url)
|
|
status = data.get("status", "error")
|
|
qr = data.get("qr")
|
|
|
|
if status == "connected":
|
|
return {
|
|
"status": "connected",
|
|
"message": "WhatsApp is already connected",
|
|
"qr": None,
|
|
}
|
|
|
|
if status == "waiting_scan" and qr:
|
|
return {
|
|
"status": "waiting_scan",
|
|
"message": "Scan the QR code with WhatsApp",
|
|
"qr": qr,
|
|
}
|
|
|
|
return {
|
|
"status": "waiting_qr",
|
|
"message": "Generating QR code...",
|
|
"qr": None,
|
|
}
|
|
|
|
except Exception as e:
|
|
return {
|
|
"status": "error",
|
|
"message": f"Bridge not reachable: {format_error(e)}",
|
|
"qr": None,
|
|
}
|