encifher-health 1.0.0

package/index.js

@@ -0,0 +1,178 @@
+const fs = require('fs');
+const { ethers } = require('ethers');
+const { TEEClient, PlaintextType } = require('@encifher-js/core');
+
+const defaultConfig = {
+    network: "baseSepolia",
+    eusdc: "0x6Bc109e6187b98B279BbbdBDdc59021dc59A5058",
+    usdc: "0x7A8cb5C885597398F7745c0403094705a4AEb1Df",
+    eusdcWrapper: "0xC97100472255aB79b7858139056f2cbA37f67252",
+    orderManager: "0x2602931AF6848e7e657b2840c1E8507C112c6129",
+    faucet: "0x2381aa52cFB6eB752d72F4A611ec43CC718ed206",
+    customAddress: "0x42ed70a51d0f4ff03ac3085c3021fc79c332f5a0"
+};
+let config = defaultConfig;
+const args = process.argv.slice(2);
+let customAddressArg = null;
+if (args.includes('--custom')) {
+    const customIndex = args.indexOf('--custom') + 1;
+    if (customIndex < args.length) {
+        customAddressArg = args[customIndex];
+    } else {
+        console.error('No custom address provided after --custom flag');
+        process.exit(1);
+    }
+}
+let configFilePath = args.find(arg => !arg.startsWith('--'));
+if (configFilePath) {
+    if (fs.existsSync(configFilePath)) {
+        try {
+            const configFile = fs.readFileSync(configFilePath, 'utf-8');
+            config = JSON.parse(configFile);
+        } catch (error) {
+            console.error('Error reading or parsing config file:', error);
+            process.exit(1);
+        }
+    } else {
+        console.error(`Config file not found at: ${configFilePath}. Using default config.`);
+    }
+}
+if (customAddressArg) {
+    config.customAddress = customAddressArg;
+}
+const { eusdc, usdc, eusdcWrapper, orderManager, faucet, customAddress } = config;
+const networks = {
+    baseSepolia: 'https://sepolia.base.org',
+}
+const network = networks[config.network];
+
+class ColoredLogger {
+    static colorText(text, colorCode) {
+        const colors = {
+            reset: '\x1b[0m',
+            red: '\x1b[31m',
+            green: '\x1b[32m',
+            yellow: '\x1b[33m',
+            cyan: '\x1b[36m',
+            orange: '\x1b[38;5;196m',
+        };
+        const color = colors[colorCode] || colors.reset;
+        return color + text + colors.reset;
+    }
+
+    static log(message, color) {
+        const coloredMessage = this.colorText(message, color);
+        console.log(coloredMessage);
+    }
+}
+
+async function requestDecrypt(handle) {
+    try {
+        const response = await fetch('https://decrypt.rpc.encifher.io/decrypt', {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json'
+            },
+            body: JSON.stringify({ handle: BigInt(handle).toString() })
+        });
+        const result = await response.json();
+        return result;
+    } catch (error) {
+        return "error";
+    }
+}
+
+async function logBalance(address, onlyToken = true) {
+    if (!address) {
+        ColoredLogger.log("no address given. Skipping...", 'red')
+        return;
+    }
+    console.log('Address:', address)
+    const provider = new ethers.JsonRpcProvider(network);
+    try {
+        if (!onlyToken) {
+            const balanceWei = await provider.getBalance(address);
+            const balanceEther = (BigInt(balanceWei).toString() / "1e18").toString();
+            ColoredLogger.log(`ETH Balance: ${balanceEther} ETH`, 'orange');
+        }
+        const eusdcContract = new ethers.Contract(
+            eusdc,
+            [
+                "function balanceOf(address) view returns (uint256)"
+            ],
+            provider
+        );
+        const balanceEusdc = await eusdcContract.balanceOf(address);
+        ColoredLogger.log(`eUSDC balance handle: ${BigInt(balanceEusdc).toString()}`, 'cyan');
+        const decrypted = await requestDecrypt(balanceEusdc);
+        if (decrypted === "error") {
+            ColoredLogger.log(`Decrypted balance: error`, 'red');
+            return;
+        }
+        const formatted = (decrypted?.toString() / "1e6");
+        ColoredLogger.log(`Decrypted balance: ${formatted} eusdc`, 'green');
+    } catch (error) {
+        console.error("Error fetching balance:", error);
+    }
+}
+
+async function logUsdcBalance(address) {
+    if (!address) {
+        ColoredLogger.log("no address given. Skipping...", 'red')
+        return;
+    }
+    console.log('Address:', address)
+    const provider = new ethers.JsonRpcProvider(network);
+    try {
+        const usdcContract = new ethers.Contract(
+            usdc,
+            [
+                "function balanceOf(address) view returns (uint256)"
+            ],
+            provider
+        );
+        const balanceUsdc = await usdcContract.balanceOf(address);
+        const formatted = (balanceUsdc?.toString() / "1e6");
+        ColoredLogger.log(`Decrypted balance: ${formatted} usdc`, 'green');
+    } catch (error) {
+        console.error("Error fetching balance:", error);
+    }
+}
+
+async function pingPong() {
+    const client = new TEEClient();
+    const handle = await client.encrypt(1, PlaintextType.uint32)
+    if (!handle) {
+        ColoredLogger.log("Encryption server down", 'red');
+        return;
+    }
+    const decrypted = await requestDecrypt('0x' + Buffer.from(handle).toString('hex'));
+    if (decrypted === "error") {
+        ColoredLogger.log("Decryption server down", 'red');
+        return;
+    }
+    ColoredLogger.log(`Encryption server is healthy`, 'green');
+    ColoredLogger.log(`Decryption server is healthy`, 'green');
+}
+
+async function main() {
+    try {
+        console.log('\n------------------------------------');
+        ColoredLogger.log('Faucet health', 'yellow')
+        await logBalance(faucet, false);
+        ColoredLogger.log('\nOrder manager health', 'yellow')
+        await logBalance(orderManager);
+        ColoredLogger.log('\nCustom address health', 'yellow')
+        await logBalance(customAddress, false);
+        ColoredLogger.log('\nEUSDC wrapper health', 'yellow')
+        await logUsdcBalance(eusdcWrapper);
+        ColoredLogger.log('\nCoprocessor health', 'yellow')
+        await pingPong();
+        console.log('------------------------------------');
+        console.log('\n');
+    } catch (error) {
+        console.error(error);
+    }
+}
+
+main();

package/package.json

@@ -0,0 +1,15 @@
+{
+  "name": "encifher-health",
+  "version": "1.0.0",
+  "main": "index.js",
+  "scripts": {
+    "start": "node index.js"
+  },
+  "author": "human",
+  "license": "ISC",
+  "description": "",
+  "dependencies": {
+    "@encifher-js/core": "^1.1.1",
+    "ethers": "^6.13.4"
+  }
+}