Add authentication using authSource (#225)

* Add authentication using authSource * Fix formatting for 153e1a1
denodrivers · Aug 2, 2021 · 88dd176 · 88dd176
1 parent 7874818
commit 88dd176
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/src/utils/uri.ts b/src/utils/uri.ts
@@ -123,9 +123,10 @@ export type SrvConnectOptions = Omit<ConnectOptions, "servers"> & {
 export function parseSrvUrl(url: string): SrvConnectOptions {
   const data = parse_url(url);
   const connectOptions: SrvConnectOptions = {
-    db: (data.pathname && data.pathname.length > 1)
-      ? data.pathname.substring(1)
-      : "admin",
+    db: new URLSearchParams(data.search).get("authSource") ??
+      ((data.pathname && data.pathname.length > 1)
+        ? data.pathname.substring(1)
+        : "admin"),
   };
 
   if (data.auth) {
@@ -184,9 +185,10 @@ function parseNormalUrl(url: string): ConnectOptions {
     server.port = server.port || 27017;
   }
 
-  connectOptions.db = (data.pathname && data.pathname.length > 1)
-    ? data.pathname.substring(1)
-    : "admin";
+  connectOptions.db = new URLSearchParams(data.search).get("authSource") ??
+    (data.pathname && (data.pathname.length > 1)
+      ? data.pathname.substring(1)
+      : "admin");
   if (data.auth) {
     connectOptions.credential = <Credential> {
       username: data.auth.user,