/*
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *     http://www.apache.org/licenses/LICENSE-2.0
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { Locator, Page } from '@playwright/test';
import { BasePage } from './base-page';

export class NotebookPage extends BasePage {
  readonly notebookContainer: Locator;
  readonly actionBar: Locator;
  readonly sidebarArea: Locator;
  readonly extensionArea: Locator;
  readonly paragraphInner: Locator;
  readonly settingsButton: Locator;

  constructor(page: Page) {
    super(page);
    this.notebookContainer = page.locator('.notebook-container');
    this.actionBar = page.locator('zeppelin-notebook-action-bar');
    this.sidebarArea = page.locator('.sidebar-area[nz-resizable]');
    this.extensionArea = page.locator('.extension-area');
    this.paragraphInner = page.locator('.paragraph-inner[nz-row]');
    this.settingsButton = page.locator('button i[nztype="setting"]').first();
  }

  async getSidebarWidth(): Promise<number> {
    const sidebarElement = await this.sidebarArea.boundingBox();
    return sidebarElement?.width || 0;
  }

  async getNotebookContainerClass(): Promise<string | null> {
    return await this.notebookContainer.getAttribute('class');
  }
}
