Steve Smith will leave Sri Lanka early to focus on South Africa tour with David Warner to take over the one-day and T20 captaincy, reports Cricket Australia.