react-native-adapty/src/mock/mock-store.ts

Adapty React Native SDK

import type {
  AdaptyProfile,
  AdaptyPaywall,
  AdaptyPaywallProduct,
  AdaptyOnboarding,
  AdaptyProfileParameters,
} from '@/types';
import type { AdaptyMockConfig } from './types';
import {
  createMockProfile,
  createMockPremiumAccessLevel,
  createMockPaywall,
  createMockProducts,
  createMockOnboarding,
  createSubscriptionDates,
  MOCK_VENDOR_PRODUCT_ID_ANNUAL,
  MOCK_ACCESS_LEVEL_PREMIUM,
} from './mock-data';

/**
 * Stateful store for mock data
 * Maintains profile state across operations
 */
export class MockStore {
  private profile: AdaptyProfile;
  private config: AdaptyMockConfig;
  private isActivated: boolean = false;

  constructor(config: AdaptyMockConfig = {}) {
    this.config = config;
    this.profile = createMockProfile(config.profile);
  }

  /**
   * Get the current profile
   */
  getProfile(): AdaptyProfile {
    return { ...this.profile };
  }

  /**
   * Update profile with custom data
   *
   * Note: In real Adapty, updateProfile() sends user attributes (firstName, lastName, email, etc.)
   * to the server for CRM and segmentation, but these fields are NOT returned by getProfile().
   * Only customAttributes are stored in the profile and returned by getProfile().
   */
  updateProfile(updates: AdaptyProfileParameters): void {
    // Only codableCustomAttributes are stored in the profile and returned by getProfile()
    if (updates.codableCustomAttributes) {
      const updatedProfile: AdaptyProfile = {
        ...this.profile,
        customAttributes: {
          ...this.profile.customAttributes,
          ...updates.codableCustomAttributes,
        },
      };
      this.profile = updatedProfile;
    }

    // Other fields (firstName, lastName, email, phoneNumber, etc.) are sent to Adapty server
    // for CRM and segmentation purposes, but are NOT stored in the profile object
    // and NOT returned by getProfile()
  }

  /**
   * Grant premium access level
   */
  grantPremiumAccess(accessLevelId: string): AdaptyProfile {
    const premiumAccessLevel = createMockPremiumAccessLevel(accessLevelId);
    const { now, expiresAt, unsubscribedAt } = createSubscriptionDates();

    const updatedProfile: AdaptyProfile = {
      ...this.profile,
      accessLevels: {
        ...this.profile.accessLevels,
        [accessLevelId]: premiumAccessLevel,
      },
      subscriptions: {
        ...this.profile.subscriptions,
        [MOCK_VENDOR_PRODUCT_ID_ANNUAL]: {
          isActive: true,
          isLifetime: false,
          vendorProductId: MOCK_VENDOR_PRODUCT_ID_ANNUAL,
          store: 'adapty',
          vendorTransactionId: '2000001082537697',
          vendorOriginalTransactionId: '2000000971279249',
          activatedAt: now,
          renewedAt: now,
          expiresAt,
          unsubscribedAt,
          willRenew: false,
          isInGracePeriod: false,
          isRefund: false,
          isSandbox: true,
        },
      },
    };
    this.profile = updatedProfile;

    return this.getProfile();
  }

  /**
   * Simulate a purchase and update profile accordingly
   *
   * @param productAccessLevelId - Access level ID from the product, or undefined if not available
   */
  makePurchase(productAccessLevelId?: string): AdaptyProfile {
    const shouldGrantPremium = this.config.autoGrantPremium !== false; // default true

    if (shouldGrantPremium) {
      const accessLevelId =
        this.config.premiumAccessLevelId ||
        productAccessLevelId ||
        MOCK_ACCESS_LEVEL_PREMIUM;
      return this.grantPremiumAccess(accessLevelId);
    }

    return this.getProfile();
  }

  /**
   * Get paywall for placement
   */
  getPaywall(placementId: string): AdaptyPaywall {
    const customPaywall = this.config.paywalls?.[placementId];
    return createMockPaywall(placementId, customPaywall);
  }

  /**
   * Get products for paywall
   */
  getPaywallProducts(
    placementId: string,
    variationId: string = 'mock_variation_id',
  ): AdaptyPaywallProduct[] {
    // If custom products are provided by variationId, use them
    const customProducts = this.config.products?.[variationId];
    if (customProducts) {
      return customProducts;
    }

    // Otherwise, generate default products
    const paywall = this.getPaywall(placementId);
    return createMockProducts(paywall);
  }

  /**
   * Get onboarding for placement
   */
  getOnboarding(placementId: string): AdaptyOnboarding {
    const customOnboarding = this.config.onboardings?.[placementId];
    return createMockOnboarding(placementId, customOnboarding);
  }

  /**
   * Set activated state
   */
  setActivated(activated: boolean): void {
    this.isActivated = activated;
  }

  /**
   * Check if SDK is activated
   */
  getIsActivated(): boolean {
    return this.isActivated;
  }

  /**
   * Reset profile to initial state
   */
  logout(): void {
    this.profile = createMockProfile(this.config.profile);
  }

  /**
   * Update customer user ID
   */
  identify(customerUserId: string): void {
    const updatedProfile: AdaptyProfile = {
      ...this.profile,
      customerUserId,
    };
    this.profile = updatedProfile;
  }
}