Politically it would be ideal to keep everything in Taiwan as this greatly increases the stakes for a potential war.
However there is a pressing shortage of electricity in Taiwan. Just yesterday there had been a major blackout as one of the coal fired power stations suddenly went off the grid. I'm sure TSMC has been given priority supply but they are cutting it extremely close. Taiwan is just not a good place to have more fabs right now.
The shortage is unlikely to get better in the long term, with existing nuclear power stations nearing the end of their service life and the replacement stuck in political limbo. And there are already plans to move all 28nm+ process to China so they could better utilize the resources in Taiwan for the more profitable products.
https://amp.scmp.com/tech/big-tech/article/3131823/why-has-t...