Move document logic to a class other sources can use

Move document logic to a class other sources can use #392