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

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

declare module '../../compass-requests' {
  interface CompassRequests {
    /**
     * Adds a collection of labels to a component.
     *
     * **Required Oauth Scopes:** `write:component:compass`
     */
    addLabels(
      input: AddCompassComponentLabelsInput,
    ): Promise<ApiPayload<LabelsPayload>>;
  }
}

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

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

    ({ errors: gqlErrors, data } = await resp.json());
    errorsResp = errorsResp.concat(mapGqlErrors(gqlErrors));
    ({
      addComponentLabels: { addedLabels: labelNames, 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: { labelNames },
  };
};