5.FancyShirts.sol

// SPDX-License-Identifier: MIT
pragma solidity 0.8.25;

contract FancyShirts {
    enum Size {
        Small,
        Medium,
        Large
    }

    enum Color {
        Red,
        Green,
        Blue
    }

    struct Shirt {
        Size size;
        Color color;
    }

    mapping(address => Shirt[]) shirts;

    modifier correctAmount(Size size, Color color) {
        require(
            getShirtPrice(size, color) == msg.value,
            "Incorrect amount sent"
        );
        _;
    }

    function getShirtPrice(Size size, Color color)
        public
        pure
        returns (uint256)
    {
        uint256 price;
        if (size == Size.Small) {
            price += 10;
        } else if (size == Size.Medium) {
            price += 15;
        } else {
            price += 20;
        }

        if (color != Color.Red) {
            price += 5;
        }

        return price;
    }

    function buyShirt(Size size, Color color)
        public
        payable
        correctAmount(size, color)
    {
        Shirt memory shirt = Shirt(size, color);
        shirts[msg.sender].push(shirt);
    }

    function getShirts(Size size, Color color) public view returns (uint256) {
        uint256 count;

        for (uint256 idx; idx < shirts[msg.sender].length; idx++) {
            if (
                shirts[msg.sender][idx].size == size &&
                shirts[msg.sender][idx].color == color
            ) {
                count++;
            }
        }
        return count;
    }
}