import { Page, expect } from '@playwright/test';

export class StripePageObject {
  private page: Page;

  constructor(page: Page) {
    this.page = page;
  }

  plans() {
    return this.page.locator('[data-test-plan]');
  }

  getStripeCheckoutIframe() {
    return this.page.frameLocator('[name="embedded-checkout"]');
  }

  async fillForm(params: {
    billingName?: string;
    cardNumber?: string;
    expiry?: string;
    cvc?: string;
    billingCountry?: string;
  } = {}) {
    expect(() => {
      return this.getStripeCheckoutIframe().locator('form').isVisible();
    });

    const billingName = this.billingName();
    const cardNumber = this.cardNumber();
    const expiry = this.expiry();
    const cvc = this.cvc();
    const billingCountry = this.billingCountry();

    await billingName.fill(params.billingName ?? 'Mr Makerkit');
    await cardNumber.fill(params.cardNumber ?? '4242424242424242');
    await expiry.fill(params.expiry ?? '1228');
    await cvc.fill(params.cvc ?? '123');
    await billingCountry.selectOption(params.billingCountry ?? 'IT');
  }

  submitForm() {
    return this.getStripeCheckoutIframe().locator('form button').click();
  }

  cardNumber() {
    return this.getStripeCheckoutIframe().locator('#cardNumber');
  }

  cvc() {
    return this.getStripeCheckoutIframe().locator('#cardCvc');
  }

  expiry() {
    return this.getStripeCheckoutIframe().locator('#cardExpiry');
  }

  billingName() {
    return this.getStripeCheckoutIframe().locator('#billingName');
  }

  cardForm() {
    return this.getStripeCheckoutIframe().locator('form');
  }

  billingCountry() {
    return this.getStripeCheckoutIframe().locator('#billingCountry');
  }

  selectPlan(index: number = 0) {
    const plans = this.plans();

    return plans.nth(index).click();
  }

  manageBillingButton() {
    return this.page.locator('manage-billing-redirect-button');
  }

  successStatus() {
    return this.page.locator('[data-test="payment-return-success"]');
  }

  proceedToCheckout() {
    return this.page.click('[data-test="checkout-submit-button"]');
  }
}