Expand to other entity types like 'Media' instead of just node types

Right now you can only use this module to set permissions to view unpublished content per 'node' type.
It would be great if we could also do this for Media entities, so we would have permissions like:
Media: Image: View any unpublished entities
Media: Video: View any unpublished entities
etc.

Maybe create a submodule for this functionality ?