Skip to content

Commit

Permalink
[Cloud Security] Refactoring the limit error message for agentless ag…
Browse files Browse the repository at this point in the history
  • Loading branch information
seanrathier authored and Samiul-TheSoccerFan committed Dec 10, 2024
1 parent 56863de commit 63ab491
Showing 1 changed file with 14 additions and 13 deletions.
27 changes: 14 additions & 13 deletions x-pack/plugins/fleet/server/services/agents/agentless_agent.ts
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,15 @@ import { listFleetServerHosts } from '../fleet_server_host';
import type { AgentlessConfig } from '../utils/agentless';
import { prependAgentlessApiBasePathToEndpoint, isAgentlessEnabled } from '../utils/agentless';

interface AgentlessAgentErrorHandlingMessages {
[key: string]: {
[key: string]: {
log: string;
message: string;
};
};
}

class AgentlessAgentService {
public async createAgentlessAgent(
esClient: ElasticsearchClient,
Expand Down Expand Up @@ -326,14 +335,12 @@ class AgentlessAgentService {
throw this.getAgentlessAgentError(action, error.message, traceId);
}

const ERROR_HANDLING_MESSAGES = this.getErrorHandlingMessages(agentlessPolicyId);
const ERROR_HANDLING_MESSAGES: AgentlessAgentErrorHandlingMessages =
this.getErrorHandlingMessages(agentlessPolicyId);

if (error.response) {
if (error.response.status in ERROR_HANDLING_MESSAGES) {
const handledResponseErrorMessage =
ERROR_HANDLING_MESSAGES[error.response.status as keyof typeof ERROR_HANDLING_MESSAGES][
action
];
const handledResponseErrorMessage = ERROR_HANDLING_MESSAGES[error.response.status][action];
this.handleResponseError(
action,
error.response,
Expand Down Expand Up @@ -426,7 +433,7 @@ class AgentlessAgentService {
: new AgentlessAgentDeleteError(this.withRequestIdMessage(userMessage, traceId));
}

private getErrorHandlingMessages(agentlessPolicyId: string) {
private getErrorHandlingMessages(agentlessPolicyId: string): AgentlessAgentErrorHandlingMessages {
return {
400: {
create: {
Expand Down Expand Up @@ -483,13 +490,7 @@ class AgentlessAgentService {
create: {
log: '[Agentless API] Creating the agentless agent failed with a status 429 for agentless policy, agentless agent limit has been reached for this deployment or project.',
message:
'the Agentless API could not create the agentless agent, you have reached the limit of agentless agents provisioned for this deployment or project. Consider removing some agentless agents and try again or use agent-based agents for this integration.',
},
// this is likely to happen when deleting agentless agents, but covering it in case
delete: {
log: '[Agentless API] Deleting the agentless deployment failed with a status 429 for agentless policy, agentless agent limit has been reached for this deployment or project.',
message:
'the Agentless API could not delete the agentless deployment, you have reached the limit of agentless agents provisioned for this deployment or project. Consider removing some agentless agents and try again or use agent-based agents for this integration.',
'you have reached the limit for agentless provisioning. Please remove some or switch to agent-based integration.',
},
},
500: {
Expand Down

0 comments on commit 63ab491

Please sign in to comment.