infisical/backend/src/ee/controllers/v1/organizationsController.ts

import { Request, Response } from 'express';
import { getLicenseServerUrl } from '../../../config';
import { licenseServerKeyRequest } from '../../../config/request';
import { EELicenseService } from '../../services';

/**
 * Return the organization's current plan and allowed feature set
 */
export const getOrganizationPlan = async (req: Request, res: Response) => {
    const { organizationId } = req.params;
    const workspaceId = req.query.workspaceId as string;

    const plan = await EELicenseService.getPlan(organizationId, workspaceId);

    return res.status(200).send({
        plan,
    });
}

/**
 * Update the organization plan to product with id [productId]
 * @param req 
 * @param res 
 * @returns 
 */
export const updateOrganizationPlan = async (req: Request, res: Response) => {
    const {
        productId
    } = req.body;

    const { data  } = await licenseServerKeyRequest.patch(
        `${await getLicenseServerUrl()}/api/license-server/v1/customers/${req.organization.customerId}/cloud-plan`,
        {
            productId
        }
    ); 
    
    return res.status(200).send(data);
}

/**
 * Return the organization's payment methods on file
 */
export const getOrganizationPmtMethods = async (req: Request, res: Response) => {
    const { data: { pmtMethods } } = await licenseServerKeyRequest.get(
        `${await getLicenseServerUrl()}/api/license-server/v1/customers/${req.organization.customerId}/billing-details/payment-methods`
    );

    return res.status(200).send({
        pmtMethods
    }); 
}

/**
 * Return a Stripe session URL to add payment method for organization
 */
export const addOrganizationPmtMethod = async (req: Request, res: Response) => {
    const {
        success_url,
        cancel_url
    } = req.body;
    
    const { data: { url } } = await licenseServerKeyRequest.post(
        `${await getLicenseServerUrl()}/api/license-server/v1/customers/${req.organization.customerId}/billing-details/payment-methods`,
        {
            success_url,
            cancel_url
        }
    );
    
    return res.status(200).send({
        url
    }); 
}

export const deleteOrganizationPmtMethod = async (req: Request, res: Response) => {
    const { pmtMethodId } = req.params;

    const { data } = await licenseServerKeyRequest.delete(
        `${await getLicenseServerUrl()}/api/license-server/v1/customers/${req.organization.customerId}/billing-details/payment-methods/${pmtMethodId}`,
    );
        
    return res.status(200).send(data);
}