Add support for AENSv2.pointee, AENSv2.name types

aeternity · Jan 25, 2024 · 8e9c8b8 · 8e9c8b8
1 parent 9c1dffc
commit 8e9c8b8
Show file tree

Hide file tree

Showing 10 changed files with 172 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -9,6 +9,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Fixed
 ### Added
 - Add support for `bytes()` arbitrary-sized type from FATE 3.
+- Add support for `AENSv2.pointee`, `AENSv2.name` types from FATE 3.
 ### Changed
 ### Deprecated
 ### Removed
diff --git a/contracts/Test.aes b/contracts/Test.aes
@@ -87,7 +87,9 @@ main contract Test =
     entrypoint test_signature(a:signature) = a
     entrypoint test_ttl(a:Chain.ttl) = a
     entrypoint test_pointee(a:AENS.pointee) = a
+    entrypoint test_pointee_v2(a:AENSv2.pointee) = a
     entrypoint test_aens_name(a:AENS.name) = a
+    entrypoint test_aens_name_v2(a:AENSv2.name) = a
     entrypoint test_ga_meta_tx(a:Chain.ga_meta_tx) = a
     entrypoint test_paying_for_tx(a:Chain.paying_for_tx) = a
     entrypoint test_base_tx(a:Chain.base_tx) = true

