import { DeploymentResources as DeploymentResourcesV1 } from 'graphql-transformer-core';
import { $TSContext, ApiCategoryFacade } from '@aws-amplify/amplify-cli-core';
import { AmplifyGraphQLTransformerErrorConverter } from '../errors/amplify-error-converter';
import { DeploymentResources as DeploymentResourcesV2 } from './cdk-compat/deployment-resources';
import { transformGraphQLSchemaV1 } from './transform-graphql-schema-v1';
import { transformGraphQLSchemaV2 } from './transform-graphql-schema-v2';

/**
 * Determine which transformer version is in effect, and execute the appropriate transformation.
 */
export const transformGraphQLSchema = async (
  context: $TSContext,
  options: any,
): Promise<DeploymentResourcesV2 | DeploymentResourcesV1 | undefined> => {
  try {
    const transformerVersion = await ApiCategoryFacade.getTransformerVersion(context);
    return transformerVersion === 2 ? await transformGraphQLSchemaV2(context, options) : await transformGraphQLSchemaV1(context, options);
  } catch (error) {
    throw AmplifyGraphQLTransformerErrorConverter.convert(error);
  }
};