mirror of
https://github.com/yogeshojha/rengine.git
synced 2026-10-20 07:11:28 +02:00
Yogesh OjhaandClaude Opus 5
048abbf3f3
Naming an engine or a scan context with more than 200 characters reached Postgres and came back as a 500 — the column is bounded, the request model was not. A name of spaces was accepted outright by engines, contexts, schedules and projects, and a project created that way got an empty slug, which is the segment its own URL is built from. Proxies and notification channels already ran their names through clean_name; every other named entity now does the same, on update as well as on create, where none of them did. A rejected name says whether it was empty or too long instead of failing at the database. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01K6LcH8uUddjJbKrahf34ys
94 lines
2.5 KiB
Python
94 lines
2.5 KiB
Python
import uuid
|
|
from datetime import datetime
|
|
from functools import partial
|
|
|
|
from pydantic import BaseModel, field_validator
|
|
from sqlmodel import Field, SQLModel
|
|
|
|
from shared.enums.proxy import ProxyMode, ProxyProtocol
|
|
from shared.utils.datetime import utc_now
|
|
from shared.utils.validation import clean_name, clean_optional_name
|
|
|
|
PROXY_MODES = tuple(m.value for m in ProxyMode)
|
|
PROXY_SCHEMES = tuple(p.value for p in ProxyProtocol)
|
|
|
|
|
|
class ProxyEndpoint(BaseModel):
|
|
scheme: str
|
|
host: str
|
|
port: int
|
|
username: str | None = None
|
|
password: str | None = None
|
|
|
|
|
|
class Proxy(SQLModel, table=True):
|
|
__tablename__ = "proxies"
|
|
|
|
id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True, index=True)
|
|
name: str = Field(max_length=120)
|
|
description: str | None = Field(default=None, max_length=500)
|
|
mode: str = Field(default="single")
|
|
is_active: bool = Field(default=True)
|
|
is_default: bool = Field(default=False)
|
|
endpoints_encrypted: str
|
|
endpoint_count: int = Field(default=0)
|
|
created_by: uuid.UUID = Field(foreign_key="users.id")
|
|
created_at: datetime = Field(default_factory=utc_now)
|
|
updated_at: datetime = Field(default_factory=utc_now)
|
|
last_test_at: datetime | None = Field(default=None)
|
|
last_test_ok: bool | None = Field(default=None)
|
|
last_test_message: str | None = Field(default=None)
|
|
|
|
|
|
class ProxyEndpointRead(BaseModel):
|
|
scheme: str
|
|
host: str
|
|
port: int
|
|
username: str | None = None
|
|
has_password: bool
|
|
url_masked: str
|
|
|
|
|
|
class ProxyCreate(BaseModel):
|
|
name: str
|
|
description: str | None = None
|
|
mode: str = "single"
|
|
is_active: bool = True
|
|
is_default: bool = False
|
|
endpoints: list[ProxyEndpoint] = Field(min_length=1)
|
|
|
|
_validate_name = field_validator("name")(clean_name)
|
|
|
|
|
|
class ProxyUpdate(BaseModel):
|
|
name: str | None = None
|
|
description: str | None = None
|
|
mode: str | None = None
|
|
is_active: bool | None = None
|
|
is_default: bool | None = None
|
|
endpoints: list[ProxyEndpoint] | None = None
|
|
|
|
_validate_name = field_validator("name")(partial(clean_optional_name, max_len=120))
|
|
|
|
|
|
class ProxyRead(BaseModel):
|
|
id: uuid.UUID
|
|
name: str
|
|
description: str | None
|
|
mode: str
|
|
is_active: bool
|
|
is_default: bool
|
|
endpoints: list[ProxyEndpointRead]
|
|
endpoint_count: int
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
last_test_at: datetime | None
|
|
last_test_ok: bool | None
|
|
last_test_message: str | None
|
|
|
|
|
|
class ProxyTestResult(BaseModel):
|
|
success: bool
|
|
message: str
|
|
latency_ms: int | None = None
|