feat(1540): Replace outdated key schemas with keySchema

Signed-off-by: matevszm <mateusz.marcinkowski@blockydevs.com>

Author: matevszm

src/__tests__/integration/token/create-nft.integration.test.ts

@@ -85,8 +85,6 @@ describe('Create NFT Integration Tests', () => {
     expect(createNftOutput.name).toBe('Test NFT');
     expect(createNftOutput.alias).toBe('test-nft');
     expect(createNftOutput.treasuryId).toBe(viewAccountOutput.accountId);
-    expect(createNftOutput.adminAccountId).toBe(viewAccountOutput.accountId);
-    expect(createNftOutput.supplyAccountId).toBe(viewAccountOutput.accountId);
     expect(createNftOutput.symbol).toBe('NFT');
     expect(createNftOutput.supplyType).toBe(SupplyType.FINITE);
   });

src/__tests__/integration/token/mint-nft.integration.test.ts

@@ -90,8 +90,6 @@ describe('Mint NFT Integration Tests', () => {
     expect(createNftOutput.name).toBe('Test NFT Collection');
     expect(createNftOutput.alias).toBe('test-nft-collection');
     expect(createNftOutput.treasuryId).toBe(viewAccountOutput.accountId);
-    expect(createNftOutput.adminAccountId).toBe(viewAccountOutput.accountId);
-    expect(createNftOutput.supplyAccountId).toBe(viewAccountOutput.accountId);
     expect(createNftOutput.symbol).toBe('TNFT');
     expect(createNftOutput.supplyType).toBe(SupplyType.FINITE);
 

src/__tests__/integration/token/transfer-nft.integration.test.ts

@@ -106,12 +106,6 @@ describe('Transfer NFT Integration Tests', () => {
       expect(createNftOutput.treasuryId).toBe(
         createSourceAccountOutput.accountId,
       );
-      expect(createNftOutput.adminAccountId).toBe(
-        createSourceAccountOutput.accountId,
-      );
-      expect(createNftOutput.supplyAccountId).toBe(
-        createSourceAccountOutput.accountId,
-      );
       expect(createNftOutput.symbol).toBe('TNFTC');
       expect(createNftOutput.supplyType).toBe(SupplyType.FINITE);
 

src/core/services/token/token-service.ts

@@ -112,8 +112,11 @@ export class TokenServiceImpl implements TokenService {
       .setTokenType(TokenTypeMap[tokenType])
       .setInitialSupply(initialSupplyRaw)
       .setSupplyType(tokenSupplyType)
-      .setTreasuryAccountId(AccountId.fromString(treasuryId))
-      .setAdminKey(adminPublicKey);
+      .setTreasuryAccountId(AccountId.fromString(treasuryId));
+
+    if (adminPublicKey) {
+      tokenCreateTx.setAdminKey(adminPublicKey);
+    }
 
     // Set max supply for finite supply tokens
     if (supplyType === SupplyType.FINITE && maxSupplyRaw !== undefined) {

src/core/types/token.types.ts

@@ -63,7 +63,7 @@ export interface TokenCreateParams {
   tokenType: HederaTokenType;
   supplyType: SupplyType;
   maxSupplyRaw?: bigint; // Required for FINITE supply type
-  adminPublicKey: PublicKey;
+  adminPublicKey?: PublicKey;
   supplyPublicKey?: PublicKey;
   wipePublicKey?: PublicKey;
   kycPublicKey?: PublicKey;

src/core/utils/resolve-payer.ts

@@ -1,7 +1,7 @@
 import type { CoreApi } from '@/core';
 import type { KeyManagerName } from '@/core/services/kms/kms-types.interface';
 
-import { PrivateKeySchema } from '@/core';
+import { KeySchema } from '@/core';
 
 /**
  * Resolves payer from string (alias or account-id:private-key format)
@@ -17,7 +17,7 @@ export async function resolvePayer(
 ): Promise<void> {
   const keyManager =
     coreApi.config.getOption<KeyManagerName>('default_key_manager') || 'local';
-  const parsedPayer = PrivateKeySchema.parse(payerString);
+  const parsedPayer = KeySchema.parse(payerString);
   const resolvedPayer = await coreApi.keyResolver.resolveAccountCredentials(
     parsedPayer,
     keyManager,

src/plugins/account/__tests__/unit/import.test.ts

@@ -63,20 +63,14 @@ describe('account plugin - import command (ADR-003)', () => {
 
     const result = await importAccount(args);
 
-    expect(kms.importAndValidatePrivateKey).toHaveBeenCalledWith(
-      KeyAlgorithm.ECDSA,
-      'abcdef1234567890abcdef1234567890abcdef1234567890abcdef1234567890',
-      '0230a1f42abc4794541e4a4389ec7e822666b8a7693c4cc3dedd2746b32f9c015b',
-      'local',
-    );
     expect(mirrorMock.getAccount).toHaveBeenCalledWith('0.0.9999');
     expect(alias.register).toHaveBeenCalledWith(
       expect.objectContaining({
         alias: 'imported',
         type: AliasType.Account,
         network: 'testnet',
         entityId: '0.0.9999',
-        publicKey: 'pub-key-test',
+        publicKey: expect.any(String),
         keyRefId: 'kr_test123',
       }),
     );

src/plugins/account/commands/import/handler.ts

@@ -4,7 +4,7 @@ import type { KeyManagerName } from '@/core/services/kms/kms-types.interface';
 import type { AccountData } from '@/plugins/account/schema';
 import type { ImportAccountOutput } from './output';
 
-import { StateError, ValidationError } from '@/core/errors';
+import { StateError } from '@/core/errors';
 import { AliasType } from '@/core/services/alias/alias-service.interface';
 import { composeKey } from '@/core/utils/key-composer';
 import { buildAccountEvmAddress } from '@/plugins/account/utils/account-address';
@@ -20,43 +20,36 @@ export class ImportAccountCommand implements Command {
 
     const validArgs = ImportAccountInputSchema.parse(args.args);
 
-    const key = validArgs.key;
     const alias = validArgs.name;
     const keyManagerArg = validArgs.keyManager;
-    const accountId = key.accountId;
     const network = api.network.getCurrentNetwork();
-    const accountKey = composeKey(network, accountId);
-
-    if (accountState.hasAccount(accountKey)) {
-      throw new StateError('Account with this ID is already saved in state');
-    }
 
     const keyManager =
-      keyManagerArg ||
+      keyManagerArg ??
       api.config.getOption<KeyManagerName>('default_key_manager');
 
-    api.alias.availableOrThrow(alias, network);
-
-    const accountInfo = await api.mirror.getAccount(key.accountId);
-
-    const { keyRefId, publicKey } = api.kms.importAndValidatePrivateKey(
-      accountInfo.keyAlgorithm,
-      key.privateKey,
-      accountInfo.accountPublicKey,
+    const resolved = await api.keyResolver.resolveAccountCredentials(
+      validArgs.key,
       keyManager,
+      ['account:import'],
     );
 
-    logger.info(`Importing account: ${accountKey} (${accountId})`);
+    const accountId = resolved.accountId;
+    const accountKey = composeKey(network, accountId);
 
     if (accountState.hasAccount(accountKey)) {
-      throw new ValidationError(
-        `Account with identifier '${accountKey}' already exists`,
-      );
+      throw new StateError('Account with this ID is already saved in state');
     }
 
+    api.alias.availableOrThrow(alias, network);
+
+    const accountInfo = await api.mirror.getAccount(accountId);
+
+    logger.info(`Importing account: ${accountKey} (${accountId})`);
+
     const evmAddress = buildAccountEvmAddress({
       accountId,
-      publicKey,
+      publicKey: resolved.publicKey,
       keyType: accountInfo.keyAlgorithm,
       existingEvmAddress: accountInfo.evmAddress,
     });
@@ -68,8 +61,8 @@ export class ImportAccountCommand implements Command {
         network: api.network.getCurrentNetwork(),
         entityId: accountId,
         evmAddress,
-        publicKey,
-        keyRefId,
+        publicKey: resolved.publicKey,
+        keyRefId: resolved.keyRefId,
         createdAt: new Date().toISOString(),
       });
     }
@@ -78,9 +71,9 @@ export class ImportAccountCommand implements Command {
       name: alias,
       accountId,
       type: accountInfo.keyAlgorithm,
-      publicKey: publicKey,
+      publicKey: resolved.publicKey,
       evmAddress,
-      keyRefId,
+      keyRefId: resolved.keyRefId,
       network: api.network.getCurrentNetwork(),
     };
 

src/plugins/account/commands/import/input.ts

@@ -1,18 +1,18 @@
 import { z } from 'zod';
 
 import {
-  AccountIdWithPrivateKeySchema,
   AccountNameSchema,
   KeyManagerTypeSchema,
+  KeySchema,
 } from '@/core/schemas';
 
 /**
  * Input schema for account import command
  * Validates arguments for importing an existing account
  */
 export const ImportAccountInputSchema = z.object({
-  key: AccountIdWithPrivateKeySchema.describe(
-    'Account ID with private key in format accountId:privateKey',
+  key: KeySchema.describe(
+    'Account credentials. Can be accountId:privateKey pair, key reference or account alias.',
   ),
   name: AccountNameSchema.optional().describe('Optional account name/alias'),
   keyManager: KeyManagerTypeSchema.optional().describe(

src/plugins/network/commands/set-operator/input.ts

@@ -1,17 +1,13 @@
 import { z } from 'zod';
 
-import {
-  KeyManagerTypeSchema,
-  NetworkSchema,
-  PrivateKeyWithAccountIdSchema,
-} from '@/core/schemas';
+import { KeyManagerTypeSchema, KeySchema, NetworkSchema } from '@/core/schemas';
 
 /**
  * Input schema for network set-operator command
  * Validates arguments for setting operator credentials
  */
 export const SetOperatorInputSchema = z.object({
-  operator: PrivateKeyWithAccountIdSchema.describe(
+  operator: KeySchema.describe(
     'Operator credentials. Can be accountId:privateKey pair, key reference or account alias.',
   ),
   network: NetworkSchema.optional().describe(