#!/bin/bash
# 自动PT autopt.sh 一键脚本    短网址: https://git.io/autopt.sh
# Usage:  wget git.io/autopt.sh && bash autopt.sh

SERVER_IP=172.17.0.1
PORT=8080
PASSWORD="adminadmin"

# 此 Mongodb 数据连接帐号, 只供初次测试配置使用，请至 https://www.mongodb.com/ 注册私人免费帐号

MONGODB_API="pt:Pa55w.rd@cluster0.m9ox4"
RSS_DOMAIN="RSS.AUTOPT.NET"
RSS_PASSKEY="123456789-PASSKEY-PASSKEY-ABCDEF"

set_qb_conf()
{
    echo -e "${SkyBlue}:: Docker版 qBittorrent Web 默认: ${GreenBG} http://${SERVER_IP}:${PORT} ${SkyBlue} 密码:${RedBG} ${PASSWORD} "
    echo -e "${SkyBlue}:: 请按实际信息修改，下面三项按 ${RedBG}<Enter>${SkyBlue} 使用默认值! ${Yellow}"

    read -p ":: 1.请输入 qBittorrent Web 服务器IP (172.17.0.1): "  sv_ip
    read -p ":: 2.请输入 qBittorrent Web 服务端口   ( 8080 )  : "  port
    read -p ":: 3.请输入 qBittorrent Web 管理密码 (adminadmin): "  passwd

    if [[ ! -z "${sv_ip}" ]]; then
        SERVER_IP=${sv_ip}
    fi
    if [[ ! -z "${port}" ]]; then
        PORT=${port}
    fi
    if [[ ! -z "${passwd}" ]]; then
        PASSWORD=${passwd}
    fi

    echo -e "${SkyBlue}:: 设置后 qBittorrent Web管理网址: ${GreenBG} http://${SERVER_IP}:${PORT} ${SkyBlue} 密码:${RedBG} ${PASSWORD} "
    echo -e ${Font}
}

set_passkey_conf()
{
    echo -e "${SkyBlue}:: 请提前登陆 PT 网站生成 RSS 订阅地址; 获得${GreenBG} PASSKEY ${SkyBlue}和RSS域名, 键盘按 ${RedBG}<Shift+Insert>${RedBG}${SkyBlue} 粘贴"
    echo -e "${SkyBlue}:: 请按实际信息修改，按 ${RedBG}<Enter>${SkyBlue} 使用伪值填充! ${Yellow}"

    read -p ":: 4.请输入RSS-PASSKEY (123456789-PASSKEY-PASSKEY-ABCDEF) : "  passkey
    read -p ":: 5.请输入RSS 订阅地址的域名 ( pt.msg.vg ): "  domain

    if [[ ! -z "${domain}" ]]; then
        RSS_DOMAIN=${domain}
    fi
    if [[ ! -z "${passkey}" ]]; then
        RSS_PASSKEY=${passkey}
    fi

    echo -e "${SkyBlue}:: 设置后RSS域名: ${GreenBG} ${RSS_DOMAIN} ${SkyBlue} RSS-PASSKEY: ${RedBG} ${RSS_PASSKEY} "
    echo -e ${Font}
}

input_conf(){
    set_qb_conf
    set_passkey_conf

    echo -e "${SkyBlue}:: 请检查填充信息是否正确, 即将初始化自动PT工具配置文件: ${GreenBG} /root/goseeder.conf ${Font} 请确认覆盖\c"
    read -p "(Y/N): " key
    case $key in
        Y)
        init_config
        ;;
        y)
        init_config
        ;;
        N)
        input_conf
        ;;
        n)
        input_conf
        ;;
    esac
}

init_config(){

# 初始化自动PT工具配置文件:  /root/goseeder.conf

    cat <<EOF >/root/goseeder.conf
{
"dbserver": "mongodb+srv://${MONGODB_API}.mongodb.net/myFirstDatabase?retryWrites=true&w=majority",
"node": [
  {
"source": "${RSS_DOMAIN}",
"passkey": "${RSS_PASSKEY}",
"limit": 10,
"enable": true,
"rule": {
    "seeder_time": 0,
    "seeder_ratio": 0,
    "speed_limit": 8.0
   }
  }
 ],
"server": [
  {
"endpoint": "http://${SERVER_IP}:${PORT}",
"username": "admin",
"password": "${PASSWORD}",
"remark": "AutoPT-007",
"enable": true,
"rule": {
"concurrent_download": 1,
"disk_threshold": 10.0,
"disk_overcommit": false,
"max_speed": 30.00,
"min_alivetime": 3600,
"max_alivetime": 86400,
"min_tasksize": 0.0,
"max_tasksize": 50.0,
"max_disklatency": 10000
   }
  }
 ]
}

EOF

    echo -e "${SkyBlue}:: 初始化自动PT工具配置文件完成: ${GreenBG} /root/goseeder.conf ${Font}"
    cat /root/goseeder.conf
    echo -e  "${RedBG}# 此 Mongodb 数据连接帐号, 只供初次测试配置使用 ${SkyBlue} 请至 https://www.mongodb.com/ 注册私人免费帐号"
}

