Total correlations of the diagonal ensemble herald the many-body localization transition