JM.js

import fs from 'fs';
import crypto from 'crypto';
import chalk from 'chalk';
import readline from 'readline';

function displayBanner() {
    console.clear();
    
    const asciiArt = [
        "    ███████╗██╗  ██╗██████╗ ██╗███████╗███████╗",
        "    ╚════██║██║ ██╔╝██╔══██╗██║██╔════╝██╔════╝",
        "        ██╔╝█████╔╝ ██████╔╝██║███████╗███████╗",
        "       ██╔╝ ██╔═██╗ ██╔══██╗██║╚════██║╚════██║",
        "       ██║  ██║  ██╗██║  ██║██║███████║███████║",
        "       ╚═╝  ╚═╝  ╚═╝╚═╝  ╚═╝╚═╝╚══════╝╚══════╝"
    ];
    
    console.log('\n');
    const colors = [chalk.red, chalk.yellow, chalk.green, chalk.blue, chalk.magenta, chalk.cyan];
    asciiArt.forEach((line, index) => {
        console.log(colors[index % colors.length].bold(line));
    });
    console.log('\n');
    
    console.log(chalk.cyan('╔══════════════════════════════════════════════════════════════╗'));
    console.log(chalk.cyan('║') + chalk.yellow.bold('                  私钥批量加密工具V1                       ') + chalk.cyan('║'));
    console.log(chalk.cyan('║') + chalk.green('                 Crypto Private Key Encryptor               ') + chalk.cyan('║'));
    console.log(chalk.cyan('╠══════════════════════════════════════════════════════════════╣'));
    console.log(chalk.cyan('║') + chalk.magenta('  🔐 支持格式: ETH (Ethereum) / SOL (Solana)              ') + chalk.cyan('║'));
    console.log(chalk.cyan('║') + chalk.blue('  🛡️  加密算法: AES-256-CBC + PBKDF2                        ') + chalk.cyan('║'));
    console.log(chalk.cyan('║') + chalk.white('  ⚡ 批量处理: 自动识别私钥类型                           ') + chalk.cyan('║'));
    console.log(chalk.cyan('╠══════════════════════════════════════════════════════════════╣'));
    console.log(chalk.cyan('║') + chalk.gray('  开发者: ') + chalk.yellow.bold('@7KRIS5') + chalk.gray('                                   ') + chalk.cyan('║'));
    console.log(chalk.cyan('║') + chalk.gray('  版本: 2025 安全加强版                                  ') + chalk.cyan('║'));
    console.log(chalk.cyan('╚══════════════════════════════════════════════════════════════╝'));
    
    console.log('\n' + chalk.red.bold('⚠️  安全提醒:'));
    console.log(chalk.yellow('   • 请妥善保管加密密码，丢失后无法恢复私钥'));
    console.log(chalk.yellow('   • 建议在离线环境中运行此工具'));
    console.log(chalk.yellow('   • 加密完成后请安全删除原始私钥文件'));
    
    console.log('\n' + chalk.green.bold('📖 使用说明:'));
    console.log(chalk.white('   1. 将私钥放入 privateKey.txt 文件（每行一个）'));
    console.log(chalk.white('   2. 运行工具并设置加密密码'));
    console.log(chalk.white('   3. 加密结果将保存到 .env 文件'));
    
    console.log('\n' + chalk.cyan('═'.repeat(62)) + '\n');
}

const rl = readline.createInterface({
    input: process.stdin,
    output: process.stdout
});

async function askPassword(query) {
    return new Promise(resolve => {
        rl.question(query, answer => {
            resolve(answer);
        });
    });
}

function isValidBase58(str) {
    const base58Regex = /^[1-9A-HJ-NP-Za-km-z]+$/;
    return base58Regex.test(str);
}

function validatePrivateKey(privateKey) {
    const cleanedKey = privateKey.trim();
    
    const ethKey = cleanedKey.startsWith('0x') ? cleanedKey.slice(2) : cleanedKey;
    if (/^[0-9a-fA-F]{64}$/.test(ethKey)) {
        return { valid: true, type: 'ETH', key: ethKey };
    }
    
    if (cleanedKey.length >= 80 && cleanedKey.length <= 90 && isValidBase58(cleanedKey)) {
        return { valid: true, type: 'SOL', key: cleanedKey };
    }
    
    return { valid: false, type: 'UNKNOWN', key: cleanedKey };
}

