src/mocks/msw/mockers/alert.mock.ts

import { Alert, AlertSeverity } from '@/api/generated'
import { faker } from '@faker-js/faker'

const ALERT_SECRET_FIELDS = {
  trigger_string: 'foo',
  trigger_type: 'codegate-secrets',
} satisfies Pick<Alert, 'trigger_string' | 'trigger_type'>

const ALERT_PII_FIELDS = {
  trigger_string: 'fakemail@fakedomain.mock',
  trigger_type: 'codegate-pii',
} satisfies Pick<Alert, 'trigger_string' | 'trigger_type'>

const ALERT_MALICIOUS_FIELDS = {
  trigger_string: {
    name: 'invokehttp',
    type: 'pypi',
    status: 'malicious',
    description: 'Python HTTP for Humans.',
  },
  trigger_type: 'codegate-context-retriever',
} satisfies Pick<Alert, 'trigger_string' | 'trigger_type'>

const getBaseAlert = ({
  timestamp,
}: {
  timestamp: string
}): Omit<Alert, 'trigger_type' | 'trigger_string'> => ({
  id: faker.string.uuid(),
  prompt_id: faker.string.uuid(),
  code_snippet: null,
  trigger_category: AlertSeverity.CRITICAL,
  timestamp: timestamp,
})

export const mockAlert = ({
  type,
}: {
  type: 'secret' | 'malicious' | 'pii'
}): Alert => {
  const timestamp = faker.date.recent().toISOString()

  const base: Omit<Alert, 'trigger_type' | 'trigger_string'> = getBaseAlert({
    timestamp,
  })

  switch (type) {
    case 'malicious': {
      const result: Alert = {
        ...base,
        ...ALERT_MALICIOUS_FIELDS,
      }

      return result
    }
    case 'secret': {
      const result: Alert = {
        ...base,
        ...ALERT_SECRET_FIELDS,
      }

      return result
    }
    case 'pii': {
      const result: Alert = {
        ...base,
        ...ALERT_PII_FIELDS,
      }

      return result
    }
  }
}