Dynamic user agent

src/config/index.ts

@@ -18,6 +18,7 @@ export default {
     GENERAL_ENDPOINT: "https://mailtrap.io",
     USER_AGENT:
       "mailtrap-nodejs (https://github.com/railsware/mailtrap-nodejs)",
+    MCP_USER_AGENT: "mailtrap-mcp (https://github.com/railsware/mailtrap-mcp)",
     MAX_REDIRECTS: 0,
     TIMEOUT: 10000,
   },

src/lib/MailtrapClient.ts

@@ -6,12 +6,14 @@ import axios, { AxiosInstance } from "axios";
 import encodeMailBuffers from "./mail-buffer-encoder";
 import handleSendingError from "./axios-logger";
 import MailtrapError from "./MailtrapError";
+import getDynamicUserAgent from "./get-agent";
 
 import GeneralAPI from "./api/General";
 import TestingAPI from "./api/Testing";
 import ContactsBaseAPI from "./api/Contacts";
 import ContactListsBaseAPI from "./api/ContactLists";
 import TemplatesBaseAPI from "./api/Templates";
+import SuppressionsBaseAPI from "./api/Suppressions";
 
 import CONFIG from "../config";
 
@@ -22,13 +24,11 @@ import {
   BatchSendResponse,
   BatchSendRequest,
 } from "../types/mailtrap";
-import SuppressionsBaseAPI from "./api/Suppressions";
 
 const { CLIENT_SETTINGS, ERRORS } = CONFIG;
 const {
   SENDING_ENDPOINT,
   MAX_REDIRECTS,
-  USER_AGENT,
   TIMEOUT,
   TESTING_ENDPOINT,
   BULK_ENDPOINT,
@@ -66,7 +66,7 @@ export default class MailtrapClient {
       headers: {
         Authorization: `Bearer ${token}`,
         Connection: "keep-alive",
-        "User-Agent": USER_AGENT,
+        "User-Agent": getDynamicUserAgent(),
       },
       maxRedirects: MAX_REDIRECTS,
       timeout: TIMEOUT,

src/lib/get-agent.ts

@@ -0,0 +1,63 @@
+import CONFIG from "../config";
+
+const { USER_AGENT, MCP_USER_AGENT } = CONFIG.CLIENT_SETTINGS;
+
+/**
+ * Checks if the main module filename indicates MCP context.
+ * @returns true if main module contains "mailtrap-mcp" and is not in node_modules
+ */
+function isMainModuleMCP(): boolean {
+  const mainFile = require?.main?.filename;
+
+  return !!(
+    mainFile &&
+    mainFile.includes("mailtrap-mcp") &&
+    !mainFile.includes("node_modules")
+  );
+}
+
+/**
+ * Checks if the current working directory indicates MCP context.
+ * @returns true if cwd contains "mailtrap-mcp" and is not in node_modules
+ */
+function isWorkingDirectoryMCP(): boolean {
+  try {
+    const cwd = process.cwd();
+    return cwd.includes("mailtrap-mcp") && !cwd.includes("node_modules");
+  } catch {
+    return false;
+  }
+}
+
+/**
+ * Checks if the call stack indicates MCP context.
+ * @returns true if stack contains "mailtrap-mcp" and is not from node_modules/mailtrap
+ */
+function isCallStackMCP(): boolean {
+  const { stack } = new Error();
+
+  return !!(
+    stack &&
+    stack.includes("mailtrap-mcp") &&
+    !stack.includes("node_modules/mailtrap")
+  );
+}
+
+/**
+ * Determines if the code is running in a Mailtrap MCP context.
+ * Uses multiple detection methods to ensure accurate context identification.
+ * @returns true if running in MCP context, false otherwise
+ */
+function isMailtrapMCPContext(): boolean {
+  return isMainModuleMCP() || isWorkingDirectoryMCP() || isCallStackMCP();
+}
+
+/**
+ * Gets the appropriate User-Agent string based on the current context.
+ * @returns The User-Agent string for the current context
+ */
+function getDynamicUserAgent(): string {
+  return isMailtrapMCPContext() ? MCP_USER_AGENT : USER_AGENT;
+}
+
+export default getDynamicUserAgent;