diff --git a/x-pack/platform/plugins/shared/fleet/common/constants/epm.ts b/x-pack/platform/plugins/shared/fleet/common/constants/epm.ts index 1a8790b869c70..b69c56033b0f2 100644 --- a/x-pack/platform/plugins/shared/fleet/common/constants/epm.ts +++ b/x-pack/platform/plugins/shared/fleet/common/constants/epm.ts @@ -22,8 +22,10 @@ export const FLEET_KUBERNETES_PACKAGE = 'kubernetes'; export const FLEET_UNIVERSAL_PROFILING_SYMBOLIZER_PACKAGE = 'profiler_symbolizer'; export const FLEET_UNIVERSAL_PROFILING_COLLECTOR_PACKAGE = 'profiler_collector'; export const FLEET_CLOUD_SECURITY_POSTURE_PACKAGE = 'cloud_security_posture'; +export const FLEET_CLOUD_SECURITY_ASSET_PACKAGE = 'cloud_asset_inventory'; export const FLEET_CLOUD_SECURITY_POSTURE_KSPM_POLICY_TEMPLATE = 'kspm'; export const FLEET_CLOUD_SECURITY_POSTURE_CSPM_POLICY_TEMPLATE = 'cspm'; +export const FLEET_CLOUD_SECURITY_POSTURE_ASSET_INVENTORY_POLICY_TEMPLATE = 'asset_inventory'; export const FLEET_CLOUD_SECURITY_POSTURE_CNVM_POLICY_TEMPLATE = 'vuln_mgmt'; export const FLEET_CLOUD_BEAT_PACKAGE = 'cloudbeat'; export const FLEET_CONNECTORS_PACKAGE = 'elastic_connectors'; diff --git a/x-pack/platform/plugins/shared/fleet/common/index.ts b/x-pack/platform/plugins/shared/fleet/common/index.ts index 0983e76aea792..1860a983b69bf 100644 --- a/x-pack/platform/plugins/shared/fleet/common/index.ts +++ b/x-pack/platform/plugins/shared/fleet/common/index.ts @@ -17,8 +17,10 @@ export { FLEET_ELASTIC_AGENT_PACKAGE, FLEET_KUBERNETES_PACKAGE, FLEET_CLOUD_SECURITY_POSTURE_PACKAGE, + FLEET_CLOUD_SECURITY_ASSET_PACKAGE, FLEET_CLOUD_SECURITY_POSTURE_KSPM_POLICY_TEMPLATE, FLEET_CLOUD_SECURITY_POSTURE_CSPM_POLICY_TEMPLATE, + FLEET_CLOUD_SECURITY_POSTURE_ASSET_INVENTORY_POLICY_TEMPLATE, FLEET_CLOUD_SECURITY_POSTURE_CNVM_POLICY_TEMPLATE, FLEET_ENDPOINT_PACKAGE, SEARCH_AI_LAKE_PACKAGES, diff --git a/x-pack/platform/plugins/shared/fleet/public/components/agent_enrollment_flyout/hooks.tsx b/x-pack/platform/plugins/shared/fleet/public/components/agent_enrollment_flyout/hooks.tsx index 440b520a69ca7..edfdd97948748 100644 --- a/x-pack/platform/plugins/shared/fleet/public/components/agent_enrollment_flyout/hooks.tsx +++ b/x-pack/platform/plugins/shared/fleet/public/components/agent_enrollment_flyout/hooks.tsx @@ -18,7 +18,11 @@ import { useGetPackageInfoByKeyQuery, useStartServices, } from '../../hooks'; -import { FLEET_KUBERNETES_PACKAGE, FLEET_CLOUD_SECURITY_POSTURE_PACKAGE } from '../../../common'; +import { + FLEET_KUBERNETES_PACKAGE, + FLEET_CLOUD_SECURITY_POSTURE_PACKAGE, + FLEET_CLOUD_SECURITY_ASSET_PACKAGE, +} from '../../../common'; import { getTemplateUrlFromAgentPolicy, @@ -103,11 +107,15 @@ export function useCloudSecurityIntegration(agentPolicy?: AgentPolicy) { }, [agentPolicy]); const integrationVersion = cloudSecurityPackagePolicy?.package?.version; + const packageName = + cloudSecurityPackagePolicy?.package?.name === FLEET_CLOUD_SECURITY_POSTURE_PACKAGE + ? FLEET_CLOUD_SECURITY_POSTURE_PACKAGE + : FLEET_CLOUD_SECURITY_ASSET_PACKAGE; // Fetch the package info to get the CloudFormation template URL only // if the package policy is a Cloud Security policy const { data: packageInfoData, isLoading } = useGetPackageInfoByKeyQuery( - FLEET_CLOUD_SECURITY_POSTURE_PACKAGE, + packageName, integrationVersion, { full: true }, { enabled: Boolean(cloudSecurityPackagePolicy) } @@ -198,7 +206,9 @@ const getCloudSecurityPackagePolicyFromAgentPolicy = ( agentPolicy?: AgentPolicy ): PackagePolicy | undefined => { return agentPolicy?.package_policies?.find( - (input) => input.package?.name === FLEET_CLOUD_SECURITY_POSTURE_PACKAGE + (input) => + input.package?.name === FLEET_CLOUD_SECURITY_POSTURE_PACKAGE || + input.package?.name === FLEET_CLOUD_SECURITY_ASSET_PACKAGE ); };