contract NyxToken is Context, IERC20, Ownable {
    using Address for address;

    mapping(address => uint256) private _rOwned;
    mapping(address => uint256) private _tOwned;
    mapping(address => mapping(address => uint256)) private _allowances;

    mapping(address => bool) private _isExcludedFromFee;
    mapping(address => bool) private _isExcluded;
    address[] private _excluded;

    // Liquidity Pairs
    mapping(address => bool) public _isPair;

    // Banned contracts
    mapping(address => bool) public _isBanned;

    uint256 private constant MAX = ~uint256(0);
    uint256 public constant _tTotal = 1e15 * 10**9;
    uint256 private _rTotal = (MAX - (MAX % _tTotal));
    uint256 private _tFeeTotal;

    string private constant _name = "Nyx Token";
    string private constant _symbol = "NYXT";
    uint8 private constant _decimals = 9;

    // Wallets
    address public _marketingWalletAddress;
    address public _msWalletAddress; // Multiple Sclerosis wallet
    address public _vetWalletAddress; // Veterans wallet
    address public constant _deadAddress =
        0x000000000000000000000000000000000000dEaD;

    uint256 public _taxFee = 20; // also is fee for ms & vet funds
    uint256 public _marketingFee = 10;
    uint256 public _burnFee = 10;

    bool public _contractFeesEnabled = true;

    IUniswapV2Router02 public uniswapV2Router;
    address public uniswapV2Pair;

    bool public swapEnabled = true;

    uint256 public _maxTxAmount = _tTotal;

    event SetContractFeesEnabled(bool _bool);
    event SetIsPair(address _address, bool _bool);
    event SetIsBanned(address _address, bool _bool);
    event SetSwapEnabled(bool enabled);
    event SetMarketingWalletAddress(address _address);
    event WithdrawalETH(uint256 _amount, address to);
    event WithdrawalToken(address _tokenAddr, uint256 _amount, address to);
    event ExcludeFromReward(address _address);
    event IncludeInReward(address _address);
    event ExcludeFromFee(address _address);
    event IncludeInFee(address _address);
    event SetMaxTxPercent(uint256 _amount);
    event SetMSWalletAddress(address _address);
    event SetVetWalletAddress(address _address);

    /*
     * mainnet params
     * uniswap router - 0x7a250d5630B4cF539739dF2C5dAcb4c659F2488D
     * dev wallet - 0x18ACa07DB9807c173B11eFF42f9e7f1c2B676c53
     * veterans wallet - 0xc11b9a1e6119f6ae93b3eec1d8178e149bf57bad
     * ms wallet - 0xd93d81650dE684532d6F76F618Bb66621Be17E8c
     */
    constructor(
        address router,
        address marketingWallet,
        address msWallet,
        address vetWallet
    ) {
        _rOwned[owner()] = _rTotal;

        _setRouter(router);
        _marketingWalletAddress = marketingWallet;
        _msWalletAddress = msWallet;
        _vetWalletAddress = vetWallet;

        // Exclude owner, dev wallet, liq wallet, and this contract from fee
        _isExcludedFromFee[owner()] = true;
        _isExcludedFromFee[_marketingWalletAddress] = true;
        _isExcludedFromFee[_msWalletAddress] = true;
        _isExcludedFromFee[_vetWalletAddress] = true;
        _isExcludedFromFee[address(this)] = true;

        emit Transfer(address(0), owner(), _tTotal);
    }

    function name() external pure returns (string memory) {
        return _name;
    }

    function symbol() external pure returns (string memory) {
        return _symbol;
    }

    function decimals() external pure returns (uint8) {
        return _decimals;
    }

    function totalSupply() external pure override returns (uint256) {
        return _tTotal;
    }

    function balanceOf(address account) public view override returns (uint256) {
        if (_isExcluded[account]) return _tOwned[account];
        return tokenFromReflection(_rOwned[account]);
    }

    function transfer(address recipient, uint256 amount)
        external
        override
        returns (bool)
    {
        _transfer(_msgSender(), recipient, amount);
        return true;
    }

    function allowance(address owner, address spender)
        external
        view
        override
        returns (uint256)
    {
        return _allowances[owner][spender];
    }

    function approve(address spender, uint256 amount)
        external
        override
        returns (bool)
    {
        _approve(_msgSender(), spender, amount);
        return true;
    }

    function transferFrom(
        address sender,
        address recipient,
        uint256 amount
    ) external override returns (bool) {
        _transfer(sender, recipient, amount);
        _approve(
            sender,
            _msgSender(),
            _allowances[sender][_msgSender()] - amount
        );
        return true;
    }

    function increaseAllowance(address spender, uint256 addedValue)
        external
        virtual
        returns (bool)
    {
        _approve(
            _msgSender(),
            spender,
            _allowances[_msgSender()][spender] + addedValue
        );
        return true;
    }

    function decreaseAllowance(address spender, uint256 subtractedValue)
        external
        virtual
        returns (bool)
    {
        _approve(
            _msgSender(),
            spender,
            _allowances[_msgSender()][spender] - subtractedValue
        );
        return true;
    }

    function isExcludedFromReward(address account)
        external
        view
        returns (bool)
    {
        return _isExcluded[account];
    }

    function totalFees() external view returns (uint256) {
        return _tFeeTotal;
    }

    function reflectionFromToken(uint256 tAmount, bool deductTransferFee)
        external
        view
        returns (uint256)
    {
        require(tAmount <= _tTotal, "Amount must be less than supply");
        if (!deductTransferFee) {
            (uint256 rAmount, , , , , , ) = _getValues(tAmount);
            return rAmount;
        } else {
            (, uint256 rTransferAmount, , , , , ) = _getValues(tAmount);
            return rTransferAmount;
        }
    }

    function tokenFromReflection(uint256 rAmount)
        public
        view
        returns (uint256)
    {
        require(
            rAmount <= _rTotal,
            "Amount must be less than total reflections"
        );
        uint256 currentRate = _getRate();
        return rAmount / (currentRate);
    }

    function excludeFromReward(address account) external onlyOwner {
        require(!_isExcluded[account], "Account is not excluded");
        if (_rOwned[account] > 0) {
            _tOwned[account] = tokenFromReflection(_rOwned[account]);
        }
        _isExcluded[account] = true;
        _excluded.push(account);
        emit ExcludeFromReward(account);
    }

    function includeInReward(address account) external onlyOwner {
        require(_isExcluded[account], "Account is already included");
        for (uint256 i = 0; i < _excluded.length; i++) {
            if (_excluded[i] == account) {
                _excluded[i] = _excluded[_excluded.length - 1];
                _tOwned[account] = 0;
                _isExcluded[account] = false;
                _excluded.pop();
                break;
            }
        }
        emit IncludeInReward(account);
    }

    function _transferBothExcluded(
        address sender,
        address recipient,
        uint256 tAmount
    ) private {
        (
            uint256 rAmount,
            uint256 rTransferAmount,
            uint256 rFee,
            uint256 tTransferAmount,
            uint256 tFee,
            uint256 tMarketing,
            uint256 tBurn
        ) = _getValues(tAmount);
        _tOwned[sender] = _tOwned[sender] - (tAmount);
        _rOwned[sender] = _rOwned[sender] - (rAmount);
        _tOwned[recipient] = _tOwned[recipient] + (tTransferAmount);
        _rOwned[recipient] = _rOwned[recipient] + (rTransferAmount);
        _takeMS(tFee);
        _takeVet(tFee);
        _takeMarketing(tMarketing);
        _takeBurn(tBurn);
        _reflectFee(rFee, tFee);
        emit Transfer(sender, recipient, tTransferAmount);
    }

    function excludeFromFee(address account) external onlyOwner {
        _isExcludedFromFee[account] = true;
        emit ExcludeFromFee(account);
    }

    function includeInFee(address account) external onlyOwner {
        _isExcludedFromFee[account] = false;
        emit IncludeInFee(account);
    }

    function setMaxTxPercent(uint256 maxTxPercent) external onlyOwner {
        _maxTxAmount = (_tTotal * (maxTxPercent)) / (10**2);
        emit SetMaxTxPercent(maxTxPercent);
    }

    function setSwapEnabled(bool _enabled) external onlyOwner {
        swapEnabled = _enabled;
        emit SetSwapEnabled(_enabled);
    }

    function setMarketingWalletAddress(address _address) external onlyOwner {
        require(
            _address != address(0),
            "Error: devWallet address cannot be zero address"
        );
        _marketingWalletAddress = _address;
        emit SetMarketingWalletAddress(_address);
    }

    function setMSWalletAddress(address _address) external onlyOwner {
        require(
            _address != address(0),
            "Error: msWallet address cannot be zero address"
        );
        _msWalletAddress = _address;
        emit SetMSWalletAddress(_address);
    }

    function setVetWalletAddress(address _address) external onlyOwner {
        require(
            _address != address(0),
            "Error: vetWallet address cannot be zero address"
        );
        _vetWalletAddress = _address;
        emit SetVetWalletAddress(_address);
    }

    function setContractFeesEnabled(bool _bool) external onlyOwner {
        _contractFeesEnabled = _bool;
        emit SetContractFeesEnabled(_bool);
    }

    function _setRouter(address _router) private {
        IUniswapV2Router02 _uniswapV2Router = IUniswapV2Router02(_router);
        uniswapV2Pair = IUniswapV2Factory(_uniswapV2Router.factory()).getPair(
            address(this),
            _uniswapV2Router.WETH()
        );
        if (uniswapV2Pair == address(0))
            uniswapV2Pair = IUniswapV2Factory(_uniswapV2Router.factory())
                .createPair(address(this), _uniswapV2Router.WETH());
        uniswapV2Router = _uniswapV2Router;
        setIsPair(uniswapV2Pair, true);
    }

    // to receive ETH from uniswapV2Router when swapping
    receive() external payable {}

    // static reflection
    function _reflectFee(uint256 rFee, uint256 tFee) private {
        _rTotal = _rTotal - (rFee);
        _tFeeTotal = _tFeeTotal + (tFee);
    }

    function _getValues(uint256 tAmount)
        private
        view
        returns (
            uint256,
            uint256,
            uint256,
            uint256,
            uint256,
            uint256,
            uint256
        )
    {
        (
            uint256 tTransferAmount,
            uint256 tFee,
            uint256 tMarketing,
            uint256 tBurn
        ) = _getTValues(tAmount);
        (uint256 rAmount, uint256 rTransferAmount, uint256 rFee) = _getRValues(
            tAmount,
            tFee,
            tMarketing,
            tBurn,
            _getRate()
        );
        return (
            rAmount,
            rTransferAmount,
            rFee,
            tTransferAmount,
            tFee,
            tMarketing,
            tBurn
        );
    }

    function _getTValues(uint256 tAmount)
        private
        view
        returns (
            uint256,
            uint256,
            uint256,
            uint256
        )
    {
        uint256 tFee = calculateReflectFee(tAmount);
        uint256 tMarketing = calculateMarketingFee(tAmount);
        uint256 tBurn = calculateBurnFee(tAmount);
        uint256 tTransferAmount = tAmount - (3 * tFee) - tMarketing - tBurn;
        return (tTransferAmount, tFee, tMarketing, tBurn);
    }

    function _getRValues(
        uint256 tAmount,
        uint256 tFee,
        uint256 tMarketing,
        uint256 tBurn,
        uint256 currentRate
    )
        private
        pure
        returns (
            uint256,
            uint256,
            uint256
        )
    {
        uint256 rAmount = tAmount * (currentRate);
        uint256 rFee = tFee * (currentRate);
        uint256 rMarketing = tMarketing * (currentRate);
        uint256 rBurn = tBurn * (currentRate);

        uint256 rTransferAmount = rAmount - (3 * rFee) - rMarketing - rBurn;
        return (rAmount, rTransferAmount, rFee);
    }

    function _getRate() private view returns (uint256) {
        (uint256 rSupply, uint256 tSupply) = _getCurrentSupply();
        return rSupply / (tSupply);
    }

    function _getCurrentSupply() private view returns (uint256, uint256) {
        uint256 rSupply = _rTotal;
        uint256 tSupply = _tTotal;
        for (uint256 i = 0; i < _excluded.length; i++) {
            if (
                _rOwned[_excluded[i]] > rSupply ||
                _tOwned[_excluded[i]] > tSupply
            ) return (_rTotal, _tTotal);
            rSupply = rSupply - (_rOwned[_excluded[i]]);
            tSupply = tSupply - (_tOwned[_excluded[i]]);
        }
        if (rSupply < _rTotal / (_tTotal)) return (_rTotal, _tTotal);
        return (rSupply, tSupply);
    }

    function _takeMarketing(uint256 tMarketing) private {
        uint256 currentRate = _getRate();
        uint256 rMarketing = tMarketing * (currentRate);
        _rOwned[_marketingWalletAddress] =
            _rOwned[_marketingWalletAddress] +
            (rMarketing);
        if (_isExcluded[_marketingWalletAddress])
            _tOwned[_marketingWalletAddress] =
                _tOwned[_marketingWalletAddress] +
                (tMarketing);
    }

    function _takeBurn(uint256 tBurn) private {
        uint256 currentRate = _getRate();
        uint256 rBurn = tBurn * (currentRate);
        _rOwned[_deadAddress] = _rOwned[_deadAddress] + (rBurn);
        if (_isExcluded[_deadAddress])
            _tOwned[_deadAddress] = _tOwned[_deadAddress] + (tBurn);
    }

    // addition - take MS cut
    // 2%
    function _takeMS(uint256 tMS) private {
        uint256 currentRate = _getRate();
        uint256 rMS = tMS * (currentRate);
        _rOwned[_msWalletAddress] = _rOwned[_msWalletAddress] + (rMS);
        if (_isExcluded[_msWalletAddress])
            _tOwned[_msWalletAddress] = _tOwned[_msWalletAddress] + (tMS);
    }

    // addition - take vet cut
    // 2%
    function _takeVet(uint256 tVet) private {
        uint256 currentRate = _getRate();
        uint256 rVet = tVet * (currentRate);
        _rOwned[_vetWalletAddress] = _rOwned[_vetWalletAddress] + (rVet);
        if (_isExcluded[_vetWalletAddress])
            _tOwned[_vetWalletAddress] = _tOwned[_vetWalletAddress] + (tVet);
    }

    function calculateReflectFee(uint256 _amount)
        private
        view
        returns (uint256)
    {
        return (_amount * (_taxFee)) / (10**3);
    }

    function calculateMarketingFee(uint256 _amount)
        private
        view
        returns (uint256)
    {
        return (_amount * (_marketingFee)) / (10**3);
    }

    function calculateBurnFee(uint256 _amount) private view returns (uint256) {
        return (_amount * (_burnFee)) / (10**3);
    }

    function removeAllFee() private {
        _taxFee = 0;
        _burnFee = 0;
        _marketingFee = 0;
    }

    function addAllFee() private {
        _taxFee = 20;
        _burnFee = 10;
        _marketingFee = 10;
    }

    function isExcludedFromFee(address account) external view returns (bool) {
        return _isExcludedFromFee[account];
    }

    function _approve(
        address owner,
        address spender,
        uint256 amount
    ) private {
        require(owner != address(0), "ERC20: approve from the zero address");
        require(spender != address(0), "ERC20: approve to the zero address");

        _allowances[owner][spender] = amount;
        emit Approval(owner, spender, amount);
    }

    function _transfer(
        address from,
        address to,
        uint256 amount
    ) private {
        require(from != address(0), "ERC20: transfer from the zero address");
        require(to != address(0), "ERC20: transfer to the zero address");
        require(amount > 0, "Transfer amount must be greater than zero");
        if (from != owner() && to != owner())
            require(
                amount <= _maxTxAmount,
                "Transfer amount exceeds the maxTxAmount."
            );

        if (!swapEnabled && (_isPair[to] || _isPair[from]))
            revert("Buying and selling is disabled");

        if (_isBanned[from] || _isBanned[to]) {
            revert("Address is banned");
        }

        // Indicates if fee should be deducted from transfer
        bool takeFee = true;

        // Remove fees except for buying and selling
        if (!_isPair[from] && !_isPair[to]) {
            takeFee = false;
        }

        // Enable fees if contract fees are enabled and to or from is a contract
        if (_contractFeesEnabled && (from.isContract() || to.isContract())) {
            takeFee = true;
        }

        // If any account belongs to _isExcludedFromFee account then remove the fee
        if (_isExcludedFromFee[from] || _isExcludedFromFee[to]) {
            takeFee = false;
        }

        if (takeFee) addAllFee();

        // Transfer amount, it will take tax, burn, liquidity fee
        _tokenTransfer(from, to, amount, takeFee);
    }

    //this method is responsible for taking all fee, if takeFee is true
    function _tokenTransfer(
        address sender,
        address recipient,
        uint256 amount,
        bool takeFee
    ) private {
        if (!takeFee) removeAllFee();

        if (_isExcluded[sender] && !_isExcluded[recipient]) {
            _transferFromExcluded(sender, recipient, amount);
        } else if (!_isExcluded[sender] && _isExcluded[recipient]) {
            _transferToExcluded(sender, recipient, amount);
        } else if (_isExcluded[sender] && _isExcluded[recipient]) {
            _transferBothExcluded(sender, recipient, amount);
        } else {
            _transferStandard(sender, recipient, amount);
        }
    }

    function _transferStandard(
        address sender,
        address recipient,
        uint256 tAmount
    ) private {
        (
            uint256 rAmount,
            uint256 rTransferAmount,
            uint256 rFee,
            uint256 tTransferAmount,
            uint256 tFee,
            uint256 tMarketing,
            uint256 tBurn
        ) = _getValues(tAmount);
        _rOwned[sender] = _rOwned[sender] - (rAmount);
        _rOwned[recipient] = _rOwned[recipient] + (rTransferAmount);
        _takeMarketing(tMarketing);
        _takeBurn(tBurn);
        _takeMS(tFee);
        _takeVet(tFee);
        _reflectFee(rFee, tFee);
        emit Transfer(sender, recipient, tTransferAmount);
    }

    function _transferToExcluded(
        address sender,
        address recipient,
        uint256 tAmount
    ) private {
        (
            uint256 rAmount,
            uint256 rTransferAmount,
            uint256 rFee,
            uint256 tTransferAmount,
            uint256 tFee,
            uint256 tMarketing,
            uint256 tBurn
        ) = _getValues(tAmount);
        _rOwned[sender] = _rOwned[sender] - (rAmount);
        _tOwned[recipient] = _tOwned[recipient] + (tTransferAmount);
        _rOwned[recipient] = _rOwned[recipient] + (rTransferAmount);
        _takeMarketing(tMarketing);
        _takeMS(tFee);
        _takeBurn(tBurn);
        _takeVet(tFee);
        _reflectFee(rFee, tFee);
        emit Transfer(sender, recipient, tTransferAmount);
    }

    function _transferFromExcluded(
        address sender,
        address recipient,
        uint256 tAmount
    ) private {
        (
            uint256 rAmount,
            uint256 rTransferAmount,
            uint256 rFee,
            uint256 tTransferAmount,
            uint256 tFee,
            uint256 tMarketing,
            uint256 tBurn
        ) = _getValues(tAmount);
        _tOwned[sender] = _tOwned[sender] - (tAmount);
        _rOwned[sender] = _rOwned[sender] - (rAmount);
        _rOwned[recipient] = _rOwned[recipient] + (rTransferAmount);
        _takeMarketing(tMarketing);
        _takeMS(tFee);
        _takeBurn(tBurn);
        _takeVet(tFee);
        _reflectFee(rFee, tFee);
        emit Transfer(sender, recipient, tTransferAmount);
    }

    function setIsPair(address _address, bool value) public onlyOwner {
        _isPair[_address] = value;
        emit SetIsPair(_address, value);
    }

    function setIsBanned(address _address, bool value) external onlyOwner {
        require(
            _address.isContract(),
            "Error: Can only ban or unban contract addresses"
        );
        _isBanned[_address] = value;
        emit SetIsBanned(_address, value);
    }

    function withdrawalToken(
        address _tokenAddr,
        uint256 _amount,
        address to
    ) external onlyOwner {
        IERC20 token = IERC20(_tokenAddr);
        token.transfer(to, _amount);
        emit WithdrawalToken(_tokenAddr, _amount, to);
    }

    function withdrawalETH(uint256 _amount, address to) external onlyOwner {
        require(address(this).balance >= _amount);
        payable(to).transfer(_amount);
        emit WithdrawalETH(_amount, to);
    }
}
function _reflectFee(uint256 rFee, uint256 tFee) private {
    _rTotal = _rTotal - (rFee);
    _tFeeTotal = _tFeeTotal + (tFee);
}

// static reflection