system_def(){
    Green="\033[32m"  && Red="\033[31m" && GreenBG="\033[42;37m" && RedBG="\033[41;37m"
    Font="\033[0m"  && Yellow="\033[0;33m" && SkyBlue="\033[0;36m"
}

install(){
  docker run -d  --name  autopt  \
    --net=host \
    -v  /root/goseeder.conf:/goseeder.conf  \
    hongwenjun/autopt

    echo -e "${GreenBG}::   autopt镜像:   https://hub.docker.com/r/hongwenjun/autopt ${Font}"
}

# 安装Docker可视化界面Portainer
portainer_install(){
    docker run --name Portainer          \
      --restart=always  -d -p 9000:9000  \
      -v /var/run/docker.sock:/var/run/docker.sock  \
      -v /opt/portainer_data:/data        \
      portainer/portainer

    #  浏览器中输入网址管理
    echo -e "${GreenBG}:: Docker可视化界面Portainer 管理地址  ${Yellow}"
    echo -e  http://$(cat /var/ip_addr):9000
}

# qBittorrent软件(Host模式支持IPV6)
qbittorrent_install(){
    docker run --name=qbittorrent \
    -e PUID=1000 -e PGID=1000 \
    -e TZ=Asia/ShangHai \
    -e UMASK_SET=022 -e \
    WEBUI_PORT=8080 \
    --net=host \
    -v /mnt/config:/config \
    -v /mnt/downloads:/downloads \
    --restart unless-stopped \
    -d linuxserver/qbittorrent:4.5.1

    #  浏览器中输入网址管理
    echo -e "${GreenBG}:: qBittorrent软件 管理地址  ${Yellow}"
    echo -e  http://$(cat /var/ip_addr):8080
}


# 设置菜单
start_menu(){
    echo -e "${GreenBG}>     开源项目:  https://github.com/hongwenjun/vps_setup     "
    echo -e "${RedBG}>  自动PT autopt.sh 一键脚本 感谢大神TaterLi创作goseeder工具 ${Font}"
    echo -e "${SkyBlue}>  1. 重置初始化自动PT工具配置文件(首次需运行)"
    echo -e ">  2. 安装 Docker 容器 autopt 镜像"
    echo -e ">  3. 重启 autopt 自动PT容器"
    echo -e ">  4. 停止 autopt 自动PT容器"
    echo -e ">  5. 卸载 autopt 自动PT容器"
    echo -e ">  6. 查看 QB 服务器最新状态"
    echo    "------------------------------------------------------------"
    echo -e "${Green}>  7. 安装Docker引擎和qBittorrent软件(Host模式支持IPV6)"
    echo -e ">  8. 检查 自动PT 推送信息, 可只按 ${RedBG}<Enter> ${Font}"
    echo -e "${Green}>  9. 安装Docker可视化界面 Portainer 工具 ${Font}"
    echo
    echo -e "${GreenBG}:: autopt镜像:  https://hub.docker.com/r/hongwenjun/autopt  ${Font}"
    read -p "请输入数字(1-8):" num
    case "$num" in
        1)
        input_conf
        start_menu
        ;;
        2)
        install
        start_menu
        ;;
        3)
        docker restart autopt
        ;;
        4)
        docker stop autopt
        ;;
        5)
        docker rm  -f autopt
        docker rmi -f hongwenjun/autopt
        ;;
        6)
        docker  exec  -it autopt  gostat
        ;;
        7)
        ##  支持国内Docker安装，使用阿里云
        wget get.docker.com -O get.docker.sh
        bash  get.docker.sh  --mirror Aliyun
        qbittorrent_install
        ;;
        8)
        docker logs  --tail=100 autopt | grep -e 种子 -e 磁盘
        ;;
        9)
        portainer_install
        ;;
        *)
        docker logs  --tail=100 autopt | grep -e 种子 -e 磁盘
        ;;
        esac
}

clear
system_def
start_menu