forked from FOSS/Steam-Panel
v 3.0.3
This commit is contained in:
Manchik
@@ -0,0 +1,356 @@
|
||||
"""CRUD and validate endpoints for log:pass accounts."""
|
||||
|
||||
import csv
|
||||
import ctypes
|
||||
import io
|
||||
import sys
|
||||
|
||||
csv.field_size_limit(min(sys.maxsize, ctypes.c_ulong(-1).value // 2))
|
||||
|
||||
from fastapi import APIRouter, HTTPException
|
||||
from loguru import logger
|
||||
|
||||
from app.database import get_db
|
||||
from app.models import (
|
||||
LogpassAccountCreate,
|
||||
LogpassAccountOut,
|
||||
LogpassAccountUpdate,
|
||||
)
|
||||
from app.core.task_manager import task_manager
|
||||
|
||||
router = APIRouter(prefix="/api/logpass", tags=["logpass"])
|
||||
|
||||
|
||||
@router.get("", response_model=list[LogpassAccountOut])
|
||||
async def list_logpass():
|
||||
db = await get_db()
|
||||
cursor = await db.execute("SELECT * FROM logpass_accounts ORDER BY id DESC")
|
||||
rows = await cursor.fetchall()
|
||||
return [dict(r) for r in rows]
|
||||
|
||||
|
||||
@router.get("/{account_id}", response_model=LogpassAccountOut)
|
||||
async def get_logpass(account_id: int):
|
||||
db = await get_db()
|
||||
cursor = await db.execute("SELECT * FROM logpass_accounts WHERE id = ?", (account_id,))
|
||||
row = await cursor.fetchone()
|
||||
if not row:
|
||||
raise HTTPException(status_code=404, detail="Account not found")
|
||||
return dict(row)
|
||||
|
||||
|
||||
@router.post("", response_model=LogpassAccountOut, status_code=201)
|
||||
async def create_logpass(account: LogpassAccountCreate):
|
||||
db = await get_db()
|
||||
cursor = await db.execute(
|
||||
"INSERT INTO logpass_accounts (login, password, steam_id, proxy, notes) VALUES (?, ?, ?, ?, ?)",
|
||||
(account.login, account.password, account.steam_id, account.proxy, account.notes),
|
||||
)
|
||||
await db.commit()
|
||||
new_cursor = await db.execute(
|
||||
"SELECT * FROM logpass_accounts WHERE id = ?", (cursor.lastrowid,)
|
||||
)
|
||||
return dict(await new_cursor.fetchone())
|
||||
|
||||
|
||||
@router.put("/{account_id}", response_model=LogpassAccountOut)
|
||||
async def update_logpass(account_id: int, account: LogpassAccountUpdate):
|
||||
db = await get_db()
|
||||
fields = {k: v for k, v in account.model_dump().items() if v is not None}
|
||||
if not fields:
|
||||
raise HTTPException(status_code=400, detail="No fields to update")
|
||||
fields["updated_at"] = "datetime('now')"
|
||||
set_clause = ", ".join(
|
||||
f"{k} = datetime('now')" if k == "updated_at" else f"{k} = ?"
|
||||
for k in fields
|
||||
)
|
||||
values = [v for k, v in fields.items() if k != "updated_at"]
|
||||
values.append(account_id)
|
||||
await db.execute(
|
||||
f"UPDATE logpass_accounts SET {set_clause} WHERE id = ?", values
|
||||
)
|
||||
await db.commit()
|
||||
cursor = await db.execute("SELECT * FROM logpass_accounts WHERE id = ?", (account_id,))
|
||||
row = await cursor.fetchone()
|
||||
if not row:
|
||||
raise HTTPException(status_code=404, detail="Account not found")
|
||||
return dict(row)
|
||||
|
||||
|
||||
@router.delete("/{account_id}", status_code=204)
|
||||
async def delete_logpass(account_id: int):
|
||||
db = await get_db()
|
||||
await db.execute("DELETE FROM logpass_accounts WHERE id = ?", (account_id,))
|
||||
await db.commit()
|
||||
|
||||
|
||||
@router.post("/delete-bulk")
|
||||
async def delete_logpass_bulk(data: dict):
|
||||
ids: list[int] = data.get("ids", [])
|
||||
if not ids:
|
||||
return {"deleted": 0}
|
||||
db = await get_db()
|
||||
placeholders = ",".join("?" for _ in ids)
|
||||
await db.execute(f"DELETE FROM logpass_accounts WHERE id IN ({placeholders})", ids)
|
||||
await db.commit()
|
||||
return {"deleted": len(ids)}
|
||||
|
||||
|
||||
@router.post("/import")
|
||||
async def import_logpass(data: dict):
|
||||
"""Bulk import from plain text (login:pass / login|pass) or CSV with headers."""
|
||||
lines: list[str] = data.get("lines", [])
|
||||
if not lines:
|
||||
return {"imported": 0, "skipped": 0, "errors": []}
|
||||
|
||||
imported = 0
|
||||
skipped = 0
|
||||
errors: list[str] = []
|
||||
|
||||
db = await get_db()
|
||||
|
||||
async def upsert(login: str, fields: dict):
|
||||
"""Insert or update by login, works with or without UNIQUE index."""
|
||||
cursor = await db.execute(
|
||||
"SELECT id FROM logpass_accounts WHERE login = ?", (login,)
|
||||
)
|
||||
existing = await cursor.fetchone()
|
||||
if existing:
|
||||
sets = ", ".join(f"{k} = ?" for k in fields)
|
||||
vals = list(fields.values()) + [existing["id"]]
|
||||
await db.execute(
|
||||
f"UPDATE logpass_accounts SET {sets}, updated_at = datetime('now') WHERE id = ?",
|
||||
vals,
|
||||
)
|
||||
else:
|
||||
fields["login"] = login
|
||||
cols = ", ".join(fields.keys())
|
||||
placeholders = ", ".join("?" for _ in fields)
|
||||
await db.execute(
|
||||
f"INSERT INTO logpass_accounts ({cols}) VALUES ({placeholders})",
|
||||
list(fields.values()),
|
||||
)
|
||||
|
||||
# Detect CSV: first line looks like a header row
|
||||
first = lines[0].strip().lower()
|
||||
is_csv = first.startswith("login,") or first.startswith("login;")
|
||||
|
||||
try:
|
||||
if is_csv:
|
||||
text = "\n".join(lines)
|
||||
reader = csv.DictReader(io.StringIO(text))
|
||||
for row in reader:
|
||||
login = (row.get("login") or "").strip()
|
||||
password = (row.get("password") or "").strip()
|
||||
if not login or not password:
|
||||
skipped += 1
|
||||
continue
|
||||
fields = {"password": password}
|
||||
_NA = {"n/a", "na", "none", "null", ""}
|
||||
for csv_col, db_col in [
|
||||
("steam_id", "steam_id"), ("ban", "ban_status"),
|
||||
("prime", "prime"), ("trophy", "trophy"),
|
||||
("behavior", "behavior"), ("license", "license"),
|
||||
]:
|
||||
val = (row.get(csv_col) or "").strip()
|
||||
if val:
|
||||
if db_col in ("prime", "trophy", "behavior") and val.lower() in _NA:
|
||||
val = "\u2014"
|
||||
fields[db_col] = val
|
||||
try:
|
||||
await upsert(login, fields)
|
||||
imported += 1
|
||||
except Exception as exc:
|
||||
logger.error(f"Logpass CSV import error for '{login}': {exc}")
|
||||
errors.append(f"{login}: {exc}")
|
||||
else:
|
||||
for raw in lines:
|
||||
line = raw.strip()
|
||||
if not line:
|
||||
skipped += 1
|
||||
continue
|
||||
# Skip JSON/mafile lines
|
||||
if line.startswith("{") or line.startswith("["):
|
||||
skipped += 1
|
||||
continue
|
||||
if "|" in line:
|
||||
parts = line.split("|", 1)
|
||||
elif ":" in line:
|
||||
parts = line.split(":", 1)
|
||||
else:
|
||||
skipped += 1
|
||||
continue
|
||||
login, password = parts[0].strip(), parts[1].strip()
|
||||
if not login or not password:
|
||||
skipped += 1
|
||||
continue
|
||||
# Reject if password looks like JSON (mafile embedded) or is too long
|
||||
if password.startswith("{") or password.startswith("[") or len(password) > 128:
|
||||
skipped += 1
|
||||
logger.warning(f"Logpass import: skipped '{login}' — password looks like mafile or is too long ({len(password)} chars)")
|
||||
continue
|
||||
try:
|
||||
await upsert(login, {"password": password})
|
||||
imported += 1
|
||||
except Exception as exc:
|
||||
logger.error(f"Logpass import error for '{login}': {exc}")
|
||||
errors.append(f"{login}: {exc}")
|
||||
|
||||
await db.commit()
|
||||
except Exception as exc:
|
||||
logger.exception(f"Logpass import failed: {exc}")
|
||||
raise HTTPException(status_code=500, detail=str(exc))
|
||||
|
||||
return {"imported": imported, "skipped": skipped, "errors": errors}
|
||||
|
||||
|
||||
@router.post("/validate")
|
||||
async def validate_logpass(data: dict):
|
||||
"""Submit validation task for selected log:pass accounts."""
|
||||
ids: list[int] = data.get("account_ids", [])
|
||||
if not ids:
|
||||
raise HTTPException(status_code=400, detail="No account IDs provided")
|
||||
|
||||
db = await get_db()
|
||||
placeholders = ",".join("?" for _ in ids)
|
||||
cursor = await db.execute(
|
||||
f"SELECT * FROM logpass_accounts WHERE id IN ({placeholders})", ids
|
||||
)
|
||||
accounts = [dict(r) for r in await cursor.fetchall()]
|
||||
if not accounts:
|
||||
raise HTTPException(status_code=404, detail="No accounts found")
|
||||
|
||||
task_id = await task_manager.submit(
|
||||
task_type="logpass_validate",
|
||||
accounts=accounts,
|
||||
params={},
|
||||
)
|
||||
logger.info(f"logpass_validate submitted for {len(accounts)} accounts → task {task_id}")
|
||||
return {"task_id": task_id, "accounts_count": len(accounts)}
|
||||
|
||||
|
||||
@router.post("/full-parse")
|
||||
async def full_parse_logpass(data: dict):
|
||||
"""Submit full parse task (prime, trophy, behavior, licenses) for selected accounts."""
|
||||
ids: list[int] = data.get("account_ids", [])
|
||||
if not ids:
|
||||
raise HTTPException(status_code=400, detail="No account IDs provided")
|
||||
|
||||
db = await get_db()
|
||||
placeholders = ",".join("?" for _ in ids)
|
||||
cursor = await db.execute(
|
||||
f"SELECT * FROM logpass_accounts WHERE id IN ({placeholders})", ids
|
||||
)
|
||||
accounts = [dict(r) for r in await cursor.fetchall()]
|
||||
if not accounts:
|
||||
raise HTTPException(status_code=404, detail="No accounts found")
|
||||
|
||||
task_id = await task_manager.submit(
|
||||
task_type="logpass_full_parse",
|
||||
accounts=accounts,
|
||||
params={},
|
||||
)
|
||||
logger.info(f"logpass_full_parse submitted for {len(accounts)} accounts → task {task_id}")
|
||||
return {"task_id": task_id, "accounts_count": len(accounts)}
|
||||
|
||||
|
||||
@router.post("/assign-proxies")
|
||||
async def logpass_assign_proxies():
|
||||
"""Round-robin assign proxies to logpass accounts that don't have one."""
|
||||
db = await get_db()
|
||||
proxy_cursor = await db.execute("SELECT address FROM proxies ORDER BY id")
|
||||
proxies = [r["address"] for r in await proxy_cursor.fetchall()]
|
||||
if not proxies:
|
||||
raise HTTPException(status_code=400, detail="No proxies available")
|
||||
|
||||
acc_cursor = await db.execute(
|
||||
"SELECT id FROM logpass_accounts WHERE proxy IS NULL OR proxy = '' ORDER BY id"
|
||||
)
|
||||
account_ids = [r["id"] for r in await acc_cursor.fetchall()]
|
||||
if not account_ids:
|
||||
return {"assigned": 0, "proxies_used": 0}
|
||||
|
||||
for i, acc_id in enumerate(account_ids):
|
||||
await db.execute(
|
||||
"UPDATE logpass_accounts SET proxy = ?, updated_at = datetime('now') WHERE id = ?",
|
||||
(proxies[i % len(proxies)], acc_id),
|
||||
)
|
||||
await db.commit()
|
||||
return {"assigned": len(account_ids), "proxies_used": len(proxies)}
|
||||
|
||||
|
||||
@router.post("/reassign-proxies")
|
||||
async def logpass_reassign_proxies():
|
||||
"""Round-robin reassign proxies to ALL logpass accounts."""
|
||||
db = await get_db()
|
||||
proxy_cursor = await db.execute("SELECT address FROM proxies ORDER BY id")
|
||||
proxies = [r["address"] for r in await proxy_cursor.fetchall()]
|
||||
if not proxies:
|
||||
raise HTTPException(status_code=400, detail="No proxies available")
|
||||
|
||||
acc_cursor = await db.execute("SELECT id FROM logpass_accounts ORDER BY id")
|
||||
account_ids = [r["id"] for r in await acc_cursor.fetchall()]
|
||||
if not account_ids:
|
||||
return {"assigned": 0, "proxies_used": 0}
|
||||
|
||||
for i, acc_id in enumerate(account_ids):
|
||||
await db.execute(
|
||||
"UPDATE logpass_accounts SET proxy = ?, updated_at = datetime('now') WHERE id = ?",
|
||||
(proxies[i % len(proxies)], acc_id),
|
||||
)
|
||||
await db.commit()
|
||||
return {"assigned": len(account_ids), "proxies_used": len(proxies)}
|
||||
|
||||
|
||||
@router.post("/clear-proxies")
|
||||
async def logpass_clear_proxies():
|
||||
"""Remove proxy from all logpass accounts."""
|
||||
db = await get_db()
|
||||
cursor = await db.execute(
|
||||
"UPDATE logpass_accounts SET proxy = NULL, updated_at = datetime('now') WHERE proxy IS NOT NULL AND proxy != ''"
|
||||
)
|
||||
await db.commit()
|
||||
return {"cleared": cursor.rowcount}
|
||||
|
||||
|
||||
@router.post("/{account_id}/browser")
|
||||
async def open_logpass_browser(account_id: int):
|
||||
"""Open Chrome browser with saved session cookies for a logpass account."""
|
||||
import asyncio
|
||||
db = await get_db()
|
||||
cursor = await db.execute("SELECT * FROM logpass_accounts WHERE id = ?", (account_id,))
|
||||
row = await cursor.fetchone()
|
||||
if not row:
|
||||
raise HTTPException(status_code=404, detail="Account not found")
|
||||
|
||||
account = dict(row)
|
||||
if not account.get("session_cookies"):
|
||||
raise HTTPException(status_code=400, detail="No session cookies. Validate the account first.")
|
||||
|
||||
from app.services.steam_auth import check_cookies_alive, _resolve_proxy
|
||||
from app.config import read_validation_settings
|
||||
|
||||
proxy = await _resolve_proxy(account)
|
||||
alive = await check_cookies_alive(account["session_cookies"], proxy)
|
||||
if not alive:
|
||||
val_settings = read_validation_settings()
|
||||
if val_settings.get("auto_revalidate_browser"):
|
||||
task_id = await task_manager.submit(
|
||||
task_type="logpass_validate",
|
||||
accounts=[account],
|
||||
params={},
|
||||
)
|
||||
logger.info(f"Cookies dead for {account['login']}, auto-revalidating → task {task_id}")
|
||||
return {"status": "revalidating", "message": "Cookies expired. Re-validating...", "task_id": task_id}
|
||||
raise HTTPException(status_code=400, detail="Session cookies expired. Re-validate the account.")
|
||||
|
||||
from app.services.browser_login import open_browser_with_cookies
|
||||
|
||||
async def _run():
|
||||
try:
|
||||
await open_browser_with_cookies(account)
|
||||
except Exception as exc:
|
||||
logger.error(f"Browser open failed for {account['login']}: {exc}")
|
||||
|
||||
asyncio.create_task(_run())
|
||||
return {"status": "ok", "message": "Browser opening..."}
|
||||
Reference in New Issue
Block a user