Update PublicKeysRepository.ts

Author: BlobMaster41

src/src/db/repositories/PublicKeysRepository.ts

@@ -61,7 +61,12 @@ export class PublicKeysRepository extends ExtendedBaseRepository<PublicKeyDocume
                 this.network,
             );
 
-            promises.push(this.getKeyInfo(addressOrPublicKeys[i], isPublicKey));
+            const isContract = AddressVerificator.isValidP2OPAddress(
+                addressOrPublicKeys[i],
+                this.network,
+            );
+
+            promises.push(this.getKeyInfo(addressOrPublicKeys[i], isPublicKey, isContract));
         }
 
         const results = await Promise.safeAll(promises);
@@ -462,6 +467,7 @@ export class PublicKeysRepository extends ExtendedBaseRepository<PublicKeyDocume
     private async getKeyInfo(
         key: string,
         isPublicKey: boolean,
+        isContract: boolean,
     ): Promise<PublicKeyWithMLDSA | IPubKeyNotFoundError> {
         try {
             const filter: Filter<PublicKeyDocument> = {
@@ -476,7 +482,7 @@ export class PublicKeysRepository extends ExtendedBaseRepository<PublicKeyDocume
 
             return await this.getOneWithMLDSA(filter);
         } catch (e) {
-            if (!isPublicKey) {
+            if (!isContract) {
                 return {
                     error: 'Public key not found (invalid key format, for contract address)',
                 };