import { Page, Locator, test } from "@playwright/test";
import { Button } from "./Button";
import { Menu } from "./Menu";

/**
 * Class representing a menu button that extends the Button class.
 */
export class MenuButton extends Button {
  button: Button;
  menu: Menu;

  /**
   * Create a MenuButton.
   * @param {Page} page - The Playwright page object.
   * @param {Locator} locator - The locator for the MenuButton element.
   * @param {string} elementReportName - The name for reporting purposes.
   * @param {any} menuType - The type of menu associated with the button.
   */
  constructor(page: Page, locator: Locator, elementReportName: string, menuType: Menu) {
    super(page, locator, elementReportName);
    this.button = new Button(this.page, this.locator, elementReportName);
    this.menu = menuType;
  }

  /**
   * Select an item from the menu.
   * @param {string} item - The item to select.
   * @returns {Promise<void>}
   */
  async selectItem(item: string): Promise<void> {
    await test.step(`Select ${item} from ${this.elementReportName}`, async () => {
      await this.openMenu();
      await this.menu.selectItem(item);
    });
  }

  /**
   * Open the menu if it is not already expanded.
   * @returns {Promise<void>}
   */
  async openMenu(): Promise<void> {
    await test.step(`Open menu in ${this.elementReportName}`, async () => {
      if (!(await this.isExpanded())) {
        await this.button.click();
      }
    });
  }

  /**
   * Close the menu if it is currently expanded.
   * @returns {Promise<void>}
   */
  async closeMenu(): Promise<void> {
    await test.step(`Close menu in ${this.elementReportName}`, async () => {
      if (await this.isExpanded()) {
        await this.button.click();
      }
    });
  }

  /**
   * Check if the menu is expanded.
   * @returns {Promise<boolean>} True if the menu is expanded, false otherwise.
   */
  async isExpanded(): Promise<boolean> {
    let isExpanded = false;
    await test.step(`Check if menu is expanded in ${this.elementReportName}`, async () => {
      isExpanded = (await this.button.getAttributeValue("aria-expanded")) === "true";
    });
    return isExpanded;
  }
}