import type { PayloadRequest } from "payload"
import type { AccountInfo, OAuthProviderConfig } from "../../types.js"
import {
  InvalidOAuthAlgorithm,
  InvalidOAuthResource,
  InvalidProvider,
} from "../errors/consoleErrors.js"
import { OIDCAuthorization } from "../protocols/oauth/oidc_authorization.js"
import { OAuth2Authorization } from "../protocols/oauth/oauth2_authorization.js"
import { OIDCCallback } from "../protocols/oauth/oidc_callback.js"
import { OAuth2Callback } from "../protocols/oauth/oauth2_callback.js"

export function OAuthHandlers(
  request: PayloadRequest,
  resource: string,
  provider: OAuthProviderConfig,
  sessionCallBack: (oauthAccountInfo: AccountInfo) => Promise<Response>,
): Promise<Response> {
  if (!provider) {
    throw new InvalidProvider()
  }

  switch (resource) {
    case "authorization":
      switch (provider.algorithm) {
        case "oidc":
          return OIDCAuthorization(request, provider)
        case "oauth2":
          return OAuth2Authorization(request, provider)
        default:
          throw new InvalidOAuthAlgorithm()
      }
    case "callback":
      switch (provider.algorithm) {
        case "oidc":
          return OIDCCallback(request, provider, sessionCallBack)
        case "oauth2":
          return OAuth2Callback(request, provider, sessionCallBack)
        default:
          throw new InvalidOAuthAlgorithm()
      }
    default:
      throw new InvalidOAuthResource()
  }
}