There's a great fiction series called "The Regeneration Trilogy" in which the story of World War I is told from the perspective of W. H. R. Rivers, a real-life British psychiatrist who treated soldiers who returned from the front with "shell shock."

It's a great read on numerous fronts, and digs pretty deep into how soldiers were treated once they were back home.

https://en.m.wikipedia.org/wiki/Regeneration_(novel)