import { currentCtx as trackCtx } from "./lib/track";
// Client gọi backend FastAPI. Dev: Vite proxy /api -> :8000.

const BASE = "";

// Máy chủ KHÔNG phải lúc nào cũng trả JSON: Space bận, vừa khởi động lại hoặc
// cổng vào HF hết giờ chờ thì nó chen vào một trang HTML ("<!DOCTYPE html>...").
// Gọi thẳng r.json() lúc đó ném "Unexpected token '<'" — câu người dùng không
// hiểu, lại che mất chuyện đang thật sự xảy ra. Mọi lời gọi đi qua đây: đọc body
// MỘT lần, tự nhận ra HTML/không-JSON, và gắn cờ `transient` cho những lỗi chỉ
// là hụt hơi tạm thời (mạng chớp, 5xx, 429, 408) để chỗ gọi biết nên thử lại.
const BUSY_MSG = "Máy chủ đang bận hoặc vừa khởi động lại, chưa trả về dữ liệu.";

function tagged(msg, { status = 0, transient = false } = {}) {
  const e = new Error(msg);
  e.status = status;
  e.transient = transient;
  return e;
}

// Lỗi FastAPI trả về dạng {"detail": "..."} — lấy ra câu chữ cho người đọc,
// đừng ném nguyên cục JSON (hay nguyên trang HTML) vào alert.
function httpError(r, text, data) {
  const d = data?.detail ?? data?.error ?? data?.message;
  let msg = "";
  if (typeof d === "string" && d) msg = d;
  else if (Array.isArray(d) && d.length) msg = d.map((x) => x.msg || "").join("; ");
  else if (text && !/^\s*</.test(text) && text.trim().length <= 300) msg = text.trim();
  else msg = `${BUSY_MSG} (lỗi ${r.status}${r.statusText ? " " + r.statusText : ""})`;
  return tagged(msg, {
    status: r.status,
    transient: r.status >= 500 || r.status === 429 || r.status === 408,
  });
}

async function readJson(r) {
  const text = await r.text().catch(() => "");
  let data = null, isJson = false;
  if (text) { try { data = JSON.parse(text); isJson = true; } catch { /* không phải JSON */ } }
  if (!r.ok) throw httpError(r, text, data);
  if (!isJson) throw tagged(`${BUSY_MSG} Thử lại sau giây lát.`, { status: r.status, transient: true });
  return data;
}

const sleep = (ms) => new Promise((ok) => setTimeout(ok, ms));

// Mở ra cho các trang gọi thẳng (Listing, Flux2, Đăng nhập, Quản trị...) —
// đừng dùng fetch trần rồi r.json() nữa, lỗi "Unexpected token '<'" sinh ra từ đó.
//
// Lời gọi CHỈ ĐỌC (GET/HEAD) gặp trục trặc tạm thời thì tự thử lại vài nhịp:
// đọc lại thì cùng lắm tốn thêm một vòng, không đẻ ra việc mới. Lệnh GHI
// (POST/DELETE) KHÔNG tự thử lại — bấm một lần mà chạy hai job thì tai hại hơn
// là báo lỗi để người dùng tự bấm lại.
export async function request(path, init, { retries = 2, backoff = 600 } = {}) {
  const method = (init?.method || "GET").toUpperCase();
  const canRetry = method === "GET" || method === "HEAD";
  for (let attempt = 0; ; attempt++) {
    try {
      let r;
      try {
        r = await fetch(BASE + path, init);
      } catch (e) {
        // Huỷ có chủ đích (AbortController hết giờ chờ) không phải trục trặc
        // mạng — trả nguyên lỗi để chỗ gọi tự phân biệt.
        if (e?.name === "AbortError") throw e;
        // fetch chỉ ném khi mạng đứt / máy chủ chưa nghe — luôn là tạm thời.
        throw tagged("Mất kết nối tới máy chủ. Kiểm tra mạng rồi thử lại.", { transient: true });
      }
      return await readJson(r);
    } catch (e) {
      if (!canRetry || !e?.transient || attempt >= retries || init?.signal?.aborted) throw e;
      await sleep(backoff * (attempt + 1));
    }
  }
}

export const jget = (path) => request(path);

export function jpost(path, body) {
  return request(path, {
    method: "POST",
    headers: { "Content-Type": "application/json" },
    body: JSON.stringify(body),
  });
}

function upload(path, file, fields = {}) {
  const fd = new FormData();
  fd.append("file", file);
  for (const [k, v] of Object.entries(fields)) if (v != null) fd.append(k, v);
  return request(path, { method: "POST", body: fd });
}

