exit_script(){
    exit 1
}
input_error_exit(){
    if [ -z  $1 ];then 
        echo -e "\033[31m "Incorrect input!!!" \033[0m"
        exit_script
    fi
}

#從現(xiàn)有物理卷擴(kuò)展邏輯卷
extend_lv_from_pv(){
    PVS_NAME=`pvs | grep -v "PFree" | awk '{if($6>0){print $1;}}'`
    VG_NAME=`vgdisplay | awk '/VG Name/ {print $3}'`
    LVS_NAME=`lvdisplay | awk '/LV Name/ {print $3}'`
    if [ ${#PVS_NAME} == 0 ];then
        echo -e "\033[31m "Don\'t available physics volume!" \033[0m"
        exit_script
    fi
    echo -e "\033[34m Please select physics volume to operate: \033[0m"
    select OPERATE_DISK in $PVS_NAME
    do
    break
    done
    input_error_exit $OPERATE_DISK    

    echo -e "\033[34m Please select logical volume to extend: \033[0m"
    echo -e "\n"
    select LV_NAME in $LVS_NAME
    do
        break
    done
    input_error_exit $LV_NAME

    echo -e "\n"
    #判斷邏輯卷是根分區(qū)還是swap分區(qū)
    unset TEMP
    TEMP=`cat /etc/fstab  | grep $LV_NAME | awk '{print $2}'`
    if [ $TEMP = "/" ];then
        lvextend /dev/$VG_NAME/$LV_NAME $OPERATE_DISK
    unset TEST
        TEST=`cat /etc/fstab  | grep $LV_NAME | awk '{print $3}'`
        if [ "$TEST" = "ext4" ] || [ "$TEST" = "ext3" ];then
            resize2fs /dev/$VG_NAME/$LV_NAME
        elif [ "$TEST" = "xfs" ];then
            xfs_growfs /dev/$VG_NAME/$LV_NAME
        fi
    elif [ $TEMP = "swap" ] || [ $TEMP = "none" ];then
    view_vg_size
    read -p "Please input size to extend(G,g,M,m):" SIZE
        swapoff /dev/$VG_NAME/$LV_NAME
        lvextend -L +${SIZE} /dev/$VG_NAME/$LV_NAME
        mkswap /dev/$VG_NAME/$LV_NAME
        swapon /dev/$VG_NAME/$LV_NAME
    fi
}

#從未分區(qū)的硬盤(pán)擴(kuò)展邏輯卷
extend_lv_from_disk(){
    #搜索所有/dev/sd開(kāi)頭的硬盤(pán)
    DISKS=`fdisk -l | awk '/Disk/ {print $2}' | cut -d: -f1 | grep '\/dev\/sd*'`
    I=0
    #篩選未分區(qū)的硬盤(pán),并將未分區(qū)的硬盤(pán)存入數(shù)組中
    for J in $DISKS;do
        TEST1=`fdisk -l $J | grep ^$J.`
        FLAG1=$?
        if [ $FLAG1 != 0 ];then
        DISK[$I]=$J
        I=`expr $I + 1` 
        fi
    done
    #如果沒(méi)有可用硬盤(pán)瞒津，則退出腳本
    if [ ${#DISK[*]}  == 0 ];then
        echo -e "\033[31m "Don\'t available disk!" \033[0m"
        exit_script
    fi
    echo -e "\n"
    echo -e "\033[34m "The following are unused hard drives." \033[0m"
    #從未分區(qū)數(shù)組中選擇要操作的硬盤(pán)
    select OPERATION_DISK in ${DISK[*]} 
    do
        break
    done
    input_error_exit $OPERATION_DISK

    #創(chuàng)建分區(qū)
    fdisk $OPERATION_DISK<<EOF
n
p
1


t
8e
w
EOF
    PV_NAME=`fdisk -l $OPERATION_DISK | grep ^$OPERATION_DISK | awk '{print $1}'`
    VG_NAME=`vgdisplay | awk '/VG Name/ {print $3}'`
    LVS_NAME=`lvdisplay | awk '/LV Name/ {print $3}'`
    #創(chuàng)建物理卷
    pvcreate $PV_NAME
    #將新建的物理卷加入到已有卷組中
    vgextend $VG_NAME $PV_NAME
    echo -e "\n"
    echo -e "\033[34m Please select logical volume to extend: \033[0m"
    select LV_NAME in $LVS_NAME
    do
        break
    done

    input_error_exit $LV_NAME
    
    #判斷邏輯卷是根分區(qū)還是swap分區(qū)
    unset TEMP
    TEMP=`cat /etc/fstab  | grep $LV_NAME | awk '{print $2}'`
    if [ $TEMP = "/" ];then
    lvextend /dev/$VG_NAME/$LV_NAME $PV_NAME
    unset TEST
        TEST=`cat /etc/fstab  | grep $LV_NAME | awk '{print $3}'`
        if [ "$TEST" = "ext4" ] || [ "$TEST" = "ext3" ];then
            resize2fs /dev/$VG_NAME/$LV_NAME
        elif [ "$TEST" = "xfs" ];then
            xfs_growfs /dev/$VG_NAME/$LV_NAME
        fi
    elif [ $TEMP = "swap" ] || [ $TEMP = "none" ];then
    view_vg_size
        read -p "Please input size to extend(G,g,M,m):" SIZE
        echo -e "\n"
        swapoff /dev/$VG_NAME/$LV_NAME
        lvextend -L +${SIZE}G /dev/$VG_NAME/$LV_NAME
        mkswap /dev/$VG_NAME/$LV_NAME
        swapon /dev/$VG_NAME/$LV_NAME
    fi
}

#查看可用的卷組大小
view_vg_size(){
    VG_SIZE=`vgs |grep -v VG |awk '{print $7}'`
    echo -e "\n"
    echo -e "\033[36m "The size of the available logical group is:" \033[0m"
    echo -e "\033[33m $VG_SIZE \033[0m"
    echo -e "\n"
}
OPTION1='Extend Logical volume using a unpartitioned hard disk.'
OPTION2='Extend Logical volume using a free physics volume.'

echo -e "\n"
select OPTION in "$OPTION1" "$OPTION2"
do
    echo -e "\n"
    break
done
echo -e "\n"
input_error_exit $OPTION

case  "$OPTION" in
"$OPTION1")
    extend_lv_from_disk
    ;;
"$OPTION2")
    extend_lv_from_pv
    ;;
*)
    echo -e "\033[31m "Incorrect input!!!" \033[0m"
    exit_script
    ;;
esac