Text this: Diffusion of heavy quarks in the early stage of high-energy nuclear collisions