Modrinth-plus/packages/ui/src/composables/stripe.ts

import type { Labrinth } from '@modrinth/api-client'
import { loadStripe, type Stripe as StripeJs, type StripeElements } from '@stripe/stripe-js'
import type { ContactOption } from '@stripe/stripe-js/dist/stripe-js/elements/address'
import type Stripe from 'stripe'
import { computed, type Ref, ref } from 'vue'

import type { ServerBillingInterval } from '../components/billing/ModrinthServersPurchaseModal.vue'
import { getPriceForInterval } from '../utils/product-utils'

// export type CreateElements = (
//   paymentMethods: Stripe.PaymentMethod[],
//   options: StripeElementsOptionsMode,
// ) => {
//   elements: StripeElements
//   paymentElement: StripePaymentElement
//   addressElement: StripeAddressElement
// }

export const useStripe = (
	publishableKey: string,
	customer: Stripe.Customer,
	paymentMethods: Stripe.PaymentMethod[],
	currency: string,
	product: Ref<Labrinth.Billing.Internal.Product | undefined>,
	interval: Ref<ServerBillingInterval>,
	region: Ref<string | undefined>,
	project: Ref<string | undefined>,
	initiatePayment: (
		body: Labrinth.Billing.Internal.InitiatePaymentRequest,
	) => Promise<
		| Labrinth.Billing.Internal.InitiatePaymentResponse
		| Labrinth.Billing.Internal.EditSubscriptionResponse
		| null
	>,
	onError: (err: Error) => void,
	affiliateCode?: Ref<string | null>,
) => {
	const stripe = ref<StripeJs | null>(null)

	let elements: StripeElements | undefined = undefined
	const elementsLoaded = ref<0 | 1 | 2>(0)
	const loadingElementsFailed = ref<boolean>(false)

	const paymentMethodLoading = ref(false)
	const loadingFailed = ref<string>()
	const paymentIntentId = ref<string>()
	const tax = ref<number>()
	const total = ref<number>()
	const confirmationToken = ref<string>()
	const submittingPayment = ref(false)
	const selectedPaymentMethod = ref<Stripe.PaymentMethod>()
	const inputtedPaymentMethod = ref<Stripe.PaymentMethod>()
	const clientSecret = ref<string>()
	const completingPurchase = ref<boolean>(false)
	const noPaymentRequired = ref<boolean>(false)

	async function initialize() {
		stripe.value = await loadStripe(publishableKey)
	}

	const planPrices = computed(() => {
		return product.value?.prices.find((x) => x.currency_code === currency)
	})

	const createElements = (options) => {
		const styles = getComputedStyle(document.body)

		if (!stripe.value) {
			throw new Error('Stripe API not yet loaded')
		}

		elements = stripe.value.elements({
			appearance: {
				variables: {
					colorPrimary: styles.getPropertyValue('--color-brand'),
					colorBackground: styles.getPropertyValue('--color-button-bg'),
					colorText: styles.getPropertyValue('--color-base'),
					colorTextPlaceholder: styles.getPropertyValue('--color-secondary'),
					colorDanger: styles.getPropertyValue('--color-red'),
					fontFamily: styles.getPropertyValue('--font-standard'),
					spacingUnit: '0.25rem',
					borderRadius: '0.75rem',
				},
			},
			loader: 'never',
			...options,
		})

		const paymentElement = elements.create('payment', {
			layout: {
				type: 'tabs',
				defaultCollapsed: false,
			},
		})
		paymentElement.mount('#payment-element')

		const contacts: ContactOption[] = []

		paymentMethods.forEach((method) => {
			const addr = method.billing_details?.address
			if (
				addr &&
				addr.line1 &&
				addr.city &&
				addr.postal_code &&
				addr.country &&
				addr.state &&
				method.billing_details.name
			) {
				contacts.push({
					address: {
						line1: addr.line1,
						line2: addr.line2 ?? undefined,
						city: addr.city,
						state: addr.state,
						postal_code: addr.postal_code,
						country: addr.country,
					},
					name: method.billing_details.name,
				})
			}
		})

		const addressElement = elements.create('address', {
			mode: 'billing',
			contacts: contacts.length > 0 ? contacts : undefined,
		})
		addressElement.mount('#address-element')

		return { elements, paymentElement, addressElement }
	}

	const primaryPaymentMethodId = computed<string | null>(() => {
		if (customer && customer.invoice_settings && customer.invoice_settings.default_payment_method) {
			const method = customer.invoice_settings.default_payment_method
			if (typeof method === 'string') {
				return method
			} else {
				return method.id
			}
		} else if (paymentMethods && paymentMethods[0] && paymentMethods[0].id) {
			return paymentMethods[0].id
		} else {
			return null
		}
	})

	const loadStripeElements = async () => {
		loadingFailed.value = undefined
		try {
			if (!customer && primaryPaymentMethodId.value) {
				paymentMethodLoading.value = true
				await refreshPaymentIntent(primaryPaymentMethodId.value, false)
				paymentMethodLoading.value = false
			}

			if (!selectedPaymentMethod.value) {
				elementsLoaded.value = 0

				const {
					elements: newElements,
					addressElement,
					paymentElement,
				} = createElements({
					mode: 'payment',
					currency: currency.toLowerCase(),
					amount: product.value
						? getPriceForInterval(product.value, currency, interval.value)
						: undefined,
					paymentMethodCreation: 'manual',
					setupFutureUsage: 'off_session',
				})

				elements = newElements
				paymentElement.on('ready', () => {
					elementsLoaded.value += 1
				})
				addressElement.on('ready', () => {
					elementsLoaded.value += 1
				})
			}
		} catch (err) {
			loadingFailed.value = String(err)
			console.log(err)
		}
	}

	async function refreshPaymentIntent(id: string, confirmation: boolean) {
		try {
			paymentMethodLoading.value = true
			if (!confirmation) {
				selectedPaymentMethod.value = paymentMethods.find((x) => x.id === id)
			}

			if (!product.value) {
				return handlePaymentError('No product selected')
			}

			const request: Labrinth.Billing.Internal.InitiatePaymentRequest = {
				type: confirmation ? 'confirmation_token' : 'payment_method',
				...(confirmation ? { token: id } : { id }),
				charge: {
					type: 'new',
					product_id: product.value.id,
					interval: interval.value as Labrinth.Billing.Internal.PriceDuration,
				},
				...(paymentIntentId.value ? { existing_payment_intent: paymentIntentId.value } : {}),
				metadata: {
					type: 'pyro',
					server_region: region.value,
					source: project.value
						? {
								project_id: project.value,
							}
						: {},
					...(affiliateCode?.value ? { affiliate_code: affiliateCode.value } : {}),
				},
			}

			const result = await initiatePayment(request)

			if (!result) {
				tax.value = 0
				total.value = 0
				noPaymentRequired.value = true
			} else {
				if (result.payment_intent_id) {
					paymentIntentId.value = result.payment_intent_id
				}
				if (result.client_secret) {
					clientSecret.value = result.client_secret
				}
				tax.value = result.tax
				total.value = result.total
				noPaymentRequired.value = false

				console.log(
					`${paymentIntentId.value ? 'Updated' : 'Created'} payment intent: ${interval.value} for ${result.total}`,
				)
			}

			if (confirmation) {
				confirmationToken.value = id
				if (result && 'payment_method' in result && result.payment_method) {
					const paymentMethod = (
						result as {
							payment_method?: string | Stripe.PaymentMethod
						}
					).payment_method
					if (typeof paymentMethod === 'string') {
						const method = paymentMethods.find((x) => x.id === paymentMethod)
						if (method) {
							inputtedPaymentMethod.value = method
						}
					} else if (paymentMethod) {
						inputtedPaymentMethod.value = paymentMethod
					}
				}
			}
		} catch (err) {
			handlePaymentError(err as string)
		}
		paymentMethodLoading.value = false
	}

	async function createConfirmationToken() {
		if (!elements) {
			return handlePaymentError('No elements')
		}
		if (!stripe.value) {
			return handlePaymentError('No stripe')
		}

		const { error, confirmationToken: confirmation } = await stripe.value.createConfirmationToken({
			elements,
		})

		if (error) {
			handlePaymentError(error.message ?? 'Unknown error creating confirmation token')
			return
		}

		return confirmation.id
	}

	function handlePaymentError(err: string | Error) {
		paymentMethodLoading.value = false
		completingPurchase.value = false
		onError(typeof err === 'string' ? new Error(err) : err)
	}

	async function createNewPaymentMethod() {
		paymentMethodLoading.value = true

		if (!elements) {
			return handlePaymentError('No elements')
		}

		const { error: submitError } = await elements.submit()

		if (submitError) {
			return handlePaymentError(submitError.message ?? 'Unknown error creating payment method')
		}

		const token = await createConfirmationToken()
		if (!token) {
			return handlePaymentError('Failed to create confirmation token')
		}
		await refreshPaymentIntent(token, true)

		if (!planPrices.value) {
			return handlePaymentError('No plan prices')
		}
		if (!total.value) {
			return handlePaymentError('No total amount')
		}

		elements.update({
			currency: planPrices.value.currency_code.toLowerCase(),
			amount: total.value,
		})

		elementsLoaded.value = 0
		confirmationToken.value = token
		paymentMethodLoading.value = false

		return token
	}

	async function selectPaymentMethod(paymentMethod: Stripe.PaymentMethod | undefined) {
		selectedPaymentMethod.value = paymentMethod
		if (paymentMethod === undefined) {
			await loadStripeElements()
		} else {
			refreshPaymentIntent(paymentMethod.id, false)
		}
	}

	const loadingElements = computed(() => elementsLoaded.value < 2)

	async function submitPayment(returnUrl?: string): Promise<boolean> {
		if (noPaymentRequired.value) {
			completingPurchase.value = false
			return true
		}
		completingPurchase.value = true
		const secret = clientSecret.value

		if (!secret) {
			handlePaymentError('No client secret')
			return false
		}

		if (!stripe.value) {
			handlePaymentError('No stripe')
			return false
		}

		submittingPayment.value = true
		const productPrice = product.value?.prices.find((x) => x.currency_code === currency)

		const { error } = returnUrl
			? await stripe.value.confirmPayment({
					clientSecret: secret,
					confirmParams: {
						confirmation_token: confirmationToken.value,
						return_url: `${returnUrl}?priceId=${productPrice?.id}&plan=${interval.value}`,
					},
				})
			: await stripe.value.confirmPayment({
					clientSecret: secret,
					redirect: 'if_required',
					confirmParams: {
						confirmation_token: confirmationToken.value,
					},
				})

		if (error) {
			handlePaymentError(error.message ?? 'Unknown error submitting payment')
			return false
		}
		submittingPayment.value = false
		completingPurchase.value = false
		return true
	}

	async function reloadPaymentIntent() {
		console.log('selected:', selectedPaymentMethod.value)
		console.log('token:', confirmationToken.value)
		if (selectedPaymentMethod.value) {
			await refreshPaymentIntent(selectedPaymentMethod.value.id, false)
		} else if (confirmationToken.value) {
			await refreshPaymentIntent(confirmationToken.value, true)
		} else {
			throw new Error('No payment method selected')
		}
	}

	const hasPaymentMethod = computed(
		() => selectedPaymentMethod.value || confirmationToken.value || noPaymentRequired.value,
	)

	return {
		initializeStripe: initialize,
		selectPaymentMethod,
		reloadPaymentIntent,
		primaryPaymentMethodId,
		selectedPaymentMethod,
		inputtedPaymentMethod,
		hasPaymentMethod,
		createNewPaymentMethod,
		loadingElements,
		loadingElementsFailed,
		paymentMethodLoading,
		loadStripeElements,
		tax,
		total,
		submitPayment,
		completingPurchase,
		noPaymentRequired,
	}
}