maybe what you want to do is restrict transfer where to or from is the pair address or router address

may want unlock button on there in case something wrong with buy/sell, allow staking etc