Приладил... полез внутрь, проверить нельзя ли всёже отрефакторить... всё ещё хуже чем мне казалось, там ещё один слой кода :(
А для стыковки между этими слоями костыли, костыли, костыли... ёёёёё