types.ts

/*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the Elastic License;
 * you may not use this file except in compliance with the Elastic License.
 */

import * as t from 'io-ts';
import { FrameworkUser } from '../../adapters/framework/adapter_types';

export const RuntimeAgentEventType = t.union([
  t.literal('STATE'),
  t.literal('ERROR'),
  t.literal('ACTION_RESULT'),
  t.literal('ACTION'),
]);

export const RuntimeAgentEventSubtype = t.union([
  // State
  t.literal('RUNNING'),
  t.literal('STARTING'),
  t.literal('IN_PROGRESS'),
  t.literal('CONFIG'),
  t.literal('FAILED'),
  t.literal('STOPPED'),
  // Action results
  t.literal('DATA_DUMP'),
  // Actions
  t.literal('ACKNOWLEDGED'),
  t.literal('UNKNOWN'),
]);

export const RuntimeAgentEvent = t.intersection(
  [
    t.interface({
      type: RuntimeAgentEventType,
      subtype: RuntimeAgentEventSubtype,
      timestamp: t.string,
      message: t.string,
    }),
    t.partial({
      payload: t.any,
      data: t.string,
      action_id: t.string,
      policy_id: t.string,
      stream_id: t.string,
    }),
  ],
  'AgentEvent'
);
export type AgentEvent = t.TypeOf<typeof RuntimeAgentEvent>;

export const RuntimeAgentEventSOAttributes = t.intersection(
  [
    t.interface({
      type: RuntimeAgentEventType,
      subtype: RuntimeAgentEventSubtype,
      timestamp: t.string,
      message: t.string,
      agent_id: t.string,
    }),
    t.partial({
      payload: t.string,
      data: t.string,
      action_id: t.string,
    }),
  ],
  'AgentEventSOAttribute'
);
export type AgentEventSOAttributes = t.TypeOf<typeof RuntimeAgentEventSOAttributes>;

export interface AgentEventsRepository {
  createEventsForAgent(user: FrameworkUser, agentId: string, events: AgentEvent[]): Promise<void>;
  getEventsForAgent(
    user: FrameworkUser,
    agentId: string,
    options?: {
      search?: string;
      page?: number;
      perPage?: number;
    }
  ): Promise<{ items: AgentEvent[]; total: number }>;
  list(
    user: FrameworkUser,
    options: {
      agentId?: string;
      search?: string;
      page?: number;
      perPage?: number;
    }
  ): Promise<{ items: AgentEvent[]; total: number }>;
  deleteEventsForAgent(user: FrameworkUser, agentId: string): Promise<void>;
}