diff --git a/src/DataFactory/VariantDataFactory.js b/src/DataFactory/VariantDataFactory.js
@@ -6,7 +6,9 @@ const TYPES = [
     'variant',
     'Chain.ttl',
     'AENS.pointee',
-    'AENS.name'
+    'AENSv2.pointee',
+    'AENS.name',
+    'AENSv2.name'
 ]
 
 class VariantDataFactory extends BaseDataFactory {

diff --git a/src/FateTypes.js b/src/FateTypes.js
@@ -187,6 +187,18 @@ const FateTypeAENSPointee = () => {
     return FateTypeVariant(variants)
 }
 
+const FateTypeAENSv2Pointee = () => {
+    const variants = [
+        {'AENSv2.AccountPt': [FateTypeAccountAddress()]},
+        {'AENSv2.OraclePt': [FateTypeAccountAddress()]},
+        {'AENSv2.ContractPt': [FateTypeAccountAddress()]},
+        {'AENSv2.ChannelPt': [FateTypeAccountAddress()]},
+        {'AENSv2.DataPt': [FateTypeBytes()]}, // max 1024 bytes
+    ]
+
+    return FateTypeVariant(variants)
+}
+
 const FateTypeAENSName = () => {
     const variants = [{
         'AENS.Name': [
@@ -199,6 +211,18 @@ const FateTypeAENSName = () => {
     return FateTypeVariant(variants)
 }
 
+const FateTypeAENSv2Name = () => {
+    const variants = [{
+        'AENSv2.Name': [
+            FateTypeAccountAddress(),
+            FateTypeChainTTL(),
+            FateTypeMap(FateTypeString(), FateTypeAENSv2Pointee())
+        ]
+    }]
+
+    return FateTypeVariant(variants)
+}
+
 const FateTypeEvent = (variantType, topics) => {
     return {
         name: 'event',
@@ -262,7 +286,9 @@ module.exports = {
     FateTypeChainPayingForTx,
     FateTypeChainBaseTx,
     FateTypeAENSPointee,
+    FateTypeAENSv2Pointee,
     FateTypeAENSName,
+    FateTypeAENSv2Name,
     FateTypeEvent,
     FateTypeBls12381Fr,
     FateTypeBls12381Fp,

diff --git a/src/TypeResolver.js b/src/TypeResolver.js
@@ -24,7 +24,9 @@ const {
     FateTypeChainPayingForTx,
     FateTypeChainBaseTx,
     FateTypeAENSPointee,
+    FateTypeAENSv2Pointee,
     FateTypeAENSName,
+    FateTypeAENSv2Name,
     FateTypeBls12381Fr,
     FateTypeBls12381Fp
 } = require('./FateTypes')
@@ -135,10 +137,18 @@ class TypeResolver {
             return FateTypeAENSPointee()
         }
 
+        if (key === 'AENSv2.pointee') {
+            return FateTypeAENSv2Pointee()
+        }
+
         if (key === 'AENS.name') {
             return FateTypeAENSName()
         }
 
+        if (key === 'AENSv2.name') {
+            return FateTypeAENSv2Name()
+        }
+
         if (key === 'Set.set') {
             return FateTypeSet(...resolvedTypes)
         }

diff --git a/tests/Decoder.js b/tests/Decoder.js
@@ -522,6 +522,28 @@ test('Decode singleton record (optimized)', t => {
     )
 })
 
+test('Decode AENSv2.name with DataPr', t => {
+    t.plan(1)
+    t.deepEqual(
+        encoder.decode(
+            CONTRACT,
+            'test_aens_name_v2',
+            'cb_rwMAO58AoGcmh1wvzEWYjmsxUe1udTT9DCHfC0A1YAxTFxGd3/Lar4IBAQEbb4MCv60vAhlvcmFjbGWvhQEBAQEBARufAKBnJodcL8xFmI5rMVHtbnU0/Qwh3wtANWAMUxcRnd/y2iF0ZXN0IGtlea+FAQEBAQEEG58BKXRlc3QgdmFsdWUZ3wys'
+        ),
+        {
+            'AENSv2.Name': [
+                'ak_nRqnePWC6yGWBmR4wfN3AvQnqbv2TizxKJdvGXj8p7YZrUZ5J',
+                { FixedTTL: [180205n] },
+                new Map([
+                    ['oracle', { 'AENSv2.OraclePt': ['ak_nRqnePWC6yGWBmR4wfN3AvQnqbv2TizxKJdvGXj8p7YZrUZ5J'] }],
+                    ['test key', { 'AENSv2.DataPt': [Uint8Array.from(Buffer.from('test value'))] }],
+                ]),
+            ],
+        },
+        'test_aens_name_v2(AENSv2.Name(ak_nRqnePWC6yGWBmR4wfN3AvQnqbv2TizxKJdvGXj8p7YZrUZ5J, FixedTTL(180205), {["oracle"] = AENSv2.OraclePt(ak_nRqnePWC6yGWBmR4wfN3AvQnqbv2TizxKJdvGXj8p7YZrUZ5J), ["test key"] = AENSv2.DataPt(Bytes.to_any_size(#746573742076616c7565))}))'
+    )
+})
+
 test('Decode Set.set', t => {
     t.plan(1)
     t.deepEqual(

diff --git a/tests/Encoder.js b/tests/Encoder.js
@@ -561,6 +561,65 @@ test('Encode AENS.pointee arguments', t => {
     )
 })
 
+test('Encode AENSv2.pointee arguments', t => {
+    t.plan(5)
+    const encoded1 = encoder.encode(
+        CONTRACT,
+        'test_pointee_v2',
+        [{'AENSv2.AccountPt': ['ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt']}]
+    )
+    t.is(
+        encoded1,
+        'cb_KxElnEHfG6+FAQEBAQEAG58AoN5ov+GyA+UfUjUboIf3m3go5qFA8MMUpnDHADs/9XB10byw+g==',
+        'test_pointee_v2(AENSv2.AccountPt(ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt))'
+    )
+
+    const encoded2 = encoder.encode(
+        CONTRACT,
+        'test_pointee_v2',
+        [{'AENSv2.OraclePt': ['ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt']}]
+    )
+    t.is(
+        encoded2,
+        'cb_KxElnEHfG6+FAQEBAQEBG58AoN5ov+GyA+UfUjUboIf3m3go5qFA8MMUpnDHADs/9XB1XO0VzQ==',
+        'test_pointee_v2(AENSv2.OraclePt(ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt))'
+    )
+
+    const encoded3 = encoder.encode(
+        CONTRACT,
+        'test_pointee_v2',
+        [{'AENSv2.ContractPt': ['ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt']}]
+    )
+    t.is(
+        encoded3,
+        'cb_KxElnEHfG6+FAQEBAQECG58AoN5ov+GyA+UfUjUboIf3m3go5qFA8MMUpnDHADs/9XB1UFjoyQ==',
+        'test_pointee_v2(AENSv2.ContractPt(ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt))'
+    )
+
+    const encoded4 = encoder.encode(
+        CONTRACT,
+        'test_pointee_v2',
+        [{'AENSv2.ChannelPt': ['ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt']}]
+    )
+    t.is(
+        encoded4,
+        'cb_KxElnEHfG6+FAQEBAQEDG58AoN5ov+GyA+UfUjUboIf3m3go5qFA8MMUpnDHADs/9XB1547G3Q==',
+        'test_pointee_v2(AENSv2.ChannelPt(ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt))'
+    )
+
+    const dataPt = Buffer.from('http://example.com')
+    const encoded5 = encoder.encode(
+        CONTRACT,
+        'test_pointee_v2',
+        [{'AENSv2.DataPt': [dataPt]}]
+    )
+    t.is(
+        encoded5,
+        'cb_KxElnEHfG6+FAQEBAQEEG58BSWh0dHA6Ly9leGFtcGxlLmNvbYixc9Q=',
+        `test_pointee_v2(AENSv2.DataPt(Bytes.to_any_size(#${dataPt.toString('hex')})))`
+    )
+})
+
 test('Encode AENS.name arguments', t => {
     t.plan(1)
     const encoded1 = encoder.encode(
@@ -585,6 +644,30 @@ test('Encode AENS.name arguments', t => {
     )
 })
 
+test('Encode AENSv2.name arguments', t => {
+    t.plan(1)
+    const encoded1 = encoder.encode(
+        CONTRACT,
+        'test_aens_name_v2',
+        [
+            {
+                'AENSv2.Name': [
+                    'ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt',
+                    {FixedTTL: [100]},
+                    new Map([
+                        ["pt1", {'AENSv2.AccountPt': ['ak_2dATVcZ9KJU5a8hdsVtTv21pYiGWiPbmVcU1Pz72FFqpk9pSRR']}]
+                    ])
+                ]
+            }
+        ]
+    )
+    t.is(
+        encoded1,
+        'cb_KxFjVV5kG68DADufAKDeaL/hsgPlH1I1G6CH95t4KOahQPDDFKZwxwA7P/Vwda+CAQEBG28kLwENcHQxr4UBAQEBAQAbnwCg1c8IQA6YgiLybrSwLI+JB3RXRnIRpubZVe23B0nGoztVU63L',
+        'test_aens_name_v2(AENSv2.Name(ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt, RelativeTTL(100), {["pt1"] = AENSv2.AccountPt(ak_2dATVcZ9KJU5a8hdsVtTv21pYiGWiPbmVcU1Pz72FFqpk9pSRR)}))'
+    )
+})
+
 test('Encode Chain.ga_meta_tx arguments', t => {
     t.plan(1)
     const encoded1 = encoder.encode(

diff --git a/tests/TypeResolver.js b/tests/TypeResolver.js
@@ -25,7 +25,9 @@ const {
     FateTypeChainPayingForTx,
     FateTypeChainBaseTx,
     FateTypeAENSPointee,
+    FateTypeAENSv2Pointee,
     FateTypeAENSName,
+    FateTypeAENSv2Name,
     FateTypeSet,
     FateTypeBls12381Fr,
     FateTypeBls12381Fp,
@@ -336,6 +338,14 @@ test('Resolve AENS.pointee', t => {
     )
 })
 
+test('Resolve AENSv2.pointee', t => {
+    t.plan(1)
+    t.deepEqual(
+        resolver.resolveType('AENSv2.pointee'),
+        FateTypeAENSv2Pointee()
+    )
+})
+
 test('Resolve AENS.name', t => {
     t.plan(1)
     t.deepEqual(
@@ -344,6 +354,14 @@ test('Resolve AENS.name', t => {
     )
 })
 
+test('Resolve AENSv2.name', t => {
+    t.plan(1)
+    t.deepEqual(
+        resolver.resolveType('AENSv2.name'),
+        FateTypeAENSv2Name()
+    )
+})
+
 test('Resolve Set.set', t => {
     t.plan(1)
     t.deepEqual(

diff --git a/tests/integration/decoder.sh b/tests/integration/decoder.sh
@@ -62,6 +62,7 @@ test_decoder 'test_primitives_tuple' 'cb_ewL/EXRlc3RPAJ8BCb7vnwGBAAECAwQFBgcICQo
 test_decoder 'test_addresses_tuple' 'cb_S58AoN5ov+GyA+UfUjUboIf3m3go5qFA8MMUpnDHADs/9XB1nwKgH8DQmexaE8uTKKMX/OzYUrH3SJ5eALoJVzw8LbaYVVOfA6DK8iokTtrAPSbwKhfZI5QtBVz8hiMoslpRwoS8nUIOSZ8EoO0e59wCeNBc6VCUc8UKk9Q6KLKNUDL2332qRC/hNxNI/pLnKw=='
 test_decoder 'test_complex_tuple' 'cb_WysCAq+EAAABAAIbBjMCBAYvAgIEBggrCgyRsE4R'
 test_decoder 'test_set' 'cb_LwQKPxA/Gj8qPwZjWoo='
+test_decoder 'test_aens_name_v2' 'cb_rwMAO58AoGcmh1wvzEWYjmsxUe1udTT9DCHfC0A1YAxTFxGd3/Lar4IBAQEbb4MCv60vAhlvcmFjbGWvhQEBAQEBARufAKBnJodcL8xFmI5rMVHtbnU0/Qwh3wtANWAMUxcRnd/y2iF0ZXN0IGtlea+FAQEBAQEEG58BKXRlc3QgdmFsdWUZ3wys'
 # https://github.com/aeternity/aesophia_cli/issues/61
 # test_decoder 'test_bls12_381_fr' 'cb_nwGBxHFEFDqOu+s9/hWhDQxmnzSFvfQ6dkSwETqb5Rgy7EblqaTu'
 # test_decoder 'test_bls12_381_fp' 'cb_nwHBA/Xt2rba+aow52i/Vr/nGSDbg1ErD/oOVs0LOV1F83sEMEfueZoY+Ng3idzNLTcU1fPMEQ=='

diff --git a/tests/integration/encoder.sh b/tests/integration/encoder.sh
@@ -70,7 +70,13 @@ test_encoder 'test_pointee(AENS.AccountPt(ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvB
 test_encoder 'test_pointee(AENS.OraclePt(ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt))'
 test_encoder 'test_pointee(AENS.ContractPt(ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt))'
 test_encoder 'test_pointee(AENS.ChannelPt(ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt))'
+test_encoder 'test_pointee_v2(AENSv2.AccountPt(ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt))'
+test_encoder 'test_pointee_v2(AENSv2.OraclePt(ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt))'
+test_encoder 'test_pointee_v2(AENSv2.ContractPt(ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt))'
+test_encoder 'test_pointee_v2(AENSv2.ChannelPt(ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt))'
+test_encoder 'test_pointee_v2(AENSv2.DataPt(Bytes.to_any_size(#687474703a2f2f6578616d706c652e636f6d)))'
 test_encoder 'test_aens_name(AENS.Name(ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt, FixedTTL(100), {["pt1"] = AENS.AccountPt(ak_2dATVcZ9KJU5a8hdsVtTv21pYiGWiPbmVcU1Pz72FFqpk9pSRR)}))'
+test_encoder 'test_aens_name_v2(AENSv2.Name(ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt, FixedTTL(100), {["pt1"] = AENSv2.AccountPt(ak_2dATVcZ9KJU5a8hdsVtTv21pYiGWiPbmVcU1Pz72FFqpk9pSRR)}))'
 test_encoder 'test_ga_meta_tx(Chain.GAMetaTx(ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt, 42))'
 test_encoder 'test_paying_for_tx(Chain.PayingForTx(ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt, 42))'
 test_encoder 'test_base_tx(Chain.SpendTx(ak_2gx9MEFxKvY9vMG5YnqnXWv1hCsX7rgnfvBLJS4aQurustR1rt, 42, "foo"))'