-
Notifications
You must be signed in to change notification settings - Fork 11
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: implement minimal billing integration with LemonSqueezy.
- Loading branch information
Showing
19 changed files
with
467 additions
and
67 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,173 @@ | ||
| import { env } from '$env/dynamic/private'; | ||
| import { env as pubenv } from '$env/dynamic/public'; | ||
| import * as lemon from '@lemonsqueezy/lemonsqueezy.js'; | ||
| import { redis } from './redis'; | ||
| import { usernames } from './usernames'; | ||
|
|
||
| const REDIS_PREFIX = 'weird:billing:lemon:'; | ||
| const REDIS_SUBSCRIPTIONS_PREFIX = REDIS_PREFIX + 'subscriptions:'; | ||
|
|
||
| type WebhookEventKind = | ||
| | 'order_created' | ||
| | 'order_refunded' | ||
| | 'subscription_created' | ||
| | 'subscription_updated' | ||
| | 'subscription_cancelled' | ||
| | 'subscription_resumed' | ||
| | 'subscription_expired' | ||
| | 'subscription_paused' | ||
| | 'subscription_unpaused' | ||
| | 'subscription_payment_success' | ||
| | 'subscription_payment_failed' | ||
| | 'subscription_payment_recovered' | ||
| | 'subscription_payment_refunded' | ||
| | 'license_key_created' | ||
| | 'license_key_updated'; | ||
|
|
||
| type WebhookPayloadKind<Kind extends WebhookEventKind, Data extends { data: any }> = { | ||
| meta: { | ||
| event_name: Kind; | ||
| custom_data?: Record<string, string>; | ||
| }; | ||
| data: Data['data']; | ||
| }; | ||
| type WebhookPayload = | ||
| | WebhookPayloadKind<'order_created', lemon.Order> | ||
| | WebhookPayloadKind<'order_refunded', lemon.Order> | ||
| | WebhookPayloadKind<'subscription_created', lemon.Subscription> | ||
| | WebhookPayloadKind<'subscription_updated', lemon.Subscription> | ||
| | WebhookPayloadKind<'subscription_cancelled', lemon.Subscription> | ||
| | WebhookPayloadKind<'subscription_resumed', lemon.Subscription> | ||
| | WebhookPayloadKind<'subscription_expired', lemon.Subscription> | ||
| | WebhookPayloadKind<'subscription_paused', lemon.Subscription> | ||
| | WebhookPayloadKind<'subscription_unpaused', lemon.Subscription> | ||
| | WebhookPayloadKind<'subscription_payment_success', lemon.Subscription> | ||
| | WebhookPayloadKind<'subscription_payment_failed', lemon.Subscription> | ||
| | WebhookPayloadKind<'subscription_payment_recovered', lemon.Subscription> | ||
| | WebhookPayloadKind<'subscription_payment_refunded', lemon.Subscription> | ||
| | WebhookPayloadKind<'license_key_created', lemon.LicenseKey> | ||
| | WebhookPayloadKind<'license_key_updated', lemon.LicenseKey>; | ||
|
|
||
| export type SubscriptionInfo = { | ||
| id: string; | ||
| attributes: Omit<lemon.Subscription['data']['attributes'], 'urls'>; | ||
| }; | ||
|
|
||
| class BillingEngine { | ||
| constructor() { | ||
| lemon.lemonSqueezySetup({ | ||
| apiKey: env.LEMONSQUEEZY_API_KEY, | ||
| onError: (e) => { | ||
| console.error('LemonSqueezy.js error:', e); | ||
| } | ||
| }); | ||
| } | ||
|
|
||
| async getWeirdNerdCheckoutLink(userEmail: string, rauthyId: string): Promise<string> { | ||
| const checkout = await lemon.createCheckout( | ||
| env.LEMONSQUEEZY_STORE_ID, | ||
| env.LEMONSQUEEZY_WEIRD_NERD_VARIANT_ID, | ||
| { | ||
| checkoutOptions: { embed: true }, | ||
| checkoutData: { | ||
| email: userEmail, | ||
| discountCode: 'WEIRD1', | ||
| custom: { | ||
| rauthyId | ||
| } | ||
| }, | ||
| productOptions: { | ||
| redirectUrl: pubenv.PUBLIC_URL + '/my-profile' | ||
| } | ||
| } | ||
| ); | ||
|
|
||
| if (checkout.data) { | ||
| return checkout.data.data.attributes.url; | ||
| } else { | ||
| throw `Error creating checkout link: ${checkout.error}`; | ||
| } | ||
| } | ||
|
|
||
| async getWeirdNerdSubscriptionInfo(rauthyId: string): Promise<SubscriptionInfo[]> { | ||
| const subscriptions: SubscriptionInfo[] = []; | ||
|
|
||
| const prefix = REDIS_SUBSCRIPTIONS_PREFIX + rauthyId + ':'; | ||
| for await (const key of redis.scanIterator({ | ||
| MATCH: prefix + '*' | ||
| })) { | ||
| const s = await redis.get(key); | ||
| if (!s) throw `Subscription not found at ${key} in redis.`; | ||
| subscriptions.push(JSON.parse(s)); | ||
| } | ||
|
|
||
| return subscriptions; | ||
| } | ||
|
|
||
| async getBillingMethodUpdateLink(rauthyId: string): Promise<string | undefined> { | ||
| // Check for subscriptions for this user | ||
| const subscriptions = await this.getWeirdNerdSubscriptionInfo(rauthyId); | ||
| if (subscriptions.length > 0) { | ||
| // Get the subscription ID | ||
| let subscriptionId = subscriptions[0].id; | ||
| // Get an up-to-date reference to the subscription, which will include a signed customer | ||
| // portal URL. | ||
| const upToDateSubscription = await lemon.getSubscription(subscriptionId); | ||
| console.log(JSON.stringify(upToDateSubscription, null, ' ')); | ||
| if (upToDateSubscription.data) { | ||
| return upToDateSubscription.data.data.attributes.urls.update_payment_method; | ||
| } | ||
| } | ||
| } | ||
|
|
||
| async cancelBillingSubscription(rauthyId: string) { | ||
| const subscriptions = await this.getWeirdNerdSubscriptionInfo(rauthyId); | ||
| const activeSubscriptions = subscriptions.filter((x) => x.attributes.status == 'active'); | ||
| if (activeSubscriptions.length != 1) { | ||
| throw 'More than one active subscription, not sure how to cancel.'; | ||
| } | ||
| const resp = await lemon.cancelSubscription(activeSubscriptions[0].id); | ||
| if (resp.error) { | ||
| console.error(`Error cancelling lemonsqueezy subscription: ${resp.error}`); | ||
| } | ||
| } | ||
|
|
||
| async resumeBillingSubscription(rauthyId: string) { | ||
| const subscriptions = await this.getWeirdNerdSubscriptionInfo(rauthyId); | ||
| const cancelledSubscriptions = subscriptions.filter((x) => x.attributes.status == 'cancelled'); | ||
| if (cancelledSubscriptions.length != 1) { | ||
| throw 'More than one cancelled subscription, not sure how to cancel.'; | ||
| } | ||
| const resp = await lemon.updateSubscription(cancelledSubscriptions[0].id, { cancelled: false }); | ||
| if (resp.error) { | ||
| console.error(`Error resuming lemonsqueezy subscription: ${resp.error}`); | ||
| } | ||
| } | ||
|
|
||
| async #updateSubscriptionInfo(rauthyId: string, subscription: lemon.Subscription['data']) { | ||
| if (subscription.type == 'subscriptions') { | ||
| const info: SubscriptionInfo = { | ||
| id: subscription.id, | ||
| attributes: { ...subscription.attributes, ...{ urls: undefined } } | ||
| }; | ||
|
|
||
| await redis.set( | ||
| REDIS_SUBSCRIPTIONS_PREFIX + rauthyId + ':' + subscription.id, | ||
| JSON.stringify(info) | ||
| ); | ||
| } | ||
| } | ||
|
|
||
| async handleWebhook(webhook: WebhookPayload) { | ||
| if (webhook.meta.event_name.startsWith('subscription_')) { | ||
| // WARNING: this is unintuitive, but apparently lemonsqueezy converts our custom | ||
| // metadata to snake case. Watch out! | ||
| const rauthyId = webhook.meta.custom_data?.['rauthy_id']; | ||
| const data = webhook.data as lemon.Subscription['data']; | ||
| if (!rauthyId) throw 'rauthyId metadata missing from webhook.'; | ||
| await this.#updateSubscriptionInfo(rauthyId, data); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| export const billing = new BillingEngine(); |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.