add stripe

dashboard/server/api/pay/[project_id]/create.post.ts

@@ -0,0 +1,45 @@
+import { PREMIUM_PLANS, STRIPE_PLANS } from "@data/PREMIUM_LIMITS";
+import { ProjectModel } from "@schema/ProjectSchema";
+import { getUserProjectFromId } from "~/server/LIVE_DEMO_DATA";
+import StripeService from '~/server/services/StripeService';
+
+
+export default defineEventHandler(async event => {
+
+    const project_id = getRequestProjectId(event);
+    if (!project_id) return;
+
+    const user = getRequestUser(event);
+    const project = await getUserProjectFromId(project_id, user);
+    if (!project) return;
+
+    const body = await readBody(event);
+
+    const { planId } = body;
+
+    const plan = PREMIUM_PLANS.find(e => e.id == planId);
+
+    if (!plan) {
+        console.error('PLAN', planId, 'NOT EXIST');
+        return setResponseStatus(event, 400, 'Plan not exist');
+    }
+
+    const { price } = STRIPE_PLANS[plan.tag];
+
+    const checkout = await StripeService.cretePayment(
+        price,
+        'https://dashboard.litlyx.com/payment_ok',
+        project.customer_id
+    );
+
+    if (!checkout) {
+        console.error('Cannot create payment', { plan, price });
+        return setResponseStatus(event, 400, 'Cannot create payment');
+    }
+
+    const customer = checkout.customer;
+    await ProjectModel.updateOne({ _id: project_id }, { customer_id: customer });
+
+    return checkout.url;
+
+});

dashboard/server/api/pay/webhook.post.ts

@@ -0,0 +1,77 @@
+
+import StripeService from '~/server/services/StripeService';
+import type Event from 'stripe';
+import { ProjectModel } from '@schema/ProjectSchema';
+import { PREMIUM_LIMITS, getPlanFromPremiumTag, getPlanTagFromStripePrice } from '@data/PREMIUM_LIMITS';
+import { ProjectCountModel } from '@schema/ProjectsCounts';
+
+async function onPaymentSuccess(event: Event.InvoicePaidEvent) {
+
+    if (event.data.object.status === 'paid') {
+        const customer = event.data.object.customer;
+
+        const project = await ProjectModel.findOne({ customer_id: customer });
+        if (!project) return { error: 'Project not found' }
+
+        const subscriptionId = event.data.object.subscription;
+        if (!subscriptionId) return { error: 'SubscriptionId not found' }
+
+        const subscription = await StripeService.getSubscription(subscriptionId as string);
+        if (!subscription) return { error: 'Subscription not found' }
+
+        const price = subscription.items.data[0].plan.id;
+
+        
+        const premiumTag = getPlanTagFromStripePrice(price);
+        if (!premiumTag) return { error: 'Premium tag not found' }
+
+        const plan = getPlanFromPremiumTag(premiumTag);
+        if (!plan) return { error: 'Plan not found' }
+
+        await ProjectModel.updateOne({ customer_id: customer }, {
+            premium: true,
+            premium_type: plan.id,
+            premium_expire_at: subscription.current_period_end
+        });
+
+        const limits = PREMIUM_LIMITS[premiumTag];
+
+        await ProjectCountModel.create({
+            project_id: project._id,
+            events: 0,
+            visits: 0,
+            ai_messages: 0,
+            limit: limits.COUNT_LIMIT,
+            ai_limit: limits.AI_MESSAGE_LIMIT,
+            billing_start_at: subscription.current_period_start,
+            billing_expire_at: subscription.current_period_end,
+        });
+
+        return { ok: true }
+    }
+
+    return { received: true }
+}
+
+async function onSubscriptionCreated(event: Event.CustomerSubscriptionCreatedEvent) {
+    return { received: true }
+}
+
+async function onSubscriptionDeleted(event: Event.CustomerSubscriptionDeletedEvent) {
+    return { received: true }
+}
+
+
+export default defineEventHandler(async event => {
+    
+    const body = await readRawBody(event);
+    const signature = getHeader(event, 'stripe-signature') || '';
+
+    const eventData = StripeService.parseWebhook(body, signature);
+    if (!eventData) return;
+    if (eventData.type === 'invoice.paid') return await onPaymentSuccess(eventData);
+    if (eventData.type === 'customer.subscription.deleted') return await onSubscriptionDeleted(eventData);
+    if (eventData.type === 'customer.subscription.created') return await onSubscriptionCreated(eventData);
+
+    return { received: true }
+});

dashboard/server/init.ts

@@ -1,6 +1,7 @@
 import mongoose from "mongoose";
 import { Redis } from "~/server/services/CacheService";
 import EmailService from '@services/EmailService';
+import StripeService from '~/server/services/StripeService';
 
 const config = useRuntimeConfig();
 let connection: mongoose.Mongoose;
@@ -16,6 +17,8 @@ export default async () => {
         config.EMAIL_PASS,
     );
 
+    StripeService.init(config.STRIPE_SECRET, config.STRIPE_WH_SECRET);
+
 
     if (!connection || connection.connection.readyState == mongoose.ConnectionStates.disconnected) {
         console.log('[DATABASE] Connecting');

dashboard/server/services/StripeService.ts

@@ -0,0 +1,54 @@
+import Stripe from 'stripe';
+
+class StripeService {
+    private stripe?: Stripe;
+    private privateKey?: string;
+    private webhookSecret?: string;
+
+    init(privateKey: string, webhookSecret: string) {
+        this.privateKey = privateKey;
+        this.webhookSecret = webhookSecret;
+        this.stripe = new Stripe(this.privateKey);
+    }
+    parseWebhook(body: any, sig: string) {
+        if (!this.stripe) {
+            console.error('Stripe not initialized')
+            return;
+        }
+        if (!this.webhookSecret) {
+            console.error('Stripe not initialized')
+            return;
+        }
+        return this.stripe.webhooks.constructEvent(body, sig, this.webhookSecret);
+    }
+
+    async cretePayment(price: string, success_url: string, customer?: string) {
+        if (!this.stripe) {
+            console.error('Stripe not initialized')
+            return;
+        }
+        const checkout = await this.stripe.checkout.sessions.create({
+            payment_method_types: ['card'],
+            line_items: [
+                { price, quantity: 1 }
+            ],
+            customer,
+            success_url,
+            mode: 'subscription'
+        });
+
+        return checkout;
+    }
+
+    async getSubscription(subscriptionId: string) {
+        if (!this.stripe) {
+            console.error('Stripe not initialized')
+            return;
+        }
+        const subscription = await this.stripe.subscriptions.retrieve(subscriptionId);
+        return subscription;
+    }
+}
+
+const instance = new StripeService();
+export default instance;