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

@@ -1,14 +1,9 @@
-/**
- * Account Import Command Handler
- * Handles importing existing accounts using the Core API
- * Follows ADR-003 contract: returns CommandExecutionResult
- */
 import type { CommandHandlerArgs, CommandResult } from '@/core';
 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';
@@ -21,53 +16,40 @@ export async function importAccount(
 ): Promise<CommandResult> {
   const { api, logger } = args;
 
-  // Initialize Zustand state helper
   const accountState = new ZustandAccountStateHelper(api.state, logger);
 
-  // Parse and validate command arguments
   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');
-  }
 
-  // Get keyManager from args or fallback to config
   const keyManager =
-    keyManagerArg ||
+    keyManagerArg ??
     api.config.getOption<KeyManagerName>('default_key_manager');
 
-  // Check if account name already exists
-  api.alias.availableOrThrow(alias, network);
-
-  // Get account info from mirror node
-  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);
 
-  // Check if account name already exists
   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,
   });
@@ -79,27 +61,24 @@ export async function importAccount(
       network: api.network.getCurrentNetwork(),
       entityId: accountId,
       evmAddress,
-      publicKey,
-      keyRefId,
+      publicKey: resolved.publicKey,
+      keyRefId: resolved.keyRefId,
       createdAt: new Date().toISOString(),
     });
   }
 
-  // Create account object (no private key in plugin state)
   const account: AccountData = {
     name: alias,
     accountId,
     type: accountInfo.keyAlgorithm,
-    publicKey: publicKey,
+    publicKey: resolved.publicKey,
     evmAddress,
-    keyRefId,
+    keyRefId: resolved.keyRefId,
     network: api.network.getCurrentNetwork(),
   };
 
-  // Store account in state using the helper
   accountState.saveAccount(accountKey, account);
 
-  // Prepare output data
   const outputData: ImportAccountOutput = {
     accountId,
     name: alias,

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(