local provided_path="${1:-$(pwd)}"

# Find git root directory - this ensures we only operate within the project

local git_root

if ! git_root=$(cd "$provided_path" && git rev-parse --show-toplevel 2>/dev/null); then

echo -e "${YELLOW}Warning: Not in a git repository, skipping cleanup${NC}" >&2

return 0

fi

# Normalize paths for comparison

git_root=$(cd "$git_root" && pwd)

local normalized_path=$(cd "$provided_path" && pwd)

# Safety check: ensure the provided path is within git root

if [[ "$normalized_path" != "$git_root"* ]]; then

echo -e "${RED}Error: Path '$normalized_path' is outside git root '$git_root', aborting cleanup${NC}" >&2

return 1

fi

# Use git root as the base for cleanup operations

local project_root="$git_root"

echo -e "${YELLOW}Cleaning up files created by Docker container in: $project_root${NC}"

# Find and remove root-owned files/directories that shouldn't persist

# Use sudo if available, otherwise try without (may fail silently)

if command -v sudo &> /dev/null; then

# Remove Python cache files (only within git root)

sudo find "$project_root" -type d -name "__pycache__" -user root -exec rm -rf {} + 2>/dev/null || true

sudo find "$project_root" -type f \( -name "*.pyc" -o -name "*.pyo" \) -user root -delete 2>/dev/null || true

# Remove temporary test scripts (only within git root)

sudo find "$project_root" -type f -name ".tmp_run_test_*.sh" -user root -delete 2>/dev/null || true

# Remove test artifacts (only within git root)

sudo rm -rf "$project_root/test-logs" "$project_root/results" 2>/dev/null || true

sudo rm -f "$project_root/test-results.json" "$project_root/coverage.xml" 2>/dev/null || true

# Remove Vim swap files (only within git root)

sudo find "$project_root" -type f \( -name "*.swp" -o -name "*.swo" -o -name ".*.swp" -o -name ".*.swo" \) -user root -delete 2>/dev/null || true

else

# Without sudo, try to remove files we can access (only within git root)

find "$project_root" -type d -name "__pycache__" -exec rm -rf {} + 2>/dev/null || true

find "$project_root" -type f \( -name "*.pyc" -o -name "*.pyo" -o -name ".tmp_run_test_*.sh" -o -name "*.swp" -o -name "*.swo" \) -delete 2>/dev/null || true

rm -rf "$project_root/test-logs" "$project_root/results" 2>/dev/null || true

rm -f "$project_root/test-results.json" "$project_root/coverage.xml" 2>/dev/null || true

fi