check contract lock in transfertFrom

massalabs · Dec 10, 2024 · 46df192 · 46df192
1 parent 5985664
commit 46df192
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/smart-contract/assembly/contracts/main.ts b/smart-contract/assembly/contracts/main.ts
@@ -23,6 +23,8 @@ import {
   _constructor,
   _ownerOf,
 } from '@massalabs/sc-standards/assembly/contracts/MRC721/enumerable/MRC721Enumerable-internals';
+// eslint-disable-next-line max-len
+import { transferFrom as _transferFrom } from '@massalabs/sc-standards/assembly/contracts/MRC721/enumerable/MRC721Enumerable';
 
 import {
   _setOwner,
@@ -451,6 +453,11 @@ export function ownerOf(binaryArgs: StaticArray<u8>): StaticArray<u8> {
   return stringToBytes(owner);
 }
 
+export function transferFrom(binaryArgs: StaticArray<u8>): void {
+  assert(!Storage.has(buildLockedKey()), 'Contract is locked');
+  _transferFrom(binaryArgs);
+}
+
 export {
   setOwner,
   ownerAddress,
@@ -461,8 +468,8 @@ export {
   setApprovalForAll,
   getApproved,
   approve,
-  transferFrom,
   balanceOf,
   symbol,
   name,
+  totalSupply,
 } from '@massalabs/sc-standards/assembly/contracts/MRC721/enumerable/MRC721Enumerable';