// Upload 1 file kèm tiến độ BYTE (fetch không có progress event -> dùng XHR).
// onProgress(loaded, total) gọi liên tục khi bytes đẩy lên.
export function uploadWithProgress(path, file, fields = {}, onProgress) {
  return new Promise((resolve, reject) => {
    const fd = new FormData();
    fd.append("file", file);
    for (const [k, v] of Object.entries(fields)) if (v != null) fd.append(k, v);
    const xhr = new XMLHttpRequest();
    xhr.open("POST", BASE + path);
    // Upload đi bằng XHR (để có % tiến độ) nên KHÔNG qua bản vá fetch trong
    // lib/track.js -> gắn tay ngữ cảnh theo dõi để backend vẫn quy được về
    // đúng task của đúng người.
    try {
      const c = trackCtx();
      if (c.user_id) {
        xhr.setRequestHeader("X-TNT-User", c.user_id);
        xhr.setRequestHeader("X-TNT-Session", c.session_id);
        if (c.task_id) xhr.setRequestHeader("X-TNT-Task", c.task_id);
        if (c.feature) xhr.setRequestHeader("X-TNT-Feature", c.feature);
      }
    } catch { /* thiếu header theo dõi không được chặn upload */ }
    if (xhr.upload && onProgress) {
      xhr.upload.onprogress = (e) => {
        if (e.lengthComputable) onProgress(e.loaded, e.total);
      };
    }
    xhr.onload = () => {
      if (xhr.status >= 200 && xhr.status < 300) {
        // 200 mà thân là HTML (cổng vào chen vào) thì coi như hỏng tạm thời,
        // đừng resolve {} rồi để chỗ gọi vấp vào chỗ thiếu trường.
        try { resolve(JSON.parse(xhr.responseText)); }
        catch { reject(tagged(`${BUSY_MSG} Thử lại sau giây lát.`, { status: xhr.status, transient: true })); }
      } else {
        // Non-2xx cũng có thể là trang HTML của cổng vào, không phải JSON.
        let msg = "";
        try {
          const j = JSON.parse(xhr.responseText);
          const d = j?.detail ?? j?.error ?? j?.message;
          if (typeof d === "string") msg = d;
        } catch { /* không phải JSON */ }
        if (!msg) {
          const t = (xhr.responseText || "").trim();
          msg = t && !t.startsWith("<") && t.length <= 300
            ? t
            : `${BUSY_MSG} (lỗi ${xhr.status}${xhr.statusText ? " " + xhr.statusText : ""})`;
        }
        reject(tagged(msg, { status: xhr.status, transient: xhr.status >= 500 }));
      }
    };
    xhr.onerror = () => reject(new Error("Lỗi mạng khi tải lên"));
    xhr.onabort = () => reject(new Error("Đã huỷ tải lên"));
    xhr.send(fd);
  });
}

// Chạy nhiều tác vụ song song có GIỚI HẠN luồng (mặc định 4) — nhanh hơn tuần tự
// mà không làm nghẽn trình duyệt (mỗi host chỉ ~6 kết nối đồng thời).
export async function runPool(count, worker, limit = 4) {
  let next = 0;
  const run = async () => {
    while (next < count) {
      const i = next++;
      await worker(i);
    }
  };
  await Promise.all(Array.from({ length: Math.min(limit, count) }, run));
}

export const fileUrl = (u) => BASE + u;

