如何使用Shell写一个显示目录结构的命令?

#!/usr/bin/env bash
# 本命令用于显示指定路径或者当前路径的文件结构，支持搜索
# tf 显示当前目录的文件结构
# tf 接关键词  搜索当前目录
# tf 目录 关键词 搜索指定目录

# 本命令依赖于 tree 命令
DIR=
KEYWORD=

# 不传路径 默认为搜索
if [ $# -eq 1 ]; then
  KEYWORD=$1
fi

# 指定路径搜索
if [ $# -eq 2 ]; then
  DIR=$1
  KEYWORD=$2
fi

if [ -n "$DIR" ]; then
   cd "$DIR" || exit 1
fi

pwd

tree -C -f | grep "$KEYWORD"

tf

tf 关键词

// 比如
tf lib

tf /usr/local/ lib