How do I use a JButton's Model?

Java Swing builds on the (approximate) concept of MVC and its UI components usually have some kind of Model associated to them.

So what is the Model for a JButton? I'm not very familiar to Java Swing. Is it an instance of the javax.swing.DefaultButtonModel class? Or an implementation of the javax.swing.ButtonModel interface?

Also, some people say we should rarely extend a Java Swing component.

So within that spirit I have the concept of a "piece button" in a simple color-guessing game which is a button tied to a Piece (the piece simply represents a single color, I could as well have used Color instead). It should not extend JButton because it is not modifying any of its functionality. Also using composition doesn't seem to make sense in that case.

So what should I do? Should I add Piece to the button's Model? If so, how should I do that?

Or should I ignore the previous statements and extend JButton instead? I would then have something like this:

public final class PieceButton extends JButton {
    private static final long serialVersionUID = 1L;
    
    private final Piece piece;
    public PieceButton(Piece piece, int iconSize) {
        this.piece = Objects.requireNonNull(piece);
        setIcon(new ColorRoundIcon(iconSize, piece.getColor()));
    }
    
    public Piece getPiece() {
        return piece;
    }
}

So what is the best (OOP-wise) approach?