65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
import logging
|
|
from pathlib import Path
|
|
from typing import Any, Dict
|
|
from tools.utils import format_relative_time, ToolError, get_file_info_list, sort_file_list
|
|
|
|
logger = logging.getLogger("MooseCP")
|
|
|
|
async def handle(args: Dict[str, Any]):
|
|
"""
|
|
Lists the contents of a directory in a file browser format.
|
|
Directories are listed first, then files.
|
|
Supports pagination and sorting.
|
|
"""
|
|
path_str = args.get("path")
|
|
page = int(args.get("page", 1))
|
|
page_size = int(args.get("page_size", 64))
|
|
sort_by = args.get("sort_by", "mtime")
|
|
|
|
if not path_str:
|
|
raise ToolError("Missing path argument")
|
|
|
|
path = Path(path_str)
|
|
if not path.is_dir():
|
|
raise ToolError(f"{path_str} is not a directory.")
|
|
|
|
all_items = get_file_info_list(path)
|
|
all_items = sort_file_list(all_items, sort_by)
|
|
|
|
total_items = len(all_items)
|
|
|
|
# Custom Pagination: show all if <= 100, otherwise paginate by page_size
|
|
if total_items <= 100:
|
|
start_idx = 0
|
|
end_idx = total_items
|
|
effective_page = 1
|
|
total_pages = 1
|
|
else:
|
|
start_idx = (page - 1) * page_size
|
|
end_idx = start_idx + page_size
|
|
total_pages = (total_items + page_size - 1) // page_size
|
|
effective_page = page
|
|
|
|
paged_items = all_items[start_idx:end_idx]
|
|
|
|
if not paged_items:
|
|
return {"text": f"No items found on page {page}."}
|
|
|
|
output = []
|
|
for item in paged_items:
|
|
mtime_rel = format_relative_time(item["mtime"])
|
|
if item["is_dir"]:
|
|
try:
|
|
count = len(list(item["path"].iterdir()))
|
|
info = f"{count} items"
|
|
except:
|
|
info = "0 items"
|
|
output.append(f"[DIR] {item['name']} | {info} | {mtime_rel}")
|
|
else:
|
|
size_kb = item["size"] / 1024
|
|
output.append(f" {item['name']} | {size_kb:.1f}KB | {mtime_rel}")
|
|
|
|
header = f"Listing of {path_str}\nPage {effective_page} of {total_pages} ({total_items} total items). Showing {start_idx+1}-{min(end_idx, total_items)}."
|
|
|
|
return {"text": f"{header}\n\n" + "\n".join(output)}
|