Auto join group based on code

Is there a method to join users to certain group based on registration code? Just build simple if else based on this: https://github.com/eggemplo/Users-auto-Group/blob/master/function.php for example?

https://wordpress.org/plugins/groups/