diff --git a/x-pack/plugins/ingest_manager/server/plugin.ts b/x-pack/plugins/ingest_manager/server/plugin.ts index fcdb6387fed3a..1ae9528f34410 100644 --- a/x-pack/plugins/ingest_manager/server/plugin.ts +++ b/x-pack/plugins/ingest_manager/server/plugin.ts @@ -54,7 +54,12 @@ import { AgentService, datasourceService, } from './services'; -import { getAgentStatusById, authenticateAgentWithAccessToken } from './services/agents'; +import { + getAgentStatusById, + authenticateAgentWithAccessToken, + listAgents, + getAgent, +} from './services/agents'; import { CloudSetup } from '../../cloud/server'; import { agentCheckinState } from './services/agents/checkin/state'; @@ -236,7 +241,7 @@ export class IngestManagerPlugin plugins: { encryptedSavedObjects: EncryptedSavedObjectsPluginStart; } - ) { + ): Promise { await appContextService.start({ encryptedSavedObjectsStart: plugins.encryptedSavedObjects, encryptedSavedObjectsSetup: this.encryptedSavedObjectsSetup, @@ -255,6 +260,8 @@ export class IngestManagerPlugin return { esIndexPatternService: new ESIndexPatternSavedObjectService(), agentService: { + getAgent, + listAgents, getAgentStatusById, authenticateAgentWithAccessToken, }, diff --git a/x-pack/plugins/ingest_manager/server/services/index.ts b/x-pack/plugins/ingest_manager/server/services/index.ts index 1a0fb262eeb7f..49896959f3c3a 100644 --- a/x-pack/plugins/ingest_manager/server/services/index.ts +++ b/x-pack/plugins/ingest_manager/server/services/index.ts @@ -4,8 +4,8 @@ * you may not use this file except in compliance with the Elastic License. */ -import { SavedObjectsClientContract } from 'kibana/server'; -import { AgentStatus } from '../types'; +import { SavedObjectsClientContract, KibanaRequest } from 'kibana/server'; +import { AgentStatus, Agent } from '../types'; import * as settingsService from './settings'; export { ESIndexPatternSavedObjectService } from './es_index_pattern'; @@ -24,12 +24,38 @@ export interface ESIndexPatternService { * A service that provides exported functions that return information about an Agent */ export interface AgentService { + /** + * Get an Agent by id + */ + getAgent(soClient: SavedObjectsClientContract, agentId: string): Promise; + /** + * Authenticate an agent with access toekn + */ + authenticateAgentWithAccessToken( + soClient: SavedObjectsClientContract, + request: KibanaRequest + ): Promise; /** * Return the status by the Agent's id - * @param soClient - * @param agentId */ getAgentStatusById(soClient: SavedObjectsClientContract, agentId: string): Promise; + /** + * List agents + */ + listAgents( + soClient: SavedObjectsClientContract, + options: { + page: number; + perPage: number; + kuery?: string; + showInactive: boolean; + } + ): Promise<{ + agents: Agent[]; + total: number; + page: number; + perPage: number; + }>; } // Saved object services diff --git a/x-pack/plugins/security_solution/server/endpoint/mocks.ts b/x-pack/plugins/security_solution/server/endpoint/mocks.ts index 5435eff4ef150..aca8a4e0ce78f 100644 --- a/x-pack/plugins/security_solution/server/endpoint/mocks.ts +++ b/x-pack/plugins/security_solution/server/endpoint/mocks.ts @@ -35,6 +35,9 @@ export const createMockEndpointAppContextServiceStartContract = (): jest.Mocked< export const createMockAgentService = (): jest.Mocked => { return { getAgentStatusById: jest.fn(), + authenticateAgentWithAccessToken: jest.fn(), + getAgent: jest.fn(), + listAgents: jest.fn(), }; };