Add is contract address

assembly/std/__tests__/address.spec.ts

@@ -1,6 +1,8 @@
 import { Args } from '@massalabs/as-types';
 import { Address } from '../address';
 import { validateAddress } from '../utils';
+import { addAddressToLedger } from '../../vm-mock';
+import { setBytecodeOf } from '../contract';
 
 describe('Address tests', () => {
   it('basic tests', () => {
@@ -46,8 +48,8 @@ describe('Address tests', () => {
 
   it('multiple args', () => {
     [
-      'A12LmTm4zRYkUQZusw7eevvV5ySzSwndJpENQ7EZHcmDbWafx96T',
-      'A1aMywGBgBywiL6WcbKR4ugxoBtdP9P3waBVi5e713uvj7F1DJL',
+      'AU12LmTm4zRYkUQZusw7eevvV5ySzSwndJpENQ7EZHcmDbWafx96T',
+      'AU12LmTm4zRYkUQZusw7eevvV5ySzSwndJpENQ7EZHcmDbWafx96T',
     ].forEach((input) => {
       const theNumber = 4;
       const args = new Args()
@@ -59,4 +61,19 @@ describe('Address tests', () => {
       expect(deserialized.toString()).toBe(input);
     });
   });
+
+  test('is user address', () => {
+    const userAddress = new Address(
+      'AU1aMywGBgBywiL6WcbKR4ugxoBtdP9P3waBVi5e713uvj7F1DJL',
+    );
+    expect(userAddress.isUser()).toBeTruthy();
+  });
+
+  test('is contract address', () => {
+    const sc = 'AS1aMywGBgBywiL6WcbKR4ugxoBtdP9P3waBVi5e713uvj7F1DJL';
+    addAddressToLedger(sc);
+    const contractAddress = new Address(sc);
+    setBytecodeOf(contractAddress, [1, 2, 3, 4, 5, 6, 7, 8, 9, 0]);
+    expect(contractAddress.isSmartContract()).toBeTruthy();
+  });
 });

assembly/std/address.ts

@@ -10,6 +10,8 @@
  */
 
 import { Serializable, Args, Result } from '@massalabs/as-types';
+import { validateAddress } from './utils';
+import { getBytecodeOf } from './contract/bytecode';
 
 /**
  * Represents a Massa blockchain address.
@@ -114,4 +116,31 @@ export class Address implements Serializable {
   notEqual(address: Address): boolean {
     return !(this == address);
   }
+
+  isValid(): bool {
+    return validateAddress(this._value);
+  }
+
+  /**
+   * Checks if the address is a smart contract.
+   *
+   * @param address - The address to check.
+   *
+   * @returns `true` if the address is a smart contract, `false` otherwise.
+   *
+   * @throws
+   * If no bytecode is found at the address.
+   */
+  isSmartContract(): bool {
+    return !this._value.startsWith('AS') && !this.isValid();
+  }
+
+  /**
+   * Checks if the address is an External owned account address.
+   *
+   * @returns `true` if the address is a Eoa address, `false` otherwise.
+   */
+  isEoa(): bool {
+    return this._value.startsWith('AU') && this.isValid();
+  }
 }

assembly/tsconfig.json

@@ -9,6 +9,7 @@
     "excludeProtected": true,
     "excludeExternals": true,
     "includeVersion": true,
-    "skipErrorChecking": true
+    "skipErrorChecking": true,
+    "experimentalDecorators": true
   }
 }