/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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 type { Page, TestInfo } from '@playwright/test';
import type { TestAssets } from '../../../helpers/fixtures/testAssets';
import { apiPostChart } from '../../../helpers/api/chart';
import { getDatasetByName } from '../../../helpers/api/dataset';

interface TestChartResult {
  id: number;
  name: string;
}

interface CreateTestChartOptions {
  /** Prefix for generated name (default: 'test_chart') */
  prefix?: string;
}

/**
 * Creates a test chart via the API for E2E testing.
 * Uses the members_channels_2 dataset (loaded via --load-examples).
 *
 * @example
 * const { id, name } = await createTestChart(page, testAssets, test.info());
 *
 * @example
 * const { id, name } = await createTestChart(page, testAssets, test.info(), {
 *   prefix: 'test_delete',
 * });
 */
export async function createTestChart(
  page: Page,
  testAssets: TestAssets,
  testInfo: TestInfo,
  options?: CreateTestChartOptions,
): Promise<TestChartResult> {
  const prefix = options?.prefix ?? 'test_chart';
  const name = `${prefix}_${Date.now()}_${testInfo.parallelIndex}`;

  // Look up the members_channels_2 dataset for chart creation
  const dataset = await getDatasetByName(page, 'members_channels_2');
  if (!dataset) {
    throw new Error(
      'members_channels_2 dataset not found — run Superset with --load-examples',
    );
  }

  const response = await apiPostChart(page, {
    slice_name: name,
    datasource_id: dataset.id,
    datasource_type: 'table',
    viz_type: 'table',
    params: '{}',
  });

  if (!response.ok()) {
    throw new Error(`Failed to create test chart: ${response.status()}`);
  }

  const body = await response.json();
  // Handle both response shapes: { id } or { result: { id } }
  const id = body.result?.id ?? body.id;
  if (!id) {
    throw new Error(
      `Chart creation returned no id. Response: ${JSON.stringify(body)}`,
    );
  }

  testAssets.trackChart(id);

  return { id, name };
}