SSO sign up with referral id

This takes us to sign up form where there is a hidden field for referral and is populated using JS.

The is then captured by an observer and processes it.

Now, my question is how can I achieve the same using an third party identify provider such as Azure IdP in or there are any better way to do this.