Simplify code that gets source generated or Razor documents in cohosting

[davidwengier]

Follow up for #12614, see this comment in particular: #12614 (comment)

Now that the bug is fixed and we have tests etc. lets review the APIs we have and see if there is some simplifications possible, as Todd suggested.