750 lines
24 KiB
PHP
750 lines
24 KiB
PHP
<?php
|
|
|
|
if ( ! defined( 'ABSPATH' ) ) exit; // Exit if accessed directly
|
|
|
|
/**
|
|
* Handle frontend forms
|
|
*
|
|
* @class WC_Form_Handler
|
|
* @version 2.1.0
|
|
* @package WooCommerce/Classes/
|
|
* @category Class
|
|
* @author WooThemes
|
|
*/
|
|
class WC_Form_Handler {
|
|
|
|
/**
|
|
* Constructor
|
|
*/
|
|
public function __construct() {
|
|
add_action( 'template_redirect', array( $this, 'save_address' ) );
|
|
add_action( 'template_redirect', array( $this, 'save_account_details' ) );
|
|
|
|
add_action( 'init', array( $this, 'checkout_action' ), 20 );
|
|
add_action( 'init', array( $this, 'process_login' ) );
|
|
add_action( 'init', array( $this, 'process_registration' ) );
|
|
add_action( 'init', array( $this, 'process_reset_password' ) );
|
|
|
|
add_action( 'init', array( $this, 'cancel_order' ) );
|
|
add_action( 'init', array( $this, 'order_again' ) );
|
|
|
|
add_action( 'init', array( $this, 'update_cart_action' ) );
|
|
add_action( 'init', array( $this, 'add_to_cart_action' ) );
|
|
|
|
add_action( 'wp', array( $this, 'pay_action' ), 20 );
|
|
}
|
|
|
|
/**
|
|
* Save and and update a billing or shipping address if the
|
|
* form was submitted through the user account page.
|
|
*/
|
|
public function save_address() {
|
|
global $woocommerce, $wp;
|
|
|
|
if ( 'POST' !== strtoupper( $_SERVER[ 'REQUEST_METHOD' ] ) )
|
|
return;
|
|
|
|
if ( empty( $_POST[ 'action' ] ) || ( 'edit_address' !== $_POST[ 'action' ] ) )
|
|
return;
|
|
|
|
wp_verify_nonce( $_POST['_wpnonce'], 'woocommerce-edit_address' );
|
|
|
|
$validation = $woocommerce->validation();
|
|
|
|
$user_id = get_current_user_id();
|
|
|
|
if ( $user_id <= 0 ) return;
|
|
|
|
$load_address = ( $wp->query_vars['edit-address'] == 'billing' || $wp->query_vars['edit-address'] == 'shipping' ) ? $wp->query_vars['edit-address'] : 'billing';
|
|
|
|
$address = $woocommerce->countries->get_address_fields( esc_attr( $_POST[ $load_address . '_country' ] ), $load_address . '_' );
|
|
|
|
foreach ($address as $key => $field) :
|
|
|
|
if (!isset($field['type'])) $field['type'] = 'text';
|
|
|
|
// Get Value
|
|
switch ($field['type']) :
|
|
case "checkbox" :
|
|
$_POST[$key] = isset($_POST[$key]) ? 1 : 0;
|
|
break;
|
|
default :
|
|
$_POST[$key] = isset($_POST[$key]) ? woocommerce_clean($_POST[$key]) : '';
|
|
break;
|
|
endswitch;
|
|
|
|
// Hook to allow modification of value
|
|
$_POST[$key] = apply_filters('woocommerce_process_myaccount_field_' . $key, $_POST[$key]);
|
|
|
|
// Validation: Required fields
|
|
if ( isset($field['required']) && $field['required'] && empty($_POST[$key]) ) wc_add_error( $field['label'] . ' ' . __( 'is a required field.', 'woocommerce' ) );
|
|
|
|
// Postcode
|
|
if ($key=='billing_postcode' || $key=='shipping_postcode') :
|
|
if ( ! $validation->is_postcode( $_POST[$key], $_POST[ $load_address . '_country' ] ) ) :
|
|
wc_add_error( __( 'Please enter a valid postcode/ZIP.', 'woocommerce' ) );
|
|
else :
|
|
$_POST[$key] = $validation->format_postcode( $_POST[$key], $_POST[ $load_address . '_country' ] );
|
|
endif;
|
|
endif;
|
|
|
|
endforeach;
|
|
|
|
if ( wc_error_count() == 0 ) {
|
|
|
|
foreach ($address as $key => $field) :
|
|
update_user_meta( $user_id, $key, $_POST[$key] );
|
|
endforeach;
|
|
|
|
wc_add_message( __( 'Address changed successfully.', 'woocommerce' ) );
|
|
|
|
do_action( 'woocommerce_customer_save_address', $user_id );
|
|
|
|
wp_safe_redirect( get_permalink( woocommerce_get_page_id('myaccount') ) );
|
|
exit;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Save the password/account details and redirect back to the my account page.
|
|
*/
|
|
public function save_account_details() {
|
|
global $woocommerce;
|
|
|
|
if ( 'POST' !== strtoupper( $_SERVER[ 'REQUEST_METHOD' ] ) )
|
|
return;
|
|
|
|
if ( empty( $_POST[ 'action' ] ) || ( 'save_account_details' !== $_POST[ 'action' ] ) )
|
|
return;
|
|
|
|
wp_verify_nonce( $_POST['_wpnonce'], 'woocommerce-save_account_details' );
|
|
|
|
$update = true;
|
|
$errors = new WP_Error();
|
|
$user = new stdClass();
|
|
|
|
$user->ID = (int) get_current_user_id();
|
|
$current_user = get_user_by( 'id', $user->ID );
|
|
|
|
if ( $user->ID <= 0 )
|
|
return;
|
|
|
|
$account_first_name = ! empty( $_POST[ 'account_first_name' ] ) ? woocommerce_clean( $_POST[ 'account_first_name' ] ) : '';
|
|
$account_last_name = ! empty( $_POST[ 'account_last_name' ] ) ? woocommerce_clean( $_POST[ 'account_last_name' ] ) : '';
|
|
$account_email = ! empty( $_POST[ 'account_email' ] ) ? woocommerce_clean( $_POST[ 'account_email' ] ) : '';
|
|
$pass1 = ! empty( $_POST[ 'password_1' ] ) ? woocommerce_clean( $_POST[ 'password_1' ] ) : '';
|
|
$pass2 = ! empty( $_POST[ 'password_2' ] ) ? woocommerce_clean( $_POST[ 'password_2' ] ) : '';
|
|
|
|
$user->first_name = $account_first_name;
|
|
$user->last_name = $account_last_name;
|
|
$user->user_email = $account_email;
|
|
$user->display_name = $user->first_name;
|
|
|
|
if ( $pass1 )
|
|
$user->user_pass = $pass1;
|
|
|
|
if ( empty( $account_first_name ) || empty( $account_last_name ) )
|
|
wc_add_error( __( 'Please enter your name.', 'woocommerce' ) );
|
|
|
|
if ( empty( $account_email ) || ! is_email( $account_email ) )
|
|
wc_add_error( __( 'Please provide a valid email address.', 'woocommerce' ) );
|
|
|
|
elseif ( email_exists( $account_email ) && $account_email !== $current_user->user_email )
|
|
wc_add_error( __( 'This email address is already registered.', 'woocommerce' ) );
|
|
|
|
if ( ! empty( $pass1 ) && empty( $pass2 ) )
|
|
wc_add_error( __( 'Please re-enter your password.', 'woocommerce' ) );
|
|
|
|
elseif ( ! empty( $pass1 ) && $pass1 !== $pass2 )
|
|
wc_add_error( __( 'Passwords do not match.', 'woocommerce' ) );
|
|
|
|
// Allow plugins to return their own errors.
|
|
do_action_ref_array( 'user_profile_update_errors', array ( &$errors, $update, &$user ) );
|
|
|
|
if ( $errors->get_error_messages() )
|
|
foreach( $errors->get_error_messages() as $error )
|
|
wc_add_error( $error );
|
|
|
|
if ( wc_error_count() == 0 ) {
|
|
|
|
wp_update_user( $user ) ;
|
|
|
|
wc_add_message( __( 'Account details changed successfully.', 'woocommerce' ) );
|
|
|
|
do_action( 'woocommerce_save_account_details', $user->ID );
|
|
|
|
wp_safe_redirect( get_permalink( woocommerce_get_page_id( 'myaccount' ) ) );
|
|
exit;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Process the checkout form.
|
|
*/
|
|
public function checkout_action() {
|
|
if ( isset( $_POST['woocommerce_checkout_place_order'] ) || isset( $_POST['woocommerce_checkout_update_totals'] ) ) {
|
|
|
|
if ( sizeof( WC()->cart->get_cart() ) == 0 ) {
|
|
wp_redirect( get_permalink( woocommerce_get_page_id( 'cart' ) ) );
|
|
exit;
|
|
}
|
|
|
|
if ( ! defined( 'WOOCOMMERCE_CHECKOUT' ) )
|
|
define( 'WOOCOMMERCE_CHECKOUT', true );
|
|
|
|
$woocommerce_checkout = WC()->checkout();
|
|
$woocommerce_checkout->process_checkout();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Process the pay form.
|
|
*/
|
|
public function pay_action() {
|
|
global $wp;
|
|
|
|
if ( isset( $_POST['woocommerce_pay'] ) && wp_verify_nonce( $_POST['_wpnonce'], 'woocommerce-pay' ) ) {
|
|
|
|
ob_start();
|
|
|
|
// Pay for existing order
|
|
$order_key = urldecode( $_GET['key'] );
|
|
$order_id = absint( $wp->query_vars['order-pay'] );
|
|
$order = new WC_Order( $order_id );
|
|
|
|
if ( $order->id == $order_id && $order->order_key == $order_key && in_array( $order->status, array( 'pending', 'failed' ) ) ) {
|
|
|
|
// Set customer location to order location
|
|
if ( $order->billing_country )
|
|
WC()->customer->set_country( $order->billing_country );
|
|
if ( $order->billing_state )
|
|
WC()->customer->set_state( $order->billing_state );
|
|
if ( $order->billing_postcode )
|
|
WC()->customer->set_postcode( $order->billing_postcode );
|
|
if ( $order->billing_city )
|
|
WC()->customer->set_city( $order->billing_city );
|
|
|
|
// Update payment method
|
|
if ( $order->order_total > 0 ) {
|
|
$payment_method = woocommerce_clean( $_POST['payment_method'] );
|
|
|
|
$available_gateways = WC()->payment_gateways->get_available_payment_gateways();
|
|
|
|
// Update meta
|
|
update_post_meta( $order_id, '_payment_method', $payment_method );
|
|
|
|
if ( isset( $available_gateways[ $payment_method ] ) )
|
|
$payment_method_title = $available_gateways[ $payment_method ]->get_title();
|
|
|
|
update_post_meta( $order_id, '_payment_method_title', $payment_method_title);
|
|
|
|
// Validate
|
|
$available_gateways[ $payment_method ]->validate_fields();
|
|
|
|
// Process
|
|
if ( wc_error_count() == 0 ) {
|
|
|
|
$result = $available_gateways[ $payment_method ]->process_payment( $order_id );
|
|
|
|
// Redirect to success/confirmation/payment page
|
|
if ( $result['result'] == 'success' ) {
|
|
wp_redirect( $result['redirect'] );
|
|
exit;
|
|
}
|
|
|
|
}
|
|
|
|
} else {
|
|
// No payment was required for order
|
|
$order->payment_complete();
|
|
wp_safe_redirect( $order->get_checkout_order_received_url() );
|
|
exit;
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Remove from cart/update.
|
|
*/
|
|
public function update_cart_action() {
|
|
// Remove from cart
|
|
if ( ! empty( $_GET['remove_item'] ) && wp_verify_nonce( $_GET['_wpnonce'], 'woocommerce-cart' ) ) {
|
|
|
|
WC()->cart->set_quantity( $_GET['remove_item'], 0 );
|
|
|
|
wc_add_message( __( 'Cart updated.', 'woocommerce' ) );
|
|
|
|
$referer = ( wp_get_referer() ) ? wp_get_referer() : WC()->cart->get_cart_url();
|
|
wp_safe_redirect( $referer );
|
|
exit;
|
|
|
|
// Update Cart
|
|
} elseif ( ( ! empty( $_POST['update_cart'] ) || ! empty( $_POST['proceed'] ) ) && wp_verify_nonce( $_POST['_wpnonce'], 'woocommerce-cart' ) ) {
|
|
|
|
$cart_totals = isset( $_POST['cart'] ) ? $_POST['cart'] : '';
|
|
|
|
if ( sizeof( WC()->cart->get_cart() ) > 0 ) {
|
|
foreach ( WC()->cart->get_cart() as $cart_item_key => $values ) {
|
|
|
|
$_product = $values['data'];
|
|
|
|
// Skip product if no updated quantity was posted
|
|
if ( ! isset( $cart_totals[$cart_item_key]['qty'] ) )
|
|
continue;
|
|
|
|
// Sanitize
|
|
$quantity = apply_filters( 'woocommerce_stock_amount_cart_item', apply_filters( 'woocommerce_stock_amount', preg_replace( "/[^0-9\.]/", "", $cart_totals[ $cart_item_key ]['qty'] ) ), $cart_item_key );
|
|
|
|
if ( "" === $quantity )
|
|
continue;
|
|
|
|
// Update cart validation
|
|
$passed_validation = apply_filters( 'woocommerce_update_cart_validation', true, $cart_item_key, $values, $quantity );
|
|
|
|
// is_sold_individually
|
|
if ( $_product->is_sold_individually() && $quantity > 1 ) {
|
|
wc_add_error( sprintf( __( 'You can only have 1 %s in your cart.', 'woocommerce' ), $_product->get_title() ) );
|
|
$passed_validation = false;
|
|
}
|
|
|
|
if ( $passed_validation )
|
|
WC()->cart->set_quantity( $cart_item_key, $quantity );
|
|
|
|
}
|
|
}
|
|
|
|
if ( ! empty( $_POST['proceed'] ) ) {
|
|
wp_safe_redirect( WC()->cart->get_checkout_url() );
|
|
exit;
|
|
} else {
|
|
wc_add_message( __( 'Cart updated.', 'woocommerce' ) );
|
|
|
|
$referer = ( wp_get_referer() ) ? wp_get_referer() : WC()->cart->get_cart_url();
|
|
$referer = remove_query_arg( 'remove_discounts', $referer );
|
|
wp_safe_redirect( $referer );
|
|
exit;
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Place a previous order again.
|
|
*/
|
|
public function order_again() {
|
|
global $woocommerce;
|
|
|
|
// Nothing to do
|
|
if ( ! isset( $_GET['order_again'] ) || ! is_user_logged_in() || ! $woocommerce->verify_nonce( 'order_again', '_GET' ) || ! wp_verify_nonce( $_GET['_wpnonce'], 'woocommerce-order_again' ) )
|
|
return;
|
|
|
|
// Clear current cart
|
|
$woocommerce->cart->empty_cart();
|
|
|
|
// Load the previous order - Stop if the order does not exist
|
|
$order = new WC_Order( absint( $_GET['order_again'] ) );
|
|
|
|
if ( empty( $order->id ) )
|
|
return;
|
|
|
|
if ( $order->status != 'completed' )
|
|
return;
|
|
|
|
// Make sure the previous order belongs to the current customer
|
|
if ( $order->user_id != get_current_user_id() )
|
|
return;
|
|
|
|
// Copy products from the order to the cart
|
|
foreach ( $order->get_items() as $item ) {
|
|
// Load all product info including variation data
|
|
$product_id = (int) apply_filters( 'woocommerce_add_to_cart_product_id', $item['product_id'] );
|
|
$quantity = (int) $item['qty'];
|
|
$variation_id = (int) $item['variation_id'];
|
|
$variations = array();
|
|
$cart_item_data = apply_filters( 'woocommerce_order_again_cart_item_data', array(), $item, $order );
|
|
|
|
foreach ( $item['item_meta'] as $meta_name => $meta_value ) {
|
|
if ( taxonomy_is_product_attribute( $meta_name ) )
|
|
$variations[ $meta_name ] = $meta_value[0];
|
|
}
|
|
|
|
// Add to cart validation
|
|
if ( ! apply_filters( 'woocommerce_add_to_cart_validation', true, $product_id, $quantity, $variation_id, $variations, $cart_item_data ) ) continue;
|
|
|
|
$woocommerce->cart->add_to_cart( $product_id, $quantity, $variation_id, $variations, $cart_item_data );
|
|
}
|
|
|
|
do_action( 'woocommerce_ordered_again', $order->id );
|
|
|
|
// Redirect to cart
|
|
wc_add_message( __( 'The cart has been filled with the items from your previous order.', 'woocommerce' ) );
|
|
wp_safe_redirect( $woocommerce->cart->get_cart_url() );
|
|
exit;
|
|
}
|
|
|
|
/**
|
|
* Cancel a pending order.
|
|
*/
|
|
public function cancel_order() {
|
|
|
|
global $woocommerce;
|
|
|
|
if ( isset($_GET['cancel_order']) && isset($_GET['order']) && isset($_GET['order_id']) ) :
|
|
|
|
$order_key = urldecode( $_GET['order'] );
|
|
$order_id = (int) $_GET['order_id'];
|
|
|
|
$order = new WC_Order( $order_id );
|
|
|
|
if ( $order->id == $order_id && $order->order_key == $order_key && in_array( $order->status, array( 'pending', 'failed' ) ) && wp_verify_nonce( $_GET['_wpnonce'], 'woocommerce-cancel_order' ) ) :
|
|
|
|
// Cancel the order + restore stock
|
|
$order->cancel_order( __('Order cancelled by customer.', 'woocommerce' ) );
|
|
|
|
// Message
|
|
wc_add_message( __( 'Your order was cancelled.', 'woocommerce' ) );
|
|
|
|
do_action( 'woocommerce_cancelled_order', $order->id );
|
|
|
|
elseif ( $order->status != 'pending' ) :
|
|
|
|
wc_add_error( __( 'Your order is no longer pending and could not be cancelled. Please contact us if you need assistance.', 'woocommerce' ) );
|
|
|
|
else :
|
|
|
|
wc_add_error( __( 'Invalid order.', 'woocommerce' ) );
|
|
|
|
endif;
|
|
|
|
wp_safe_redirect($woocommerce->cart->get_cart_url());
|
|
exit;
|
|
|
|
endif;
|
|
}
|
|
|
|
/**
|
|
* Add to cart action
|
|
*
|
|
* Checks for a valid request, does validation (via hooks) and then redirects if valid.
|
|
*
|
|
* @param bool $url (default: false)
|
|
*/
|
|
public function add_to_cart_action( $url = false ) {
|
|
if ( empty( $_REQUEST['add-to-cart'] ) || ! is_numeric( $_REQUEST['add-to-cart'] ) )
|
|
return;
|
|
|
|
$product_id = apply_filters( 'woocommerce_add_to_cart_product_id', absint( $_REQUEST['add-to-cart'] ) );
|
|
$was_added_to_cart = false;
|
|
$added_to_cart = array();
|
|
$adding_to_cart = get_product( $product_id );
|
|
$add_to_cart_handler = apply_filters( 'woocommerce_add_to_cart_handler', $adding_to_cart->product_type, $adding_to_cart );
|
|
|
|
// Variable product handling
|
|
if ( 'variable' === $add_to_cart_handler ) {
|
|
|
|
$variation_id = empty( $_REQUEST['variation_id'] ) ? '' : absint( $_REQUEST['variation_id'] );
|
|
$quantity = empty( $_REQUEST['quantity'] ) ? 1 : apply_filters( 'woocommerce_stock_amount', $_REQUEST['quantity'] );
|
|
$all_variations_set = true;
|
|
$variations = array();
|
|
|
|
// Only allow integer variation ID - if its not set, redirect to the product page
|
|
if ( empty( $variation_id ) ) {
|
|
wc_add_error( __( 'Please choose product options…', 'woocommerce' ) );
|
|
return;
|
|
}
|
|
|
|
$attributes = $adding_to_cart->get_attributes();
|
|
$variation = get_product( $variation_id );
|
|
|
|
// Verify all attributes
|
|
foreach ( $attributes as $attribute ) {
|
|
if ( ! $attribute['is_variation'] )
|
|
continue;
|
|
|
|
$taxonomy = 'attribute_' . sanitize_title( $attribute['name'] );
|
|
|
|
if ( ! empty( $_REQUEST[ $taxonomy ] ) ) {
|
|
|
|
// Get value from post data
|
|
// Don't use woocommerce_clean as it destroys sanitized characters
|
|
$value = sanitize_title( trim( stripslashes( $_REQUEST[ $taxonomy ] ) ) );
|
|
|
|
// Get valid value from variation
|
|
$valid_value = $variation->variation_data[ $taxonomy ];
|
|
|
|
// Allow if valid
|
|
if ( $valid_value == '' || $valid_value == $value ) {
|
|
if ( $attribute['is_taxonomy'] )
|
|
$variations[ esc_html( $attribute['name'] ) ] = $value;
|
|
else {
|
|
// For custom attributes, get the name from the slug
|
|
$options = array_map( 'trim', explode( '|', $attribute['value'] ) );
|
|
foreach ( $options as $option ) {
|
|
if ( sanitize_title( $option ) == $value ) {
|
|
$value = $option;
|
|
break;
|
|
}
|
|
}
|
|
$variations[ esc_html( $attribute['name'] ) ] = $value;
|
|
}
|
|
continue;
|
|
}
|
|
|
|
}
|
|
|
|
$all_variations_set = false;
|
|
}
|
|
|
|
if ( $all_variations_set ) {
|
|
// Add to cart validation
|
|
$passed_validation = apply_filters( 'woocommerce_add_to_cart_validation', true, $product_id, $quantity, $variation_id, $variations );
|
|
|
|
if ( $passed_validation ) {
|
|
if ( WC()->cart->add_to_cart( $product_id, $quantity, $variation_id, $variations ) ) {
|
|
woocommerce_add_to_cart_message( $product_id );
|
|
$was_added_to_cart = true;
|
|
$added_to_cart[] = $product_id;
|
|
}
|
|
}
|
|
} else {
|
|
wc_add_error( __( 'Please choose product options…', 'woocommerce' ) );
|
|
return;
|
|
}
|
|
|
|
// Grouped Products
|
|
} elseif ( 'grouped' === $add_to_cart_handler ) {
|
|
|
|
if ( ! empty( $_REQUEST['quantity'] ) && is_array( $_REQUEST['quantity'] ) ) {
|
|
|
|
$quantity_set = false;
|
|
|
|
foreach ( $_REQUEST['quantity'] as $item => $quantity ) {
|
|
if ( $quantity <= 0 )
|
|
continue;
|
|
|
|
$quantity_set = true;
|
|
|
|
// Add to cart validation
|
|
$passed_validation = apply_filters( 'woocommerce_add_to_cart_validation', true, $item, $quantity );
|
|
|
|
if ( $passed_validation ) {
|
|
if ( WC()->cart->add_to_cart( $item, $quantity ) ) {
|
|
$was_added_to_cart = true;
|
|
$added_to_cart[] = $item;
|
|
}
|
|
}
|
|
}
|
|
|
|
if ( $was_added_to_cart ) {
|
|
woocommerce_add_to_cart_message( $added_to_cart );
|
|
}
|
|
|
|
if ( ! $was_added_to_cart && ! $quantity_set ) {
|
|
wc_add_error( __( 'Please choose the quantity of items you wish to add to your cart…', 'woocommerce' ) );
|
|
return;
|
|
}
|
|
|
|
} elseif ( $product_id ) {
|
|
|
|
/* Link on product archives */
|
|
wc_add_error( __( 'Please choose a product to add to your cart…', 'woocommerce' ) );
|
|
return;
|
|
|
|
}
|
|
|
|
// Simple Products
|
|
} else {
|
|
|
|
$quantity = empty( $_REQUEST['quantity'] ) ? 1 : apply_filters( 'woocommerce_stock_amount', $_REQUEST['quantity'] );
|
|
|
|
// Add to cart validation
|
|
$passed_validation = apply_filters( 'woocommerce_add_to_cart_validation', true, $product_id, $quantity );
|
|
|
|
if ( $passed_validation ) {
|
|
// Add the product to the cart
|
|
if ( WC()->cart->add_to_cart( $product_id, $quantity ) ) {
|
|
woocommerce_add_to_cart_message( $product_id );
|
|
$was_added_to_cart = true;
|
|
$added_to_cart[] = $product_id;
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
// If we added the product to the cart we can now optionally do a redirect.
|
|
if ( $was_added_to_cart && wc_error_count() == 0 ) {
|
|
|
|
$url = apply_filters( 'add_to_cart_redirect', $url );
|
|
|
|
// If has custom URL redirect there
|
|
if ( $url ) {
|
|
wp_safe_redirect( $url );
|
|
exit;
|
|
}
|
|
|
|
// Redirect to cart option
|
|
elseif ( get_option('woocommerce_cart_redirect_after_add') == 'yes' ) {
|
|
wp_safe_redirect( WC()->cart->get_cart_url() );
|
|
exit;
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
/**
|
|
* Process the login form.
|
|
*/
|
|
public function process_login() {
|
|
if ( ! empty( $_POST['login'] ) ) {
|
|
|
|
wp_verify_nonce( $_POST['_wpnonce'], 'woocommerce-login' );
|
|
|
|
try {
|
|
$creds = array();
|
|
|
|
if ( empty( $_POST['username'] ) )
|
|
throw new Exception( '<strong>' . __( 'Error', 'woocommerce' ) . ':</strong> ' . __( 'Username is required.', 'woocommerce' ) );
|
|
if ( empty( $_POST['password'] ) )
|
|
throw new Exception( '<strong>' . __( 'Error', 'woocommerce' ) . ':</strong> ' . __( 'Password is required.', 'woocommerce' ) );
|
|
|
|
if ( is_email( $_POST['username'] ) ) {
|
|
$user = get_user_by( 'email', $_POST['username'] );
|
|
|
|
if ( isset( $user->user_login ) )
|
|
$creds['user_login'] = $user->user_login;
|
|
else
|
|
throw new Exception( '<strong>' . __( 'Error', 'woocommerce' ) . ':</strong> ' . __( 'A user could not be found with this email address.', 'woocommerce' ) );
|
|
} else {
|
|
$creds['user_login'] = $_POST['username'];
|
|
}
|
|
|
|
$creds['user_password'] = $_POST['password'];
|
|
$creds['remember'] = true;
|
|
$secure_cookie = is_ssl() ? true : false;
|
|
$user = wp_signon( $creds, $secure_cookie );
|
|
|
|
if ( is_wp_error( $user ) ) {
|
|
throw new Exception( $user->get_error_message() );
|
|
} else {
|
|
|
|
if ( ! empty( $_POST['redirect'] ) ) {
|
|
$redirect = esc_url( $_POST['redirect'] );
|
|
} elseif ( wp_get_referer() ) {
|
|
$redirect = esc_url( wp_get_referer() );
|
|
} else {
|
|
$redirect = esc_url( get_permalink( woocommerce_get_page_id( 'myaccount' ) ) );
|
|
}
|
|
|
|
// Feedback
|
|
wc_add_message( sprintf( __( 'You are now logged in as <strong>%s</strong>', 'woocommerce' ), $user->display_name ) );
|
|
|
|
wp_redirect( apply_filters( 'woocommerce_login_redirect', $redirect, $user ) );
|
|
exit;
|
|
}
|
|
} catch (Exception $e) {
|
|
|
|
wc_add_error( apply_filters('login_errors', $e->getMessage() ) );
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Handle reset password form
|
|
*/
|
|
public function process_reset_password() {
|
|
if ( ! isset( $_POST['wc_reset_password'] ) )
|
|
return;
|
|
|
|
// process lost password form
|
|
if ( isset( $_POST['user_login'] ) ) {
|
|
|
|
wp_verify_nonce( $_POST['_wpnonce'], 'woocommerce-lost_password' );
|
|
|
|
WC_Shortcode_My_Account::retrieve_password();
|
|
}
|
|
|
|
// process reset password form
|
|
if( isset( $_POST['password_1'] ) && isset( $_POST['password_2'] ) && isset( $_POST['reset_key'] ) && isset( $_POST['reset_login'] ) ) {
|
|
|
|
// verify reset key again
|
|
$user = WC_Shortcode_My_Account::check_password_reset_key( $_POST['reset_key'], $_POST['reset_login'] );
|
|
|
|
if ( is_object( $user ) ) {
|
|
|
|
// save these values into the form again in case of errors
|
|
$args['key'] = woocommerce_clean( $_POST['reset_key'] );
|
|
$args['login'] = woocommerce_clean( $_POST['reset_login'] );
|
|
|
|
wp_verify_nonce( $_POST['_wpnonce'], 'woocommerce-reset_password' );
|
|
|
|
if( empty( $_POST['password_1'] ) || empty( $_POST['password_2'] ) ) {
|
|
wc_add_error( __( 'Please enter your password.', 'woocommerce' ) );
|
|
$args['form'] = 'reset_password';
|
|
}
|
|
|
|
if( $_POST[ 'password_1' ] !== $_POST[ 'password_2' ] ) {
|
|
wc_add_error( __( 'Passwords do not match.', 'woocommerce' ) );
|
|
$args['form'] = 'reset_password';
|
|
}
|
|
|
|
if( 0 == wc_error_count() && ( $_POST['password_1'] == $_POST['password_2'] ) ) {
|
|
|
|
WC_Shortcode_My_Account::reset_password( $user, woocommerce_clean( $_POST['password_1'] ) );
|
|
|
|
do_action( 'woocommerce_customer_reset_password', $user );
|
|
|
|
wc_add_message( __( 'Your password has been reset.', 'woocommerce' ) . ' <a href="' . get_permalink( woocommerce_get_page_id( 'myaccount' ) ) . '">' . __( 'Log in', 'woocommerce' ) . '</a>' );
|
|
|
|
wp_redirect( remove_query_arg( array( 'key', 'login' ) ) );
|
|
exit;
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Process the registration form.
|
|
*/
|
|
public function process_registration() {
|
|
if ( ! empty( $_POST['register'] ) ) {
|
|
|
|
WC()->verify_nonce( 'register' );
|
|
|
|
$username = ! empty( $_POST['username'] ) ? woocommerce_clean( $_POST['username'] ) : '';
|
|
$email = ! empty( $_POST['email'] ) ? woocommerce_clean( $_POST['email'] ) : '';
|
|
$password = ! empty( $_POST['password'] ) ? woocommerce_clean( $_POST['password'] ) : '';
|
|
|
|
// Anti-spam trap
|
|
if ( ! empty( $_POST['email_2'] ) ) {
|
|
wc_add_error( '<strong>' . __( 'ERROR', 'woocommerce' ) . '</strong>: ' . __( 'Anti-spam field was filled in.', 'woocommerce' ) );
|
|
return;
|
|
}
|
|
|
|
$new_customer = woocommerce_create_new_customer( $email, $username, $password );
|
|
|
|
if ( is_wp_error( $new_customer ) ) {
|
|
wc_add_error( $new_customer->get_error_message() );
|
|
return;
|
|
}
|
|
|
|
woocommerce_set_customer_auth_cookie( $new_customer );
|
|
|
|
// Redirect
|
|
if ( wp_get_referer() ) {
|
|
$redirect = esc_url( wp_get_referer() );
|
|
} else {
|
|
$redirect = esc_url( get_permalink( woocommerce_get_page_id( 'myaccount' ) ) );
|
|
}
|
|
|
|
wp_redirect( apply_filters( 'woocommerce_registration_redirect', $redirect ) );
|
|
exit;
|
|
}
|
|
}
|
|
}
|
|
|
|
new WC_Form_Handler(); |