signoz/frontend/src/container/LogDetailedView/TableView/useAsyncJSONProcessing.ts

import { useEffect, useRef, useState } from 'react';
import { FeatureKeys } from 'constants/features';
import { ChangeViewFunctionType } from 'container/ExplorerOptions/types';
import { isEmpty } from 'lodash-es';
import { useAppContext } from 'providers/App/App';

import { jsonToDataNodes, recursiveParseJSON } from '../utils';

const MAX_BODY_BYTES = 100 * 1024; // 100 KB

// Hook for async JSON processing
const useAsyncJSONProcessing = (
	value: string | Record<string, unknown>,
	shouldProcess: boolean,
	handleChangeSelectedView?: ChangeViewFunctionType,
): {
	isLoading: boolean;
	treeData: any[] | null;
	error: string | null;
} => {
	const [jsonState, setJsonState] = useState<{
		isLoading: boolean;
		treeData: any[] | null;
		error: string | null;
	}>({
		isLoading: false,
		treeData: null,
		error: null,
	});

	const processingRef = useRef<boolean>(false);
	const { featureFlags } = useAppContext();
	const isBodyJsonQueryEnabled =
		featureFlags?.find((flag) => flag.name === FeatureKeys.USE_JSON_BODY)
			?.active || false;

	// eslint-disable-next-line sonarjs/cognitive-complexity
	useEffect((): (() => void) => {
		if (!shouldProcess || processingRef.current) {
			return (): void => {};
		}

		// When value is already a parsed object skip the size check and JSON parsing
		const parseBody = (): Record<string, unknown> | null => {
			if (typeof value === 'object' && value !== null) {
				return value as Record<string, unknown>;
			}
			const byteSize = new Blob([value as string]).size;
			if (byteSize > MAX_BODY_BYTES) {
				return null;
			}
			return recursiveParseJSON(value as string);
		};

		processingRef.current = true;
		setJsonState({ isLoading: true, treeData: null, error: null });

		// Option 1: Using setTimeout for non-blocking processing
		const processAsync = (): void => {
			setTimeout(() => {
				try {
					const parsedBody = parseBody();
					if (parsedBody && !isEmpty(parsedBody)) {
						const treeData = jsonToDataNodes(parsedBody, {
							isBodyJsonQueryEnabled,
							handleChangeSelectedView,
						});
						setJsonState({ isLoading: false, treeData, error: null });
					} else {
						setJsonState({ isLoading: false, treeData: null, error: null });
					}
				} catch (error) {
					setJsonState({
						isLoading: false,
						treeData: null,
						error: error instanceof Error ? error.message : 'Parsing failed',
					});
				} finally {
					processingRef.current = false;
				}
			}, 0);
		};

		// Option 2: Using requestIdleCallback for better performance
		const processWithIdleCallback = (): void => {
			if ('requestIdleCallback' in window) {
				requestIdleCallback(
					// eslint-disable-next-line sonarjs/no-identical-functions
					(): void => {
						try {
							const parsedBody = parseBody();
							if (parsedBody && !isEmpty(parsedBody)) {
								const treeData = jsonToDataNodes(parsedBody, {
									isBodyJsonQueryEnabled,
									handleChangeSelectedView,
								});
								setJsonState({ isLoading: false, treeData, error: null });
							} else {
								setJsonState({ isLoading: false, treeData: null, error: null });
							}
						} catch (error) {
							setJsonState({
								isLoading: false,
								treeData: null,
								error: error instanceof Error ? error.message : 'Parsing failed',
							});
						} finally {
							processingRef.current = false;
						}
					},
					{ timeout: 1000 },
				);
			} else {
				processAsync();
			}
		};

		processWithIdleCallback();

		// Cleanup function
		return (): void => {
			processingRef.current = false;
		};
	}, [value, shouldProcess, isBodyJsonQueryEnabled, handleChangeSelectedView]);

	return jsonState;
};

export default useAsyncJSONProcessing;