shell 递归遍历文件夹 对文件行首和末尾进行修改

功能：

    递归遍历文件夹下的cs文件或者lua文件，并对文件的行首或者末尾插入代码

#!/bin/bash
str=""
luafile=false
csfile=false
dfs(){
	#echo $1
	for file in $1/* 
	do
		if [ -d $file ]
		then
			dfs $file
		elif [ -f $file ]
		then
			#文件 
			str=`echo "${file##*.}"`
			if [ $str == "lua" ]&&[ $luafile == true ]; then
				echo $file
				#在lua文件末尾加入一行注释
				#echo '-- insert some code ' >> $file
				#在lua文件行首加入一行注释
				sed -i '1i\-- insert some code ' $file
			elif [ $str == "cs" ]&&[ $csfile == true ]; then
				echo $file
				#在js文件末尾加入一行注释
				#echo '// insert some code ' >> $file
				sed -i '1i\// insert some code ' $file
			fi
		fi
	done
}

run(){
	dir=../Unity/Assets
	dfs $dir
}

if [ $# == 0 ]; then
	echo "传参数 指定操作文件"
elif [ $# == 1 ]; then
	if [ $1 == 'all' ]; then
		luafile=true
		csfile=true
	elif [ $1 == 'lua' ]; then
		luafile=true
	elif [ $1 == 'cs' ]; then
		csfile=true
	fi
	run
fi