import SchemaBuilder, { isThenable, type MaybePromise, type SchemaTypes } from '@pothos/core';
import { ForbiddenError } from './errors';
import RequestCache from './request-cache';
import type { AuthFailure } from './types';

const schemaBuilderProto = SchemaBuilder.prototype as PothosSchemaTypes.SchemaBuilder<SchemaTypes>;

schemaBuilderProto.runAuthScopes = function runAuthScopes(
  context,
  scopes,
  unauthorizedError = (result) => new ForbiddenError(result.message, result.failure),
): MaybePromise<void> {
  const cache = RequestCache.fromContext(context, this);

  const resultOrPromise = cache.evaluateScopeMap(scopes);

  if (isThenable(resultOrPromise)) {
    return resultOrPromise.then(handleScopeResult);
  }

  handleScopeResult(resultOrPromise);

  function handleScopeResult(result: AuthFailure | null) {
    if (result) {
      const error = unauthorizedError({
        message: 'Unauthorized',
        failure: result,
      });

      if (typeof error === 'string') {
        throw new ForbiddenError(error, result);
      }

      throw error;
    }
  }
};