/*
 * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
 */

import {
  MutationKey,
  useMutation,
  UseMutationResult,
} from "@tanstack/react-query";
import { GustoEmbeddedCore } from "../core.js";
import { timeOffPoliciesUpdateBalance } from "../funcs/timeOffPoliciesUpdateBalance.js";
import { combineSignals } from "../lib/primitives.js";
import { RequestOptions } from "../lib/sdks.js";
import { GustoEmbeddedError } from "../models/errors/gustoembeddederror.js";
import {
  ConnectionError,
  InvalidRequestError,
  RequestAbortedError,
  RequestTimeoutError,
  UnexpectedClientError,
} from "../models/errors/httpclienterrors.js";
import { NotFoundErrorObject } from "../models/errors/notfounderrorobject.js";
import { ResponseValidationError } from "../models/errors/responsevalidationerror.js";
import { SDKValidationError } from "../models/errors/sdkvalidationerror.js";
import { UnprocessableEntityError } from "../models/errors/unprocessableentityerror.js";
import {
  PutV1TimeOffPoliciesTimeOffPolicyUuidBalanceRequest,
  PutV1TimeOffPoliciesTimeOffPolicyUuidBalanceResponse,
} from "../models/operations/putv1timeoffpoliciestimeoffpolicyuuidbalance.js";
import { unwrapAsync } from "../types/fp.js";
import { useGustoEmbeddedContext } from "./_context.js";
import { MutationHookOptions } from "./_types.js";

export type TimeOffPoliciesUpdateBalanceMutationVariables = {
  request: PutV1TimeOffPoliciesTimeOffPolicyUuidBalanceRequest;
  options?: RequestOptions;
};

export type TimeOffPoliciesUpdateBalanceMutationData =
  PutV1TimeOffPoliciesTimeOffPolicyUuidBalanceResponse;

export type TimeOffPoliciesUpdateBalanceMutationError =
  | NotFoundErrorObject
  | UnprocessableEntityError
  | GustoEmbeddedError
  | ResponseValidationError
  | ConnectionError
  | RequestAbortedError
  | RequestTimeoutError
  | InvalidRequestError
  | UnexpectedClientError
  | SDKValidationError;

/**
 * Update employee time off balances
 *
 * @remarks
 * Updates time off hours balances for employees for a time off policy.
 *
 * scope: `time_off_policies:write`
 */
export function useTimeOffPoliciesUpdateBalanceMutation(
  options?: MutationHookOptions<
    TimeOffPoliciesUpdateBalanceMutationData,
    TimeOffPoliciesUpdateBalanceMutationError,
    TimeOffPoliciesUpdateBalanceMutationVariables
  >,
): UseMutationResult<
  TimeOffPoliciesUpdateBalanceMutationData,
  TimeOffPoliciesUpdateBalanceMutationError,
  TimeOffPoliciesUpdateBalanceMutationVariables
> {
  const client = useGustoEmbeddedContext();
  return useMutation({
    ...buildTimeOffPoliciesUpdateBalanceMutation(client, options),
    ...options,
  });
}

export function mutationKeyTimeOffPoliciesUpdateBalance(): MutationKey {
  return ["@gusto/embedded-api", "timeOffPolicies", "updateBalance"];
}

export function buildTimeOffPoliciesUpdateBalanceMutation(
  client$: GustoEmbeddedCore,
  hookOptions?: RequestOptions,
): {
  mutationKey: MutationKey;
  mutationFn: (
    variables: TimeOffPoliciesUpdateBalanceMutationVariables,
  ) => Promise<TimeOffPoliciesUpdateBalanceMutationData>;
} {
  return {
    mutationKey: mutationKeyTimeOffPoliciesUpdateBalance(),
    mutationFn: function timeOffPoliciesUpdateBalanceMutationFn({
      request,
      options,
    }): Promise<TimeOffPoliciesUpdateBalanceMutationData> {
      const mergedOptions = {
        ...hookOptions,
        ...options,
        fetchOptions: {
          ...hookOptions?.fetchOptions,
          ...options?.fetchOptions,
          signal: combineSignals(
            hookOptions?.fetchOptions?.signal,
            options?.fetchOptions?.signal,
          ),
        },
      };
      return unwrapAsync(timeOffPoliciesUpdateBalance(
        client$,
        request,
        mergedOptions,
      ));
    },
  };
}