This discord bot can automatically assign a role to a new user based on invitation link value. It is made with Go and can be used for your own discord bot. You can freely take it and modify it to your needs. Since the API token is your own, this script should work if you add a credentials.json and invite-roles.json.

React Roles is a simple discord bot for allowing self-managing user roles in discord.