async function generateEncryptionKey() {
    const password = await askPassword(chalk.magenta('请输入加密密码（至少8个字符）: '));
    if (!password || password.length < 8) {
        console.log(chalk.red('❌ 密码必须至少8个字符'));
        rl.close();
        process.exit(1);
    }
    
    const salt = crypto.randomBytes(16);
    const key = crypto.pbkdf2Sync(password, salt, 100000, 32, 'sha256');
    return { key: key.toString('hex'), salt: salt.toString('hex') };
}

async function encryptPrivateKeys() {
    const inputFile = 'privateKey.txt';
    const envFile = '.env';
    
    if (!fs.existsSync(inputFile)) {
        console.log(chalk.red(`❌ 输入文件 ${inputFile} 未找到`));
        rl.close();
        process.exit(1);
    }
    
    const privateKeys = fs.readFileSync(inputFile, 'utf8')
        .split('\n')
        .map(key => key.trim())
        .filter(key => key.length > 0);
    
    if (privateKeys.length === 0) {
        console.log(chalk.red('❌ privateKey.txt 文件为空'));
        rl.close();
        process.exit(1);
    }
    
    const { key: encryptionKey, salt } = await generateEncryptionKey();
    
    const envContent = [`SALT=${salt}`];
    
    try {
        let validKeyCount = 0;
        
        privateKeys.forEach((privateKey, index) => {
            const validation = validatePrivateKey(privateKey);
            
            if (!validation.valid) {
                console.log(chalk.red(`❌ 无效私钥格式: ${privateKey.slice(0, 10)}...${privateKey.slice(-10)}`));
                console.log(chalk.yellow('   支持格式: ETH (64位十六进制) 或 SOL (Base58编码)'));
                return;
            }
            
            const iv = crypto.randomBytes(16);
            const cipher = crypto.createCipheriv('aes-256-cbc', Buffer.from(encryptionKey, 'hex'), iv);
            let encrypted = cipher.update(validation.key, 'utf8', 'hex');
            encrypted += cipher.final('hex');
            
            envContent.push(`ENCRYPTED_KEY_${index}=${encrypted}:${iv.toString('hex')}:${validation.type}`);
            
            const displayKey = validation.key.length > 20 
                ? `${validation.key.slice(0, 10)}...${validation.key.slice(-10)}`
                : `${validation.key.slice(0, 6)}...${validation.key.slice(-6)}`;
            
            console.log(chalk.green(`✅ 成功加密 ${validation.type} 私钥 ${index}: ${displayKey}`));
            validKeyCount++;
        });
        
        if (validKeyCount === 0) {
            console.log(chalk.red('❌ 没有找到有效的私钥'));
            rl.close();
            process.exit(1);
        }
        
        fs.writeFileSync(envFile, envContent.join('\n'), 'utf8');
        console.log(chalk.green(`✅ 盐和 ${validKeyCount} 个加密私钥已保存到 ${envFile}`));
        
        console.log(chalk.yellow('\n已清空并更新 .env 文件，格式说明：'));
        console.log(chalk.cyan('ENCRYPTED_KEY_X=加密数据:IV:类型'));
        console.log(chalk.cyan('类型标识: ETH=以太坊私钥, SOL=Solana私钥\n'));
        
        const ethCount = envContent.filter(line => line.includes(':ETH')).length;
        const solCount = envContent.filter(line => line.includes(':SOL')).length;
        console.log(chalk.blue(`📊 加密统计: ETH私钥 ${ethCount} 个, SOL私钥 ${solCount} 个`));
        
    } catch (error) {
        console.log(chalk.red(`❌ 加密失败: ${error.message}`));
        rl.close();
        process.exit(1);
    } finally {
        rl.close();
    }
}

async function main() {
    displayBanner();
    
    await new Promise(resolve => {
        rl.question(chalk.green('按 Enter 键开始加密...'), () => {
            resolve();
        });
    });
    
    await encryptPrivateKeys();
}

main().catch(error => {
    console.error(chalk.red('程序运行出错:'), error);
    process.exit(1);
});