John domains for the control distance of diagonal vector fields