export const api = {
  job: (id) => jget(`/api/jobs/${id}`),
  listInputs: (sid) => jget(`/api/sessions/${sid}/inputs`),
  cancel: (id) => jpost(`/api/jobs/${id}/cancel`, {}),
  dropSession: (sid) => request(`/api/sessions/${sid}`, { method: "DELETE" }),

  // Down list video (không dùng cookie)
  parseLinks: (raw) => jpost("/api/download/parse", { raw }),
  startDownload: (urls, zip_output, username = "") =>
    jpost("/api/download/start", { urls, zip_output, username }),
  push: (session, paths, target_tool) =>
    jpost("/api/download/push", { session, paths, target_tool }),

  // Cut video
  cutUpload: (file, session) => upload("/api/cut/upload", file, { session }),
  cutPreview: (req) => jpost("/api/cut/preview", req),
  cutExecute: (req) => jpost("/api/cut/execute", req),
  cutExecuteBatch: (req) => jpost("/api/cut/execute-batch", req),
  // segments = mảng nhóm: segments[i] = các khúc thứ i của mọi video -> cùng 1 vị trí Xào
  cutPushShuffle: (session, segments) =>
    jpost("/api/cut/push-shuffle", { session, segments }),
  // Tách video -> video không lời + mp3
  cutSplitAudio: (session, file, username = "") =>
    jpost("/api/cut/split-audio", { session, file, username }),

  // Xào video
  shuffleUpload: (file, segment, session) =>
    upload("/api/shuffle/upload", file, { segment, session }),
  shuffleUploadAudio: (file, session) =>
    upload("/api/shuffle/upload-audio", file, { session }),
  // Bản có tiến độ byte (dùng cho loader nhanh + chuẩn khi thả clip/thư mục).
  shuffleUploadP: (file, segment, session, onProgress) =>
    uploadWithProgress("/api/shuffle/upload", file, { segment, session }, onProgress),
  shuffleUploadAudioP: (file, session, onProgress) =>
    uploadWithProgress("/api/shuffle/upload-audio", file, { session }, onProgress),
  shuffleSegments: (sid) => jget(`/api/shuffle/segments/${sid}`),
  shuffleEstimate: (req) => jpost("/api/shuffle/estimate", req),
  shufflePlan: (req) => jpost("/api/shuffle/plan", req),
  shuffleGenerate: (req) => jpost("/api/shuffle/generate", req),

  // Dịch / đổi giọng
  dubUpload: (file, session) => upload("/api/dub/upload", file, { session }),
  dubVoiceSample: (file, session) =>
    upload("/api/dub/voice-sample", file, { session }),
  dubVoices: (lang) => jget(`/api/dub/voices?lang=${lang}`),
  dubStart: (req) => jpost("/api/dub/start", req),
  // 2 pha: dịch trước (xem/sửa) rồi mới tạo giọng
  dubPrepare: (req) => jpost("/api/dub/prepare", req),
  dubSynthesize: (req) => jpost("/api/dub/synthesize", req),
  dubTts: (req) => jpost("/api/dub/tts", req),
  // Nạp SRT tự tải (thay bản dịch tự động) — trả lines đã bỏ nhãn [tên]
  dubSrtImport: (file, session) => upload("/api/dub/subtitle/import", file, { session }),
  // Nhận dạng lời gốc trong video (ASR, không dịch) — cho tab "Chèn chữ vào video"
  dubTranscribe: (req) => jpost("/api/dub/subtitle/transcribe", req),
  // Hard subtitle
  subtitlePlan: (session, srt) => jpost("/api/dub/subtitle/plan", { session, srt }),
  subtitleBurn: (req) => jpost("/api/dub/subtitle", req),
  // Xem trước KHỚP 100% bản burn: server render đúng 1 khung tại `time`.
  subtitlePreviewFrame: (session, video, segments, mode, time) =>
    jpost("/api/dub/subtitle/preview-frame", { session, video, segments, mode, time }),

  // Drive
  driveStatus: (username) => jget(`/api/drive/status?username=${encodeURIComponent(username)}`),
  driveSetup: (username, folder_link) => jpost("/api/drive/setup", { username, folder_link }),
  driveRemove: (username) => request(`/api/drive/setup?username=${encodeURIComponent(username)}`, { method: "DELETE" }),

  // History
  getHistory: (username) => jget(`/api/history?username=${encodeURIComponent(username)}`),
};

// Poll một job tới khi xong. onTick(job) gọi mỗi lần cập nhật.
// `signal` (AbortSignal, tuỳ chọn): khi component unmount/huỷ, truyền signal đã
// abort để DỪNG vòng poll — tránh rò rỉ setTimeout và setState trên component đã
// gỡ. Khi bị abort, promise KHÔNG resolve (để code đang await không chạy tiếp).
// Chịu đựng máy chủ mất tiếng tối đa chừng này rồi mới báo hỏng.
const POLL_GRACE_MS = 90_000;

export function pollJob(id, onTick, interval = 1000, signal) {
  return new Promise((resolve, reject) => {
    let timer = null;
    const stop = () => { if (timer) clearTimeout(timer); };
    if (signal) signal.addEventListener("abort", stop, { once: true });
    // Job chạy TRÊN MÁY CHỦ, vòng này chỉ đứng ngoài hỏi thăm. Một cú 502 của
    // cổng vào Space hay mạng chớp giữa chừng KHÔNG có nghĩa là job hỏng — nên
    // hỏi lại, chỉ bỏ cuộc khi mất liên lạc quá lâu. (Trước đây lỗi đầu tiên là
    // văng luôn: người dùng mất việc đang chạy dở ở 60%.)
    let failedSince = 0;
    const tick = async () => {
      if (signal && signal.aborted) return;
      try {
        const job = await api.job(id);
        if (signal && signal.aborted) return;
        failedSince = 0;
        onTick && onTick(job);
        if (["done", "error", "cancelled"].includes(job.status)) {
          resolve(job);
        } else {
          timer = setTimeout(tick, interval);
        }
      } catch (e) {
        if (signal && signal.aborted) return;
        // Máy chủ trả lời rành mạch "không có job này" -> nó vừa khởi động lại,
        // tiến trình trong RAM mất rồi, hỏi thêm cũng vô ích.
        if (e?.status === 404) {
          return reject(new Error("Máy chủ vừa khởi động lại nên không còn tiến trình này. Hãy chạy lại."));
        }
        if (!e?.transient) return reject(e);
        const now = Date.now();
        if (!failedSince) failedSince = now;
        if (now - failedSince > POLL_GRACE_MS) {
          return reject(new Error(`${e.message} (mất liên lạc với máy chủ quá lâu)`));
        }
        // Máy chủ đang hụt hơi -> giãn nhịp hỏi ra, đừng dồn thêm.
        timer = setTimeout(tick, Math.min(5000, Math.max(interval, interval * 2)));
      }
    };
    tick();
  });
}