Merge pull request #313 from massalabs/feature/chainid

Add chainId external function

README.md

@@ -1,6 +1,6 @@
 # Massa-as-sdk
 
-![check-code-coverage](https://img.shields.io/badge/coverage-83%%25-red)
+![check-code-coverage](https://img.shields.io/badge/coverage-82%25-green)
 
 Massa-as-sdk is a collection of tools, objects, and functions specifically designed for Massa smart contracts in AssemblyScript. This SDK enables you to import object classes, such as address and storage objects, and use them without having to write them from scratch every time. Additionally, it allows you to use Massa's ABI functions.
 

assembly/env/env.ts

@@ -222,4 +222,7 @@ export namespace env {
 
   @external("massa", "assembly_script_validate_address")
   export declare function validateAddress(address: string): bool;
+
+  @external("massa", "assembly_script_chain_id")
+  export declare function chainId(): u64;
 }

assembly/std/context.ts

@@ -209,3 +209,12 @@ export function currentPeriod(): u64 {
 export function currentThread(): u8 {
   return env.currentThread();
 }
+
+/**
+ * Retrieve the current chain id
+ *
+ * @returns the current chain id.
+ */
+export function chainId(): u64 {
+  return env.chainId();
+}

assembly/vm-mock/env.ts

@@ -19,6 +19,9 @@
 @external("massa", "assembly_script_mock_call")
 export declare function mockScCall(value: StaticArray<u8>): void;
 
+@external("massa", "assembly_script_set_chain_id")
+export declare function mockSetChainId(value: number): void;
+
 /**
  * Add a new smart contract address to the ledger
  *

vm-mock/vm.js

@@ -87,6 +87,7 @@ function resetLedger() {
 let webModule;
 
 const scCallMockStack = [];
+let chainIdMock = BigInt(77658366); // Default value, chain id for chain id
 
 /**
  * Create a mock vm to simulate calls and responses of Massa WebAssembly sdk.
@@ -774,6 +775,12 @@ export default function createMockedABI(
         const hash = sha3.keccak256.arrayBuffer(data);
         return newArrayBuffer(hash);
       },
+      assembly_script_set_chain_id(value) {
+        chainIdMock = value;
+      },
+      assembly_script_chain_id() {
+        return chainIdMock;
+      },
     },
   };