// SPDX-License-Identifier: MIT

pragma solidity 0.6.12;
pragma experimental ABIEncoderV2;

import '../modules/FoldingAccount/FoldingAccountStorage.sol';
import './interfaces/IFoldingConnectorProvider.sol';

contract FoldingAccount is FoldingAccountStorage {
    constructor(address foldingRegistry) public {
        aStore().foldingRegistry = foldingRegistry;
    }

    /// @notice Find the connector for `msg.sig` and delegate call it with `msg.data`
    function delegate() private {
        bool firstCall = false; // We need to delete the entryCaller on exit

        AccountStore storage accountStorage = aStore();
        if (accountStorage.entryCaller == address(0)) {
            accountStorage.entryCaller = msg.sender;
            firstCall = true;
        }
        // Check if a connector expects a callback or find connector
        address impl = accountStorage.callbackTarget;
        if (impl != address(0)) {
            require(accountStorage.expectedCallbackSig == msg.sig, 'FA1');
        } else {
            impl = IFoldingConnectorProvider(accountStorage.foldingRegistry).getImplementation(msg.sig);
        }

        /// @dev This assembly code returns directly to caller
        assembly {
            calldatacopy(0, 0, calldatasize())
            let result := delegatecall(gas(), impl, 0, calldatasize(), 0, 0)
            let size := returndatasize()
            returndatacopy(0, 0, size)
            switch result
            case 0 {
                revert(0, size)
            }
            default {
                /// @dev if this is the first call, set the entryCaller to 0
                if firstCall {
                    sstore(accountStorage_slot, 0)
                }
                return(0, size)
            }
        }
    }

    fallback() external payable {
        if (msg.sig != bytes4(0)) delegate();
    }
}

// SPDX-License-Identifier: MIT

pragma solidity 0.6.12;

contract FoldingAccountStorage {
    bytes32 constant ACCOUNT_STORAGE_POSITION = keccak256('folding.account.storage');

    /**
     * entryCaller:         address of the caller of the account, during a transaction
     *
     * callbackTarget:      address of logic to be run when expecting a callback
     *
     * expectedCallbackSig: signature of function to be run when expecting a callback
     *
     * foldingRegistry      address of factory creating FoldingAccount
     *
     * nft:                 address of the nft contract.
     *
     * owner:               address of the owner of this FoldingAccount.
     */
    struct AccountStore {
        address entryCaller;
        address callbackTarget;
        bytes4 expectedCallbackSig;
        address foldingRegistry;
        address nft;
        address owner;
    }

    modifier onlyAccountOwner() {
        AccountStore storage s = aStore();
        require(s.entryCaller == s.owner, 'FA2');
        _;
    }

    modifier onlyNFTContract() {
        AccountStore storage s = aStore();
        require(s.entryCaller == s.nft, 'FA3');
        _;
    }

    modifier onlyAccountOwnerOrRegistry() {
        AccountStore storage s = aStore();
        require(s.entryCaller == s.owner || s.entryCaller == s.foldingRegistry, 'FA4');
        _;
    }

    function aStore() internal pure returns (AccountStore storage s) {
        bytes32 position = ACCOUNT_STORAGE_POSITION;
        assembly {
            s_slot := position
        }
    }

    function accountOwner() internal view returns (address) {
        return aStore().owner;
    }
}

// SPDX-License-Identifier: MIT

pragma solidity 0.6.12;

interface IFoldingConnectorProvider {
    function getImplementation(bytes4 functionSignature) external view returns (address implementation);
}

{
  "optimizer": {
    "enabled": true,
    "runs": 9999
  },
  "outputSelection": {
    "*": {
      "*": [
        "evm.bytecode",
        "evm.deployedBytecode",
        "devdoc",
        "userdoc",
        "metadata",
        "abi"
      ]
    }
  },
  "libraries": {}
}

• No Contract Security Audit Submitted- Submit Audit Here

[{"inputs":[{"internalType":"address","name":"foldingRegistry","type":"address"}],"stateMutability":"nonpayable","type":"constructor"},{"stateMutability":"payable","type":"fallback"}]

