<?php

add_filter('ms_frontend_handle_registration', '__return_false');


// Allow gravity forms registration to pass its information to the membership plugin to add new accounts into the membership plugin as users
// See here for more: https://premium.wpmudev.org/forums/topic/using-gravity-forms-with-membership-to-handle-registration#post-1148550

add_action( 'set_user_role', 'assign_membership_on_register', 10, 1 );
function assign_membership_on_register( $user_id ) {

    $membership = array(
        // 'role' => 'membership ID'
        'company' => 1976,
        'employee' => 1977,
		'vendor' => 1983
    );

  $user_meta = get_userdata( $user_id );
    $user_roles = $user_meta->roles;

    $member = MS_Factory::load( 'MS_Model_Member', $user_id );
    $subscription = $member->add_membership( $membership[ $user_roles[0] ] );

    if ( $member->has_membership() ) {
        $member->is_member = true;
    } else {
        $member->is_member = false;
    }
    $member->save();
}