This is how you deal with misbehaving children. Define a boundary (don’t do X), establish consequences (If you do X, I will do Y), and then follow through consistently, exactly as you described.

Ideally you don’t have to treat world leaders like they’re children, but these are the times we find ourselves in.