Update PublicKeysRepository.ts

Author: BlobMaster41

src/src/db/repositories/PublicKeysRepository.ts

@@ -1,4 +1,12 @@
-import { AnyBulkWriteOperation, Binary, ClientSession, Collection, Db, Document, Filter, } from 'mongodb';
+import {
+    AnyBulkWriteOperation,
+    Binary,
+    ClientSession,
+    Collection,
+    Db,
+    Document,
+    Filter,
+} from 'mongodb';
 import { OPNetCollections } from '../indexes/required/IndexedCollection.js';
 import { PublicKeyDocument } from '../interfaces/PublicKeyDocument.js';
 import { ExtendedBaseRepository } from './ExtendedBaseRepository.js';
@@ -384,11 +392,14 @@ export class PublicKeysRepository extends ExtendedBaseRepository<PublicKeyDocume
 
     private async getKeyInfoFromContracts(
         key: string,
+        isPublicKey: boolean,
     ): Promise<PublicKeyWithMLDSA | IPubKeyNotFoundError> {
         try {
-            const filter: Filter<IContractDocument> = {
-                $or: [{ contractAddress: key }, { contractPublicKey: new Binary(fromHex(key)) }],
-            };
+            const filter: Filter<IContractDocument> = isPublicKey
+                ? {
+                      contractPublicKey: new Binary(fromHex(key)),
+                  }
+                : { contractAddress: key };
 
             const resp = await this.getContractCollection().findOne(filter, {
                 projection: {
@@ -482,7 +493,7 @@ export class PublicKeysRepository extends ExtendedBaseRepository<PublicKeyDocume
                 };
             }
 
-            return await this.getKeyInfoFromContracts(key);
+            return await this.getKeyInfoFromContracts(key, isPublicKey);
         }
     }