How do I "adopt" Rigify's IK/FK snapping to my own rig?

I’d like to have IK/FK snapping functionality on arms/legs in my own rig. I know Rigify has that feature and I am wondering how can I rip out that part of the script and use it with my own rig.

Thanks!