zkt安装脚本

#!/bin/bash

function print_tips
{
    echo '========================================================='
    echo "(1) 安装docker和docker-compose"
    echo "(2) 安装单节点zktube"
    echo "(3) 复制单节点zktube，多开节点"
    echo "(4) 重启zktube节点（官方升级后，重启即可升级为最新版）"
    echo "(5) 停止并删除zktube节点"
    echo "(6) 停止并删除所有docer container network image"
    echo "(Q/q) 退出"
    echo '========================================================='
}

#安装docker
function docker_install
{
    wget http://download.swarmeth.org/docker/docker_install.sh && chmod a+x docker_install.sh && ./docker_install.sh
}

#新建zkt存储文件夹
function zkt_init
{
    rm -rf /zktdata
    cd / && mkdir zktdata && mkdir zktdata/zkt1
    cd /zktdata/zkt1
    echo '========================================================='
    echo "请输入您的ETH钱包地址"
    read eth
    echo ${eth} >>/zktdata/zkt1/.revenue_address
    echo "您的钱包地址是：${eth}"
    echo '========================================================='
}


function zkt_start
{
    wget https://file.zktube.io/docker/prover/docker-compose.yml
    sed -i 's#~/.revenue_address#/zktdata/zkt1/.revenue_address#g' /zktdata/zkt1/docker-compose.yml
    #启动
    docker-compose up -d
    #查看日志
    echo '========================================================='
    echo '查看进程：docker ps'
    echo '存放目录：/zktdata/zkt1'
    echo '查看日志：docker logs -f zkt1_zktube-prover_1'
    echo '============================================== Aven7 ====' 
}

function zkt_loop
{
    echo "请输入要搭建的zkt节点数量"
    read tCnt
    zCnt=`ls -l /zktdata|grep "zkt"|wc -l`
    tCnt=$(($tCnt + $zCnt))
    zCnt=$(($zCnt + 1))
    for ((i=zCnt; i<=tCnt; i ++))
    do
    if [ ! -f /zktdata/zkt${i} ]; then
        cp -r /zktdata/zkt1 /zktdata/zkt${i}
        cd /zktdata/zkt${i}
        docker-compose up -d
        echo '========================================================='
        echo '查看进程：docker ps'
        echo "存放目录：/zktdata/zkt${i} "
        echo "查看日志：docker logs -f zkt${i}_zktube-prover_1"
        echo '============================================== Aven7 ====' 
    else 
        echo '========================================================='
        echo "zkt${i} 已经存在, 跳过"
        echo '========================================================='
    fi
    done
}

function zkt_rm
{
    tCnt=`ls -l /zktdata|grep "zkt"|wc -l`
    for ((i=1; i<=tCnt; i ++))    
    do
        if [ ! -f /zktdata/zkt${i} ]; then
            cd /zktdata/zkt${i}
            docker-compose down
            echo '========================================================='
            echo "此zkt${i}节点已删除"
            echo '========================================================='
        fi
    done
    rm -rf /zktdata
    docker rmi zktubelabs/zktube-prover
}

function zkt_restart
{
    tCnt=`ls -l /zktdata|grep "zkt"|wc -l`
    for ((i=1; i<=tCnt; i ++))    
    do
        if [ ! -f /zktdata/zkt${i} ]; then
            cd /zktdata/zkt${i}
            docker-compose down
            echo '========================================================='
            echo "zkt${i}节点已停止"
            echo '========================================================='
        fi
        docker rmi zktubelabs/zktube-prover
        if [ ! -f /zktdata/zkt${i} ]; then
            cd /zktdata/zkt${i}
            docker-compose up -d
            echo '========================================================='
            echo "zkt${i}节点已重启"
            echo '========================================================='
        fi
    done
}

function zkt_docker_stop
{
    echo '========================================================='
    echo '停止所有docker container'
    docker stop $(docker container ls -aq)
    echo '删除所有docker container'
    docker rm $(docker container ls -aq)
    echo '删除所有docker image'
    docker rmi $(docker image ls -aq)
    echo '删除所有zkt network'
    tCnt=`ls -l /zktdata|grep "zkt"|wc -l`
    for ((i=1; i<=tCnt; i ++))    
    do
        docker network rm "zkt${i}_default"
    done
    echo '============================================== Aven7 ====' 
}

while true
do
    print_tips
    read -p "请输入你的选项(1|2|3|4|5|6|q|Q):" choice

    case $choice in
        1)
            docker_install
            ;;
        2)
            zkt_init
            zkt_start
            ;;
        3)
            zkt_loop
            exit
            ;;
        4)
            zkt_restart
            ;;
        5)
            zkt_rm
            ;;
        6)
            zkt_docker_stop
            ;;
        q|Q)
            exit
            ;;
        *)
            echo "Error,input only in {1|2|3|4|5|6|q|Q}"
            ;;
    esac
done