/
class.llms.controller.registration.php
109 lines (82 loc) · 2.47 KB
/
class.llms.controller.registration.php
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
<?php
/**
* User Registration Forms
*
* @package LifterLMS/Forms/Controllers/Classes
*
* @since 3.0.0
* @version 3.24.0
*/
defined( 'ABSPATH' ) || exit;
/**
* User Registration Forms (excludes checkout registration)
*
* @since 3.0.0
*/
class LLMS_Controller_Registration {
/**
* Constructor
*
* @since 3.0.0
*
* @return void
*/
public function __construct() {
add_action( 'init', array( $this, 'register' ) );
add_action( 'lifterlms_user_registered', array( $this, 'voucher' ), 10, 3 );
}
/**
* Attempt to redeem a voucher on user registration if a voucher was submitted during registration
*
* @since 3.0.0
* @since 3.19.4 Unknown.
*
* @param int $person_id WP_User ID of the newly registered user.
* @param array $data $_POST data.
* @param string $screen Screen user registered from [checkout|registration].
* @return void
*/
public function voucher( $person_id, $data, $screen ) {
if ( 'registration' === $screen && ! empty( $data['llms_voucher'] ) ) {
$voucher = new LLMS_Voucher();
$redeemed = $voucher->use_voucher( $data['llms_voucher'], $person_id );
if ( is_wp_error( $redeemed ) ) {
llms_add_notice( $redeemed->get_error_message(), 'error' );
}
}
}
/**
* Handle submission of user registration forms
*
* @since 3.0.0
* @since 3.24.0 Unknown.
*
* @return void
*/
public function register() {
if ( ! llms_verify_nonce( '_llms_register_person_nonce', 'llms_register_person' ) ) {
return;
}
do_action( 'lifterlms_before_new_user_registration' );
// Already logged in can't register!
// This shouldn't happen but let's check anyway.
if ( get_current_user_id() ) {
return llms_add_notice( __( 'Already logged in! Please log out and try again.', 'lifterlms' ), 'error' );
}
$person_id = llms_register_user( $_POST, 'registration', true );
// Validation or registration issues.
if ( is_wp_error( $person_id ) ) {
foreach ( $person_id->get_error_messages() as $msg ) {
llms_add_notice( $msg, 'error' );
}
return;
} elseif ( ! is_numeric( $person_id ) ) {
// Catch unexpected returns from llms_register_user().
return llms_add_notice( __( 'An unknown error occurred when attempting to create an account, please try again.', 'lifterlms' ), 'error' );
} else {
// Handle redirect.
llms_redirect_and_exit( apply_filters( 'lifterlms_registration_redirect', llms_get_page_url( 'myaccount' ) ) );
}
}
}
return new LLMS_Controller_Registration();