payment-kit 1.19.0 → 1.19.1

package/api/src/libs/notification/template/subscription-trial-start.ts

@@ -1,28 +1,14 @@
 /* eslint-disable @typescript-eslint/brace-style */
 /* eslint-disable @typescript-eslint/indent */
-import { fromUnitToToken, toDid } from '@ocap/util';
+import { toDid } from '@ocap/util';
 import pWaitFor from 'p-wait-for';
-import prettyMsI18n from 'pretty-ms-i18n';
-
 import isEmpty from 'lodash/isEmpty';
-import { getUserLocale } from '../../../integrations/blocklet/notification';
+
 import { translate } from '../../../locales';
-import {
-  CheckoutSession,
-  Customer,
-  NFTMintChainType,
-  Invoice,
-  NftMintItem,
-  PaymentMethod,
-  Subscription,
-} from '../../../store/models';
-import { PaymentCurrency } from '../../../store/models/payment-currency';
-import { getCustomerInvoicePageUrl, getOneTimeProductInfo } from '../../invoice';
-import { getMainProductName } from '../../product';
-import { getCustomerSubscriptionPageUrl } from '../../subscription';
-import { formatTime, getPrettyMsI18nLocale } from '../../time';
-import type { BaseEmailTemplate, BaseEmailTemplateType } from './base';
-import { getSubscriptionNotificationCustomActions } from '../../util';
+import { CheckoutSession, NFTMintChainType, NftMintItem } from '../../../store/models';
+import { getOneTimeProductInfo } from '../../invoice';
+import { formatTime } from '../../time';
+import { BaseSubscriptionEmailTemplate, BaseEmailTemplateType } from './base';
 
 export interface SubscriptionTrialStartEmailTemplateOptions {
   subscriptionId: string;
@@ -51,31 +37,27 @@ interface SubscriptionTrialStartEmailTemplateContext {
   viewInvoiceLink: string;
   oneTimeProductInfo?: Array<OneTimeProductInfo>;
   customActions: any[];
+  isCreditSubscription: boolean;
 }
 
-export class SubscriptionTrialStartEmailTemplate
-  implements BaseEmailTemplate<SubscriptionTrialStartEmailTemplateContext>
-{
+export class SubscriptionTrialStartEmailTemplate extends BaseSubscriptionEmailTemplate<SubscriptionTrialStartEmailTemplateContext> {
   options: SubscriptionTrialStartEmailTemplateOptions;
 
   constructor(options: SubscriptionTrialStartEmailTemplateOptions) {
+    super();
     this.options = options;
   }
 
   async getContext(): Promise<SubscriptionTrialStartEmailTemplateContext> {
-    const subscription: Subscription | null = await Subscription.findByPk(this.options.subscriptionId);
-    if (!subscription) {
-      throw new Error(`Subscription not found: ${this.options.subscriptionId}`);
-    }
+    // 获取基础订阅数据
+    const basicData = await this.getSubscriptionBasicData(this.options.subscriptionId);
+    const { subscription, userDid, locale, productName, paymentCurrency, isCreditSubscription } = basicData;
+
     if (subscription.status !== 'trialing') {
       throw new Error(`Subscription not trialing: ${this.options.subscriptionId}`);
     }
 
-    const customer = await Customer.findByPk(subscription.customer_id);
-    if (!customer) {
-      throw new Error(`Customer not found: ${subscription.customer_id}`);
-    }
-
+    // 等待 NFT 铸造完成
     await pWaitFor(
       async () => {
         const checkoutSession = await CheckoutSession.findBySubscriptionId(subscription.id);
@@ -84,57 +66,39 @@ export class SubscriptionTrialStartEmailTemplate
       { timeout: 1000 * 10, interval: 1000 }
     );
 
-    const invoice: Invoice = (await Invoice.findByPk(subscription.latest_invoice_id)) as Invoice;
-    if (!invoice) {
-      throw new Error(`Invoice not found in subscription: ${subscription.id}`);
-    }
-
-    const paymentCurrency = (await PaymentCurrency.findOne({
-      where: {
-        id: subscription.currency_id,
-      },
-    })) as PaymentCurrency;
-
+    // 获取 NFT 和支付信息
+    const paymentInfoResult = await this.getPaymentInfo(
+      subscription,
+      paymentCurrency,
+      isCreditSubscription,
+      subscription.latest_invoice_id
+    );
     const oneTimeProductInfo = await getOneTimeProductInfo(subscription.latest_invoice_id as string, paymentCurrency);
 
     const checkoutSession = await CheckoutSession.findBySubscriptionId(subscription.id);
-
-    const userDid: string = customer.did;
-    const locale = await getUserLocale(userDid);
-    const productName = await getMainProductName(subscription.id);
-    const subscriptionTrialEnd: string = formatTime((subscription.trial_end as number) * 1000);
-
-    const hasNft: boolean = checkoutSession?.nft_mint_status === 'minted';
-    const nftMintItem: NftMintItem | undefined = hasNft
+    const hasNft = checkoutSession?.nft_mint_status === 'minted';
+    const nftMintItem = hasNft
       ? checkoutSession?.nft_mint_details?.[checkoutSession?.nft_mint_details?.type as NFTMintChainType]
       : undefined;
-    const paymentMethod: PaymentMethod | null = await PaymentMethod.findByPk(invoice.default_payment_method_id);
-    const chainHost: string | undefined =
-      paymentMethod?.settings?.[checkoutSession?.nft_mint_details?.type as NFTMintChainType]?.api_host;
-    const paymentInfo: string = `${fromUnitToToken('0', paymentCurrency.decimal)} ${paymentCurrency.symbol}`;
-    const currentPeriodStart: string = formatTime((subscription.trial_start as number) * 1000);
-    const currentPeriodEnd: string = formatTime((subscription.trial_end as number) * 1000);
-    const duration: string = prettyMsI18n(
-      new Date(currentPeriodEnd).getTime() - new Date(currentPeriodStart).getTime(),
-      {
-        locale: getPrettyMsI18nLocale(locale),
-      }
-    );
+    const chainHost =
+      paymentInfoResult.paymentMethod?.settings?.[checkoutSession?.nft_mint_details?.type as NFTMintChainType]
+        ?.api_host;
 
-    const viewSubscriptionLink = getCustomerSubscriptionPageUrl({
-      subscriptionId: subscription.id,
+    // 获取链接
+    const links = this.generateSubscriptionLinks(
+      subscription,
       locale,
       userDid,
-    });
-    const viewInvoiceLink = getCustomerInvoicePageUrl({
-      invoiceId: invoice.id,
-      userDid,
-      locale,
-    });
-
-    const customActions = getSubscriptionNotificationCustomActions(
-      subscription,
       'customer.subscription.trial_start',
+      subscription.latest_invoice_id,
+      isCreditSubscription
+    );
+
+    // 计算时间信息
+    const subscriptionTrialEnd = formatTime((subscription.trial_end as number) * 1000);
+    const periodInfo = this.formatSubscriptionPeriod(
+      subscription.trial_start as number,
+      subscription.trial_end as number,
       locale
     );
 
@@ -142,23 +106,22 @@ export class SubscriptionTrialStartEmailTemplate
       locale,
       productName,
       subscriptionTrialEnd,
-
       nftMintItem,
       chainHost,
       userDid,
-      paymentInfo,
-      currentPeriodStart,
-      currentPeriodEnd,
-      duration,
-
-      viewSubscriptionLink,
-      viewInvoiceLink,
+      paymentInfo: '0',
+      currentPeriodStart: periodInfo.currentPeriodStart,
+      currentPeriodEnd: periodInfo.currentPeriodEnd,
+      duration: periodInfo.duration,
+      viewSubscriptionLink: links.viewSubscriptionLink,
+      viewInvoiceLink: links.viewInvoiceLink!,
       oneTimeProductInfo,
-      customActions,
+      customActions: links.customActions,
+      isCreditSubscription,
     };
   }
 
-  getOneTimeProductTemplate(oneTimeProductInfo: Array<OneTimeProductInfo>, locale: string = 'en') {
+  private getOneTimeProductTemplate(oneTimeProductInfo: Array<OneTimeProductInfo>, locale: string = 'en') {
     return [
       {
         type: 'divider',
@@ -226,134 +189,125 @@ export class SubscriptionTrialStartEmailTemplate
   }
 
   async getTemplate(): Promise<BaseEmailTemplateType> {
+    const context = await this.getContext();
     const {
       locale,
       productName,
       subscriptionTrialEnd,
-
       nftMintItem,
       chainHost,
       userDid,
       currentPeriodStart,
       currentPeriodEnd,
       duration,
-
       viewSubscriptionLink,
       viewInvoiceLink,
       oneTimeProductInfo,
       customActions,
-    } = await this.getContext();
+      isCreditSubscription,
+    } = context;
 
     const hasOneTimeProduct = !isEmpty(oneTimeProductInfo);
-    const template: BaseEmailTemplateType = {
-      title: `${translate('notification.subscriptionTrialStart.title', locale, {
-        productName,
-      })}`,
-      body: `${translate('notification.subscriptionTrialStart.body', locale, {
-        subscriptionTrialEnd,
-        productName,
-        trialDuration: duration,
-      })}`,
-      // @ts-expect-error
-      attachments: [
-        nftMintItem &&
-          chainHost && {
-            type: 'asset',
+
+    // 构建字段
+    const commonFields = this.buildCommonFields(userDid, productName, locale);
+
+    // 试用期字段
+    const trialPeriodFields = isCreditSubscription
+      ? []
+      : [
+          {
+            type: 'text',
             data: {
-              chainHost,
-              did: nftMintItem.address,
+              type: 'plain',
+              color: '#9397A1',
+              text: translate('notification.common.trialPeriod', locale),
             },
           },
-        hasOneTimeProduct && {
-          type: 'text',
-          data: {
-            type: 'plain',
-            text: translate('notification.common.trialProduct', locale),
-          },
-        },
-        {
-          type: 'section',
-          fields: [
-            {
-              type: 'text',
-              data: {
-                type: 'plain',
-                color: '#9397A1',
-                text: translate('notification.common.account', locale),
-              },
-            },
-            {
-              type: 'text',
-              data: {
-                type: 'plain',
-                text: userDid,
-              },
-            },
-            {
-              type: 'text',
-              data: {
-                type: 'plain',
-                color: '#9397A1',
-                text: translate('notification.common.product', locale),
-              },
-            },
-            {
-              type: 'text',
-              data: {
-                type: 'plain',
-                text: productName,
-              },
-            },
-            {
-              type: 'text',
-              data: {
-                type: 'plain',
-                color: '#9397A1',
-                text: translate('notification.common.trialPeriod', locale),
-              },
-            },
-            {
-              type: 'text',
-              data: {
-                type: 'plain',
-                text: `${currentPeriodStart} ~ ${currentPeriodEnd}(${duration})`,
-              },
+          {
+            type: 'text',
+            data: {
+              type: 'plain',
+              text: `${currentPeriodStart} ~ ${currentPeriodEnd}(${duration})`,
             },
-            nftMintItem && {
-              type: 'text',
-              data: {
-                type: 'plain',
-                color: '#9397A1',
-                text: translate('notification.common.nftAddress', locale),
-              },
+          },
+        ];
+
+    // NFT 地址字段
+    const nftFields = nftMintItem
+      ? [
+          {
+            type: 'text',
+            data: {
+              type: 'plain',
+              color: '#9397A1',
+              text: translate('notification.common.nftAddress', locale),
             },
-            nftMintItem && {
-              type: 'text',
-              data: {
-                type: 'plain',
-                text: toDid(nftMintItem.address),
-              },
+          },
+          {
+            type: 'text',
+            data: {
+              type: 'plain',
+              text: toDid(nftMintItem.address),
             },
-          ].filter(Boolean),
-        },
-        ...(hasOneTimeProduct
-          ? this.getOneTimeProductTemplate(oneTimeProductInfo as Array<OneTimeProductInfo>, locale)
-          : []),
-      ].filter(Boolean),
-      // @ts-ignore
-      actions: [
-        {
-          name: translate('notification.common.viewSubscription', locale),
-          title: translate('notification.common.viewSubscription', locale),
-          link: viewSubscriptionLink,
+          },
+        ]
+      : [];
+
+    // 构建附件
+    const attachments = [
+      nftMintItem &&
+        chainHost && {
+          type: 'asset',
+          data: {
+            chainHost,
+            did: nftMintItem.address,
+          },
         },
-        {
-          name: translate('notification.common.viewInvoice', locale),
-          title: translate('notification.common.viewInvoice', locale),
-          link: viewInvoiceLink,
+      hasOneTimeProduct && {
+        type: 'text',
+        data: {
+          type: 'plain',
+          text: translate('notification.common.trialProduct', locale),
         },
-        ...customActions,
-      ].filter(Boolean),
+      },
+      {
+        type: 'section',
+        fields: [...commonFields, ...trialPeriodFields, ...nftFields].filter(Boolean),
+      },
+      ...(hasOneTimeProduct
+        ? this.getOneTimeProductTemplate(oneTimeProductInfo as Array<OneTimeProductInfo>, locale)
+        : []),
+    ].filter(Boolean);
+
+    // 构建操作按钮
+    const actions = [
+      {
+        name: translate('notification.common.viewSubscription', locale),
+        title: translate('notification.common.viewSubscription', locale),
+        link: viewSubscriptionLink,
+      },
+      {
+        name: translate('notification.common.viewInvoice', locale),
+        title: translate('notification.common.viewInvoice', locale),
+        link: viewInvoiceLink,
+      },
+      ...customActions,
+    ].filter(Boolean);
+
+    const template: BaseEmailTemplateType = {
+      title: translate('notification.subscriptionTrialStart.title', locale, {
+        productName,
+      }),
+      body: translate('notification.subscriptionTrialStart.body', locale, {
+        subscriptionTrialEnd,
+        productName,
+        trialDuration: duration,
+      }),
+      // @ts-expect-error
+      attachments,
+      // @ts-ignore
+      actions,
     };
 
     return template;