R

README.md

@@ -2,7 +2,7 @@
 
 ## Overview
 
-The USTB Token project is a Solidity-based smart contract system designed to provide a rebase token with cross-chain and Layer Zero functionalities. It's built with upgradeability in mind, ensuring that the token logic can evolve over time.
+The USTB Token project is a Solidity-based smart contract system designed to provide a rebase token with cross-chain and Layer Zero functionalities. It's built with upgradeability in mind, ensuring that the token logic can evolve.
 
 ## Key Components
 

foundry.toml

@@ -37,4 +37,6 @@ unreal = { key = "", url = "https://unreal.blockscout.com/api" }
 [fmt]
 wrap_comments = true
 
+[invariant]
+
 # See more config options https://github.com/foundry-rs/foundry/blob/master/crates/config/README.md#all-options

src/USTB.sol

@@ -34,7 +34,8 @@ contract USTB is IUSTB, LayerZeroRebaseTokenUpgradeable, UUPSUpgradeable {
     }
 
     // keccak256(abi.encode(uint256(keccak256("tangible.storage.USTB")) - 1)) & ~bytes32(uint256(0xff))
-    bytes32 private constant USTBStorageLocation = 0x56cb630b12f1f031f72de1d734e98085323517cc6515c1c85452dc02f218dd00;
+    bytes32 private constant USTBStorageLocation =
+        0x56cb630b12f1f031f72de1d734e98085323517cc6515c1c85452dc02f218dd00;
 
     function _getUSTBStorage() private pure returns (USTBStorage storage $) {
         // slither-disable-next-line assembly
@@ -70,15 +71,18 @@ contract USTB is IUSTB, LayerZeroRebaseTokenUpgradeable, UUPSUpgradeable {
      * @param endpoint The Layer Zero endpoint for cross-chain operations.
      * @custom:oz-upgrades-unsafe-allow constructor
      */
-    constructor(address underlying, uint256 mainChainId, address endpoint)
-        CrossChainToken(mainChainId)
-        LayerZeroRebaseTokenUpgradeable(endpoint)
-    {
+    constructor(
+        address underlying,
+        uint256 mainChainId,
+        address endpoint
+    ) CrossChainToken(mainChainId) LayerZeroRebaseTokenUpgradeable(endpoint) {
         UNDERLYING = underlying;
         _disableInitializers();
     }
 
-    function _authorizeUpgrade(address newImplementation) internal override onlyOwner {}
+    function _authorizeUpgrade(
+        address newImplementation
+    ) internal override onlyOwner {}
 
     /**
      * @notice Initializes the USTB contract with essential parameters.
@@ -146,11 +150,20 @@ contract USTB is IUSTB, LayerZeroRebaseTokenUpgradeable, UUPSUpgradeable {
         _disableRebase(account, disable);
     }
 
-    function rebaseIndexManager() external view override returns (address _rebaseIndexManager) {
+    function rebaseIndexManager()
+        external
+        view
+        override
+        returns (address _rebaseIndexManager)
+    {
         USTBStorage storage $ = _getUSTBStorage();
         _rebaseIndexManager = $.rebaseIndexManager;
     }
 
+    function isNotRebase(address account) external view returns (bool) {
+        return _isRebaseDisabled(account);
+    }
+
     /**
      * @notice Sets the rebase index and its corresponding nonce on non-main chains.
      * @dev This function allows the rebase index manager to manually update the rebase index and nonce when not on the
@@ -162,7 +175,10 @@ contract USTB is IUSTB, LayerZeroRebaseTokenUpgradeable, UUPSUpgradeable {
      * @param index The new rebase index to set.
      * @param nonce The new nonce corresponding to the rebase index.
      */
-    function setRebaseIndex(uint256 index, uint256 nonce) public onlyIndexManager mainChain(false) {
+    function setRebaseIndex(
+        uint256 index,
+        uint256 nonce
+    ) public onlyIndexManager mainChain(false) {
         _setRebaseIndex(index, nonce);
     }
 
@@ -210,7 +226,11 @@ contract USTB is IUSTB, LayerZeroRebaseTokenUpgradeable, UUPSUpgradeable {
      * @param to The address to which tokens are being transferred or minted.
      * @param amount The amount of tokens being transferred, minted, or burned.
      */
-    function _update(address from, address to, uint256 amount) internal virtual override {
+    function _update(
+        address from,
+        address to,
+        uint256 amount
+    ) internal virtual override {
         refreshRebaseIndex();
         super._update(from, to, amount);
     }