/*!
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
import { Heading, IconButton, useDisclosure } from "@chakra-ui/react";
import { useTranslation } from "react-i18next";
import { FiEdit } from "react-icons/fi";

import type { PoolResponse } from "openapi/requests/types.gen";
import { Dialog, Tooltip } from "src/components/ui";
import { useEditPool } from "src/queries/useEditPool";

import PoolForm, { type PoolBody } from "./PoolForm";

type Props = {
  readonly pool: PoolResponse;
};

const EditPoolButton = ({ pool }: Props) => {
  const { t: translate } = useTranslation("admin");
  const { onClose, onOpen, open } = useDisclosure();
  const initialPoolValue: PoolBody = {
    description: pool.description ?? "",
    include_deferred: pool.include_deferred,
    name: pool.name,
    slots: pool.slots,
    team_name: pool.team_name ?? "",
  };
  const { editPool, error, isPending, setError } = useEditPool(initialPoolValue, {
    onSuccessConfirm: onClose,
  });

  const handleClose = () => {
    setError(undefined);
    onClose();
  };

  return (
    <>
      <Tooltip content={translate("pools.edit")}>
        <IconButton
          aria-label={translate("pools.edit")}
          colorPalette="brand"
          onClick={onOpen}
          size="md"
          variant="ghost"
        >
          <FiEdit />
        </IconButton>
      </Tooltip>

      <Dialog.Root onOpenChange={handleClose} open={open} size="xl">
        <Dialog.Content backdrop>
          <Dialog.Header>
            <Heading size="xl">{translate("pools.edit")}</Heading>
          </Dialog.Header>

          <Dialog.CloseTrigger />

          <Dialog.Body>
            <PoolForm
              error={error}
              initialPool={initialPoolValue}
              isPending={isPending}
              manageMutate={editPool}
              setError={setError}
            />
          </Dialog.Body>
        </Dialog.Content>
      </Dialog.Root>
    </>
  );
};

export default EditPoolButton;
