That's a great idea, I think that's the best option. I haven't looked into the Factorio modding API, but I think it should be possible with mod.