mirror of
https://github.com/FunnyWolf/agentic-soc-platform.git
synced 2026-08-22 13:12:56 +02:00
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
|
from integrations.cmdb.models import CMDBQueryOutput
|
||
|
|
from integrations.cmdb.providers import get_providers
|
||
|
|
|
||
|
|
|
||
|
|
def list_providers():
|
||
|
|
return list(get_providers().keys())
|
||
|
|
|
||
|
|
|
||
|
|
def lookup_artifact_context(artifact_type, artifact_value, provider=None):
|
||
|
|
artifact_type = artifact_type.value if hasattr(artifact_type, "value") else str(artifact_type)
|
||
|
|
artifact_value = str(artifact_value or "").strip()
|
||
|
|
active_providers = get_providers()
|
||
|
|
if provider is not None:
|
||
|
|
if provider not in active_providers:
|
||
|
|
raise ValueError(f"Unknown CMDB provider: {provider}")
|
||
|
|
providers = {provider: active_providers[provider]}
|
||
|
|
else:
|
||
|
|
providers = active_providers
|
||
|
|
|
||
|
|
results = []
|
||
|
|
errors = []
|
||
|
|
for provider_name, provider_instance in providers.items():
|
||
|
|
try:
|
||
|
|
result = provider_instance.lookup(artifact_type, artifact_value)
|
||
|
|
results.append(result)
|
||
|
|
if result.error:
|
||
|
|
errors.append(f"[{provider_name}] {result.error}")
|
||
|
|
except Exception as exc:
|
||
|
|
errors.append(f"[{provider_name}] {type(exc).__name__}: {exc}")
|
||
|
|
|
||
|
|
return CMDBQueryOutput(
|
||
|
|
artifact_type=artifact_type,
|
||
|
|
artifact_value=artifact_value,
|
||
|
|
results=results,
|
||
|
|
errors=errors,
|
||
|
|
)
|