Agent Zero
66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
"""Entity Links plugin — link files to companies/contacts, reverse links, event cleanup."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from app.plugins.base import BasePlugin
|
|
from app.plugins.manifest import PluginManifest, PluginRouteDef
|
|
|
|
|
|
class EntityLinksPlugin(BasePlugin):
|
|
"""Entity Links plugin for N:M file-to-entity links with event-driven cleanup."""
|
|
|
|
manifest = PluginManifest(
|
|
name="entity_links",
|
|
version="1.0.0",
|
|
display_name="Entity Links",
|
|
description="Link files to companies/contacts (N:M), reverse links, event cleanup on deletion.",
|
|
dependencies=[],
|
|
routes=[
|
|
PluginRouteDef(
|
|
path="/api/v1/dms",
|
|
module="app.plugins.builtins.entity_links.routes",
|
|
router_attr="router",
|
|
),
|
|
PluginRouteDef(
|
|
path="/api/v1/contacts",
|
|
module="app.plugins.builtins.entity_links.routes",
|
|
router_attr="contact_router",
|
|
),
|
|
],
|
|
events=["contact.deleted"],
|
|
migrations=["0001_initial.sql"],
|
|
permissions=[
|
|
"entity_links:read",
|
|
"entity_links:write",
|
|
"entity_links:delete",
|
|
],
|
|
is_core=True,
|
|
)
|
|
|
|
async def on_contact_deleted(self, payload: dict[str, Any]) -> None:
|
|
"""Handle contact.deleted event — remove all EntityLink rows for that contact."""
|
|
from sqlalchemy import delete
|
|
|
|
from app.core.db import get_session_factory
|
|
from app.plugins.builtins.entity_links.models import EntityLink
|
|
|
|
entity_id = payload.get("entity_id") or payload.get("contact_id")
|
|
tenant_id = payload.get("tenant_id")
|
|
if entity_id is None or tenant_id is None:
|
|
return
|
|
|
|
import uuid as _uuid
|
|
|
|
factory = get_session_factory()
|
|
async with factory() as session:
|
|
await session.execute(
|
|
delete(EntityLink).where(
|
|
EntityLink.tenant_id == _uuid.UUID(tenant_id),
|
|
EntityLink.entity_type == "contact",
|
|
EntityLink.entity_id == _uuid.UUID(str(entity_id)),
|
|
)
|
|
)
|
|
await session.commit()
|