pwap/shared/lib/wallet.ts

// ========================================
// PezkuwiChain - Substrate/Polkadot.js Configuration
// ========================================
// This file configures wallet connectivity for Substrate-based chains

import type { InjectedAccountWithMeta } from '@pezkuwi/extension-inject/types';
import { getCurrentNetworkConfig } from '../blockchain/endpoints';

// Helper to get environment variables that works in both web (Vite) and React Native (Expo)
const getEnv = (key: string, defaultValue: string = ''): string => {
  // Check for Vite environment (web)
  if (typeof import.meta !== 'undefined' && import.meta.env) {
    return (import.meta.env as any)[key] || defaultValue;
  }
  // Check for Expo environment (React Native)
  if (typeof process !== 'undefined' && process.env) {
    // Expo uses EXPO_PUBLIC_ prefix, Vite uses VITE_ prefix
    const expoKey = key.replace('VITE_', 'EXPO_PUBLIC_');
    return process.env[expoKey] || process.env[key] || defaultValue;
  }
  return defaultValue;
};

// ========================================
// CHAIN CONFIGURATION
// ========================================
export const CHAIN_CONFIG = {
  name: getEnv('VITE_CHAIN_NAME', 'PezkuwiChain'),
  symbol: getEnv('VITE_CHAIN_TOKEN_SYMBOL', 'PEZ'),
  decimals: parseInt(getEnv('VITE_CHAIN_TOKEN_DECIMALS', '12')),
  ss58Format: parseInt(getEnv('VITE_CHAIN_SS58_FORMAT', '42')),
};

// ========================================
// SUBSTRATE ASSET IDs (Assets Pallet on Asset Hub)
// ========================================
// ⚠️ IMPORTANT: HEZ is the native token and does NOT have an Asset ID
// Only wrapped/asset tokens are listed here
//
// Asset ID Allocation on Asset Hub:
// - Asset 1: PEZ (Pezkuwi Token)
// - Asset 2: wHEZ (Wrapped HEZ via tokenWrapper pallet)
// - Asset 1000: wUSDT (Wrapped USDT)
export const ASSET_IDS = {
  WHEZ: parseInt(getEnv('VITE_ASSET_WHEZ', '2')),  // Wrapped HEZ (tokenWrapper creates Asset 2)
  PEZ: parseInt(getEnv('VITE_ASSET_PEZ', '1')),    // PEZ utility token
  WUSDT: parseInt(getEnv('VITE_ASSET_WUSDT', '1000')), // Wrapped USDT (6 decimals, Asset ID 1000)
  USDT: parseInt(getEnv('VITE_ASSET_USDT', '3')),
  BTC: parseInt(getEnv('VITE_ASSET_BTC', '4')),
  ETH: parseInt(getEnv('VITE_ASSET_ETH', '5')),
  DOT: parseInt(getEnv('VITE_ASSET_DOT', '6')),
} as const;

// ========================================
// ASSET CONFIGURATIONS
// ========================================
export const ASSET_CONFIGS = {
  WHEZ: {
    id: ASSET_IDS.WHEZ,
    symbol: 'wHEZ',
    name: 'Wrapped HEZ',
    decimals: 12,
    minBalance: 1_000_000_000, // 0.001 HEZ
  },
  PEZ: {
    id: ASSET_IDS.PEZ,
    symbol: 'PEZ',
    name: 'PEZ Utility Token',
    decimals: 12,
    minBalance: 1_000_000_000, // 0.001 PEZ
  },
  WUSDT: {
    id: ASSET_IDS.WUSDT,
    symbol: 'wUSDT',
    name: 'Wrapped USDT',
    decimals: 6, // ⚠️ CRITICAL: wUSDT uses 6 decimals (USDT standard), not 12!
    minBalance: 1_000, // 0.001 USDT
  },
} as const;

// ========================================
// EXPLORER URLS
// ========================================
export const EXPLORER_URLS = {
  pezkuwiJs: getEnv('VITE_EXPLORER_URL', 'https://js.pezkuwichain.io'),
  custom: getEnv('VITE_CUSTOM_EXPLORER_URL', 'https://explorer.pezkuwichain.io'),
};

