import type { MesProBagCodeApi } from '#/api/mes/pro/bagCode';

import { onBeforeUnmount, ref } from 'vue';

import { message } from 'ant-design-vue';

import {
  cancelBagCodeGenerateTask,
  getBagCodeGenerateTask,
} from '#/api/mes/pro/bagCode';

/** 弹窗最小接口，避免依赖内部 ModalApi 类型 */
interface GenerateModalApi {
  close: () => Promise<void>;
  setState: (state: {
    showCancelButton?: boolean;
    showConfirmButton?: boolean;
  }) => unknown;
}

interface GenerateTaskCallbacks {
  /** 生成成功（状态 30），调用方刷新列表 */
  onSuccess: () => void;
  /** 生成失败（状态 50），调用方更换幂等键后允许重新提交 */
  onFailed?: () => void;
}

/** 终态：已完成 / 已取消 / 已失败 */
const FINAL_STATUSES = new Set([30, 40, 50]);

export function isFinalTaskStatus(status?: number) {
  return status !== undefined && FINAL_STATUSES.has(status);
}

/**
 * 大批量袋码后台生成：轮询进度、取消任务，并把弹窗按钮切换为进度展示态。
 */
export function useGenerateTask(
  modalApi: GenerateModalApi,
  callbacks: GenerateTaskCallbacks,
) {
  const task = ref<MesProBagCodeApi.GenerateTask>();
  const canceling = ref(false);
  let timer: ReturnType<typeof setTimeout> | undefined;

  function stopPolling() {
    if (timer) clearTimeout(timer);
    timer = undefined;
  }

  function restoreButtons() {
    modalApi.setState({ showConfirmButton: true, showCancelButton: true });
  }

  /** 重置到可编辑态：关闭弹窗、重新打开、失败重试前调用 */
  function reset() {
    stopPolling();
    task.value = undefined;
    canceling.value = false;
    restoreButtons();
  }

  /** 进入进度展示态：隐藏提交/取消按钮，进度区自行提供取消操作 */
  function start(initial: MesProBagCodeApi.GenerateTask) {
    stopPolling();
    task.value = initial;
    modalApi.setState({ showConfirmButton: false, showCancelButton: false });
    void refresh();
  }

  async function refresh() {
    if (!task.value) return;
    stopPolling();
    try {
      const current = await getBagCodeGenerateTask(task.value.id);
      task.value = current;
      if (isFinalTaskStatus(current.status)) {
        finish(current);
        return;
      }
    } catch {
      // 一次查询失败不终止轮询，避免网络抖动把进度卡死
    }
    timer = setTimeout(refresh, 2000);
  }

  function finish(current: MesProBagCodeApi.GenerateTask) {
    if (current.status === 30) {
      message.success(
        `生成完成：共 ${current.totalCount ?? 0} 条袋码` +
          (current.palletCount
            ? `，并自动生成 ${current.palletCount} 个托盘`
            : ''),
      );
      callbacks.onSuccess();
      void modalApi.close();
      return;
    }
    if (current.status === 40) {
      message.info('生成任务已取消，已生成的数据已清理');
      void modalApi.close();
      return;
    }
    // 失败：回到可编辑态，由调用方更换 requestId 后重新提交
    message.error(current.errorMsg ?? '生成失败，请重新提交');
    task.value = undefined;
    restoreButtons();
    callbacks.onFailed?.();
  }

  async function cancel() {
    if (!task.value || canceling.value) return;
    canceling.value = true;
    try {
      await cancelBagCodeGenerateTask(task.value.id);
      message.info('已申请取消，正在停止并清理已生成数据');
    } catch {
      message.error('取消生成失败，请稍后重试');
    } finally {
      canceling.value = false;
    }
  }

  onBeforeUnmount(stopPolling);

  return {
    task,
    canceling,
    cancel,
    reset,
    stopPolling,
    start,
  };
}