git.delta.rocks / unique-network / refs/commits / 1fe93b2e72cb

difftreelog

fix PR

Trubnikov Sergey2022-12-16parent: #ea778c4.patch.diff
in: master

4 files changed

modifiedpallets/nonfungible/src/erc.rsdiffbeforeafterboth
--- a/pallets/nonfungible/src/erc.rs
+++ b/pallets/nonfungible/src/erc.rs
@@ -90,6 +90,7 @@
 	/// @notice Set permissions for token property.
 	/// @dev Throws error if `msg.sender` is not admin or owner of the collection.
 	/// @param permissions Permissions for keys.
+	#[weight(<SelfWeightOf<T>>::set_token_property_permissions(permissions.len() as u32))]
 	fn set_token_property_permissions(
 		&mut self,
 		caller: caller,
@@ -98,7 +99,7 @@
 		let caller = T::CrossAccountId::from_eth(caller);
 		const PERMISSIONS_FIELDS_COUNT: usize = 3;
 
-		let mut perms = <Vec<_>>::new();
+		let mut perms = Vec::new();
 
 		for (key, pp) in permissions {
 			if pp.len() > PERMISSIONS_FIELDS_COUNT {
@@ -112,11 +113,7 @@
 				.into());
 			}
 
-			let mut token_permission = PropertyPermission {
-				mutable: false,
-				collection_admin: false,
-				token_owner: false,
-			};
+			let mut token_permission = PropertyPermission::default();
 
 			for (perm, value) in pp {
 				match perm {
@@ -129,9 +126,7 @@
 			}
 
 			perms.push(PropertyKeyPermission {
-				key: <Vec<u8>>::from(key)
-					.try_into()
-					.map_err(|_| "too long key")?,
+				key: key.into_bytes().try_into().map_err(|_| "too long key")?,
 				permission: token_permission,
 			});
 		}
@@ -144,17 +139,19 @@
 	fn token_property_permissions(
 		&self,
 	) -> Result<Vec<(string, Vec<(EthTokenPermissions, bool)>)>> {
-		let mut res = <Vec<_>>::new();
-		for (key, pp) in <Pallet<T>>::token_property_permission(self.id) {
-			let key = string::from_utf8(key.into_inner()).unwrap();
-			let pp = vec![
-				(EthTokenPermissions::Mutable, pp.mutable),
-				(EthTokenPermissions::TokenOwner, pp.token_owner),
-				(EthTokenPermissions::CollectionAdmin, pp.collection_admin),
-			];
-			res.push((key, pp));
-		}
-		Ok(res)
+		let perms = <Pallet<T>>::token_property_permission(self.id);
+		Ok(perms
+			.into_iter()
+			.map(|(key, pp)| {
+				let key = string::from_utf8(key.into_inner()).expect("Stored key must be valid");
+				let pp = vec![
+					(EthTokenPermissions::Mutable, pp.mutable),
+					(EthTokenPermissions::TokenOwner, pp.token_owner),
+					(EthTokenPermissions::CollectionAdmin, pp.collection_admin),
+				];
+				(key, pp)
+			})
+			.collect())
 	}
 
 	/// @notice Set token property value.
modifiedpallets/refungible/src/erc.rsdiffbeforeafterboth
--- a/pallets/refungible/src/erc.rs
+++ b/pallets/refungible/src/erc.rs
@@ -93,6 +93,7 @@
 	/// @notice Set permissions for token property.
 	/// @dev Throws error if `msg.sender` is not admin or owner of the collection.
 	/// @param permissions Permissions for keys.
+	#[weight(<SelfWeightOf<T>>::set_token_property_permissions(permissions.len() as u32))]
 	fn set_token_property_permissions(
 		&mut self,
 		caller: caller,
@@ -101,7 +102,7 @@
 		let caller = T::CrossAccountId::from_eth(caller);
 		const PERMISSIONS_FIELDS_COUNT: usize = 3;
 
-		let mut perms = <Vec<_>>::new();
+		let mut perms = Vec::new();
 
 		for (key, pp) in permissions {
 			if pp.len() > PERMISSIONS_FIELDS_COUNT {
@@ -132,9 +133,7 @@
 			}
 
 			perms.push(PropertyKeyPermission {
-				key: <Vec<u8>>::from(key)
-					.try_into()
-					.map_err(|_| "too long key")?,
+				key: key.into_bytes().try_into().map_err(|_| "too long key")?,
 				permission: token_permission,
 			});
 		}
@@ -147,18 +146,19 @@
 	fn token_property_permissions(
 		&self,
 	) -> Result<Vec<(string, Vec<(EthTokenPermissions, bool)>)>> {
-		let mut res = <Vec<_>>::new();
-		for (key, pp) in <Pallet<T>>::token_property_permission(self.id) {
-			let key = string::from_utf8(key.into_inner()).unwrap();
-			let pp = [
-				(EthTokenPermissions::Mutable, pp.mutable),
-				(EthTokenPermissions::TokenOwner, pp.token_owner),
-				(EthTokenPermissions::CollectionAdmin, pp.collection_admin),
-			]
-			.into();
-			res.push((key, pp));
-		}
-		Ok(res)
+		let perms = <Pallet<T>>::token_property_permission(self.id);
+		Ok(perms
+			.into_iter()
+			.map(|(key, pp)| {
+				let key = string::from_utf8(key.into_inner()).expect("Stored key must be valid");
+				let pp = vec![
+					(EthTokenPermissions::Mutable, pp.mutable),
+					(EthTokenPermissions::TokenOwner, pp.token_owner),
+					(EthTokenPermissions::CollectionAdmin, pp.collection_admin),
+				];
+				(key, pp)
+			})
+			.collect())
 	}
 
 	/// @notice Set token property value.
modifiedprimitives/data-structs/src/lib.rsdiffbeforeafterboth
--- a/primitives/data-structs/src/lib.rs
+++ b/primitives/data-structs/src/lib.rs
@@ -1010,7 +1010,7 @@
 pub type PropertyValue = BoundedBytes<ConstU32<MAX_PROPERTY_VALUE_LENGTH>>;
 
 /// Property permission.
-#[derive(Encode, Decode, TypeInfo, Debug, MaxEncodedLen, PartialEq, Clone)]
+#[derive(Encode, Decode, TypeInfo, Debug, MaxEncodedLen, PartialEq, Clone, Default)]
 #[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]
 pub struct PropertyPermission {
 	/// Permission to change the property and property permission.
modifiedtests/src/eth/tokenProperties.test.tsdiffbeforeafterboth
before · tests/src/eth/tokenProperties.test.ts
1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617import {IKeyringPair} from '@polkadot/types/types';18import {Contract} from 'web3-eth-contract';19import {itEth, usingEthPlaygrounds, expect} from './util';20import {ITokenPropertyPermission} from '../util/playgrounds/types';21import {Pallets} from '../util';22import {UniqueNFTCollection, UniqueNFToken, UniqueRFTCollection} from '../util/playgrounds/unique';23import {EthTokenPermissions} from './util/playgrounds/types';2425describe('EVM token properties', () => {26  let donor: IKeyringPair;27  let alice: IKeyringPair;2829  before(async function() {30    await usingEthPlaygrounds(async (helper, privateKey) => {31      donor = await privateKey({filename: __filename});32      [alice] = await helper.arrange.createAccounts([100n], donor);33    });34  });3536  [37    {mode: 'nft' as const, requiredPallets: []},38    {mode: 'rft' as const, requiredPallets: [Pallets.ReFungible]},39  ].map(testCase =>40    itEth.ifWithPallets(`[${testCase.mode}] Set and get token property permissions`, testCase.requiredPallets, async({helper}) => {41      const owner = await helper.eth.createAccountWithBalance(donor);42      const caller = await helper.ethCrossAccount.createAccountWithBalance(donor);43      for(const [mutable,collectionAdmin, tokenOwner] of cartesian([], [false, true], [false, true], [false, true])) {44        const {collectionId, collectionAddress} = await helper.eth.createCollection(testCase.mode, owner, 'A', 'B', 'C');45        const collection = await helper.ethNativeContract.collection(collectionAddress, testCase.mode, owner);46        await collection.methods.addCollectionAdminCross(caller).send({from: owner});4748        await collection.methods.setTokenPropertyPermissions([49          ['testKey', [50            [EthTokenPermissions.Mutable, mutable], 51            [EthTokenPermissions.TokenOwner, tokenOwner], 52            [EthTokenPermissions.CollectionAdmin, collectionAdmin]],53          ],54        ]).send({from: caller.eth});55      56        expect(await helper[testCase.mode].getPropertyPermissions(collectionId)).to.be.deep.equal([{57          key: 'testKey',58          permission: {mutable, collectionAdmin, tokenOwner},59        }]);6061        expect(await collection.methods.tokenPropertyPermissions().call({from: caller.eth})).to.be.like([62          ['testKey', [63            [EthTokenPermissions.Mutable.toString(), mutable], 64            [EthTokenPermissions.TokenOwner.toString(), tokenOwner], 65            [EthTokenPermissions.CollectionAdmin.toString(), collectionAdmin]],66          ],67        ]);68      }69    }));7071  [72    {73      method: 'setProperties',74      methodParams: [[{key: 'testKey1', value: Buffer.from('testValue1')}, {key: 'testKey2', value: Buffer.from('testValue2')}]],75      expectedProps: [{key: 'testKey1', value: 'testValue1'}, {key: 'testKey2', value: 'testValue2'}],76    },77    {78      method: 'setProperty' /*Soft-deprecated*/, 79      methodParams: ['testKey1', Buffer.from('testValue1')],80      expectedProps: [{key: 'testKey1', value: 'testValue1'}],81    },82  ].map(testCase => 83    itEth(`[${testCase.method}] Can be set`, async({helper}) => {84      const caller = await helper.eth.createAccountWithBalance(donor);85      const collection = await helper.nft.mintCollection(alice, {86        tokenPropertyPermissions: [{87          key: 'testKey1',88          permission: {89            collectionAdmin: true,90          },91        }, {92          key: 'testKey2',93          permission: {94            collectionAdmin: true,95          },96        }],97      });9899      await collection.addAdmin(alice, {Ethereum: caller});100      const token = await collection.mintToken(alice);101  102      const collectionEvm = helper.ethNativeContract.collectionById(collection.collectionId, 'nft', caller, testCase.method === 'setProperty');103  104      await collectionEvm.methods[testCase.method](token.tokenId, ...testCase.methodParams).send({from: caller});105  106      const properties = await token.getProperties();107      expect(properties).to.deep.equal(testCase.expectedProps);108    }));109  110  [111    {mode: 'nft' as const, requiredPallets: []},112    {mode: 'rft' as const, requiredPallets: [Pallets.ReFungible]},113  ].map(testCase => 114    itEth.ifWithPallets(`Can be multiple set/read for ${testCase.mode}`, testCase.requiredPallets, async({helper}) => {115      const caller = await helper.eth.createAccountWithBalance(donor);116      117      const properties = Array(5).fill(0).map((_, i) => { return {key: `key_${i}`, value: Buffer.from(`value_${i}`)}; });118      const permissions: ITokenPropertyPermission[] = properties.map(p => { return {key: p.key, permission: {tokenOwner: true,119        collectionAdmin: true,120        mutable: true}}; });121      122      const collection = await helper[testCase.mode].mintCollection(alice, {123        tokenPrefix: 'ethp',124        tokenPropertyPermissions: permissions,125      }) as UniqueNFTCollection | UniqueRFTCollection;126      127      const token = await collection.mintToken(alice);128      129      const valuesBefore = await token.getProperties(properties.map(p => p.key));130      expect(valuesBefore).to.be.deep.equal([]);131      132      133      await collection.addAdmin(alice, {Ethereum: caller});134      135      const address = helper.ethAddress.fromCollectionId(collection.collectionId);136      const contract = helper.ethNativeContract.collection(address, testCase.mode, caller);137      138      expect(await contract.methods.properties(token.tokenId, []).call()).to.be.deep.equal([]);139  140      await contract.methods.setProperties(token.tokenId, properties).send({from: caller});141  142      const values = await token.getProperties(properties.map(p => p.key));143      expect(values).to.be.deep.equal(properties.map(p => { return {key: p.key, value: p.value.toString()}; }));144      145      expect(await contract.methods.properties(token.tokenId, []).call()).to.be.like(properties146        .map(p => { return helper.ethProperty.property(p.key, p.value.toString()); }));147      148      expect(await contract.methods.properties(token.tokenId, [properties[0].key]).call())149        .to.be.like([helper.ethProperty.property(properties[0].key, properties[0].value.toString())]);150    }));151  152  [153    {mode: 'nft' as const, requiredPallets: []},154    {mode: 'rft' as const, requiredPallets: [Pallets.ReFungible]},155  ].map(testCase => 156    itEth.ifWithPallets(`Can be deleted for ${testCase.mode}`, testCase.requiredPallets, async({helper}) => {157      const caller = await helper.eth.createAccountWithBalance(donor);158      const collection = await helper[testCase.mode].mintCollection(alice, {159        tokenPropertyPermissions: [{160          key: 'testKey',161          permission: {162            mutable: true,163            collectionAdmin: true,164          },165        },166        {167          key: 'testKey_1',168          permission: {169            mutable: true,170            collectionAdmin: true,171          },172        }],173      });174    175      const token = await collection.mintToken(alice);176      await token.setProperties(alice, [{key: 'testKey', value: 'testValue'}, {key: 'testKey_1', value: 'testValue_1'}]);177      expect(await token.getProperties()).to.has.length(2);178179      await collection.addAdmin(alice, {Ethereum: caller});180181      const address = helper.ethAddress.fromCollectionId(collection.collectionId);182      const contract = helper.ethNativeContract.collection(address, testCase.mode, caller);183184      await contract.methods.deleteProperties(token.tokenId, ['testKey', 'testKey_1']).send({from: caller});185186      const result = await token.getProperties(['testKey', 'testKey_1']);187      expect(result.length).to.equal(0);188    }));189190  itEth('Can be read', async({helper}) => {191    const caller = helper.eth.createAccount();192    const collection = await helper.nft.mintCollection(alice, {193      tokenPropertyPermissions: [{194        key: 'testKey',195        permission: {196          collectionAdmin: true,197        },198      }],199    });200  201    const token = await collection.mintToken(alice);202    await token.setProperties(alice, [{key: 'testKey', value: 'testValue'}]);203204    const address = helper.ethAddress.fromCollectionId(collection.collectionId);205    const contract = helper.ethNativeContract.collection(address, 'nft', caller);206207    const value = await contract.methods.property(token.tokenId, 'testKey').call();208    expect(value).to.equal(helper.getWeb3().utils.toHex('testValue'));209  });210});211212describe('EVM token properties negative', () => {213  let donor: IKeyringPair;214  let alice: IKeyringPair;215  let caller: string;216  let aliceCollection: UniqueNFTCollection;217  let token: UniqueNFToken;218  const tokenProps = [{key: 'testKey_1', value: 'testValue_1'}, {key: 'testKey_2', value: 'testValue_2'}];219  let collectionEvm: Contract;220221  before(async function() {222    await usingEthPlaygrounds(async (helper, privateKey) => {223      donor = await privateKey({filename: __filename});224      [alice] = await helper.arrange.createAccounts([100n], donor);225    });226  });227228  beforeEach(async () => {229    // 1. create collection with props: testKey_1, testKey_2230    // 2. create token and set props testKey_1, testKey_2231    await usingEthPlaygrounds(async (helper) => {232      aliceCollection = await helper.nft.mintCollection(alice, {233        tokenPropertyPermissions: [{234          key: 'testKey_1',235          permission: {236            mutable: true,237            collectionAdmin: true,238          },239        },240        {241          key: 'testKey_2',242          permission: {243            mutable: true,244            collectionAdmin: true,245          },246        }],247      }); 248      token = await aliceCollection.mintToken(alice);249      await token.setProperties(alice, tokenProps);250      collectionEvm = helper.ethNativeContract.collection(helper.ethAddress.fromCollectionId(aliceCollection.collectionId), 'nft', caller, true);251    });252  });253254  [255    {method: 'setProperty', methodParams: [tokenProps[1].key, Buffer.from('newValue')]},256    {method: 'setProperties', methodParams: [[{key: tokenProps[1].key, value: Buffer.from('newValue')}]]},257  ].map(testCase =>258    itEth(`[${testCase.method}] Cannot set properties of non-owned collection`, async ({helper}) => {259      caller = await helper.eth.createAccountWithBalance(donor);260      collectionEvm = helper.ethNativeContract.collection(helper.ethAddress.fromCollectionId(aliceCollection.collectionId), 'nft', caller, true);261      // Caller not an owner and not an admin, so he cannot set properties:262      await expect(collectionEvm.methods[testCase.method](token.tokenId, ...testCase.methodParams).call({from: caller})).to.be.rejectedWith('NoPermission');263      await expect(collectionEvm.methods[testCase.method](token.tokenId, ...testCase.methodParams).send({from: caller})).to.be.rejected;264265      // Props have not changed:266      const expectedProps = tokenProps.map(p => helper.ethProperty.property(p.key, p.value.toString()));267      const actualProps = await collectionEvm.methods.properties(token.tokenId, []).call();268      expect(actualProps).to.deep.eq(expectedProps);269    }));270271  [272    {method: 'setProperty', methodParams: ['testKey_3', Buffer.from('testValue3')]},273    {method: 'setProperties', methodParams: [[{key: 'testKey_3', value: Buffer.from('testValue3')}]]},274  ].map(testCase =>275    itEth(`[${testCase.method}] Cannot set non-existing properties`, async ({helper}) => {276      caller = await helper.eth.createAccountWithBalance(donor);277      collectionEvm = helper.ethNativeContract.collection(helper.ethAddress.fromCollectionId(aliceCollection.collectionId), 'nft', caller, true);278      await helper.collection.addAdmin(alice, aliceCollection.collectionId, {Ethereum: caller});279280      await expect(collectionEvm.methods[testCase.method](token.tokenId, ...testCase.methodParams).call({from: caller})).to.be.rejectedWith('NoPermission');281      await expect(collectionEvm.methods[testCase.method](token.tokenId, ...testCase.methodParams).send({from: caller})).to.be.rejected;282283      // Props have not changed:284      const expectedProps = tokenProps.map(p => helper.ethProperty.property(p.key, p.value.toString()));285      const actualProps = await collectionEvm.methods.properties(token.tokenId, []).call();286      expect(actualProps).to.deep.eq(expectedProps);287    }));288289  [290    {method: 'deleteProperty', methodParams: ['testKey_2']},291    {method: 'deleteProperties', methodParams: [['testKey_2']]},292  ].map(testCase =>  293    itEth(`[${testCase.method}] Cannot delete properties of non-owned collection`, async ({helper}) => {294      caller = await helper.eth.createAccountWithBalance(donor);295      collectionEvm = helper.ethNativeContract.collection(helper.ethAddress.fromCollectionId(aliceCollection.collectionId), 'nft', caller, testCase.method == 'deleteProperty');296      // Caller not an owner and not an admin, so he cannot set properties:297      await expect(collectionEvm.methods[testCase.method](token.tokenId, ...testCase.methodParams).call({from: caller})).to.be.rejectedWith('NoPermission');298      await expect(collectionEvm.methods[testCase.method](token.tokenId, ...testCase.methodParams).send({from: caller})).to.be.rejected;299300      // Props have not changed:301      const expectedProps = tokenProps.map(p => helper.ethProperty.property(p.key, p.value.toString()));302      const actualProps = await collectionEvm.methods.properties(token.tokenId, []).call();303      expect(actualProps).to.deep.eq(expectedProps);304    }));305   306  [307    {method: 'deleteProperty', methodParams: ['testKey_3']},308    {method: 'deleteProperties', methodParams: [['testKey_3']]},309  ].map(testCase =>  310    itEth(`[${testCase.method}] Cannot delete non-existing properties`, async ({helper}) => {311      caller = await helper.eth.createAccountWithBalance(donor);312      collectionEvm = helper.ethNativeContract.collection(helper.ethAddress.fromCollectionId(aliceCollection.collectionId), 'nft', caller, testCase.method == 'deleteProperty');313      await helper.collection.addAdmin(alice, aliceCollection.collectionId, {Ethereum: caller});314      // Caller cannot delete non-existing properties:315      await expect(collectionEvm.methods[testCase.method](token.tokenId, ...testCase.methodParams).call({from: caller})).to.be.rejectedWith('NoPermission');316      await expect(collectionEvm.methods[testCase.method](token.tokenId, ...testCase.methodParams).send({from: caller})).to.be.rejected;317      // Props have not changed:318      const expectedProps = tokenProps.map(p => helper.ethProperty.property(p.key, p.value.toString()));319      const actualProps = await collectionEvm.methods.properties(token.tokenId, []).call();320      expect(actualProps).to.deep.eq(expectedProps);321    }));322});323324325type ElementOf<A> = A extends readonly (infer T)[] ? T : never;326function* cartesian<T extends Array<Array<any>>, R extends Array<any>>(internalRest: [...R], ...args: [...T]): Generator<[...R, ...{[K in keyof T]: ElementOf<T[K]>}]> {327  if(args.length === 0) {328    yield internalRest as any;329    return;330  }331  for(const value of args[0]) {332    yield* cartesian([...internalRest, value], ...args.slice(1)) as any;333  }334}