Merge pull request #147 from blacklanternsecurity/fix-bad-mongo-uri-parsing

Fix bad mongo URI parsing

Author: TheTechromancer

bbot_server/modules/server/server_cli.py

@@ -6,6 +6,8 @@
 from subprocess import run
 from contextlib import suppress
 
+from pymongo import uri_parser
+
 from bbot_server.config import BBOT_SERVER_CONFIG as bbcfg, BBOT_SERVER_DIR
 from bbot_server.cli.base import BaseBBCTL, subcommand, Option, Annotated
 
@@ -158,7 +160,7 @@ def cleardb(
             )
 
         for store_name, store_config, data_desc in stores_to_clear:
-            db_name = store_config.uri.split("/")[-1]
+            db_name = uri_parser.parse_uri(store_config.uri)["database"]
             prefix = store_config.collection_prefix
             if not db_name:
                 raise self.BBOTServerError(f"{store_name.title()} database not found in config")

bbot_server/store.py

@@ -7,7 +7,7 @@
 class BaseMongoStore(BaseDB):
     async def setup(self):
         self.client = AsyncMongoClient(self.uri)
-        self.db = self.client.get_database(self.db_name)
+        self.db = self.client.get_default_database()
         self.collection_prefix = getattr(self.db_config, "collection_prefix", "")
         bucket_name = f"{self.collection_prefix}fs" if self.collection_prefix else "fs"
         self.fs = AsyncGridFSBucket(self.db, bucket_name=bucket_name)

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "bbot-server"
-version = "0.1.3"
+version = "0.1.4"
 description = ""
 authors = [{name = "TheTechromancer"}]
 license = "AGPL-3.0"