woothemes / woocommerce

function wc_customer_has_capability( $allcaps, $caps, $args ) {

	if ( isset( $caps[0] ) ) {

		switch ( $caps[0] ) {

			case 'view_order' :

				$user_id = $args[1];

				$order   = wc_get_order( $args[2] );

				if ( $order && $user_id == $order->user_id ) {

					$allcaps['view_order'] = true;

				}

			break;

			case 'pay_for_order' :

				$user_id  = $args[1];

				$order_id = isset( $args[2] ) ? $args[2] : null;

					$allcaps['pay_for_order'] = true;

				}

			break;

			case 'order_again' :

				$user_id = $args[1];

				$order   = wc_get_order( $args[2] );

				if ( $user_id == $order->user_id ) {

					$allcaps['order_again'] = true;

				}

			break;

			case 'cancel_order' :

				$user_id = $args[1];

				$order   = wc_get_order( $args[2] );

					$allcaps['order_again'] = true;

				}

			break;

			case 'cancel_order' :

				$user_id = $args[1];

				$order   = wc_get_order( $args[2] );

				if ( $user_id == $order->user_id ) {

					$allcaps['cancel_order'] = true;

				}

			break;

			case 'download_file' :

				$user_id  = $args[1];

				$download = $args[2];