Bash玩转脚本1之自己的脚本安装程序

版权声明:本文为博主Atany原创文章,未经博主允许不得转载。博客地址:http://blog.csdn.net/yang8456211 https://blog.csdn.net/yang8456211/article/details/51020797

Bash之打造自己的脚本安装器

前言


还是理所当然的前言,我一直想找一套管理脚本的“框架”,能让自己杂乱的脚本有点规整,无奈眼界尚浅,未能找到。因此萌生自己写一点优化脚本的工具来,新手可学习,高手请指正。今天先写一个脚本的安装器,目的在于写完并新脚本之后能够在shell的任何位置都能够便捷使用。

安装器干了啥?

一、配置文件

config.ini主要用于配置两个目录。
  • 脚本的读取目录
  • 生成软链接的存放目录

二、读取脚本

    递归遍历读取scriptPath目录下的脚本文件,排除掉install.sh和config.ini。
do_file()
{
    for file in $1/*
    do 
        if [[ -d "$file" ]]; then
            do_file "$file"
        else
            basename=`basename $file`
            if [[ ! $basename == "install.sh" && ! $basename == "config.ini" ]];then
                link_file $file
            fi
        fi
    done
}

三、创建软链接

    为每一个脚本在binPath目录下创建软链接,如果之前存在则会首先删除掉,并对软链接加上执行权限(这里直接加了777)
link_file()
{
    filePath=$1
    fileName=`basename $1`
    linkName=${fileName%.*}
    linkPath=$binPath"/"$linkName
    if [[ -L $linkPath ]];then
        echo "===>(warn):"$linkPath" is exist,remove it!"
        rm $linkPath
    fi
    ln -s $filePath $linkPath
    echo "===>(info):link file "$filePath" -----> "$linkName" successful!"
    chmod 777 $linkPath
}

四、配置环境变量

    把binPath目录添加到环境变量中(~/.bash_profile),这样就可以随时的访问脚本了~
add_profile()
{
    isIn=`cat ~/.bash_profile | grep $1`
    echo_test "isIn is "$isIn
    if [[ x"$isIn" == x    ]];then
        echo "\n#Setting PATH FOR LOCAL SCRIPT" >> ~/.bash_profile
        echo "export PATH=\"$1:\${PATH}\"" >> ~/.bash_profile
        echo "===>(info)"$binPath" is added to bash_profile successful!"
        export PATH=$1:${PATH} 
    else
        echo "===>(info)"$binPath" is already in the bash_profile!<SKIP>"
    fi
}

尝试

每次新加的脚本便可以放在scriptPath目录,执行install.sh之后便会在binPath里面生成对应的软链接,然后就可以在终端中自由的使用了~

1.可以看到,我的目录下面有五个文件(包括安装脚本的配置文件)

2.执行sh install.sh run 之后

3.在binPath目录下生成了三个软链接~

4.并在~/.bash_profile里生成了对应的Path

5.可以看到我们在Shell的任何位置已经可以是用自己编写的脚本指令了~(例如pyversion,是自己写的一个修改本地python版本的小脚本)

6.完整代码:

#!/bin/bash

# 读取config.ini
source ./config.ini
isTest=$isTest
binPath=$binPath
scriptPath=$scriptPath

editor(){
    echo '''
        @auther: 杨光
        @blog: http://blog.csdn.net/yang8456211
        @email: [email protected]
    '''
}

help_fun(){
cat << ENTER
     ============= 脚本安装工具 =============
     Version: 0.1
     Date: 20160330
     Usage: 用作初始安装自己的脚本环境
     e.g.: sh install.sh run
     ============= 脚本安装工具 =============
ENTER
}

echo_emp(){
    echo -e "\033[31m"$1"\033[0m" 
}

echo_test(){
    [[ $isTest == true ]] && echo $1
}

exit_pro(){
    echo "==用户退出== Abort(1)"
    exit 1
}

link_file()
{
    filePath=$1
    fileName=`basename $1`
    linkName=${fileName%.*}
    linkPath=$binPath"/"$linkName
    if [[ -L $linkPath ]];then
        echo "===>(warn):"$linkPath" is exist,remove it!"
        rm $linkPath
    fi
    ln -s $filePath $linkPath
    echo "===>(info):link file "$filePath" -----> "$linkName" successful!"
    chmod 777 $linkPath
}

do_file()
{
    for file in $1/*
    do 
        if [[ -d "$file" ]]; then
            do_file "$file"
        else
            basename=`basename $file`
            if [[ ! $basename == "install.sh" && ! $basename == "config.ini" ]];then
                link_file $file
            fi
        fi
    done
}

add_profile()
{
    isIn=`cat ~/.bash_profile | grep $1`
    echo_test "isIn is "$isIn
    if [[ x"$isIn" == x    ]];then
        echo "\n#Setting PATH FOR LOCAL SCRIPT" >> ~/.bash_profile
        echo "export PATH=\"$1:\${PATH}\"" >> ~/.bash_profile
        echo "===>(info)"$binPath" is added to bash_profile successful!"
        export PATH=$1:${PATH} #只是加到了内存中,新开终端失效
    else
        echo "===>(info)"$binPath" is already in the bash_profile!<SKIP>"
    fi
}


if [[ $# != 1 || $1 != "run" ]];then
    help_fun
    editor
    exit 2
fi

echo "是否对"$scriptPath"目录下的脚本进行安装?"
echo "安装目录为:"$binPath"(y/n)"
read
if [[ $REPLY == "y" || $REPLY == "Y" ]];then    
    do_file $scriptPath
    add_profile $binPath
    echo "脚本环境安装成功!!"
else
    echo "用户终止exit (Abort)"
    exit 0
fi

杨光(atany)原创,转载请注明博主与博文链接,未经博主允许,禁止任何商业用途。
博客地址:http://blog.csdn.net/yang8456211
博文地址:http://blog.csdn.net/yang8456211/article/details/51020797
本文遵循“署名-非商业用途-保持一致”创作公用协议

猜你喜欢

转载自blog.csdn.net/yang8456211/article/details/51020797