I ended up making a very similar mod, although since I don't have any icons to use, I ended up reusing the vanilla mining drill lamp icon. The problem is that it's sometimes hard to see... Do you have any tips on that? I considered using the train station white/black icon, but I don't know how light types interact with that.
Also, I looked at the code, and I think there's a trick with string.find () you would find useful... There's an optional 4th parameter, which if set to true, ignores special characters, and looks at the string raw. Useful to match names with dashes in them.