refactor(aiguard): slim down sdk file (#9753)

Author: IlyasShabi

packages/dd-trace/src/aiguard/client.js

@@ -1,5 +1,27 @@
 'use strict'
 
+const tracerVersion = require('../../../../package.json').version
+const { AIGuardClientError } = require('./errors')
+const { parseEvaluationResponse } = require('./evaluation')
+const TAGS = require('./tags')
+
+/**
+ * Resolves the AI Guard host for a Datadog site.
+ *
+ * @param {string} site
+ * @returns {string}
+ */
+function aiGuardHost (site) {
+  return site.split('.').length === 2 ? `app.${site}` : site
+}
+
+/**
+ * Sends a request to the AI Guard service.
+ *
+ * @param {object} body
+ * @param {{ url: string, headers: Record<string, string>, timeout: number }} opts
+ * @returns {Promise<{ status: number, body: unknown }>}
+ */
 async function executeRequest (body, opts) {
   const postData = JSON.stringify(body)
   const headers = {
@@ -22,4 +44,81 @@ async function executeRequest (body, opts) {
   }
 }
 
-module.exports = executeRequest
+class AIGuardClient {
+  #headers
+  #evaluateUrl
+  #timeout
+
+  /**
+   * @param {import('../config/config-base')} config
+   */
+  constructor (config) {
+    this.#headers = {
+      'DD-API-KEY': config.DD_API_KEY,
+      'DD-APPLICATION-KEY': config.DD_APP_KEY,
+      'DD-AI-GUARD-VERSION': tracerVersion,
+      'DD-AI-GUARD-SOURCE': 'SDK',
+      'DD-AI-GUARD-LANGUAGE': 'nodejs',
+    }
+    const endpoint = config.experimental.aiguard.endpoint || `https://${aiGuardHost(config.site)}/api/v2/ai-guard`
+    this.#evaluateUrl = `${endpoint}/evaluate`
+    this.#timeout = config.experimental.aiguard.timeout
+  }
+
+  /**
+   * Evaluates messages and converts the service response to the internal evaluation contract.
+   *
+   * @param {import('../../../../index').aiguard.Message[]} messages
+   * @param {{ service: string, env: string }} meta
+   * @returns {Promise<NonNullable<ReturnType<typeof parseEvaluationResponse>>>}
+   */
+  evaluate (messages, meta) {
+    const payload = {
+      data: {
+        attributes: {
+          messages,
+          meta,
+        },
+      },
+    }
+    return executeRequest(
+      payload,
+      { url: this.#evaluateUrl, headers: this.#headers, timeout: this.#timeout }
+    )
+      .then(response => this.#parseResponse(response))
+      .catch(cause => {
+        if (cause instanceof AIGuardClientError) throw cause
+
+        throw new AIGuardClientError(`Unexpected error calling AI Guard service: ${cause.message}`, {
+          cause,
+          telemetryType: TAGS.ERROR_TYPE_CLIENT,
+        })
+      })
+  }
+
+  /**
+   * Validates an AI Guard HTTP response.
+   *
+   * @param {{ status: number, body: unknown }} response
+   * @returns {NonNullable<ReturnType<typeof parseEvaluationResponse>>}
+   */
+  #parseResponse (response) {
+    if (response.status !== 200) {
+      throw new AIGuardClientError(`AI Guard service call failed, status ${response.status}`, {
+        errors: response.body?.errors,
+        telemetryType: TAGS.ERROR_TYPE_STATUS,
+      })
+    }
+
+    const evaluation = parseEvaluationResponse(response.body)
+    if (!evaluation) {
+      throw new AIGuardClientError(`AI Guard service returned unexpected response : ${response.body}`, {
+        telemetryType: TAGS.ERROR_TYPE_RESPONSE,
+      })
+    }
+
+    return evaluation
+  }
+}
+
+module.exports = AIGuardClient

packages/dd-trace/src/aiguard/errors.js

@@ -0,0 +1,41 @@
+'use strict'
+
+class AIGuardAbortError extends Error {
+  /**
+   * @param {string|undefined} reason
+   * @param {Array<unknown>} tags
+   * @param {Record<string, unknown>} tagProbabilities
+   * @param {Array<unknown>} sds
+   */
+  constructor (reason, tags, tagProbabilities, sds) {
+    super(reason)
+    this.name = 'AIGuardAbortError'
+    this.reason = reason
+    this.tags = tags
+    this.tagProbabilities = tagProbabilities
+    this.sds = sds || []
+  }
+}
+
+class AIGuardClientError extends Error {
+  /**
+   * @param {string} message
+   * @param {{ telemetryType: string, errors?: Array<unknown>, cause?: Error }} opts
+   */
+  constructor (message, opts) {
+    super(message)
+    this.name = 'AIGuardClientError'
+    this.telemetryType = opts.telemetryType
+    if (opts.errors) {
+      this.errors = opts.errors
+    }
+    if (opts.cause) {
+      this.cause = opts.cause
+    }
+  }
+}
+
+module.exports = {
+  AIGuardAbortError,
+  AIGuardClientError,
+}