Add a new identifyServerName function

Author: kraenhansen

packages/mongodb-build-info/src/index.ts

@@ -1,5 +1,7 @@
 import ConnectionString from 'mongodb-connection-string-url';
 
+type Document = Record<string, unknown>;
+
 const ATLAS_REGEX = /\.mongodb(-dev|-qa|-stage)?\.net$/i;
 const ATLAS_STREAM_REGEX = /^atlas-stream-.+/i;
 const LOCALHOST_REGEX =
@@ -134,6 +136,28 @@ export function getGenuineMongoDB(uri: string): {
   };
 }
 
+export async function identifyServerName(
+  uri: string,
+  runCommand: (command: Document) => Promise<Document>,
+): Promise<string> {
+  const hostname = getHostnameFromUrl(uri);
+  if (hostname.match(COSMOS_DB_REGEX)) {
+    return 'cosmosdb';
+  }
+
+  if (hostname.match(DOCUMENT_DB_REGEX)) {
+    return 'documentdb';
+  }
+
+  const buildInfo = await runCommand({ buildInfo: 1 });
+
+  if ('ferretdb' in buildInfo) {
+    return 'ferretdb';
+  }
+
+  return 'mongodb';
+}
+
 export function getBuildEnv(buildInfo: unknown): {
   serverOs: string | null;
   serverArch: string | null;