110 lines
3.0 KiB
JavaScript
110 lines
3.0 KiB
JavaScript
import { useState } from "react";
|
|
import { useDispatch, useSelector } from "react-redux";
|
|
import { setFile, setResult, setValidatedData } from "../../../store/slices/uploadSlice";
|
|
import { uploadFile, uploadPdf, saveToDatabase } from "./service_admin_upload";
|
|
import { useNavigate } from "react-router-dom";
|
|
import * as pdfjsLib from "pdfjs-dist";
|
|
pdfjsLib.GlobalWorkerOptions.workerSrc = new URL(
|
|
"pdfjs-dist/build/pdf.worker.mjs",
|
|
import.meta.url
|
|
).toString();
|
|
|
|
// pdfjsLib.GlobalWorkerOptions.workerSrc = pdfjsWorker;
|
|
|
|
export function useUploadController() {
|
|
const dispatch = useDispatch();
|
|
const navigate = useNavigate();
|
|
const { file, result } = useSelector((state) => state.upload);
|
|
|
|
const [loading, setLoading] = useState(false);
|
|
const [selectedTable, setSelectedTable] = useState(null);
|
|
const [selectedPages, setSelectedPages] = useState("");
|
|
const [tableTitle, setTableTitle] = useState("");
|
|
const [pdfPageCount, setPdfPageCount] = useState(null); // 👈 halaman pdf
|
|
|
|
// 🔹 handle drop file
|
|
const handleFileSelect = async (f) => {
|
|
dispatch(setFile(f));
|
|
const ext = f.name.split(".").pop().toLowerCase();
|
|
|
|
if (ext === "pdf") {
|
|
try {
|
|
const reader = new FileReader();
|
|
reader.onload = async (e) => {
|
|
const typedArray = new Uint8Array(e.target.result);
|
|
const pdf = await pdfjsLib.getDocument({ data: typedArray }).promise;
|
|
setPdfPageCount(pdf.numPages);
|
|
console.log(`📄 PDF terdeteksi dengan ${pdf.numPages} halaman`);
|
|
};
|
|
reader.readAsArrayBuffer(f);
|
|
} catch (err) {
|
|
console.error("Gagal membaca PDF:", err);
|
|
}
|
|
} else {
|
|
setPdfPageCount(null);
|
|
}
|
|
};
|
|
|
|
// 🔹 upload file
|
|
const handleUpload = async () => {
|
|
if (!file) return;
|
|
setLoading(true);
|
|
try {
|
|
const res = await uploadFile(file, selectedPages);
|
|
dispatch(setResult(res));
|
|
console.log('gtw',res);
|
|
|
|
if (res.file_type !== ".pdf" || (res.file_type === ".pdf" && !res.tables)) {
|
|
navigate("/admin/upload/validate");
|
|
}
|
|
} finally {
|
|
setLoading(false);
|
|
}
|
|
};
|
|
|
|
const handleNextPdf = async () => {
|
|
if (!selectedTable) return;
|
|
setLoading(true);
|
|
try {
|
|
const res = await uploadPdf(selectedTable);
|
|
dispatch(setResult(res));
|
|
navigate("/admin/upload/validate");
|
|
} finally {
|
|
setLoading(false);
|
|
}
|
|
};
|
|
|
|
const handleConfirmUpload = async () => {
|
|
setLoading(true);
|
|
try {
|
|
const data = {
|
|
title: tableTitle,
|
|
columns: result.columns,
|
|
rows: result.preview,
|
|
};
|
|
const res = await saveToDatabase(data);
|
|
dispatch(setValidatedData(res));
|
|
navigate("/admin/upload/success");
|
|
} finally {
|
|
setLoading(false);
|
|
}
|
|
};
|
|
|
|
return {
|
|
loading,
|
|
file,
|
|
result,
|
|
tableTitle,
|
|
selectedTable,
|
|
selectedPages,
|
|
pdfPageCount, // 👈 halaman pdf
|
|
setSelectedTable,
|
|
setSelectedPages,
|
|
setTableTitle,
|
|
handleFileSelect,
|
|
handleUpload,
|
|
handleNextPdf,
|
|
handleConfirmUpload,
|
|
};
|
|
}
|