diagnostics: Refactor diagnostics removal

Instead of doing it in 3 places, move the logic to DocumentDiagnosticItem.
This allows us to reuse the logic and even make things a bit faster.


(cherry picked from commit 507384f4)