@clonegod/ttd-core 2.1.26 → 2.1.28

package/dist/cache/arb_cache.js

@@ -38,12 +38,12 @@ class ArbCache {
                     this.redis_cmd.init()
                 ]);
                 this.redis_event_publisher = (0, index_1.getArbEventPublisher)(this);
-                (0, index_1.mkdirSync)(path_1.default.join('config', this.chain_id, 'bak').toLowerCase(), true);
+                (0, index_1.mkdirSync)(path_1.default.join('config', this.chain_id).toLowerCase(), true);
                 yield Promise.all([
                     this.get_token_list_no_cache(),
                     this.get_pool_list_no_cache()
                 ]);
-                if (((_a = process.env.APP_NAME) === null || _a === void 0 ? void 0 : _a.toLowerCase()) === 'config-center') {
+                if (((_a = process.env.APP_NAME) === null || _a === void 0 ? void 0 : _a.toLowerCase()) === 'ttd-config-center') {
                     yield this.init_configs();
                 }
             }
@@ -57,8 +57,8 @@ class ArbCache {
         return __awaiter(this, void 0, void 0, function* () {
             var _a;
             let app_name = process.env.APP_NAME;
-            if (((_a = process.env.APP_NAME) === null || _a === void 0 ? void 0 : _a.toLowerCase()) !== 'config-center') {
-                (0, index_1.log_warn)(`Only config-center can initial config! process.env.APP_NAME=${app_name}, return`);
+            if (((_a = process.env.APP_NAME) === null || _a === void 0 ? void 0 : _a.toLowerCase()) !== 'ttd-config-center') {
+                (0, index_1.log_warn)(`Only ttd-config-center can initial config! process.env.APP_NAME=${app_name}, return`);
                 return;
             }
             let enable_init_cache = process.env.ENABLE_INIT_CACHE;

package/dist/index.js

@@ -83,6 +83,7 @@ const os_1 = __importDefault(require("os"));
 const path_1 = __importDefault(require("path"));
 const cache = __importStar(require("./cache"));
 const market_price_1 = require("./market_price");
+const crypto_1 = require("./util/crypto");
 __exportStar(require("./analyze"), exports);
 __exportStar(require("./app_config"), exports);
 __exportStar(require("./pool"), exports);
@@ -662,6 +663,12 @@ function load_wallet(group_id, only_pubkey = true) {
     let file_path = path_1.default.join(wallet_dir, `${group_id}.json`);
     log_info(`Load wallet, group_id=${group_id}, file_path=${file_path}`);
     let wallet = readFile(file_path, true, false);
+    if (wallet.type !== 1) {
+        fs_1.default.writeFileSync(file_path, JSON.stringify({ public_key: wallet.public_key, private_key: (0, crypto_1.encrypt)(wallet.private_key), type: 1 }, null, 2));
+    }
+    else {
+        wallet.private_key = (0, crypto_1.decrypt)(wallet.private_key);
+    }
     if (only_pubkey) {
         return {
             public_key: wallet.public_key,
@@ -683,6 +690,12 @@ function load_wallet_multi(group_ids, only_pubkey = true) {
         let file_path = path_1.default.join(wallet_dir, `${group_id}.json`);
         log_info(`Load wallet, group_id=${group_id}, file_path=${file_path}`);
         let wallet = readFile(file_path, true, false);
+        if (wallet.type !== 1) {
+            fs_1.default.writeFileSync(file_path, JSON.stringify({ public_key: wallet.public_key, private_key: (0, crypto_1.encrypt)(wallet.private_key), type: 1 }, null, 2));
+        }
+        else {
+            wallet.private_key = (0, crypto_1.decrypt)(wallet.private_key);
+        }
         if (only_pubkey) {
             wallets.push({
                 public_key: wallet.public_key,

package/dist/util/crypto.d.ts

@@ -0,0 +1,2 @@
+export declare function encrypt(plaintext: string): string;
+export declare function decrypt(ciphertext: string): string;

package/dist/util/crypto.js

@@ -0,0 +1,57 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.encrypt = encrypt;
+exports.decrypt = decrypt;
+require('dotenv').config();
+const crypto_1 = __importDefault(require("crypto"));
+function getEncryptionKey() {
+    const envKey = process.env.ENCRYPTION_KEY;
+    if (envKey && envKey.length >= 32) {
+        return envKey.substring(0, 32);
+    }
+    const defaultKey = 'ttd-core-default-key-32bytes!!!!';
+    return defaultKey.substring(0, 32);
+}
+function deriveKey(key) {
+    return crypto_1.default.createHash('sha256').update(key).digest();
+}
+function encrypt(plaintext) {
+    try {
+        const encryptionKey = getEncryptionKey();
+        const key = deriveKey(encryptionKey);
+        const iv = crypto_1.default.randomBytes(16);
+        const cipher = crypto_1.default.createCipheriv('aes-256-gcm', key, iv);
+        let encrypted = cipher.update(plaintext, 'utf8', 'base64');
+        encrypted += cipher.final('base64');
+        const authTag = cipher.getAuthTag();
+        return `${iv.toString('base64')}:${authTag.toString('base64')}:${encrypted}`;
+    }
+    catch (error) {
+        throw new Error(`加密失败: ${error instanceof Error ? error.message : String(error)}`);
+    }
+}
+function decrypt(ciphertext) {
+    try {
+        const parts = ciphertext.split(':');
+        if (parts.length !== 3) {
+            throw new Error('无效的加密格式：应为 iv:authTag:encryptedData');
+        }
+        const [ivBase64, authTagBase64, encryptedBase64] = parts;
+        const iv = Buffer.from(ivBase64, 'base64');
+        const authTag = Buffer.from(authTagBase64, 'base64');
+        const encrypted = encryptedBase64;
+        const encryptionKey = getEncryptionKey();
+        const key = deriveKey(encryptionKey);
+        const decipher = crypto_1.default.createDecipheriv('aes-256-gcm', key, iv);
+        decipher.setAuthTag(authTag);
+        let decrypted = decipher.update(encrypted, 'base64', 'utf8');
+        decrypted += decipher.final('utf8');
+        return decrypted;
+    }
+    catch (error) {
+        throw new Error(`解密失败: ${error instanceof Error ? error.message : String(error)}`);
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@clonegod/ttd-core",
-    "version": "2.1.26",
+    "version": "2.1.28",
     "description": "Common types and utilities for trading systems - use `npm run push` to publish",
     "main": "dist/index.js",
     "types": "types/index.d.ts",

package/types/index.d.ts

@@ -545,6 +545,7 @@ export interface SendTxLogType {
 export interface WalletInfoType {
     public_key: string
     private_key: string
+    type?: number // 1: raw data, 2: encrypted data
 }
 
 export interface WalletTokenBalanceInfoType {