#!/usr/bin/env python3
# Copyright (C) 2024 Spearhead Systems SRL

def get_params(params):
    return params["tenant"], params["client"], params["secret"]

def agent_azure_keyvault(params, hostname, ipaddress):
    tenant, client, secret = get_params(params)
    return ["keyvault", tenant, client, secret]

def agent_azure_firewall(params, hostname, ipaddress):
    tenant, client, secret = get_params(params)
    return ["firewall", tenant, client, secret]

def agent_azure_defender(params, hostname, ipaddress):
    tenant, client, secret = get_params(params)
    return ["defender", tenant, client, secret]

special_agent_info["azure_keyvault"] = agent_azure_keyvault
special_agent_info["azure_firewall"] = agent_azure_firewall
special_agent_info["azure_defender"] = agent_azure_defender