diff --git a/agent/files.php b/agent/files.php
index 626beed3..6205ca17 100644
--- a/agent/files.php
+++ b/agent/files.php
@@ -73,7 +73,7 @@ function is_ancestor_folder($folder_id, $current_folder_id, $client_id) {
}
function display_folders($parent_folder_id, $client_id, $indent = 0) {
- global $mysqli, $get_folder_id, $session_user_role;
+ global $mysqli, $get_folder_id, $session_user_role, $archive_query, $archived;
$sql_folders = mysqli_query($mysqli, "SELECT * FROM folders WHERE parent_folder = $parent_folder_id AND folder_client_id = $client_id ORDER BY folder_name ASC");
while ($row = mysqli_fetch_array($sql_folders)) {
@@ -81,11 +81,11 @@ function display_folders($parent_folder_id, $client_id, $indent = 0) {
$folder_name = nullable_htmlentities($row['folder_name']);
// Count files in folder
- $row_files = mysqli_fetch_assoc(mysqli_query($mysqli, "SELECT COUNT('file_id') AS num FROM files WHERE file_folder_id = $folder_id AND file_client_id = $client_id AND file_archived_at IS NULL"));
+ $row_files = mysqli_fetch_assoc(mysqli_query($mysqli, "SELECT COUNT('file_id') AS num FROM files WHERE file_folder_id = $folder_id AND file_client_id = $client_id AND file_$archive_query"));
$num_files = intval($row_files['num']);
// Count documents in folder
- $row_docs = mysqli_fetch_assoc(mysqli_query($mysqli, "SELECT COUNT('document_id') AS num FROM documents WHERE document_folder_id = $folder_id AND document_client_id = $client_id AND document_archived_at IS NULL"));
+ $row_docs = mysqli_fetch_assoc(mysqli_query($mysqli, "SELECT COUNT('document_id') AS num FROM documents WHERE document_folder_id = $folder_id AND document_client_id = $client_id AND document_$archive_query"));
$num_docs = intval($row_docs['num']);
$num_total = $num_files + $num_docs;
@@ -99,7 +99,7 @@ function display_folders($parent_folder_id, $client_id, $indent = 0) {
echo '