/*!
 * 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 { useDisclosure } from "@chakra-ui/react";
import { useTranslation } from "react-i18next";
import { CgRedo } from "react-icons/cg";

import type { LightGridTaskInstanceSummary, TaskInstanceResponse } from "openapi/requests/types.gen";
import { ClearGroupTaskInstanceDialog } from "src/components/Clear/TaskInstance/ClearGroupTaskInstanceDialog";
import { IconButton } from "src/components/ui";
import { SHORTCUTS } from "src/context/keyboardShortcuts";
import { useShortcut } from "src/hooks/useShortcut";

import ClearTaskInstanceDialog from "./ClearTaskInstanceDialog";

type Props = {
  // Clear all mapped TIs of a task at once. When set, dagId/dagRunId/taskId
  // are required; taskInstance/groupTaskInstance must be absent.
  readonly allMapped?: boolean;
  readonly dagId?: string;
  readonly dagRunId?: string;
  readonly groupTaskInstance?: LightGridTaskInstanceSummary;
  readonly isHotkeyEnabled?: boolean;
  // Optional: allow parent to handle opening a stable, page-level dialog
  readonly onOpen?: (ti: LightGridTaskInstanceSummary | TaskInstanceResponse) => void;
  readonly taskId?: string;
  readonly taskInstance?: TaskInstanceResponse;
};

const ClearTaskInstanceButton = ({
  allMapped = false,
  dagId,
  dagRunId,
  groupTaskInstance,
  isHotkeyEnabled = false,
  onOpen,
  taskId,
  taskInstance,
}: Props) => {
  const { onClose, onOpen: onOpenInternal, open } = useDisclosure();
  const { t: translate } = useTranslation();
  const isGroup = groupTaskInstance && !taskInstance;
  const useInternalDialog = !Boolean(onOpen);

  const selectedInstance = taskInstance ?? groupTaskInstance;

  useShortcut({
    ...SHORTCUTS.runActions.clearTaskInstance,
    callback: () => {
      if (onOpen && selectedInstance) {
        onOpen(selectedInstance);
      } else {
        onOpenInternal();
      }
    },
    options: { enabled: isHotkeyEnabled },
  });

  const label = allMapped
    ? isHotkeyEnabled
      ? translate("dags:runAndTaskActions.clearAllMapped.buttonTooltip")
      : translate("dags:runAndTaskActions.clearAllMapped.button")
    : isHotkeyEnabled
      ? translate("dags:runAndTaskActions.clear.buttonTooltip")
      : translate("dags:runAndTaskActions.clear.button", { type: translate("taskInstance_one") });

  return (
    <>
      <IconButton
        label={label}
        onClick={() => (onOpen && selectedInstance ? onOpen(selectedInstance) : onOpenInternal())}
      >
        <CgRedo />
      </IconButton>

      {useInternalDialog && isGroup ? (
        <ClearGroupTaskInstanceDialog onClose={onClose} open={open} taskInstance={groupTaskInstance} />
      ) : undefined}

      {useInternalDialog &&
      !isGroup &&
      allMapped &&
      dagId !== undefined &&
      dagRunId !== undefined &&
      taskId !== undefined ? (
        <ClearTaskInstanceDialog
          allMapped
          dagId={dagId}
          dagRunId={dagRunId}
          onClose={onClose}
          open={open}
          taskId={taskId}
        />
      ) : undefined}

      {useInternalDialog && !isGroup && !allMapped && taskInstance ? (
        <ClearTaskInstanceDialog onClose={onClose} open={open} taskInstance={taskInstance} />
      ) : undefined}
    </>
  );
};

export default ClearTaskInstanceButton;
