litlyx/dashboard/server/ai/functions/AI_Billing.ts

import { ProjectLimitModel } from "@schema/project/ProjectsLimits";
import { AIPlugin } from "../Plugin";
import { MAX_LOG_LIMIT_PERCENT } from "@data/broker/Limits";
import { ProjectModel } from "@schema/project/ProjectSchema";
import StripeService from "~/server/services/StripeService";
import { InvoiceData } from "~/server/api/pay/invoices";

export class AiBilling extends AIPlugin<[
    'getBillingInfo',
    'getLimits',
    'getInvoices'
]> {

    constructor() {
        super({

            'getInvoices': {
                handler: async (data: { project_id: string }) => {

                    const project = await ProjectModel.findOne({ _id: data.project_id });
                    if (!project) return { error: 'Project not found' };
                    const invoices = await StripeService.getInvoices(project.customer_id);
                    if (!invoices) return [];

                    return invoices?.data.map(e => {
                        const result: InvoiceData = {
                            link: e.invoice_pdf || '',
                            id: e.number || '',
                            date: e.created * 1000,
                            status: e.status || 'NO_STATUS',
                            cost: e.amount_due
                        }
                        return result;
                    });
                },
                tool: {
                    type: 'function',
                    function: {
                        name: 'getInvoices',
                        description: 'Gets the invoices of the user project',
                        parameters: {}
                    }
                }
            },

            'getBillingInfo': {
                handler: async (data: { project_id: string }) => {

                    const project = await ProjectModel.findOne({ _id: data.project_id });
                    if (!project) return { error: 'Project not found' };

                    if (project.subscription_id === 'onetime') {

                        const projectLimits = await ProjectLimitModel.findOne({ project_id: data.project_id });
                        if (!projectLimits) return { error: 'Limits not found' }

                        const result = {
                            premium: project.premium,
                            premium_type: project.premium_type,
                            billing_start_at: projectLimits.billing_start_at,
                            billing_expire_at: projectLimits.billing_expire_at,
                            limit: projectLimits.limit,
                            count: projectLimits.events + projectLimits.visits,
                            subscription_status: StripeService.isDisabled() ? 'Disabled mode' : ('One time payment')
                        }

                        return result;
                    }

                    const subscription = await StripeService.getSubscription(project.subscription_id);

                    const projectLimits = await ProjectLimitModel.findOne({ project_id: data.project_id });
                    if (!projectLimits) return { error: 'Limits not found' }


                    const result = {
                        premium: project.premium,
                        premium_type: project.premium_type,
                        billing_start_at: projectLimits.billing_start_at,
                        billing_expire_at: projectLimits.billing_expire_at,
                        limit: projectLimits.limit,
                        count: projectLimits.events + projectLimits.visits,
                        subscription_status: StripeService.isDisabled() ? 'Disabled mode' : (subscription?.status ?? '?')
                    }

                    return result;
                },
                tool: {
                    type: 'function',
                    function: {
                        name: 'getBillingInfo',
                        description: 'Gets the informations about the billing of the user project, limits, count, subscription_status, is premium, premium type, billing start at, billing expire at',
                        parameters: {}
                    }
                }
            },

            'getLimits': {
                handler: async (data: { project_id: string }) => {
                    const projectLimits = await ProjectLimitModel.findOne({ project_id: data.project_id });
                    if (!projectLimits) return { error: 'Project limits not found' };
                    const TOTAL_COUNT = projectLimits.events + projectLimits.visits;
                    const COUNT_LIMIT = projectLimits.limit;
                    return {
                        total: TOTAL_COUNT,
                        limit: COUNT_LIMIT,
                        limited: TOTAL_COUNT > COUNT_LIMIT * MAX_LOG_LIMIT_PERCENT,
                        percent: Math.round(100 / COUNT_LIMIT * TOTAL_COUNT)
                    }
                },
                tool: {
                    type: 'function',
                    function: {
                        name: 'getLimits',
                        description: 'Gets the informations about the limits of the user project',
                        parameters: {}
                    }
                }
            },


        })
    }
}

export const AiBillingInstance = new AiBilling();