mssql_database_user.py

"""MSSQLDatabaseUser: one node_mssql_database_user row -> one MSSQL_DatabaseUser node."""
import logging

from openhound.core.asset import BaseAsset
from pydantic import ConfigDict, Field

from ..graph import MSSQLDatabaseUserProperties, SCCMNode, domain_environment_id
from ..kinds import nodes as nk

logger = logging.getLogger(__name__)


class MSSQLDatabaseUser(BaseAsset):
    model_config = ConfigDict(populate_by_name=True, extra="ignore")

    dbuser_id: str | None = None
    dbuser_name: str | None = None
    login_id: str | None = None
    login_name: str | None = None
    database_id: str | None = None
    database: str | None = None
    server_id: str | None = None
    host_sid: str | None = None
    port: str | None = None
    sql_server: str | None = None
    sccm_site: str | None = None
    collection_source: list[str] = Field(default_factory=list)
    assumed: bool = False
    assumption_basis: str | None = None

    @property
    def as_node(self) -> SCCMNode | None:
        if not self.dbuser_id:
            logger.warning("MSSQLDatabaseUser: dropping row with no dbuser_id")
            return None
        display = self.dbuser_name or self.dbuser_id
        roles = [f"db_owner@{self.database_id}"] if self.database_id else []
        return SCCMNode(
            id=self.dbuser_id,
            kinds=[nk.MSSQL_DATABASE_USER],
            properties=MSSQLDatabaseUserProperties(
                name=display, displayname=display,
                environmentid=domain_environment_id(self.host_sid or ""),
                collectionSource=list(self.collection_source),
                database=self.database, login=self.login_name, memberOfRoles=roles,
                SCCMInfra=True, SCCMSite=self.sccm_site, SQLServer=self.sql_server,
                assumed=self.assumed or None, assumptionBasis=self.assumption_basis,
            ),
        )

    @property
    def edges(self):
        return iter(())