update user files info

src/app/ui/marking-page/protection-summary.tsx

@@ -3,15 +3,43 @@
 import { useTranslations } from 'next-intl';
 import { useQuery } from '@tanstack/react-query';
 import { getUserFilesData } from '@/app/actions/fileEntity';
+import { getUserFilesInfo } from '@/app/actions/action';
+import { convertBytes } from '@/app/lib/convertBytes';
 
 type FilesInfo = {
-	totalSize: string;
+	totalSize: number;
 	totalCount: number;
 };
 
 
-export default function ProtectionSummary() {
+export default function ProtectionSummary({ fileType }: { fileType: string }) {
 	const t = useTranslations('Global');
+	console.log(fileType);
+
+	const getFileTypeFields = (fileType: string): { totalSize: string, totalCount: string } => {
+		switch (fileType) {
+			case 'image':
+				return {
+					totalSize: 'images_size',
+					totalCount: 'images_quantity'
+				};
+			case 'video':
+				return {
+					totalSize: 'videos_size',
+					totalCount: 'videos_quantity'
+				};
+			case 'audio':
+				return {
+					totalSize: 'audios_size',
+					totalCount: 'audios_quantity'
+				};
+			default:
+				return {
+					totalSize: 'all_files_size',
+					totalCount: 'all_files_quantity'
+				};
+		}
+	};
 
 	const {
 		data: filesInfo,
@@ -19,24 +47,26 @@ export default function ProtectionSummary() {
 		isError,
 		error,
 	} = useQuery<{
-		formattedTotalSize: string,
-		totalCount: number
+		all_files_size: number,
+		all_files_quantity: number
 	}, Error, FilesInfo>({
-		queryKey: ['userFilesData'],
-		queryFn: getUserFilesData,
+		queryKey: ['userFilesInfo'],
+		queryFn: getUserFilesInfo,
 		select: (data): FilesInfo => {
 			if (!data) {
 				return {
-					totalSize: '0',
+					totalSize: 0,
 					totalCount: 0
 				}
 			}
 
+			const fields = getFileTypeFields(fileType);
+
 			return {
-				totalSize: data.formattedTotalSize,
-				totalCount: (data.totalCount)
+				totalSize: data[fields.totalSize as keyof typeof data],
+				totalCount: data[fields.totalCount as keyof typeof data]
 			}
-		},
+		}
 	});
 
 
@@ -56,7 +86,7 @@ export default function ProtectionSummary() {
 					<div className="protection-stat-label">{t('violations')}</div>
 				</div>
 				<div className="protection-stat total-usage">
-					<div className="protection-stat-value">{filesInfo?.totalSize ? filesInfo?.totalSize : 0} / 0</div>
+					<div className="protection-stat-value">{filesInfo?.totalSize ? convertBytes(filesInfo?.totalSize) : 0} / 0</div>
 					<div className="protection-stat-label">
 						{t('disk-space-used')}
 					</div>