/*!
 * 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 { Flex } from "@chakra-ui/react";
import { useTranslation } from "react-i18next";
import { BiTargetLock } from "react-icons/bi";
import { FiBarChart, FiUser, FiDatabase } from "react-icons/fi";
import { LuBrackets } from "react-icons/lu";
import {
  MdDateRange,
  MdSearch,
  MdHistory,
  MdHourglassEmpty,
  MdCode,
  MdPlayArrow,
  MdCheckCircle,
  MdBuild,
  MdComputer,
} from "react-icons/md";
import { PiQueue } from "react-icons/pi";

import type { DagRunState, DagRunType, TaskInstanceState } from "openapi/requests/types.gen";
import { DagIcon } from "src/assets/DagIcon";
import { TaskIcon } from "src/assets/TaskIcon";
import type { FilterConfig } from "src/components/FilterBar";
import { RunTypeIcon } from "src/components/RunTypeIcon";
import { StateBadge } from "src/components/StateBadge";
import {
  dagRunStateOptions,
  dagRunTypeOptions,
  jobStateOptions,
  jobTypeOptions,
  taskInstanceStateOptions,
} from "src/constants/stateOptions";

import { SearchParamsKeys } from "./searchParams";

export enum FilterTypes {
  DATE = "date",
  DATERANGE = "daterange",
  NUMBER = "number",
  SELECT = "select",
  TEXT = "text",
}

export const useFilterConfigs = () => {
  const { t: translate } = useTranslation(["browse", "common", "components", "admin", "hitl"]);

  const filterConfigMap = {
    [SearchParamsKeys.ASSET_EVENT_DATE_RANGE]: {
      endKey: SearchParamsKeys.END_DATE,
      icon: <MdDateRange />,
      label: translate("components:backfill.dateRange"),
      startKey: SearchParamsKeys.START_DATE,
      type: FilterTypes.DATERANGE,
    },
    [SearchParamsKeys.BODY_SEARCH]: {
      hotkeyDisabled: true,
      icon: <MdSearch />,
      label: translate("hitl:filters.body"),
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.BUNDLE_VERSION]: {
      hotkeyDisabled: true,
      icon: <MdCode />,
      label: translate("components:versionDetails.bundleVersion"),
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.CONF_CONTAINS]: {
      hotkeyDisabled: true,
      icon: <MdCode />,
      label: translate("common:dagRun.conf"),
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.CONSUMING_ASSET_PATTERN]: {
      hotkeyDisabled: true,
      icon: <FiDatabase />,
      label: translate("common:consumingAsset"),
      placeholder: translate("common:filters.searchAsset"),
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.CREATED_AT_RANGE]: {
      endKey: SearchParamsKeys.CREATED_AT_LTE,
      icon: <MdDateRange />,
      label: translate("hitl:filters.createdAt"),
      startKey: SearchParamsKeys.CREATED_AT_GTE,
      type: FilterTypes.DATERANGE,
    },
    [SearchParamsKeys.DAG_DISPLAY_NAME_PATTERN]: {
      hotkeyDisabled: true,
      icon: <DagIcon />,
      label: translate("common:dagId"),
      supportsAdvancedSearch: true,
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.DAG_ID]: {
      hotkeyDisabled: true,
      icon: <DagIcon />,
      label: translate("common:dagId"),
      supportsAdvancedSearch: true,
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.DAG_ID_PATTERN]: {
      hotkeyDisabled: true,
      icon: <DagIcon />,
      label: translate("common:dagId"),
      supportsAdvancedSearch: true,
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.DAG_VERSION]: {
      hotkeyDisabled: true,
      icon: <MdHistory />,
      label: translate("common:dagRun.dagVersions"),
      min: 1,
      type: FilterTypes.NUMBER,
    },
    [SearchParamsKeys.DEADLINE_TIME_RANGE]: {
      endKey: SearchParamsKeys.DEADLINE_TIME_LTE,
      icon: <MdDateRange />,
      label: translate("browse:deadlines.columns.deadlineTime"),
      startKey: SearchParamsKeys.DEADLINE_TIME_GTE,
      type: FilterTypes.DATERANGE,
    },
    [SearchParamsKeys.DURATION_GTE]: {
      icon: <MdHourglassEmpty />,
      label: translate("common:filters.durationFrom"),
      min: 0,
      type: FilterTypes.NUMBER,
    },
    [SearchParamsKeys.DURATION_LTE]: {
      icon: <MdHourglassEmpty />,
      label: translate("common:filters.durationTo"),
      min: 0,
      type: FilterTypes.NUMBER,
    },
    [SearchParamsKeys.END_DATE_RANGE]: {
      endKey: SearchParamsKeys.END_DATE_LTE,
      icon: <MdDateRange />,
      label: translate("common:endDate"),
      startKey: SearchParamsKeys.END_DATE_GTE,
      type: FilterTypes.DATERANGE,
    },
    [SearchParamsKeys.EVENT_DATE_RANGE]: {
      endKey: SearchParamsKeys.BEFORE,
      icon: <MdDateRange />,
      label: translate("common:logicalDate"),
      startKey: SearchParamsKeys.AFTER,
      type: FilterTypes.DATERANGE,
    },
    [SearchParamsKeys.EVENT_TYPE]: {
      label: translate("browse:auditLog.filters.eventType"),
      supportsAdvancedSearch: true,
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.EXECUTOR_CLASS]: {
      hotkeyDisabled: true,
      icon: <MdBuild />,
      label: translate("admin:jobs.columns.executorClass"),
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.HOSTNAME]: {
      hotkeyDisabled: true,
      icon: <MdComputer />,
      label: translate("admin:jobs.columns.hostname"),
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.JOB_STATE]: {
      icon: <MdCheckCircle />,
      label: translate("common:state"),
      options: jobStateOptions.items.map((option) => ({
        label: translate(option.label),
        value: option.value === "all" ? "" : option.value,
      })),
      type: FilterTypes.SELECT,
    },
    [SearchParamsKeys.JOB_TYPE]: {
      icon: <MdBuild />,
      label: translate("admin:jobs.columns.jobType"),
      options: jobTypeOptions.items.map((option) => ({
        label: translate(option.label),
        value: option.value === "all" ? "" : option.value,
      })),
      type: FilterTypes.SELECT,
    },
    [SearchParamsKeys.KEY_PATTERN]: {
      icon: <MdSearch />,
      label: translate("admin:columns.key"),
      supportsAdvancedSearch: true,
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.LOGICAL_DATE_RANGE]: {
      endKey: SearchParamsKeys.LOGICAL_DATE_LTE,
      icon: <MdDateRange />,
      label: translate("common:logicalDate"),
      startKey: SearchParamsKeys.LOGICAL_DATE_GTE,
      type: FilterTypes.DATERANGE,
    },
    [SearchParamsKeys.MAP_INDEX]: {
      icon: <LuBrackets />,
      label: translate("common:mapIndex"),
      min: -1,
      type: FilterTypes.NUMBER,
    },
    [SearchParamsKeys.MISSED]: {
      icon: <MdCheckCircle />,
      label: translate("browse:deadlines.filters.status"),
      options: [
        { label: translate("browse:deadlines.filters.statusOptions.all"), value: "" },
        { label: translate("browse:deadlines.filters.statusOptions.pending"), value: "false" },
        { label: translate("browse:deadlines.filters.statusOptions.missed"), value: "true" },
      ],
      type: FilterTypes.SELECT,
    },
    [SearchParamsKeys.NAME_PATTERN]: {
      hotkeyDisabled: true,
      icon: <TaskIcon />,
      label: translate("common:taskId"),
      supportsAdvancedSearch: true,
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.OPERATOR_NAME_PATTERN]: {
      hotkeyDisabled: true,
      icon: <MdBuild />,
      label: translate("common:task.operator"),
      supportsAdvancedSearch: true,
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.PARTITION_KEY_PATTERN]: {
      hotkeyDisabled: true,
      icon: <MdSearch />,
      label: translate("common:dagRun.partitionKey"),
      supportsAdvancedSearch: true,
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.POOL_NAME_PATTERN]: {
      hotkeyDisabled: true,
      icon: <BiTargetLock />,
      label: translate("common:taskInstance.pool"),
      supportsAdvancedSearch: true,
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.QUEUE_NAME_PATTERN]: {
      hotkeyDisabled: true,
      icon: <PiQueue />,
      label: translate("common:taskInstance.queue"),
      supportsAdvancedSearch: true,
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.RENDERED_MAP_INDEX]: {
      hotkeyDisabled: true,
      icon: <MdSearch />,
      label: translate("common:taskInstance.renderedMapIndex"),
      supportsAdvancedSearch: true,
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.RESPONDED_BY_USER_NAME]: {
      hotkeyDisabled: true,
      icon: <FiUser />,
      label: translate("hitl:response.responded_by_user_name"),
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.RESPONSE_RECEIVED]: {
      icon: <FiUser />,
      label: translate("hitl:requiredActionState"),
      options: [
        { label: translate("hitl:filters.response.all"), value: "all" },
        {
          label: <StateBadge state="awaiting_input">{translate("hitl:filters.response.pending")}</StateBadge>,
          value: "false",
        },
        {
          label: <StateBadge state="success">{translate("hitl:filters.response.received")}</StateBadge>,
          value: "true",
        },
      ],
      type: FilterTypes.SELECT,
    },
    [SearchParamsKeys.RUN_AFTER_RANGE]: {
      endKey: SearchParamsKeys.RUN_AFTER_LTE,
      icon: <MdDateRange />,
      label: translate("common:dagRun.runAfter"),
      startKey: SearchParamsKeys.RUN_AFTER_GTE,
      type: FilterTypes.DATERANGE,
    },
    [SearchParamsKeys.RUN_ID]: {
      hotkeyDisabled: true,
      icon: <FiBarChart />,
      label: translate("common:runId"),
      supportsAdvancedSearch: true,
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.RUN_ID_PATTERN]: {
      hotkeyDisabled: true,
      icon: <FiBarChart />,
      label: translate("common:runId"),
      supportsAdvancedSearch: true,
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.RUN_TYPE]: {
      icon: <MdPlayArrow />,
      label: translate("common:dagRun.runType"),
      options: dagRunTypeOptions.items.map((option) => ({
        label:
          option.value === "all" ? (
            translate(option.label)
          ) : (
            <Flex alignItems="center" gap={1}>
              <RunTypeIcon runType={option.value as DagRunType} />
              {translate(option.label)}
            </Flex>
          ),
        value: option.value === "all" ? "" : option.value,
      })),
      type: FilterTypes.SELECT,
    },
    [SearchParamsKeys.START_DATE_RANGE]: {
      endKey: SearchParamsKeys.START_DATE_LTE,
      icon: <MdDateRange />,
      label: translate("common:startDate"),
      startKey: SearchParamsKeys.START_DATE_GTE,
      type: FilterTypes.DATERANGE,
    },
    [SearchParamsKeys.STATE]: {
      icon: <MdCheckCircle />,
      label: translate("common:state"),
      options: dagRunStateOptions.items.map((option) => ({
        label:
          option.value === "all" ? (
            translate(option.label)
          ) : (
            <StateBadge state={option.value as DagRunState}>{translate(option.label)}</StateBadge>
          ),
        value: option.value === "all" ? "" : option.value,
      })),
      type: FilterTypes.SELECT,
    },
    [SearchParamsKeys.SUBJECT_SEARCH]: {
      icon: <MdSearch />,
      label: translate("hitl:subject"),
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.TASK_ID]: {
      hotkeyDisabled: true,
      icon: <TaskIcon />,
      label: translate("common:taskId"),
      supportsAdvancedSearch: true,
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.TASK_ID_PATTERN]: {
      hotkeyDisabled: true,
      icon: <TaskIcon />,
      label: translate("common:taskId"),
      supportsAdvancedSearch: true,
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.TASK_STATE]: {
      icon: <MdCheckCircle />,
      label: translate("common:state"),
      options: taskInstanceStateOptions.items.map((option) => ({
        label:
          option.value === "all" ? (
            translate(option.label)
          ) : (
            <StateBadge state={option.value as TaskInstanceState}>{translate(option.label)}</StateBadge>
          ),
        value: option.value === "all" ? "" : option.value,
      })),
      type: FilterTypes.SELECT,
    },
    [SearchParamsKeys.TRIGGERING_USER_NAME_PATTERN]: {
      hotkeyDisabled: true,
      icon: <FiUser />,
      label: translate("common:dagRun.triggeringUser"),
      supportsAdvancedSearch: true,
      type: FilterTypes.TEXT,
    },
    [SearchParamsKeys.TRY_NUMBER]: {
      label: translate("common:tryNumber"),
      min: 1,
      type: FilterTypes.NUMBER,
    },
    [SearchParamsKeys.USER]: {
      icon: <FiUser />,
      label: translate("common:user"),
      supportsAdvancedSearch: true,
      type: FilterTypes.TEXT,
    },
  };

  const getFilterConfig = (key: keyof typeof filterConfigMap): FilterConfig => ({
    key,
    ...filterConfigMap[key],
  });

  return { getFilterConfig };
};
