wp-graphql-woocommerce/includes/data/mutation/class-customer-mutation.php

<?php
/**
 * Defines functions for processing customer mutations.
 *
 * @package WPGraphQL\WooCommerce\Data\Mutation
 * @since 0.1.0
 */

namespace WPGraphQL\WooCommerce\Data\Mutation;

/**
 * Class Customer_Mutation
 */
class Customer_Mutation {
	/**
	 * Maps the GraphQL input to a format that the used by WooCommerce's WC_Customer class
	 *
	 * @param array  $input    Data coming from the GraphQL mutation query input.
	 * @param string $mutation Mutation being performed.
	 *
	 * @access public
	 * @return array
	 */
	public static function prepare_customer_props( $input, $mutation ) {
		$customer_args = [];

		if ( ! empty( $input['billing'] ) ) {
			$customer_args['billing'] = self::address_input_mapping( $input['billing'], 'billing' );
		}

		if ( ! empty( $input['shipping'] ) ) {
			$customer_args['shipping'] = self::address_input_mapping( $input['shipping'], 'shipping' );
		}

		/**
		 * Filters the mappings for input to arguments
		 *
		 * @var array  $customer_args The arguments to ultimately be passed to the WC_Customer::set_props function
		 * @var array  $input         Input data from the GraphQL mutation
		 * @var string $mutation      What customer mutation is being performed for context
		 */
		$customer_args = apply_filters( 'woocommerce_new_customer_data', $customer_args, $input, $mutation ); // phpcs:ignore WordPress.NamingConventions.PrefixAllGlobals.NonPrefixedHooknameFound

		return $customer_args;
	}

	/**
	 * Formats CustomerAddressInput into a address object to be used by WC_Customer object
	 *
	 * @param array  $input Customer address input.
	 * @param string $type  Address type.
	 *
	 * @return array;
	 */
	public static function address_input_mapping( $input, $type = 'billing' ) {
		// Map GQL input to address props array.
		$key_mapping = [
			'firstName' => 'first_name',
			'lastName'  => 'last_name',
			'address1'  => 'address_1',
			'address2'  => 'address_2',
		];

		$skip = apply_filters( 'graphql_woocommerce_customer_address_input_mapping_skipped', [ 'overwrite' ] );

		$type    = 'empty_' . $type;
		$address = ! empty( $input['overwrite'] ) && true === $input['overwrite']
			? self::{$type}()
			: [];
		foreach ( $input as $input_field => $value ) {
			if ( in_array( $input_field, array_keys( $key_mapping ), true ) ) {
				$address[ $key_mapping[ $input_field ] ] = $value;
			} elseif ( in_array( $input_field, $skip, true ) ) {
				continue;
			} else {
				$address[ $input_field ] = $value;
			}
		}

		return $address;
	}

	/**
	 * Returns default customer shipping address data
	 *
	 * @return array
	 */
	public static function empty_shipping() {
		return [
			'first_name' => '',
			'last_name'  => '',
			'company'    => '',
			'address_1'  => '',
			'address_2'  => '',
			'city'       => '',
			'state'      => '',
			'postcode'   => '',
			'country'    => '',
			'phone'      => '',
		];
	}

	/**
	 * Returns default customer billing address data
	 *
	 * @return array
	 */
	public static function empty_billing() {
		return array_merge(
			self::empty_shipping(),
			[ 'email' => '' ]
		);
	}

	/**
	 * Processes Customer meta data input.
	 *
	 * @param \WC_Customer $customer  Customer object.
	 * @param array        $inputs    Incoming meta data.
	 *
	 * @return void
	 */
	public static function input_meta_data_mapping( $customer, $inputs ) {
		if ( is_array( $inputs ) ) {
			foreach ( $inputs as $meta ) {
				$customer->update_meta_data( $meta['key'], $meta['value'], isset( $meta['id'] ) ? $meta['id'] : '' );
			}
		}
	}
}