#!/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 git@github.com:${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 "代码检出完成!"