#!/usr/bin/env bash

#

# 检查脚本参数，如必要参数未传入，退出脚本。

#

checkInput() {

if [ "${repository}" == "" ] || [ "${branch}" == "" ]; then

echo "请输入脚本参数：repository branch [source] [target]"

echo " repository: git 仓储（必填）。"

echo " branch: git 分支（必填）。如 master/develop"

echo " source: 代码存放目录。默认为/home/zp/source。"

echo " target: 代码存放目录。默认为脚本所在目录。"

exit 0

fi

}

#

# 判断 git 版本库是否存在。根据实际结果修改 ${gitok} 值。

#

gitok=false

isGitExist() {

cd ${SOURCE_PATH}

if [ -d "${SOURCE_PATH}/${repository}/${target}" ]; then

cd ${SOURCE_PATH}/${repository}/${target}

#(1)删除git状态零时文件

if [ -f "gitstatus.tmp" ]; then

rm -rf gitstatus.tmp

fi

#(2) 判断是否存在.git目录

if [ -d "./.git" ]; then

#(3) 判断git是否可用

git status &> gitstatus.tmp

grep -iwq 'not a git repository' gitstatus.tmp && gitok=false || gitok=true

fi

#返回到主目录

cd ${SOURCE_PATH}

fi

}

#

# 如果 git 版本库存在（根据 ${gitok} 值），执行 fetch 操作；反之，执行 clone 操作。

#

doFetchOrClone() {

if [ ! -d "${SOURCE_PATH}" ]; then

mkdir -p ${SOURCE_PATH}

fi

if ${gitok}; then

cd ${SOURCE_PATH}/${repository}/${target}

git reset --hard

git clean -ffdx

git fetch

echo "git fetch ${repository} remote repository 到本地成功"

else

#删除所有内容,便于重新进行git clone

rm -rf ${repository}

git clone --no-checkout [email protected]:${ACCOUNT}/${repository}.git ${SOURCE_PATH}/${repository}/${target}

echo "git clone ${repository} remote repository 到本地成功"

cd ${SOURCE_PATH}/${repository}/${target}

fi

}

#

# 切换到 ${branch} 分支

#

doCheckout() {

echo "检出 ${repository} ${branch} 分支代码"

isRemoteBranch=false

gitRemoteBranch=`git branch -r`

echo -e "$gitRemoteBranch" | grep -iwq ${branch} && isRemoteBranch=true || isRemoteBranch=false

if ${isRemoteBranch}; then

echo "找到 ${branch} 对应的远端分支"

git checkout -f 'origin/'${branch}

else

git checkout -f ${branch}

fi

echo "更新子模块代码"

git submodule update --init --recursive --force

}

##############################__MAIN__########################################

export LANG="zh_CN.UTF-8"

# Github 账户

ACCOUNT=dunwu

# 源码默认根目录

SOURCE_PATH=/home/zp/source

# 必填输入参数

repository=`echo $1`

branch=`echo $2`

# 可选输入参数

source=`echo $3`

target=`echo $4`

if [ "${source}" != "" ]; then

SOURCE_PATH=${source}

fi

# 0. 检查传入的参数

checkInput

# 1. 判断本地是否已存在 Git 仓库

isGitExist

# 2. 如果本地已有代码，执行 fetch；反之，从远程 clone

doFetchOrClone

# 3. 切换到指定分支

doCheckout

echo "代码检出完成！"