apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-04-27 01:36:30 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: update document processing status handling and improve sidebar components

Browse source
This commit is contained in:
DESKTOP-RTLN3BA\$punk 2026-03-31 21:29:46 -07:00
parent d8f403efba
commit 0201fd319d
7 changed files with 141 additions and 49 deletions
Download patch file Download diff file Expand all files Collapse all files

65
surfsense_web/hooks/use-documents-processing.ts
View file

@ -4,18 +4,31 @@ import { useQuery } from "@rocicorp/zero/react";
import { useEffect, useRef, useState } from "react";
import { queries } from "@/zero/queries";
export type DocumentsProcessingStatus = "idle" | "processing" | "success" | "error";
export type DocumentsProcessingStatus =
| "idle"
| "processing"
| "background_sync"
| "success"
| "error";
const SUCCESS_LINGER_MS = 5000;
interface UseDocumentsProcessingOptions {
hasPeriodicSyncEnabled?: boolean;
}
/**
* Returns the processing status of documents in the search space:
* - "processing" — at least one doc is pending/processing (show spinner)
* - "processing" — docs are queued or actively being prepared for search
* - "background_sync" — existing docs are being refreshed in the background
* - "error" — nothing processing, but failed docs exist (show red icon)
* - "success" — just transitioned from processing → all clear (green check, auto-dismisses)
* - "idle" — nothing noteworthy (show normal icon)
*/
export function useDocumentsProcessing(searchSpaceId: number | null): DocumentsProcessingStatus {
export function useDocumentsProcessing(
searchSpaceId: number | null,
{ hasPeriodicSyncEnabled = false }: UseDocumentsProcessingOptions = {}
): DocumentsProcessingStatus {
const [status, setStatus] = useState<DocumentsProcessingStatus>("idle");
const wasProcessingRef = useRef(false);
const successTimerRef = useRef<ReturnType<typeof setTimeout> | null>(null);
@ -25,38 +38,56 @@ export function useDocumentsProcessing(searchSpaceId: number | null): DocumentsP
useEffect(() => {
if (!searchSpaceId || !documents) return;
const clearSuccessTimer = () => {
if (successTimerRef.current) {
clearTimeout(successTimerRef.current);
successTimerRef.current = null;
}
};
let pendingCount = 0;
let processingCount = 0;
let failedCount = 0;
let readyCount = 0;
for (const doc of documents) {
// Keep the nav indicator aligned with what the Documents sidebar actually renders.
// Some connectors can create temporary untitled placeholder rows that remain hidden
// from the sidebar, and those should not keep the whole section looking "stuck".
if (!doc.title || doc.title.trim() === "") {
continue;
}
const state = (doc.status as { state?: string } | null)?.state;
if (state === "pending" || state === "processing") {
if (state === "pending") {
pendingCount++;
} else if (state === "processing") {
processingCount++;
} else if (state === "failed") {
failedCount++;
} else {
readyCount++;
}
}
if (processingCount > 0) {
if (pendingCount > 0) {
wasProcessingRef.current = true;
if (successTimerRef.current) {
clearTimeout(successTimerRef.current);
successTimerRef.current = null;
}
clearSuccessTimer();
setStatus("processing");
} else if (processingCount > 0) {
wasProcessingRef.current = true;
clearSuccessTimer();
const isBackgroundSync = hasPeriodicSyncEnabled && readyCount > 0;
setStatus(isBackgroundSync ? "background_sync" : "processing");
} else if (failedCount > 0) {
wasProcessingRef.current = false;
if (successTimerRef.current) {
clearTimeout(successTimerRef.current);
successTimerRef.current = null;
}
clearSuccessTimer();
setStatus("error");
} else if (wasProcessingRef.current) {
wasProcessingRef.current = false;
setStatus("success");
if (successTimerRef.current) {
clearTimeout(successTimerRef.current);
}
clearSuccessTimer();
successTimerRef.current = setTimeout(() => {
setStatus("idle");
successTimerRef.current = null;
@ -64,7 +95,7 @@ export function useDocumentsProcessing(searchSpaceId: number | null): DocumentsP
} else {
setStatus("idle");
}
}, [searchSpaceId, documents]);
}, [searchSpaceId, documents, hasPeriodicSyncEnabled]);
useEffect(() => {
return () => {