It really sounds like this whole bit needs to be a plugin