// ========================================
// WALLET ERROR MESSAGES
// ========================================
export const WALLET_ERRORS = {
  NO_EXTENSION: 'No Polkadot.js extension detected. Please install Polkadot.js or compatible wallet.',
  NO_ACCOUNTS: 'No accounts found. Please create an account in your wallet extension.',
  CONNECTION_FAILED: 'Failed to connect wallet. Please try again.',
  TRANSACTION_FAILED: 'Transaction failed. Please check your balance and try again.',
  USER_REJECTED: 'User rejected the request.',
  INSUFFICIENT_BALANCE: 'Insufficient balance to complete transaction.',
  INVALID_ADDRESS: 'Invalid address format.',
  API_NOT_READY: 'Blockchain API not ready. Please wait...',
};

// ========================================
// UTILITY FUNCTIONS
// ========================================

/**
 * Format Substrate address for display (SS58 format)
 * @param address - Full substrate address
 * @returns Shortened address string (e.g., "5GrwV...xQjz")
 */
export const formatAddress = (address: string): string => {
  if (!address) return '';
  return `${address.slice(0, 6)}...${address.slice(-4)}`;
};

/**
 * Format balance from planck to human-readable format
 * @param balance - Balance in smallest unit (planck)
 * @param decimals - Token decimals (default 12 for PEZ)
 * @returns Formatted balance string
 */
export const formatBalance = (balance: string | number | bigint, decimals = 12): string => {
  if (!balance && balance !== 0) return '0';

  let value: bigint;

  if (typeof balance === 'bigint') {
    value = balance;
  } else if (typeof balance === 'string') {
    // Handle hex strings (from blockchain) and decimal strings
    if (balance.startsWith('0x')) {
      value = BigInt(balance);
    } else {
      // Try to parse as BigInt, fall back to float conversion
      try {
        value = BigInt(balance);
      } catch {
        // If it's a decimal string like "1.5", convert via float
        value = BigInt(Math.floor(parseFloat(balance)));
      }
    }
  } else {
    value = BigInt(Math.floor(balance));
  }

  // Convert to decimal using string manipulation to avoid float precision issues
  const divisor = BigInt(10 ** decimals);
  const wholePart = value / divisor;
  const fractionalPart = value % divisor;

  // Format fractional part with leading zeros
  const fractionalStr = fractionalPart.toString().padStart(decimals, '0').slice(0, 4);

  return `${wholePart}.${fractionalStr}`;
};

/**
 * Parse human-readable amount to planck (smallest unit)
 * @param amount - Human-readable amount
 * @param decimals - Token decimals
 * @returns Amount in planck
 */
export const parseAmount = (amount: string | number, decimals = 12): bigint => {
  const value = typeof amount === 'string' ? parseFloat(amount) : amount;
  return BigInt(Math.floor(value * Math.pow(10, decimals)));
};

/**
 * Get asset symbol by ID
 * @param assetId - Asset ID from Assets pallet
 * @returns Asset symbol or 'UNKNOWN'
 */
export const getAssetSymbol = (assetId: number): string => {
  const entry = Object.entries(ASSET_IDS).find(([_, id]) => id === assetId);
  return entry ? entry[0] : 'UNKNOWN';
};

/**
 * Get current network endpoint based on VITE_NETWORK env
 * @returns WebSocket endpoint URL
 */
export const getCurrentEndpoint = (): string => {
  return getCurrentNetworkConfig().wsEndpoint;
};

// ========================================
// TYPE DEFINITIONS
// ========================================

export interface PolkadotWalletState {
  isConnected: boolean;
  accounts: InjectedAccountWithMeta[];
  selectedAccount: InjectedAccountWithMeta | null;
  balance: string;
  error: string | null;
}

export const initialPolkadotWalletState: PolkadotWalletState = {
  isConnected: false,
  accounts: [],
  selectedAccount: null,
  balance: '0',
  error: null,
};