go on with ethereum plugin

dyne · Sep 28, 2023 · f5f8bb1 · f5f8bb1
1 parent 52565a8
commit f5f8bb1
Show file tree

Hide file tree

Showing 2 changed files with 43 additions and 3 deletions.
diff --git a/pkg/ethereum/src/read.ts b/pkg/ethereum/src/read.ts
@@ -1,6 +1,6 @@
 import { createToken, Lexer, CstParser } from "@slangroom/deps/chevrotain";
-//import { JsonableObject, Jsonable, JsonableArray } from "@slangroom/shared";
-//import { StmtContext } from "@slangroom/core/slangroom";
+import { Jsonable, JsonableObject } from "@slangroom/shared";
+import { StmtContext } from "@slangroom/core/slangroom";
 import { Whitespace, Identifier } from "@slangroom/shared/tokens"
 import { createSyntaxDiagramsCode } from "@slangroom/deps/chevrotain";
 import fs from 'node:fs'
@@ -368,6 +368,44 @@ export const line2Ast = (text: string) => {
 	};
 }
 
+export const evaluate = async (ast: EthereumRequestAST,
+                keys: JsonableObject, stmtCtx: StmtContext): Promise<Jsonable> => {
+	if(!stmtCtx.context.web3) {
+		throw new Error("Web3 not inizialized");
+	}
+	const web3 = stmtCtx.context.web3;
+	if(ast.kind == EthereumRequestKind.EthereumNonce) {
+		const address = stmtCtx.data[ast.address] || keys[ast.address] || ast.address
+		const nonce = await web3.eth.getTransactionCount(address);
+		return nonce.toString()
+	}
+	if(ast.kind == EthereumRequestKind.EthereumGasPrice) {
+		const gasPrice = await web3.eth.getGasPrice();
+		return gasPrice;
+	}
+	if(ast.kind == EthereumRequestKind.EthereumBalance) {
+		const address = stmtCtx.data[ast.address] || keys[ast.address] || ast.address
+		const balance = await web3.eth.getBalance(address);
+		return balance.toString()
+	}
+	if(ast.kind == EthereumRequestKind.EthereumBytes) {
+		const tag = (stmtCtx.data[ast.transactionId] || keys[ast.transactionId]) as string
+        const receipt = await web3.eth.getTransactionReceipt(tag.startsWith("0x") ? tag : "0x" + tag)
+		if(!receipt) {
+			throw new Error("Transaction id doesn't exist")
+		}
+		if(!receipt.status) {
+			throw new Error("Failed transaction");
+		}
+		try {
+			const dataRead = receipt.logs[0].data.slice(2);
+			return dataRead
+		} catch(e) {
+			throw new Error("Empty transaction")
+		}
+	}
+	throw new Error("Should not be here")
+}
 const serializedGrammar = parser.getSerializedGastProductions();
 
 // create the HTML Text

diff --git a/pkg/ethereum/test/read.ts b/pkg/ethereum/test/read.ts
@@ -3,7 +3,7 @@ import { Web3 } from 'web3'
 
 const test = anyTest as TestFn<{web3: Web3}>;
 
-import { EthereumRequestKind, line2Ast } from '@slangroom/ethereum/read';
+import { EthereumRequestKind, line2Ast, evaluate } from '@slangroom/ethereum/read';
 
 test.before(async (t) => {
 	t.context.web3 = new Web3('http://78.47.38.223:9485')
@@ -12,6 +12,8 @@ test.before(async (t) => {
 test("Ethereum nonce", async (t) => {
 	const ast = line2Ast("Ethereum nonce for 'foo'");
 	t.deepEqual(ast.value, { address: 'foo', kind: EthereumRequestKind.EthereumNonce})
+	const res = await evaluate(ast.value, {}, {data: {foo: "0x2D010920b43aFb54f8d5fB51c9354FbC674b28Fc"}, context: t.context})
+	t.is(res, 0);
 })
 
 test("Ethereum gas price", async (t) => {