/// <reference path="../../compass-requests.ts" />

import {
  ApiPayload,
  CreateCompassComponentRelationship,
  CreateCompassRelationshipInput,
  GqlError,
  MutationError,
  RelationshipPayload,
  SdkError,
} from '@atlassian/forge-graphql-types';
import { CompassRequests } from '../../compass-requests';
import {
  mapGqlErrors,
  mapMutationErrors,
  parsingResponseError,
} from '../../helpers';

declare module '../../compass-requests' {
  interface CompassRequests {
    /**
     * Creates a new relationship between two components.
     *
     * **Required Oauth Scopes:** `write:component:compass`
     */
    createRelationship(
      input: CreateCompassRelationshipInput,
    ): Promise<ApiPayload<RelationshipPayload>>;
  }
}

CompassRequests.prototype.createRelationship = async function (
  this: CompassRequests,
  input,
) {
  let relationship = null;
  let gqlErrors: Array<GqlError>;
  let mutationErrors: Array<MutationError>;
  let errorsResp: Array<SdkError> = [];
  let data;

  try {
    const resp = await this.api.requestGraph(
      CreateCompassComponentRelationship,
      { input },
      'createRelationship',
    );

    ({ errors: gqlErrors, data } = await resp.json());
    errorsResp = errorsResp.concat(mapGqlErrors(gqlErrors));
    ({
      createRelationship: {
        createdCompassRelationship: relationship,
        errors: mutationErrors,
      },
    } = data.compass);
    errorsResp = errorsResp.concat(mapMutationErrors(mutationErrors));
  } catch (e) {
    if (errorsResp.length === 0) {
      errorsResp.push(parsingResponseError(e));
    }
  }

  return {
    errors: errorsResp,
    success: errorsResp.length === 0,
    data: { relationship },
  };
};