feat: add support for HTTP proxy configuration and display status in dashboard
CI / lint-and-test (push) Has been cancelled
CI / lint-and-test (push) Has been cancelled
This commit is contained in:
+2
-1
@@ -1,5 +1,5 @@
|
|||||||
# Telegram Bot Token (from @BotFather)
|
# Telegram Bot Token (from @BotFather)
|
||||||
TELEGRAM_BOT_TOKEN=8653489932:AAHhyOD1jtimE7kg0zoVCUVd3l0YEz_YJgg
|
TELEGRAM_BOT_TOKEN=8653489932:AAG_Ins2_z3sNHX8ZlGI4mhyzmUhWAWCZlg
|
||||||
|
|
||||||
# Direct PostgreSQL connection
|
# Direct PostgreSQL connection
|
||||||
POSTGRES_HOST=192.168.178.3
|
POSTGRES_HOST=192.168.178.3
|
||||||
@@ -11,3 +11,4 @@ POSTGRES_DB=postgres
|
|||||||
# Worker Configuration
|
# Worker Configuration
|
||||||
DEFAULT_INTERVAL_MINUTES=60
|
DEFAULT_INTERVAL_MINUTES=60
|
||||||
ADMIN_TELEGRAM_IDS=298181113 # Comma-separated Telegram user IDs with admin access
|
ADMIN_TELEGRAM_IDS=298181113 # Comma-separated Telegram user IDs with admin access
|
||||||
|
HTTPS_PROXY=datacenter-de.floxy.io:1338:IPv4D_TZhQUiP9C3-ttl-0:70EDRDQUpo9Jc0a
|
||||||
|
|||||||
@@ -83,6 +83,7 @@ Edit `.env` before first startup. All values are read by the worker and database
|
|||||||
| `POSTGRES_DB` | Database name | `postgres` |
|
| `POSTGRES_DB` | Database name | `postgres` |
|
||||||
| `JWT_SECRET` | PostgREST JWT signing key | auto-generated default |
|
| `JWT_SECRET` | PostgREST JWT signing key | auto-generated default |
|
||||||
| `DEFAULT_INTERVAL_MINUTES`| Default scrape interval per keyword | `5` |
|
| `DEFAULT_INTERVAL_MINUTES`| Default scrape interval per keyword | `5` |
|
||||||
|
| `HTTPS_PROXY` | Optional proxy for outbound HTTP requests | empty |
|
||||||
|
|
||||||
## Architecture
|
## Architecture
|
||||||
|
|
||||||
|
|||||||
+22
-14
@@ -372,13 +372,6 @@ async def main() -> None:
|
|||||||
|
|
||||||
pool = await get_pool()
|
pool = await get_pool()
|
||||||
|
|
||||||
app = Application.builder().token(os.getenv("TELEGRAM_BOT_TOKEN")).build()
|
|
||||||
|
|
||||||
from bot import register_handlers, setup_global_commands # noqa: E402
|
|
||||||
|
|
||||||
await setup_global_commands(app)
|
|
||||||
register_handlers(app)
|
|
||||||
|
|
||||||
# ── Start healthcheck HTTP server ──────────────────────────────
|
# ── Start healthcheck HTTP server ──────────────────────────────
|
||||||
health_app = create_health_app()
|
health_app = create_health_app()
|
||||||
runner = web.AppRunner(health_app)
|
runner = web.AppRunner(health_app)
|
||||||
@@ -403,7 +396,26 @@ async def main() -> None:
|
|||||||
except Exception:
|
except Exception:
|
||||||
logger.exception("Failed to start Web UI (optional)")
|
logger.exception("Failed to start Web UI (optional)")
|
||||||
|
|
||||||
|
app = Application.builder().token(os.getenv("TELEGRAM_BOT_TOKEN")).build()
|
||||||
|
scheduler = None
|
||||||
|
poll_task = None
|
||||||
|
bot_started = False
|
||||||
|
|
||||||
|
from bot import register_handlers, setup_global_commands # noqa: E402
|
||||||
|
|
||||||
|
try:
|
||||||
|
await setup_global_commands(app)
|
||||||
|
register_handlers(app)
|
||||||
|
await app.initialize()
|
||||||
|
await app.start()
|
||||||
|
logger.info("Bot started with long polling")
|
||||||
|
|
||||||
|
set_telegram_polling(True)
|
||||||
|
poll_task = asyncio.ensure_future(app.updater.start_polling()) # type: ignore[attr-defined]
|
||||||
scheduler = asyncio.ensure_future(scheduler_task(pool, app.bot))
|
scheduler = asyncio.ensure_future(scheduler_task(pool, app.bot))
|
||||||
|
bot_started = True
|
||||||
|
except Exception:
|
||||||
|
logger.exception("Telegram bot startup failed — continuing in UI-only mode")
|
||||||
|
|
||||||
loop = asyncio.get_running_loop()
|
loop = asyncio.get_running_loop()
|
||||||
stop = loop.create_future()
|
stop = loop.create_future()
|
||||||
@@ -416,18 +428,12 @@ async def main() -> None:
|
|||||||
loop.add_signal_handler(sig, _signal_handler)
|
loop.add_signal_handler(sig, _signal_handler)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
await app.initialize()
|
|
||||||
await app.start()
|
|
||||||
logger.info("Bot started with long polling")
|
|
||||||
|
|
||||||
set_telegram_polling(True)
|
|
||||||
poll_task = asyncio.ensure_future(app.updater.start_polling()) # type: ignore[attr-defined]
|
|
||||||
|
|
||||||
await stop
|
await stop
|
||||||
logger.info("Signal received — initiating graceful shutdown...")
|
logger.info("Signal received — initiating graceful shutdown...")
|
||||||
|
|
||||||
finally:
|
finally:
|
||||||
# ── Cancel scheduler with grace period ───────────────────────
|
# ── Cancel scheduler with grace period ───────────────────────
|
||||||
|
if scheduler is not None:
|
||||||
logger.info("Cancelling scheduler task...")
|
logger.info("Cancelling scheduler task...")
|
||||||
scheduler.cancel()
|
scheduler.cancel()
|
||||||
with suppress(asyncio.CancelledError):
|
with suppress(asyncio.CancelledError):
|
||||||
@@ -437,6 +443,7 @@ async def main() -> None:
|
|||||||
logger.warning("Scheduler task did not finish within 5s — force cancelled.")
|
logger.warning("Scheduler task did not finish within 5s — force cancelled.")
|
||||||
|
|
||||||
# ── Stop Telegram polling ────────────────────────────────────
|
# ── Stop Telegram polling ────────────────────────────────────
|
||||||
|
if bot_started and poll_task is not None:
|
||||||
set_telegram_polling(False)
|
set_telegram_polling(False)
|
||||||
logger.info("Stopping Telegram poller...")
|
logger.info("Stopping Telegram poller...")
|
||||||
poll_task.cancel()
|
poll_task.cancel()
|
||||||
@@ -444,6 +451,7 @@ async def main() -> None:
|
|||||||
await poll_task
|
await poll_task
|
||||||
|
|
||||||
# ── Shutdown application ─────────────────────────────────────
|
# ── Shutdown application ─────────────────────────────────────
|
||||||
|
if bot_started:
|
||||||
logger.info("Shutting down Telegram bot application...")
|
logger.info("Shutting down Telegram bot application...")
|
||||||
await app.shutdown()
|
await app.shutdown()
|
||||||
|
|
||||||
|
|||||||
+79
-3
@@ -3,13 +3,81 @@ import logging
|
|||||||
import os
|
import os
|
||||||
from datetime import datetime, timezone
|
from datetime import datetime, timezone
|
||||||
from typing import Any
|
from typing import Any
|
||||||
from urllib.parse import quote_plus
|
from urllib.parse import quote
|
||||||
|
|
||||||
import httpx
|
import httpx
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
_client: httpx.AsyncClient | None = None
|
_client: httpx.AsyncClient | None = None
|
||||||
|
_proxy_ip_logged = False
|
||||||
|
|
||||||
|
|
||||||
|
def _get_proxy_url() -> str | None:
|
||||||
|
raw = (os.getenv("HTTPS_PROXY") or os.getenv("https_proxy") or "").strip()
|
||||||
|
if not raw:
|
||||||
|
return None
|
||||||
|
|
||||||
|
if "://" in raw:
|
||||||
|
return raw
|
||||||
|
|
||||||
|
parts = raw.split(":", 3)
|
||||||
|
if len(parts) != 4:
|
||||||
|
logger.warning(
|
||||||
|
"Ignoring invalid HTTPS_PROXY value (expected host:port:user:pass or full URL)"
|
||||||
|
)
|
||||||
|
return None
|
||||||
|
|
||||||
|
host, port, username, password = parts
|
||||||
|
return f"http://{quote(username, safe='')}:{quote(password, safe='')}@{host}:{port}"
|
||||||
|
|
||||||
|
|
||||||
|
def _redact_proxy_url(proxy_url: str) -> str:
|
||||||
|
try:
|
||||||
|
parsed = httpx.URL(proxy_url)
|
||||||
|
host = parsed.host or "unknown-host"
|
||||||
|
port = parsed.port or 80
|
||||||
|
user = parsed.username or "unknown-user"
|
||||||
|
return f"{host}:{port} (user={user}, credentials=set)"
|
||||||
|
except Exception:
|
||||||
|
return "<unparseable proxy>"
|
||||||
|
|
||||||
|
|
||||||
|
async def _fetch_public_ip(client: httpx.AsyncClient) -> str | None:
|
||||||
|
try:
|
||||||
|
resp = await client.get("https://api.ipify.org?format=json")
|
||||||
|
resp.raise_for_status()
|
||||||
|
data = resp.json()
|
||||||
|
ip = data.get("ip")
|
||||||
|
return str(ip) if ip else None
|
||||||
|
except Exception as exc:
|
||||||
|
logger.warning("Could not resolve public IP: %s", exc)
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
async def _log_proxy_ip_comparison(proxy_url: str) -> None:
|
||||||
|
global _proxy_ip_logged
|
||||||
|
|
||||||
|
if _proxy_ip_logged:
|
||||||
|
return
|
||||||
|
|
||||||
|
_proxy_ip_logged = True
|
||||||
|
|
||||||
|
direct_client = httpx.AsyncClient(timeout=10.0, trust_env=False)
|
||||||
|
proxy_client = httpx.AsyncClient(timeout=10.0, trust_env=False, proxy=proxy_url)
|
||||||
|
|
||||||
|
try:
|
||||||
|
direct_ip = await _fetch_public_ip(direct_client)
|
||||||
|
proxy_ip = await _fetch_public_ip(proxy_client)
|
||||||
|
logger.info(
|
||||||
|
"HTTPS proxy enabled: %s | public_ip_without_proxy=%s | public_ip_with_proxy=%s",
|
||||||
|
_redact_proxy_url(proxy_url),
|
||||||
|
direct_ip or "unknown",
|
||||||
|
proxy_ip or "unknown",
|
||||||
|
)
|
||||||
|
finally:
|
||||||
|
await direct_client.aclose()
|
||||||
|
await proxy_client.aclose()
|
||||||
|
|
||||||
|
|
||||||
async def get_client() -> httpx.AsyncClient:
|
async def get_client() -> httpx.AsyncClient:
|
||||||
@@ -19,6 +87,10 @@ async def get_client() -> httpx.AsyncClient:
|
|||||||
if _client is None or _client.is_closed:
|
if _client is None or _client.is_closed:
|
||||||
max_conns = int(os.getenv("HTTP_MAX_CONNECTIONS", "10"))
|
max_conns = int(os.getenv("HTTP_MAX_CONNECTIONS", "10"))
|
||||||
max_keepalive = int(os.getenv("HTTP_KEEPALIVE_CONNECTIONS", "5"))
|
max_keepalive = int(os.getenv("HTTP_KEEPALIVE_CONNECTIONS", "5"))
|
||||||
|
proxy_url = _get_proxy_url()
|
||||||
|
|
||||||
|
if proxy_url:
|
||||||
|
await _log_proxy_ip_comparison(proxy_url)
|
||||||
|
|
||||||
_client = httpx.AsyncClient(
|
_client = httpx.AsyncClient(
|
||||||
timeout=float(os.getenv("HTTP_TIMEOUT_S", "30.0")),
|
timeout=float(os.getenv("HTTP_TIMEOUT_S", "30.0")),
|
||||||
@@ -27,10 +99,14 @@ async def get_client() -> httpx.AsyncClient:
|
|||||||
max_keepalive_connections=max_keepalive,
|
max_keepalive_connections=max_keepalive,
|
||||||
keepalive_expiry=60,
|
keepalive_expiry=60,
|
||||||
),
|
),
|
||||||
|
proxy=proxy_url,
|
||||||
|
trust_env=False,
|
||||||
)
|
)
|
||||||
logger.info(
|
logger.info(
|
||||||
"Created httpx client: max_conns=%d, keepalive=%d",
|
"Created httpx client: max_conns=%d, keepalive=%d, proxy=%s",
|
||||||
max_conns, max_keepalive,
|
max_conns,
|
||||||
|
max_keepalive,
|
||||||
|
"enabled" if proxy_url else "disabled",
|
||||||
)
|
)
|
||||||
|
|
||||||
return _client
|
return _client
|
||||||
|
|||||||
@@ -38,6 +38,15 @@
|
|||||||
{{ data.last_scheduler.strftime('%Y-%m-%d %H:%M:%S') if data.last_scheduler else 'Never' }}
|
{{ data.last_scheduler.strftime('%Y-%m-%d %H:%M:%S') if data.last_scheduler else 'Never' }}
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
<div class="card">
|
||||||
|
<div style="font-size: 12px; color: #a0a0b0; text-transform: uppercase;">HTTP Proxy</div>
|
||||||
|
<div style="font-size: 32px; font-weight: 700; margin-top: 8px; color: {{ '#0f9b58' if data.proxy_enabled else '#e94560' }};">
|
||||||
|
{{ 'Enabled' if data.proxy_enabled else 'Disabled' }}
|
||||||
|
</div>
|
||||||
|
<div style="font-size: 12px; color: #a0a0b0; margin-top: 4px;">
|
||||||
|
{{ 'Outbound scraping uses the proxy' if data.proxy_enabled else 'Outbound scraping goes direct' }}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
</div>
|
</div>
|
||||||
{% endif %}
|
{% endif %}
|
||||||
{% endblock %}
|
{% endblock %}
|
||||||
@@ -74,6 +74,10 @@ def format_postcodes(postcodes: list | None) -> str:
|
|||||||
return ", ".join(str(p) for p in postcodes)
|
return ", ".join(str(p) for p in postcodes)
|
||||||
|
|
||||||
|
|
||||||
|
def _proxy_enabled() -> bool:
|
||||||
|
return bool((os.getenv("HTTPS_PROXY") or os.getenv("https_proxy") or "").strip())
|
||||||
|
|
||||||
|
|
||||||
@asynccontextmanager
|
@asynccontextmanager
|
||||||
async def lifespan(app: FastAPI):
|
async def lifespan(app: FastAPI):
|
||||||
logger.info("Web UI starting up")
|
logger.info("Web UI starting up")
|
||||||
@@ -115,6 +119,7 @@ async def dashboard(request: Request):
|
|||||||
"queue_pending": queue_pending or 0,
|
"queue_pending": queue_pending or 0,
|
||||||
"queue_dead": queue_dead or 0,
|
"queue_dead": queue_dead or 0,
|
||||||
"last_scheduler": last_scheduler["scraped_at"] if last_scheduler else None,
|
"last_scheduler": last_scheduler["scraped_at"] if last_scheduler else None,
|
||||||
|
"proxy_enabled": _proxy_enabled(),
|
||||||
}
|
}
|
||||||
return templates.TemplateResponse("dashboard.html", {"request": request, "error": None, "data": data})
|
return templates.TemplateResponse("dashboard.html", {"request": request, "error": None, "data": data})
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user