feat: add vector as output slice (#812)

* setup

* adjust rough draft

* cs

* use as

* use current code

---------

Co-authored-by: Anderson Arboleya <[email protected]>
Co-authored-by: Dhaiwat <[email protected]>

Author: 3 people

.changeset/olive-cows-accept.md

@@ -0,0 +1,5 @@
+---
+"@fuel-ts/abi-coder": patch
+---
+
+Adjust to support vector output

packages/abi-coder/src/abi-coder.ts

@@ -69,6 +69,12 @@ export default class AbiCoder {
       return new ArrayCoder(itemCoder, length);
     }
 
+    if (['raw untyped slice'].includes(param.type)) {
+      const length = 0;
+      const itemCoder = this.getCoder({ type: 'u64' });
+      return new ArrayCoder(itemCoder, length);
+    }
+
     const stringMatch = stringRegEx.exec(param.type)?.groups;
     if (stringMatch) {
       const length = parseInt(stringMatch.length, 10);
@@ -188,6 +194,9 @@ export default class AbiCoder {
     }
 
     const coders = nonEmptyTypes.map((type) => this.getCoder(type));
+    if (nonEmptyTypes[0] && nonEmptyTypes[0].type === 'raw untyped slice') {
+      (coders[0] as ArrayCoder<U64Coder>).length = bytes.length / 8;
+    }
     const coder = new TupleCoder(coders);
     const [decoded, newOffset] = coder.decode(bytes, 0);
 

packages/fuel-gauge/src/coverage-contract.test.ts

@@ -448,4 +448,22 @@ describe('Coverage Contract', () => {
     expect([logs[3], logs[4], logs[5]]).toEqual([1, 2, 3]);
     // #endregion
   });
+
+  it('should get raw_slice output [u8]', async () => {
+    const { value } = await contractInstance.functions.echo_u8_vector([100, 2, 1, 2, 3]).call();
+
+    expect(value.map((v: BN) => v.toNumber())).toStrictEqual([100, 2, 1, 2, 3]);
+  });
+
+  it('should get raw_slice output [u64]', async () => {
+    const { value } = await contractInstance.functions.echo_u64_vector([100, 2, 1, 2, 3]).call();
+
+    expect(value.map((v: BN) => v.toNumber())).toStrictEqual([100, 2, 1, 2, 3]);
+  });
+
+  it('should get raw_slice output', async () => {
+    const { value } = await contractInstance.functions.get_u64_vector().call();
+
+    expect(value.map((v: BN) => v.toNumber())).toStrictEqual([1, 2, 3]);
+  });
 });

packages/fuel-gauge/test-projects/coverage-contract/src/main.sw

@@ -84,14 +84,16 @@ abi CoverageContract {
     fn echo_option_u8(input: Option<u8>) -> Option<u8>;
     fn echo_option_extract_u32(input: Option<u32>) -> u32;
     fn echo_option_three_u8(inputA: Option<u8>, inputB: Option<u8>, inputC: Option<u8>) -> u8;
-    fn echo_u8_vector(input: Vec<u8>) -> Vec<u8>;
     fn echo_u8_vector_first(vector: Vec<u8>) -> u8;
     fn echo_u8_option_vector_first(vector: Vec<Option<u8>>) -> u8;
     fn echo_u64_vector_last(vector: Vec<u64>) -> u64;
     fn echo_u32_vector_addition_other_type(vector: Vec<u32>, input: u32) -> u32;
     fn echo_u32_vector_addition(vector_1: Vec<u32>, vector_2: Vec<u32>) -> u32;
     fn echo_struct_vector_first(vector: Vec<BigStruct>) -> BigStruct;
     fn echo_struct_vector_last(vector: Vec<ComplexStruct>) -> ComplexStruct;
+    fn get_u64_vector() -> raw_slice;
+    fn echo_u8_vector(input: Vec<u8>) -> raw_slice;
+    fn echo_u64_vector(input: Vec<u64>) -> raw_slice;
 }
 
 impl CoverageContract for Contract {
@@ -273,9 +275,6 @@ impl CoverageContract for Contract {
         value1 + value2 + value3
     }
     // #endregion
-    fn echo_u8_vector(input: Vec<u8>) -> Vec<u8> {
-        input
-    }
 
     fn echo_u8_vector_first(vector: Vec<u8>) -> u8 {
         match vector.get(0) {
@@ -320,4 +319,24 @@ impl CoverageContract for Contract {
         vector.get(vector.len() - 1).unwrap()
     }
      // #endregion
+
+    fn get_u64_vector() -> raw_slice {
+        // Convert to a vector
+        let mut vec: Vec<u64> = Vec::new();
+
+        vec.push(1);
+        vec.push(2);
+        vec.push(3);
+
+        // Return it
+        vec.as_raw_slice()
+    }
+
+    fn echo_u8_vector(input: Vec<u8>) -> raw_slice {
+        input.as_raw_slice()
+    }
+
+    fn echo_u64_vector(input: Vec<u64>) -> raw_slice {
+        input.as_raw_slice()
+    }
 }