openrag/frontend/lib/upload-utils.ts

export interface DuplicateCheckResponse {
  exists: boolean;
  [key: string]: unknown;
}

export interface UploadFileResult {
  fileId: string;
  filePath: string;
  run: unknown;
  deletion: unknown;
  unified: boolean;
  raw: unknown;
}

export async function duplicateCheck(
  file: File,
): Promise<DuplicateCheckResponse> {
  const response = await fetch(
    `/api/documents/check-filename?filename=${encodeURIComponent(file.name)}`,
  );

  if (!response.ok) {
    const errorText = await response.text();
    throw new Error(
      errorText || `Failed to check duplicates: ${response.statusText}`,
    );
  }

  return response.json();
}

export async function uploadFileForContext(
  file: File,
): Promise<UploadFileResult> {
  window.dispatchEvent(
    new CustomEvent("fileUploadStart", {
      detail: { filename: file.name },
    }),
  );

  try {
    const formData = new FormData();
    formData.append("file", file);

    const uploadResponse = await fetch("/api/upload_context", {
      method: "POST",
      body: formData,
    });

    let payload: unknown;
    try {
      payload = await uploadResponse.json();
    } catch (error) {
      throw new Error("Upload failed: unable to parse server response");
    }

    const uploadJson =
      typeof payload === "object" && payload !== null ? payload : {};

    if (!uploadResponse.ok) {
      const errorMessage =
        (uploadJson as { error?: string }).error || "Upload failed";
      throw new Error(errorMessage);
    }

    const fileId =
      (uploadJson as { response_id?: string }).response_id || "uploaded";
    const filePath =
      (uploadJson as { filename?: string }).filename || file.name;
    const pages = (uploadJson as { pages?: number }).pages;
    const contentLength = (uploadJson as { content_length?: number })
      .content_length;
    const confirmation = (uploadJson as { confirmation?: string }).confirmation;

    const result: UploadFileResult = {
      fileId,
      filePath,
      run: null,
      deletion: null,
      unified: false,
      raw: uploadJson,
    };

    window.dispatchEvent(
      new CustomEvent("fileUploaded", {
        detail: {
          file,
          result: {
            file_id: fileId,
            file_path: filePath,
            filename: filePath,
            pages: pages,
            content_length: contentLength,
            confirmation: confirmation,
            response_id: fileId,
            run: null,
            deletion: null,
            unified: false,
          },
        },
      }),
    );

    return result;
  } catch (error) {
    window.dispatchEvent(
      new CustomEvent("fileUploadError", {
        detail: {
          filename: file.name,
          error: error instanceof Error ? error.message : "Upload failed",
        },
      }),
    );
    throw error;
  } finally {
    window.dispatchEvent(new CustomEvent("fileUploadComplete"));
  }
}

export async function uploadFile(
  file: File,
  replace = false,
): Promise<UploadFileResult> {
  try {
    const formData = new FormData();
    formData.append("file", file);
    formData.append("replace_duplicates", replace.toString());

    const uploadResponse = await fetch("/api/router/upload_ingest", {
      method: "POST",
      body: formData,
    });

    let payload: unknown;
    try {
      payload = await uploadResponse.json();
    } catch (error) {
      throw new Error("Upload failed: unable to parse server response");
    }

    const uploadIngestJson =
      typeof payload === "object" && payload !== null ? payload : {};

    if (!uploadResponse.ok) {
      const errorMessage =
        (uploadIngestJson as { error?: string }).error ||
        "Upload and ingest failed";
      throw new Error(errorMessage);
    }

    const fileId =
      (uploadIngestJson as { upload?: { id?: string } }).upload?.id ||
      (uploadIngestJson as { id?: string }).id ||
      (uploadIngestJson as { task_id?: string }).task_id;
    const filePath =
      (uploadIngestJson as { upload?: { path?: string } }).upload?.path ||
      (uploadIngestJson as { path?: string }).path ||
      "uploaded";
    const runJson = (uploadIngestJson as { ingestion?: unknown }).ingestion;
    const deletionJson = (uploadIngestJson as { deletion?: unknown }).deletion;

    if (!fileId) {
      throw new Error("Upload successful but no file id returned");
    }

    if (
      runJson &&
      typeof runJson === "object" &&
      "status" in (runJson as Record<string, unknown>) &&
      (runJson as { status?: string }).status !== "COMPLETED" &&
      (runJson as { status?: string }).status !== "SUCCESS"
    ) {
      const errorMsg =
        (runJson as { error?: string }).error || "Ingestion pipeline failed";
      throw new Error(
        `Ingestion failed: ${errorMsg}. Try setting DISABLE_INGEST_WITH_LANGFLOW=true if you're experiencing Langflow component issues.`,
      );
    }

    const result: UploadFileResult = {
      fileId,
      filePath,
      run: runJson,
      deletion: deletionJson,
      unified: true,
      raw: uploadIngestJson,
    };

    return result;
  } catch (error) {
    window.dispatchEvent(
      new CustomEvent("fileUploadError", {
        detail: {
          filename: file.name,
          error: error instanceof Error ? error.message : "Upload failed",
        },
      }),
    );
    throw error;
  } finally {
    window.dispatchEvent(new CustomEvent("fileUploadComplete"));
  }
}