From 78640a804ebc65e367b836fc1397f9e3d6ec78dd Mon Sep 17 00:00:00 2001
From: David C Goldenberg <dgoldenberg@mitre.org>
Date: Tue, 29 Oct 2024 15:48:35 -0400
Subject: [PATCH] Update: Added a ETH to TRX conversion operation.

---
 src/core/operations/ETHTRXConversion.mjs    | 100 ++++++++++++++++++++
 tests/operations/tests/ETHTRXConversion.mjs |  48 ++++++++++
 2 files changed, 148 insertions(+)
 create mode 100644 src/core/operations/ETHTRXConversion.mjs
 create mode 100644 tests/operations/tests/ETHTRXConversion.mjs

diff --git a/src/core/operations/ETHTRXConversion.mjs b/src/core/operations/ETHTRXConversion.mjs
new file mode 100644
index 00000000..67f97f1e
--- /dev/null
+++ b/src/core/operations/ETHTRXConversion.mjs
@@ -0,0 +1,100 @@
+/**
+ * @author dgoldenberg [virtualcurrency@mitre.org]
+ * @copyright Crown Copyright 2024
+ * @license Apache-2.0
+ */
+
+import Operation from "../Operation.mjs";
+import {base58Decode, base58Encode, doubleSHA} from "../lib/Bitcoin.mjs";
+import Utils from "../Utils.mjs";
+import {toHex} from "crypto-api/src/encoder/hex.mjs";
+import { fromArrayBuffer } from "crypto-api/src/encoder/array-buffer.mjs";
+import JSSHA3 from "js-sha3";
+
+/**
+ * Turns an address to a checksummed address.
+ * @param {*} address ETH style address.
+ */
+function ethCheckSum(address) {
+    const lowerCaseAddress = address.toLowerCase();
+    const trimmedAddress = lowerCaseAddress.slice(2,);
+    const algo = JSSHA3.keccak256;
+    const hashResult = algo(Utils.convertToByteArray(trimmedAddress, "utf8"));
+    let finalResult = "";
+    for (let i = 0; i < trimmedAddress.length; i++) {
+        if (trimmedAddress.charAt(i).toLowerCase() >= "a" && trimmedAddress.charAt(i).toLowerCase() <= "f") {
+            if (hashResult.charAt(i)>= "8") {
+                finalResult += trimmedAddress.charAt(i).toUpperCase();
+            } else {
+                finalResult += trimmedAddress.charAt(i);
+            }
+        } else {
+            finalResult += trimmedAddress.charAt(i);
+        }
+    }
+    return "0x" + finalResult;
+}
+
+/**
+ * ETH / TRX Conversion operation
+ */
+class ETHTRXConversion extends Operation {
+
+    /**
+     * ETHTRXConversion constructor
+     */
+    constructor() {
+        super();
+
+        this.name = "ETH / TRX Conversion";
+        this.module = "Default";
+        this.description = "Converts between ETH and TRX addresses.";
+        this.infoURL = "https://tronscan.org/#/tools/code-converter/tron-ethereum-address"; // Usually a Wikipedia link. Remember to remove localisation (i.e. https://wikipedia.org/etc rather than https://en.wikipedia.org/etc)
+        this.inputType = "string";
+        this.outputType = "string";
+        this.args = [
+            {
+                name: "Direction",
+                type: "option",
+                value: ["ETH->TRX", "TRX->ETH"]
+            }
+            /* Example arguments. See the project wiki for full details.
+            {
+                name: "First arg",
+                type: "string",
+                value: "Don't Panic"
+            },
+            {
+                name: "Second arg",
+                type: "number",
+                value: 42
+            }
+            */
+        ];
+    }
+
+    /**
+     * @param {string} input
+     * @param {Object[]} args
+     * @returns {string}
+     */
+    run(input, args) {
+        const [direction] = args;
+        switch (direction) {
+            case "ETH->TRX":{
+                const unencodedAddress =  input.slice(2,);
+                const checksumHash = toHex(doubleSHA(fromArrayBuffer(Utils.convertToByteArray("41" + unencodedAddress, "hex"))));
+                const finalString = "41" + unencodedAddress + checksumHash.slice(0, 8);
+                const address = base58Encode(Utils.convertToByteArray(finalString, "hex"));
+                return address;
+            }
+            case "TRX->ETH":{
+                return ethCheckSum("0x" + toHex(fromArrayBuffer(base58Decode(input).slice(1, -4))));
+            }
+
+        }
+    }
+
+}
+
+export default ETHTRXConversion;
diff --git a/tests/operations/tests/ETHTRXConversion.mjs b/tests/operations/tests/ETHTRXConversion.mjs
new file mode 100644
index 00000000..3cc368d0
--- /dev/null
+++ b/tests/operations/tests/ETHTRXConversion.mjs
@@ -0,0 +1,48 @@
+/**
+ * Public Key to TRX Style Address Cryptocurrency Address tests.
+ *
+ * @author dgoldenberg [virtualcurrency@mitre.org]
+ * @copyright  MITRE 2023
+ * @license Apache-2.0
+ */
+
+import TestRegister from "../../lib/TestRegister.mjs";
+
+TestRegister.addTests([
+    {
+        name: "ETH To TRX",
+        input: "0xAb495f468B0bF116860E76be9f26339b2CbC33c3",
+        expectedOutput: "TRatSrEqUKLK4eNWL8NSrvEbpeht5takWd",
+        recipeConfig: [
+            {
+                "op": "ETH / TRX Conversion",
+                "args": ["ETH->TRX"]
+            },
+        ],
+
+    },
+    {
+        name: "TRX to ETH",
+        input: "TRatSrEqUKLK4eNWL8NSrvEbpeht5takWd",
+        expectedOutput: "0xAb495f468B0bF116860E76be9f26339b2CbC33c3",
+        recipeConfig: [
+            {
+                "op": "ETH / TRX Conversion",
+                "args": ["TRX->ETH"]
+            },
+        ],
+
+    },
+    {
+        name: "TRX to ETH With Checksum",
+        input: "TF8xufByTvQmf5gssBzosfmCFoJyJpzZtj",
+        expectedOutput: "0x38b10632a1F3B4363cF1852D12ff492808aA3675",
+        recipeConfig: [
+            {
+                "op": "ETH / TRX Conversion",
+                "args": ["TRX->ETH"]
+            },
+        ]
+
+    },
+]);