<?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' );

		$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 ( ! WC_Validation::is_postcode( $_POST[$key], $_POST[ $load_address . '_country' ] ) ) :
					wc_add_error( __( 'Please enter a valid postcode/ZIP.', 'woocommerce' ) );
				else :
					$_POST[$key] = wc_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 || $quantity == $values['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, false );

				}

				WC()->cart->calculate_totals();
			}

			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_coupon', $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 user is allowed to order again. By default it check if the 
		// previous order belonged to the current user.
		if ( !current_user_can( 'order_again', $order->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];
				elseif( meta_is_product_attribute( $meta_name, $meta_value, $product_id ) )
					$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 );

			$can_cancel = current_user_can( 'cancel_order', $order_id );

			if ( $can_cancel && $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 ( $can_cancel && $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&hellip;', '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&hellip;', '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&hellip;', 'woocommerce' ) );
					return;
				}

			} elseif ( $product_id ) {

				/* Link on product archives */
				wc_add_error( __( 'Please choose a product to add to your cart&hellip;', '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( apply_filters( 'woocommerce_login_credentials', $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();