0x608060405234801561001057600080fd5b506040516103f63803806103f683398101604081905261002f91610083565b8061003861005f565b60020180546001600160a01b0319166001600160a01b0392909216919091179055506100b1565b7fadbb89e904f74d44d0d104c851da4f2fa24de8f1dee64f038923bb940f37d53a90565b600060208284031215610094578081fd5b81516001600160a01b03811681146100aa578182fd5b9392505050565b610336806100c06000396000f3fe60806040526000357fffffffff00000000000000000000000000000000000000000000000000000000161561003657610036610038565b005b60008061004361023d565b805490915073ffffffffffffffffffffffffffffffffffffffff1661008f5780547fffffffffffffffffffffffff00000000000000000000000000000000000000001633178155600191505b600181015473ffffffffffffffffffffffffffffffffffffffff16801561013e5760018201546000357fffffffff000000000000000000000000000000000000000000000000000000009081167401000000000000000000000000000000000000000090920460e01b1614610139576040517f08c379a0000000000000000000000000000000000000000000000000000000008152600401610130906102c9565b60405180910390fd5b61020d565b60028201546040517fdc9cc64500000000000000000000000000000000000000000000000000000000815273ffffffffffffffffffffffffffffffffffffffff9091169063dc9cc645906101ba907fffffffff00000000000000000000000000000000000000000000000000000000600035169060040161029c565b60206040518083038186803b1580156101d257600080fd5b505afa1580156101e6573d6000803e3d6000fd5b505050506040513d601f19601f8201168201806040525081019061020a9190610261565b90505b3660008037600080366000845af43d806000803e81801561023857851561023357600085555b816000f35b816000fd5b7fadbb89e904f74d44d0d104c851da4f2fa24de8f1dee64f038923bb940f37d53a90565b600060208284031215610272578081fd5b815173ffffffffffffffffffffffffffffffffffffffff81168114610295578182fd5b9392505050565b7fffffffff0000000000000000000000000000000000000000000000000000000091909116815260200190565b60208082526003908201527f464131000000000000000000000000000000000000000000000000000000000060408201526060019056fea264697066735822122024923cd795b5543a38f56fe39c7b6ead92248d6b1e0a30cc07b17c8087c8a18264736f6c634300060c0033000000000000000000000000ec6b351778aaa2349a8726b4837e05232ef20d03

0x60806040526000357fffffffff00000000000000000000000000000000000000000000000000000000161561003657610036610038565b005b60008061004361023d565b805490915073ffffffffffffffffffffffffffffffffffffffff1661008f5780547fffffffffffffffffffffffff00000000000000000000000000000000000000001633178155600191505b600181015473ffffffffffffffffffffffffffffffffffffffff16801561013e5760018201546000357fffffffff000000000000000000000000000000000000000000000000000000009081167401000000000000000000000000000000000000000090920460e01b1614610139576040517f08c379a0000000000000000000000000000000000000000000000000000000008152600401610130906102c9565b60405180910390fd5b61020d565b60028201546040517fdc9cc64500000000000000000000000000000000000000000000000000000000815273ffffffffffffffffffffffffffffffffffffffff9091169063dc9cc645906101ba907fffffffff00000000000000000000000000000000000000000000000000000000600035169060040161029c565b60206040518083038186803b1580156101d257600080fd5b505afa1580156101e6573d6000803e3d6000fd5b505050506040513d601f19601f8201168201806040525081019061020a9190610261565b90505b3660008037600080366000845af43d806000803e81801561023857851561023357600085555b816000f35b816000fd5b7fadbb89e904f74d44d0d104c851da4f2fa24de8f1dee64f038923bb940f37d53a90565b600060208284031215610272578081fd5b815173ffffffffffffffffffffffffffffffffffffffff81168114610295578182fd5b9392505050565b7fffffffff0000000000000000000000000000000000000000000000000000000091909116815260200190565b60208082526003908201527f464131000000000000000000000000000000000000000000000000000000000060408201526060019056fea264697066735822122024923cd795b5543a38f56fe39c7b6ead92248d6b1e0a30cc07b17c8087c8a18264736f6c634300060c0033