feat: set all permissions at once

meemproject · Jan 7, 2022 · e635190 · e635190
1 parent d0dbe9c
commit e635190
Show file tree

Hide file tree

Showing 8 changed files with 463 additions and 5 deletions.
diff --git a/abi/contracts/Meem/facets/MeemPermissionsFacet.sol/MeemPermissionsFacet.json b/abi/contracts/Meem/facets/MeemPermissionsFacet.sol/MeemPermissionsFacet.json
@@ -36,6 +36,11 @@
     "name": "InvalidTotalChildren",
     "type": "error"
   },
+  {
+    "inputs": [],
+    "name": "MissingRequiredPermissions",
+    "type": "error"
+  },
   {
     "inputs": [
       {
@@ -320,6 +325,56 @@
     "stateMutability": "nonpayable",
     "type": "function"
   },
+  {
+    "inputs": [
+      {
+        "internalType": "uint256",
+        "name": "tokenId",
+        "type": "uint256"
+      },
+      {
+        "internalType": "enum PropertyType",
+        "name": "propertyType",
+        "type": "uint8"
+      },
+      {
+        "internalType": "enum PermissionType",
+        "name": "permissionType",
+        "type": "uint8"
+      },
+      {
+        "components": [
+          {
+            "internalType": "enum Permission",
+            "name": "permission",
+            "type": "uint8"
+          },
+          {
+            "internalType": "address[]",
+            "name": "addresses",
+            "type": "address[]"
+          },
+          {
+            "internalType": "uint256",
+            "name": "numTokens",
+            "type": "uint256"
+          },
+          {
+            "internalType": "address",
+            "name": "lockedBy",
+            "type": "address"
+          }
+        ],
+        "internalType": "struct MeemPermission[]",
+        "name": "permissions",
+        "type": "tuple[]"
+      }
+    ],
+    "name": "setPermissions",
+    "outputs": [],
+    "stateMutability": "nonpayable",
+    "type": "function"
+  },
   {
     "inputs": [
       {

diff --git a/abi/contracts/Meem/interfaces/MeemStandard.sol/IMeemPermissionsStandard.json b/abi/contracts/Meem/interfaces/MeemStandard.sol/IMeemPermissionsStandard.json
@@ -250,6 +250,56 @@
     "stateMutability": "nonpayable",
     "type": "function"
   },
+  {
+    "inputs": [
+      {
+        "internalType": "uint256",
+        "name": "tokenId",
+        "type": "uint256"
+      },
+      {
+        "internalType": "enum PropertyType",
+        "name": "propertyType",
+        "type": "uint8"
+      },
+      {
+        "internalType": "enum PermissionType",
+        "name": "permissionType",
+        "type": "uint8"
+      },
+      {
+        "components": [
+          {
+            "internalType": "enum Permission",
+            "name": "permission",
+            "type": "uint8"
+          },
+          {
+            "internalType": "address[]",
+            "name": "addresses",
+            "type": "address[]"
+          },
+          {
+            "internalType": "uint256",
+            "name": "numTokens",
+            "type": "uint256"
+          },
+          {
+            "internalType": "address",
+            "name": "lockedBy",
+            "type": "address"
+          }
+        ],
+        "internalType": "struct MeemPermission[]",
+        "name": "permissions",
+        "type": "tuple[]"
+      }
+    ],
+    "name": "setPermissions",
+    "outputs": [],
+    "stateMutability": "nonpayable",
+    "type": "function"
+  },
   {
     "inputs": [
       {

diff --git a/contracts/Meem/facets/MeemPermissionsFacet.sol b/contracts/Meem/facets/MeemPermissionsFacet.sol
@@ -33,6 +33,20 @@ contract MeemPermissionsFacet is IMeemPermissionsStandard {
 		LibMeem.lockChildrenPerWallet(tokenId);
 	}
 
+	function setPermissions(
+		uint256 tokenId,
+		PropertyType propertyType,
+		PermissionType permissionType,
+		MeemPermission[] memory permissions
+	) external override {
+		LibMeem.setPermissions(
+			tokenId,
+			propertyType,
+			permissionType,
+			permissions
+		);
+	}
+
 	function addPermission(
 		uint256 tokenId,
 		PropertyType propertyType,

diff --git a/contracts/Meem/interfaces/MeemStandard.sol b/contracts/Meem/interfaces/MeemStandard.sol
@@ -208,6 +208,13 @@ interface IMeemPermissionsStandard {
 		MeemPermission[] permission
 	);
 
+	function setPermissions(
+		uint256 tokenId,
+		PropertyType propertyType,
+		PermissionType permissionType,
+		MeemPermission[] memory permissions
+	) external;
+
 	function addPermission(
 		uint256 tokenId,
 		PropertyType propertyType,

diff --git a/contracts/Meem/libraries/LibMeem.sol b/contracts/Meem/libraries/LibMeem.sol
@@ -166,6 +166,36 @@ library LibMeem {
 		return tokenId;
 	}
 
+	function setPermissions(
+		uint256 tokenId,
+		PropertyType propertyType,
+		PermissionType permissionType,
+		MeemPermission[] memory permissions
+	) internal {
+		LibERC721.requireOwnsToken(tokenId);
+		MeemProperties storage props = getProperties(tokenId, propertyType);
+		permissionNotLocked(props, permissionType);
+
+		MeemPermission[] storage perms = getPermissions(props, permissionType);
+
+		// Check if there are any existing locked permissions and if so, verify they're the same as the new permissions
+		validatePermissions(permissions, perms);
+
+		if (permissionType == PermissionType.Copy) {
+			delete props.copyPermissions;
+		} else if (permissionType == PermissionType.Remix) {
+			delete props.remixPermissions;
+		} else if (permissionType == PermissionType.Read) {
+			delete props.readPermissions;
+		} else {
+			revert InvalidPermissionType();
+		}
+
+		for (uint256 i = 0; i < permissions.length; i++) {
+			perms.push(permissions[i]);
+		}
+	}
+
 	function addPermission(
 		uint256 tokenId,
 		PropertyType propertyType,

diff --git a/test/helpers/meemStandard.ts b/test/helpers/meemStandard.ts
@@ -18,6 +18,6 @@ export enum Permission {
 }
 
 export enum PropertyType {
-	Meem = 1,
-	Child = 2
+	Meem = 0,
+	Child = 1
 }