TrieDB

triedb.sol

// SPDX-License-Identifier: UNLICENSED
pragma solidity 0.8.14;

contract TrieDB {

    enum TRIE_STATUS { INACTIVE, ACTIVE }

    struct TrieNode {
        bytes32 origin;
        TRIE_STATUS status;
        Transaction [] txChange;
    }

    struct Transaction {
        uint256 amount;
        bytes32 extraData;
    }

    mapping(bytes32 => TrieNode) public _trie;
    mapping(address => uint256) public _trieCount;

    function _hash(address account, uint256 Id) public pure returns (bytes32) {
        return bytes32(abi.encode(keccak256(abi.encode(account, Id))));
    }

    function _create(address account, uint256 amount, bytes32 origin, bytes32 data) public returns (bool) {
        bytes32 TrieHash = _hash(account, _trieCount[account]);
        if (_trie[TrieHash].status == TRIE_STATUS.INACTIVE) {
            _trie[TrieHash].origin = origin;
            _trie[TrieHash].txChange.push(Transaction(amount,data));
            _trieCount[account]++;
        } 
        return true;
    }

    function _spend(address account, uint256 amount, uint256 id) public returns (bool) {
        bytes32 TrieHashOrigin = _hash(msg.sender, id);
        bytes32 TrieHash = _hash(msg.sender, id);
        TrieNode storage trie = _trie[TrieHash];
        Transaction storage transac = trie.txChange[trie.txChange.length-1];
        _create(account, amount, _trie[TrieHashOrigin].origin, "");
        trie.txChange.push(Transaction(transac.amount - amount, transac.extraData));
        return true;
    }

    function getTrie(bytes32 TireId) public view returns (TrieNode memory, Transaction memory) {
        TrieNode memory trieCache = _trie[TireId];
        return (trieCache, trieCache.txChange[trieCache.txChange.length-1]);
    }

    function getTx(bytes32 TireId) public view returns (Transaction [] memory) {
        TrieNode memory trieCache = _trie[TireId];
        return trieCache.txChange;
    }

}