tonb/sources/wallet.tact

import "@stdlib/ownable";
import "./messages";
import "./linker";
import "./constants";



@interface("org.ton.jetton.wallet")
contract TONBWallet {

    const minTonsForStorage: Int = ton("0.01");
    const gasConsumption: Int = ton("0.01");

    balance: Int;
    owner: Address;
    master: Address; 
    blacklisted: Bool = false;
    linker: Int?;
    linker_address: Address?;

    init(master: Address, owner: Address) {
        self.balance = 0;
        self.owner = owner;
        self.master = master;
    }

    receive(msg: TokenTransfer) {
        require(!self.blacklisted, "Wallet is blacklisted");

        // Check sender
        let ctx: Context = context();
        require(ctx.sender == self.owner, "Invalid sender");

        // Update balance
        self.balance = self.balance - msg.amount;
        require(self.balance >= 0, "Invalid balance");

        // Gas checks
        let fwdFee: Int = ctx.readForwardFee();
        let fwdCount: Int = 1;
        if (msg.forwardTonAmount > 0) {
            fwdCount = 2;
        }
        require(ctx.value > transfer_gas_consumption, "Invalid value");
  
        // Send tokens
        let init: StateInit = initOf TONBWallet(self.master, msg.destination);
        let walletAddress: Address = contractAddress(init);
        send(SendParameters{
            to: walletAddress, 
            value: 0,
            mode: SendRemainingValue, 
            bounce: true,
            body: TokenTransferInternal{
                amount: msg.amount,
                queryId: msg.queryId,
                from: self.owner,
                responseAddress: self.owner,
                forwardTonAmount: msg.forwardTonAmount,
                forwardPayload: msg.forwardPayload,
                setLinker: null,
                setLinkerAddress: null
            }.toCell(),
            code: init.code,
            data: init.data
        });
    }

    receive(msg: TokenTransferInternal) {
        if(self.blacklisted){
            send(SendParameters{
                to: self.master,
                value: 0,
                mode: SendRemainingValue,
                body: TokenBurnNotification{
                    queryId: msg.queryId,
                    amount: msg.amount,
                    owner: self.owner,
                    responseAddress: self.owner
                }.toCell()
            });
            return;
        }

        // Check sender
        let ctx: Context = context();
        if (self.linker == null) {
            if (msg.setLinker != null) {
                self.linker = msg.setLinker;
                self.linker_address = msg.setLinkerAddress;
            }
        }
        if (ctx.sender != self.master/* && ctx.sender != self.linker_address*/) {
            let is_from_linker: Bool = false;
            if (self.linker_address != null) {
                if (ctx.sender == self.linker_address!!) {
                    is_from_linker = true;
                }
            }
            if (!is_from_linker) {
                let sinit: StateInit = initOf TONBWallet(self.master, msg.from);
                require(contractAddress(sinit) == ctx.sender, "Invalid sender");
            }
        }

        // Update balance
        self.balance = self.balance + msg.amount;
        require(self.balance >= 0, "Invalid balance");

        // Adjust value for gas
        let msgValue: Int = ctx.value;
        let tonBalanceBeforeMsg: Int = myBalance() - msgValue;
        let storageFee: Int = self.minTonsForStorage - min(tonBalanceBeforeMsg, self.minTonsForStorage);

        if (self.linker == null) {
            // request a linker
            send(SendParameters{
                to: self.master,
                value: 0,
                mode: SendRemainingValue,
                body: RequestLinker {
                    client: self.owner
                }.toCell()
            });
            return;
        }
        msgValue = msgValue - (storageFee + self.gasConsumption);

        // Forward ton
        if (msg.forwardTonAmount > 0) {
            let fwdFee: Int = ctx.readForwardFee();
            msgValue = msgValue - (msg.forwardTonAmount + fwdFee);
            send(SendParameters{
                to: self.owner,
                value: msg.forwardTonAmount,
                bounce: false,
                body: TokenNotification{
                    queryId: msg.queryId,
                    amount: msg.amount,
                    from: msg.from,
                    forwardPayload: msg.forwardPayload
                }.toCell()
            });
        }

        // Cashback
        if (msg.responseAddress != null && msgValue > 0) {
            send(SendParameters{
                to: msg.responseAddress!!,
                value: msgValue,
                bounce: false,
                body: TokenExcesses{
                    queryId: msg.queryId
                }.toCell()
            });
        }
    }

    receive(msg: TokenBurn) {

        // Check sender
        let ctx: Context = context();
        require(ctx.sender == self.owner || ctx.sender == self.master || ctx.sender == self.linker_address, "Invalid sender");

        // Update balance
        self.balance = self.balance - msg.amount;
        require(self.balance >= 0, "Invalid balance");

        // // Gas checks
        let fwdFee: Int = ctx.readForwardFee();
        require(ctx.value > 2 * self.gasConsumption + self.minTonsForStorage, "Invalid value");

        // Burn tokens
        send(SendParameters{
            to: self.master,
            value: 0,
            mode: SendRemainingValue,
            bounce: true,
            body: TokenBurnNotification{
                queryId: msg.queryId,
                amount: msg.amount,
                owner: self.owner,
                responseAddress: self.owner
            }.toCell()
        });
    }

    receive(_msg: BlacklistWallet) {
        // Check sender
        let ctx: Context = context();
        require(ctx.sender == self.master || ctx.sender == self.linker_address, "Invalid sender");

        // Blacklist wallet
        self.blacklisted = true;
        let amount: Int = self.balance;
        self.balance = 0;

        send(SendParameters{
            to: self.master,
            value: 0,
            mode: SendRemainingValue,
            bounce: true,
            body: TokenBurnNotification{
                queryId: 0,
                amount: amount,
                owner: self.owner,
                responseAddress: self.owner
            }.toCell()
        });
    }

    bounced(msg: Slice) {

        // Parse bounced message
        msg.skipBits(32); // 0xFFFFFFFF
        let op: Int = msg.loadUint(32);
        let queryId: Int = msg.loadUint(64);
        let jettonAmount: Int = msg.loadCoins();
        require(op == 0x178d4519 || op == 0x7bdd97de, "Invalid bounced message");

        // Update balance
        self.balance = self.balance + jettonAmount;
    }

    get fun get_wallet_data(): JettonWalletData {
        return JettonWalletData{
            balance: self.balance,
            owner: self.owner,
            master: self.master,
            walletCode: (initOf TONBWallet(self.master, self.owner)).code
        };
    }
}