/*
 *  Copyright 2024 Collate.
 *  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 { Page } from '@playwright/test';
import MysqlIngestionClass from '../support/entity/ingestion/MySqlIngestionClass';
import { getApiContext, toastNotification } from './common';
import { visitServiceDetailsPage } from './service';

export const addAndTriggerAutoClassificationPipeline = async (
  page: Page,
  mysqlService: MysqlIngestionClass
) => {
  const { apiContext } = await getApiContext(page);

  await visitServiceDetailsPage(
    page,
    {
      type: mysqlService.category,
      name: mysqlService.name,
      displayName: mysqlService.name,
    },
    true
  );

  // Add auto classification ingestion
  await page.click('[data-testid="ingestions"]');

  await page.click('[data-testid="add-new-ingestion-button"]');

  await page.waitForSelector('[data-menu-id*="autoClassification"]');

  await page.click('[data-menu-id*="autoClassification"]');

  // Fill the auto classification form details
  await page.waitForSelector('#root\\/tableFilterPattern\\/includes');

  await mysqlService.fillIngestionDetails(page);

  await page.click('#root\\/enableAutoClassification');

  await page.click('[data-testid="submit-btn"]');

  // Make sure we create ingestion with None schedule to avoid conflict between Airflow and Argo behavior
  await mysqlService.scheduleIngestion(page);

  await page.click('[data-testid="view-service-button"]');

  // Header available once page loads
  await page.getByTestId('loader').waitFor({ state: 'detached' });
  await page.getByTestId('ingestions').click();
  await page
    .getByLabel('Ingestions')
    .getByTestId('loader')
    .waitFor({ state: 'detached' });

  const response = await apiContext
    .get(
      `/api/v1/services/ingestionPipelines?service=${encodeURIComponent(
        mysqlService.name
      )}&pipelineType=autoClassification&serviceType=databaseService&limit=1`
    )
    .then((res) => res.json());

  // need manual wait to settle down the deployed pipeline, before triggering the pipeline
  await page.waitForTimeout(3000);

  await page.click(
    `[data-row-key*="${response.data[0].name}"] [data-testid="more-actions"]`
  );
  await page.getByTestId('run-button').click();

  await toastNotification(page, `Pipeline triggered successfully!`);

  // need manual wait to make sure we are awaiting on latest run results
  await page.waitForTimeout(2000);

  await mysqlService.handleIngestionRetry('autoClassification', page);
};