Changing the color of blocks using data from Block Entity instead of Blockstate 1.20.1

Hey. In my mod I am changing the texture color depending on custom enum property. I do it normally through extends BlockColors (RegisterColorHandlersEvent) and everything works ok, however, I have quite a lot of colors and also other properties which increases the number of blockstats very quickly. 

E.g. Only one type of leaves generates DISTANCE * PERSISTENT * WATERLOGGED * COLOR * One More State = 7*2*2*56*2 = 3156. Few blocks have one more new state and its *2 or even *3. So 6272/9408

And with each successive block that uses this method, another large number of new blockstats come in. So I found that it would be better to do it through a block entity and save the data that way. That's what I did, the data saves fine, but I can't find a way to use this data to change the color. Can't figure out how to call. I'm losing hope that it's even possible. Would anyone be able to help and suggest something? Is there even a point to such a solution, because maybe it is not possible?

Edited February 24Feb 24 by DanteTooMayCry