DEV Community

Query Filter
Query Filter

Posted on

print tree3

!/usr/bin/env bash

Unicode tree for Linux, directories first, files second, human-readable sizes

print_tree() {
local dir="$1"
local prefix="${2:-}"
local maxdepth="${3:-0}"
local depth="${4:-1}"

[[ ! -d "$dir" ]] && return
[[ $maxdepth -ne 0 && $depth -gt $maxdepth ]] && return

echo "${prefix}${dir##*/}/"

# Directories first
local d
local dirs=()
shopt -s nullglob
for d in "$dir"/*; do
    [[ -d "$d" ]] && dirs+=("$d")
done
IFS=$'\n' dirs=($(printf "%s\n" "${dirs[@]}" | sort))
for i in "${!dirs[@]}"; do
    d="${dirs[i]}"
    local branch="├──"
    [[ $i -eq $(( ${#dirs[@]} - 1 )) ]] && branch="└──"
    local new_prefix="$prefix"
    [[ $i -eq $(( ${#dirs[@]} - 1 )) ]] && new_prefix+="    " || new_prefix+="│   "
    echo "${prefix}${branch} ${d##*/}/"
    print_tree "$d" "$new_prefix" "$maxdepth" $((depth+1))
done

# Files
local f
local files=()
for f in "$dir"/*; do
    [[ -f "$f" ]] && files+=("$f")
done
IFS=$'\n' files=($(printf "%s\n" "${files[@]}" | sort))
local file_count=0
for i in "${!files[@]}"; do
    f="${files[i]}"
    local branch="├──"
    [[ $i -eq $(( ${#files[@]} - 1 )) ]] && branch="└──"
    local size=$(stat -c "%s" "$f" 2>/dev/null || echo 0)
    local hr_size
    if [[ "$size" -lt 1024 ]]; then hr_size="${size}B"
    elif [[ "$size" -lt 1048576 ]]; then hr_size="$((size/1024))K"
    elif [[ "$size" -lt 1073741824 ]]; then hr_size="$((size/1024/1024))M"
    else hr_size="$((size/1024/1024/1024))G"; fi
    echo "${prefix}${branch} ${f##*/} [${hr_size}]"
    ((file_count++))
done

# Total files per directory
[[ $file_count -gt 0 ]] && echo "${prefix}└── Total files: $file_count"

shopt -u nullglob
Enter fullscreen mode Exit fullscreen mode

}

Usage: print_tree

print_tree . "" 0

Top comments (0)