woocommerce/plugins/woocommerce-blocks/assets/js/base/context/hooks/payment-methods/use-payment-methods.ts

101 lines
2.9 KiB
TypeScript

/**
* External dependencies
*/
import { useShallowEqual } from '@woocommerce/base-hooks';
import type {
PaymentMethods,
ExpressPaymentMethods,
PaymentMethodConfigInstance,
ExpressPaymentMethodConfigInstance,
} from '@woocommerce/types';
import {
getPaymentMethods,
getExpressPaymentMethods,
} from '@woocommerce/blocks-registry';
import { useSelect } from '@wordpress/data';
import { PAYMENT_STORE_KEY } from '@woocommerce/block-data';
interface PaymentMethodState {
paymentMethods: PaymentMethods;
isInitialized: boolean;
}
interface ExpressPaymentMethodState {
paymentMethods: ExpressPaymentMethods;
isInitialized: boolean;
}
const usePaymentMethodState = (
express = false
): PaymentMethodState | ExpressPaymentMethodState => {
const {
paymentMethodsInitialized,
expressPaymentMethodsInitialized,
availablePaymentMethods,
availableExpressPaymentMethods,
} = useSelect( ( select ) => {
const store = select( PAYMENT_STORE_KEY );
return {
paymentMethodsInitialized: store.paymentMethodsInitialized(),
expressPaymentMethodsInitialized:
store.expressPaymentMethodsInitialized(),
availableExpressPaymentMethods:
store.getAvailableExpressPaymentMethods(),
availablePaymentMethods: store.getAvailablePaymentMethods(),
};
} );
const availablePaymentMethodNames = Object.values(
availablePaymentMethods
).map( ( { name } ) => name );
const availableExpressPaymentMethodNames = Object.values(
availableExpressPaymentMethods
).map( ( { name } ) => name );
const registeredPaymentMethods = getPaymentMethods();
const registeredExpressPaymentMethods = getExpressPaymentMethods();
// Remove everything from registeredPaymentMethods that is not in availablePaymentMethodNames.
const paymentMethods = Object.keys( registeredPaymentMethods ).reduce(
( acc: Record< string, PaymentMethodConfigInstance >, key ) => {
if ( availablePaymentMethodNames.includes( key ) ) {
acc[ key ] = registeredPaymentMethods[ key ];
}
return acc;
},
{}
);
// Remove everything from registeredExpressPaymentMethods that is not in availableExpressPaymentMethodNames.
const expressPaymentMethods = Object.keys(
registeredExpressPaymentMethods
).reduce(
( acc: Record< string, ExpressPaymentMethodConfigInstance >, key ) => {
if ( availableExpressPaymentMethodNames.includes( key ) ) {
acc[ key ] = registeredExpressPaymentMethods[ key ];
}
return acc;
},
{}
);
const currentPaymentMethods = useShallowEqual( paymentMethods );
const currentExpressPaymentMethods = useShallowEqual(
expressPaymentMethods
);
return {
paymentMethods: express
? currentExpressPaymentMethods
: currentPaymentMethods,
isInitialized: express
? expressPaymentMethodsInitialized
: paymentMethodsInitialized,
};
};
export const usePaymentMethods = ():
| PaymentMethodState
| ExpressPaymentMethodState => usePaymentMethodState( false );
export const useExpressPaymentMethods = (): ExpressPaymentMethodState =>
usePaymentMethodState( true );