Fork https://mods.factorio.com/mod/automatic-train-and-station-painter has a function to color stations too. As I can see there is no problem of integrating this function to a base mod (if I understood code correctly). I'd do it myself but I don't want to create third mod with almost the same code and almost the same function.
Please, consider updating your mod with to paint stations too especially when there is ready-to-use code for this.
Thank you!