From 13276166354450a083c8621ae6c36a311be2e3c4 Mon Sep 17 00:00:00 2001 From: liaojack8 Date: Mon, 6 Jul 2020 09:58:43 +0800 Subject: [PATCH] config.js can setting sa path, and install w/ my repo in script. --- config.js | 51 ++--- gdutilsinstall.sh | 508 +++++++++++++++++++++++----------------------- src/gd.js | 4 +- 3 files changed, 283 insertions(+), 280 deletions(-) diff --git a/config.js b/config.js index a5be916..45e553b 100644 --- a/config.js +++ b/config.js @@ -1,24 +1,27 @@ -// 单次请求多少毫秒未响应以后超时(基准值,若连续超时则下次调整为上次的2倍) -const TIMEOUT_BASE = 7000 -// 最大超时设置,比如某次请求,第一次7s超时,第二次14s,第三次28s,第四次56s,第五次不是112s而是60s,后续同理 -const TIMEOUT_MAX = 60000 - -const LOG_DELAY = 5000 // 日志输出时间间隔,单位毫秒 -const PAGE_SIZE = 1000 // 每次网络请求读取目录下的文件数,数值越大,越有可能超时,不得超过1000 - -const RETRY_LIMIT = 7 // 如果某次请求失败,允许其重试的最大次数 -const PARALLEL_LIMIT = 20 // 网络请求的并行数量,可根据网络环境调整 - -const DEFAULT_TARGET = '' // 必填,拷贝默认目的地ID,如果不指定target,则会复制到此处,建议填写团队盘ID - -const AUTH = { // 如果您拥有service account的json授权文件,可将其拷贝至 sa 目录中以代替 client_id/secret/refrest_token - client_id: 'your_client_id', - client_secret: 'your_client_secret', - refresh_token: 'your_refrest_token', - expires: 0, // 可以留空 - access_token: '', // 可以留空 - tg_token: 'bot_token', // 你的 telegram robot 的 token,获取方法参见 https://core.telegram.org/bots#6-botfather - tg_whitelist: ['your_tg_username'] // 你的tg username(t.me/username),bot只会执行这个列表里的用户所发送的指令 -} - -module.exports = { AUTH, PARALLEL_LIMIT, RETRY_LIMIT, TIMEOUT_BASE, TIMEOUT_MAX, LOG_DELAY, PAGE_SIZE, DEFAULT_TARGET } +// 单次请求多少毫秒未响应以后超时(基准值,若连续超时则下次调整为上次的2倍) +const TIMEOUT_BASE = 7000 +// 最大超时设置,比如某次请求,第一次7s超时,第二次14s,第三次28s,第四次56s,第五次不是112s而是60s,后续同理 +const TIMEOUT_MAX = 60000 + +const LOG_DELAY = 5000 // 日志输出时间间隔,单位毫秒 +const PAGE_SIZE = 1000 // 每次网络请求读取目录下的文件数,数值越大,越有可能超时,不得超过1000 + +const RETRY_LIMIT = 7 // 如果某次请求失败,允许其重试的最大次数 +const PARALLEL_LIMIT = 20 // 网络请求的并行数量,可根据网络环境调整 + +const DEFAULT_TARGET = '' // 必填,拷贝默认目的地ID,如果不指定target,则会复制到此处,建议填写团队盘ID + +const AUTH = { // 如果您拥有service account的json授权文件,可将其拷贝至 sa 目录中以代替 client_id/secret/refrest_token + client_id: 'your_client_id', + client_secret: 'your_client_secret', + refresh_token: 'your_refrest_token', + expires: 0, // 可以留空 + access_token: '', // 可以留空 + tg_token: 'bot_token', // 你的 telegram robot 的 token,获取方法参见 https://core.telegram.org/bots#6-botfather + tg_whitelist: ['your_tg_username'] // 你的tg username(t.me/username),bot只会执行这个列表里的用户所发送的指令 +} + +//-------------------MOD------------------- +const SA_PATH = '../sa' //sa路徑配置, 給定絕對路徑或是以src為當前路徑給定相對路徑 + +module.exports = { AUTH, PARALLEL_LIMIT, RETRY_LIMIT, TIMEOUT_BASE, TIMEOUT_MAX, LOG_DELAY, PAGE_SIZE, DEFAULT_TARGET, SA_PATH } diff --git a/gdutilsinstall.sh b/gdutilsinstall.sh index 66a153d..21c3466 100644 --- a/gdutilsinstall.sh +++ b/gdutilsinstall.sh @@ -1,254 +1,254 @@ -#!/bin/bash -echo -echo -e "\033[1;32m===== <> =====\033[0m" -echo -e "\033[1;32m---------------[ v2.1 by oneking ]---------------\033[0m" -echo -e "\033[32m 01.\033[0m 本脚本是针对TG大神@viegg的gdutils项目一键部署脚本;" -echo -e "\033[32m 02.\033[0m 脚本包括“TD盘VPS上查询转存部署”和“Telegram机器人部署”两部分" -echo -e "\033[32m 03.\033[0m 本脚本适应CentOS/Debian/Ubuntu三种操作系统,自动识别、自动选择对应分支一键安装部署" -echo -e "\033[32m 04.\033[0m 三步即可完成部署:上传脚本到VPS → 设置脚本执行权限 → 运行" -echo -e "\033[32m 05.\033[0m 准备工作一:在TG上注册好机器人取得并记录下该机器人TOKEN" -echo -e "\033[32m 06.\033[0m 准备工作二:拥有一个域名绑定到cloudflare解析到该机器人所在服务器IP" -echo -e "\033[32m 07.\033[0m 准备工作三:向机器人@userinfobot获取个人TG账号ID并记录" -echo -e "\033[32m 08.\033[0m 准备工作四:注册好一个Google team drive加入sa并记录下该盘ID" -echo -e "\033[32m 09.\033[0m 经测试可用完美安装系统:Centos 7/8 debian 9/10 ubuntu 16.04/18.04/19.10/20.04" -echo -e "\033[32m 10.\033[0m 部署过程中有任何问题请把“错误截图”“部署VPS系统名称版本”信息发给TG:onekings 或 vitaminor@gmail.com" -echo -e "\033[1;32m------------------------------------------------\033[0m" -read -s -n1 -p "★★★ 如已做好以上[5/6/7/8]准备或不需要安装Telegram机器人请按任意键开始部署,如未做好准备请按“Ctrl+c”终止脚本 ★★★" -echo -echo -e "\033[1;32m------------------------------------------------\033[0m" - -# 识别操作系统 -aNAME="$(uname -a)" -bNAME="$(cat /proc/version)" -cNAME="$(lsb_release -a)" -if [ -f "/etc/redhat-release" ]; then - if [[ $(cat /etc/redhat-release) =~ "CentOS" ]]; then - os="CentOS" - fi -elif [ "$aNAME"=~"Debian" -o "$bNAME"=~"Debian" -o "$cNAME"=~"Debian" ]; then - os="Debian" -elif [ "$aNAME"=~"Ubuntu" -o "$bNAME"=~"Ubuntu" -o "$cNAME"=~"Ubuntu" ]; then - os="Debian" -elif [ "$aNAME"=~"CentOS" -o "$bNAME"=~"CentOS" -o "$cNAME"=~"CentOS" ]; then - os="CentOS" -elif [ "$aNAME"=~"Darwin" -o "$bNAME"=~"Darwin" -o "$cNAME"=~"Darwin" ]; then - os="mac" -else - os="$bNAME" -fi - -# 需要安装的软件工具及依赖 -insofts=(epel-release update upgrade wget curl git unzip zip python3-distutils python3 python3-pip) - -#根据操作系统设置变量 -if [[ "$os" = "Debian" ]]; then - cmd_install="apt-get" #安装命令 - cmd_install_rely="build-essential" #c++编译环境 - nodejs_curl="https://deb.nodesource.com/setup_10.x" #nodejs下载链接 - cmd_install_rpm_build="" #安装rpm-build - nginx_conf="/etc/nginx/sites-enabled/" #nginx配置文件存放路径 - rm_nginx_default="rm -f /etc/nginx/sites-enabled/default" #删除default - echo - echo -e "\033[1;32m★★★★★ 您的操作系统为Debian,即将为你开始部署gdutils项目 ★★★★★\033[0m" -elif [[ "$os" = "Ubuntu" ]]; then - cmd_install="sudo apt-get" - cmd_install_rely="build-essential" - nodejs_curl="https://deb.nodesource.com/setup_10.x" - cmd_install_rpm_build="" - nginx_conf="/etc/nginx/sites-enabled/" - rm_nginx_default="rm -f /etc/nginx/sites-enabled/default" - echo - echo -e "\033[1;32m★★★★★ 您的操作系统为Ubuntu,即将为你开始部署gdutils项目 ★★★★★\033[0m" -elif [[ "$os" = "CentOS" ]]; then - cmd_install="yum" - cmd_install_rely="gcc-c++ make" - nodejs_curl="https://rpm.nodesource.com/setup_10.x" - cmd_install_rpm_build="yum install rpm-build -y" - nginx_conf="/etc/nginx/conf.d/" - rm_nginx_default="" - echo - echo -e "\033[1;32m★★★★★ 您的操作系统为Centos,即将为你开始部署gdutils项目 ★★★★★\033[0m" -elif [[ "$os" = "mac" ]]; then - echo - echo -e "\033[1;32m★★★★★ 您的操作系统为MacOS,请在图形界面手动安装 ★★★★★\033[0m" - exit - echo - echo -else - echo - echo -e "\033[1;32m unknow os $OS, exit! \033[0m" - exit - echo - echo -fi - -echo -echo -e "\033[1;32m===== <<升级系统/更新软件/安装工具/安装依赖>> =====\033[0m" -echo - -#安装which和sudo -if [[ "$(which which)" == "" ]]; then - echo -e "\033[1;32m“which”开始安装......\033[0m" - $cmd_install install which -y - echo -e "\033[1;32m------------------------------------------------\033[0m" -elif [[ "$(which sudo)" == "" ]]; then - echo -e "\033[1;32m“sudo”开始安装......\033[0m" - $cmd_install install sudo -y - echo -e "\033[1;32m------------------------------------------------\033[0m" -fi - -#安装工具和依赖 -for ((aloop = 0; aloop < ${#insofts[@]}; aloop++)); do - if [ ${insofts[$aloop]} = "update" -o ${insofts[$aloop]} = "upgrade" ]; then - echo -e "\033[1;32m“${insofts[$aloop]}”开始安装......\033[0m" - $cmd_install ${insofts[$aloop]} -y - echo -e "\033[1;32m------------------------------------------------\033[0m" - else - echo -e "\033[1;32m“${insofts[$aloop]}”开始安装......\033[0m" - $cmd_install install ${insofts[$aloop]} -y - echo -e "\033[1;32m------------------------------------------------\033[0m" - fi -done - -echo -echo -e "\033[1;32m===== <<安装gdutils依赖-nodejs和npm/安装配置gdutils>> =====\033[0m" -echo -$cmd_install install $cmd_install_rely -y -curl -sL $nodejs_curl | bash - -$cmd_install install nodejs -y -$cmd_install_rpm_build -git clone https://github.com/iwestlin/gd-utils && cd gd-utils -npm config set unsafe-perm=true -npm i - -echo -echo -e "\033[1;32m★★★ 恭喜您!gdutils统计转存系统已经正确安装完成,请上传sa到“./gd-utils/sa/”目录下完成最后的配置 ★★★\033[0m" -echo - -################################################################################################# - -echo -e "\033[1;32m----------------------------------------------------------\033[0m" -read -s -n1 -p "★★★ 下面将部署Telegram机器人,请确保准备所需条件已准备好,按任意键开始部署机器人;如未做好准备请按“Ctrl+c”终止部署机器人 ★★★" -echo -echo -e "\033[1;32m----------------------------------------------------------\033[0m" - -echo -echo -e "\033[1;32m ===== <<开始部署gdutils查询转存TG机器人>> ===== \033[0m" -echo - -#输入“机器人token/TG账号ID/域名/转存目的盘ID” -read -p """请输入机器人token并回车 - Your Bot Token =>:""" YOUR_BOT_TOKEN -#判断token是否输入正确 -while [[ "${#YOUR_BOT_TOKEN}" != 46 ]]; do - echo -e "\033[1;32m★★★ 机器人TOKEN输入不正确,请重新输入或按“Ctrl+C”结束安装! ★★★\033[0m" - read -p """请输入机器人token并回车 - Your Bot Token =>:""" YOUR_BOT_TOKEN -done - -read -p """请输入你的域名(在cloudflare上解析到你机器人所在VPS的域名,格式:bot.abc.com)并回车 - Your Domain Name =>:""" YOUR_DOMAIN_NAME -#判断域名是否正确 -while [[ "$YOUR_DOMAIN_NAME" =~ "http" ]]; do - echo -e "\033[1;32m★★★ “Your Domain Name”输入错误,应该输入你在cloudflare上解析的域名且不包含“http”,请重新输入或按“Ctrl+C”结束安装! ★★★\033[0m" - read -p """请输入你的域名(在cloudflare上解析到你机器人所在VPS的域名,格式:bot.abc.com)并回车 - Your Domain Name =>:""" YOUR_DOMAIN_NAME -done - -read -p """请输入使用机器人的telegram账号ID(获取ID机器人@userinfobot)并回车 - Your Telegram ID =>:""" YOUR_TELEGRAM_ID -#判断telegram ID是否正确(通过判断是不是纯数字) -until [[ $YOUR_TELEGRAM_ID =~ ^-?[0-9]+$ ]]; do - echo -e "\033[1;32m★★★ 您的TG账号ID输入不正确,请重新输入或按“Ctrl+C”结束安装! ★★★\033[0m" - read -p """请输入使用机器人的telegram账号ID(获取ID机器人@userinfobot)并回车 - Your Telegram ID =>:""" YOUR_TELEGRAM_ID -done - -read -p """请输入转存默认目的地团队盘ID(不指定转存目的地默认改地址,脚本强制要求输入团队盘ID)并回车 - Your Google Team Drive ID =>:""" YOUR_GOOGLE_TEAM_DRIVE_ID -#判断google team drive ID是否正确(团队盘ID长度19位) -while [[ "${#YOUR_GOOGLE_TEAM_DRIVE_ID}" != 19 ]]; do - echo -e "\033[1;32m★★★ 您的Google team drive ID输入不正确,请重新输入或按“Ctrl+C”结束安装! ★★★\033[0m" - read -p """请输入转存默认目的地ID(不指定转存目的地默认该地址,脚本强制要求输入团队盘ID)并回车 - Your Google Team Drive ID =>:""" YOUR_GOOGLE_TEAM_DRIVE_ID -done - -cd ~ && - sed -i "s/bot_token/$YOUR_BOT_TOKEN/g" ./gd-utils/config.js && - sed -i "s/your_tg_username/$YOUR_TELEGRAM_ID/g" ./gd-utils/config.js && - sed -i "s/DEFAULT_TARGET = ''/DEFAULT_TARGET = '$YOUR_GOOGLE_TEAM_DRIVE_ID'/g" ./gd-utils/config.js -echo -e "\033[1;32m----------------------------------------------------------\033[0m" - -echo -e "\033[1;32m“进程守护程序pm2”开始安装......\033[0m" -cd /root/gd-utils && - npm i pm2 -g && pm2 l -echo -e "\033[1;32m启动守护进程......\033[0m" -pm2 start server.js --node-args="--max-old-space-size=4096" -echo -e "\033[1;32m----------------------------------------------------------\033[0m" - -echo -e "\033[1;32m“nginx”开始安装......\033[0m" -cd ~ && - $cmd_install install nginx -y -echo -echo -e "\033[1;32m===== <<配置nginx服务>> ===== \033[0m" -echo -echo -e "\033[1;32m“nginx”起一个web服务......\033[0m" - -cd $nginx_conf -echo "server { - listen 80; - server_name ecbot.nctu.me; - return 301 https://$host$request_uri; -} - -server { - listen 443 ssl; - ssl on; - ssl_certificate /etc/ssl/certificate.crt; - ssl_certificate_key /etc/ssl/private.key; - server_name $YOUR_DOMAIN_NAME; - location / { - proxy_pass http://127.0.0.1:23333/; - } -}" >${nginx_conf}gdutilsbot.conf && - $rm_nginx_default - -ls && - nginx -t && - nginx -c /etc/nginx/nginx.conf && - nginx -s reload && - netstat -tulpen -echo -e "\033[1;32m----------------------------------------------------------\033[0m" - -echo -e "\033[1;32m“检查网站是否部署成功”......\033[0m" -curl $YOUR_DOMAIN_NAME/api/gdurl/count\?fid=124pjM5LggSuwI1n40bcD5tQ13wS0M6wg -echo -echo -e "\033[1;32m设置Webhook服务......\033[0m" -print_webhook=$(curl -F "url=https://$YOUR_DOMAIN_NAME/api/gdurl/tgbot" "https://api.telegram.org/bot$YOUR_BOT_TOKEN/setWebhook") -echo - -# 判断反向代理是否部署成功 -if [[ $print_webhook =~ "true" ]]; then - echo -e "\033[1;32m★★★ 恭喜你!GoogleDrive查询转存机器人部署成功,请回到TG界面给bot发送个“/help”获取使用帮助 ★★★\033[0m" -else - echo -e "\033[32m★★★ 很遗憾!机器人设置失败,请返回检查网站是否部署成功,并重复本安装过程 ★★★\033[0m", exit! -fi -nginx -t && nginx -s reload -echo -echo - -cd ~ -rm -f gdutilsinstall.sh - -###########################gdutils功能建议################################## -# 本部分是对gdutils项目的建议,因为我主要用的是查询功能所以以下建议只涉及查询功能 -# 1-把以下参数放入配置文件设置:sa存放路径 -# 2-改sa“随机”使用为“顺序”分组使用; -# 3-增加输出模式,可以用命令行后带参数选择,具体模式建议: -# ①按一级或者二级文件夹显示数量大小 -# ②可以一次性统计多个磁盘并且输出单个磁盘文件数和大小以及几个磁盘总和 -# ③获取id对应的文件夹名或者磁盘明保存数据库,给个命令能够查询历史记录汇总或者指定汇总 -# 4-查询过程中输出方式不要每次都输出一次,可以固定+数字变化 -# 5-命令参数可加在ID前或后,如果非要固定一种的话就加在ID之前 -# 6-命令行也改为默认sa模式 -############################################################################ +#!/bin/bash +echo +echo -e "\033[1;32m===== <> =====\033[0m" +echo -e "\033[1;32m---------------[ v2.1 by oneking ]---------------\033[0m" +echo -e "\033[32m 01.\033[0m 本脚本是针对TG大神@viegg的gdutils项目一键部署脚本;" +echo -e "\033[32m 02.\033[0m 脚本包括“TD盘VPS上查询转存部署”和“Telegram机器人部署”两部分" +echo -e "\033[32m 03.\033[0m 本脚本适应CentOS/Debian/Ubuntu三种操作系统,自动识别、自动选择对应分支一键安装部署" +echo -e "\033[32m 04.\033[0m 三步即可完成部署:上传脚本到VPS → 设置脚本执行权限 → 运行" +echo -e "\033[32m 05.\033[0m 准备工作一:在TG上注册好机器人取得并记录下该机器人TOKEN" +echo -e "\033[32m 06.\033[0m 准备工作二:拥有一个域名绑定到cloudflare解析到该机器人所在服务器IP" +echo -e "\033[32m 07.\033[0m 准备工作三:向机器人@userinfobot获取个人TG账号ID并记录" +echo -e "\033[32m 08.\033[0m 准备工作四:注册好一个Google team drive加入sa并记录下该盘ID" +echo -e "\033[32m 09.\033[0m 经测试可用完美安装系统:Centos 7/8 debian 9/10 ubuntu 16.04/18.04/19.10/20.04" +echo -e "\033[32m 10.\033[0m 部署过程中有任何问题请把“错误截图”“部署VPS系统名称版本”信息发给TG:onekings 或 vitaminor@gmail.com" +echo -e "\033[1;32m------------------------------------------------\033[0m" +read -s -n1 -p "★★★ 如已做好以上[5/6/7/8]准备或不需要安装Telegram机器人请按任意键开始部署,如未做好准备请按“Ctrl+c”终止脚本 ★★★" +echo +echo -e "\033[1;32m------------------------------------------------\033[0m" + +# 识别操作系统 +aNAME="$(uname -a)" +bNAME="$(cat /proc/version)" +cNAME="$(lsb_release -a)" +if [ -f "/etc/redhat-release" ]; then + if [[ $(cat /etc/redhat-release) =~ "CentOS" ]]; then + os="CentOS" + fi +elif [ "$aNAME"=~"Debian" -o "$bNAME"=~"Debian" -o "$cNAME"=~"Debian" ]; then + os="Debian" +elif [ "$aNAME"=~"Ubuntu" -o "$bNAME"=~"Ubuntu" -o "$cNAME"=~"Ubuntu" ]; then + os="Debian" +elif [ "$aNAME"=~"CentOS" -o "$bNAME"=~"CentOS" -o "$cNAME"=~"CentOS" ]; then + os="CentOS" +elif [ "$aNAME"=~"Darwin" -o "$bNAME"=~"Darwin" -o "$cNAME"=~"Darwin" ]; then + os="mac" +else + os="$bNAME" +fi + +# 需要安装的软件工具及依赖 +insofts=(epel-release update upgrade wget curl git unzip zip python3-distutils python3 python3-pip) + +#根据操作系统设置变量 +if [[ "$os" = "Debian" ]]; then + cmd_install="apt-get" #安装命令 + cmd_install_rely="build-essential" #c++编译环境 + nodejs_curl="https://deb.nodesource.com/setup_10.x" #nodejs下载链接 + cmd_install_rpm_build="" #安装rpm-build + nginx_conf="/etc/nginx/sites-enabled/" #nginx配置文件存放路径 + rm_nginx_default="rm -f /etc/nginx/sites-enabled/default" #删除default + echo + echo -e "\033[1;32m★★★★★ 您的操作系统为Debian,即将为你开始部署gdutils项目 ★★★★★\033[0m" +elif [[ "$os" = "Ubuntu" ]]; then + cmd_install="sudo apt-get" + cmd_install_rely="build-essential" + nodejs_curl="https://deb.nodesource.com/setup_10.x" + cmd_install_rpm_build="" + nginx_conf="/etc/nginx/sites-enabled/" + rm_nginx_default="rm -f /etc/nginx/sites-enabled/default" + echo + echo -e "\033[1;32m★★★★★ 您的操作系统为Ubuntu,即将为你开始部署gdutils项目 ★★★★★\033[0m" +elif [[ "$os" = "CentOS" ]]; then + cmd_install="yum" + cmd_install_rely="gcc-c++ make" + nodejs_curl="https://rpm.nodesource.com/setup_10.x" + cmd_install_rpm_build="yum install rpm-build -y" + nginx_conf="/etc/nginx/conf.d/" + rm_nginx_default="" + echo + echo -e "\033[1;32m★★★★★ 您的操作系统为Centos,即将为你开始部署gdutils项目 ★★★★★\033[0m" +elif [[ "$os" = "mac" ]]; then + echo + echo -e "\033[1;32m★★★★★ 您的操作系统为MacOS,请在图形界面手动安装 ★★★★★\033[0m" + exit + echo + echo +else + echo + echo -e "\033[1;32m unknow os $OS, exit! \033[0m" + exit + echo + echo +fi + +echo +echo -e "\033[1;32m===== <<升级系统/更新软件/安装工具/安装依赖>> =====\033[0m" +echo + +#安装which和sudo +if [[ "$(which which)" == "" ]]; then + echo -e "\033[1;32m“which”开始安装......\033[0m" + $cmd_install install which -y + echo -e "\033[1;32m------------------------------------------------\033[0m" +elif [[ "$(which sudo)" == "" ]]; then + echo -e "\033[1;32m“sudo”开始安装......\033[0m" + $cmd_install install sudo -y + echo -e "\033[1;32m------------------------------------------------\033[0m" +fi + +#安装工具和依赖 +for ((aloop = 0; aloop < ${#insofts[@]}; aloop++)); do + if [ ${insofts[$aloop]} = "update" -o ${insofts[$aloop]} = "upgrade" ]; then + echo -e "\033[1;32m“${insofts[$aloop]}”开始安装......\033[0m" + $cmd_install ${insofts[$aloop]} -y + echo -e "\033[1;32m------------------------------------------------\033[0m" + else + echo -e "\033[1;32m“${insofts[$aloop]}”开始安装......\033[0m" + $cmd_install install ${insofts[$aloop]} -y + echo -e "\033[1;32m------------------------------------------------\033[0m" + fi +done + +echo +echo -e "\033[1;32m===== <<安装gdutils依赖-nodejs和npm/安装配置gdutils>> =====\033[0m" +echo +$cmd_install install $cmd_install_rely -y +curl -sL $nodejs_curl | bash - +$cmd_install install nodejs -y +$cmd_install_rpm_build +git clone https://github.com/liaojack8/gd-utils-cht && cd gd-utils-cht +npm config set unsafe-perm=true +npm i + +echo +echo -e "\033[1;32m★★★ 恭喜您!gdutils统计转存系统已经正确安装完成,请上传sa到“./gd-utils/sa/”目录下完成最后的配置 ★★★\033[0m" +echo + +################################################################################################# + +echo -e "\033[1;32m----------------------------------------------------------\033[0m" +read -s -n1 -p "★★★ 下面将部署Telegram机器人,请确保准备所需条件已准备好,按任意键开始部署机器人;如未做好准备请按“Ctrl+c”终止部署机器人 ★★★" +echo +echo -e "\033[1;32m----------------------------------------------------------\033[0m" + +echo +echo -e "\033[1;32m ===== <<开始部署gdutils查询转存TG机器人>> ===== \033[0m" +echo + +#输入“机器人token/TG账号ID/域名/转存目的盘ID” +read -p """请输入机器人token并回车 + Your Bot Token =>:""" YOUR_BOT_TOKEN +#判断token是否输入正确 +while [[ "${#YOUR_BOT_TOKEN}" != 46 ]]; do + echo -e "\033[1;32m★★★ 机器人TOKEN输入不正确,请重新输入或按“Ctrl+C”结束安装! ★★★\033[0m" + read -p """请输入机器人token并回车 + Your Bot Token =>:""" YOUR_BOT_TOKEN +done + +read -p """请输入你的域名(在cloudflare上解析到你机器人所在VPS的域名,格式:bot.abc.com)并回车 + Your Domain Name =>:""" YOUR_DOMAIN_NAME +#判断域名是否正确 +while [[ "$YOUR_DOMAIN_NAME" =~ "http" ]]; do + echo -e "\033[1;32m★★★ “Your Domain Name”输入错误,应该输入你在cloudflare上解析的域名且不包含“http”,请重新输入或按“Ctrl+C”结束安装! ★★★\033[0m" + read -p """请输入你的域名(在cloudflare上解析到你机器人所在VPS的域名,格式:bot.abc.com)并回车 + Your Domain Name =>:""" YOUR_DOMAIN_NAME +done + +read -p """请输入使用机器人的telegram账号ID(获取ID机器人@userinfobot)并回车 + Your Telegram ID =>:""" YOUR_TELEGRAM_ID +#判断telegram ID是否正确(通过判断是不是纯数字) +until [[ $YOUR_TELEGRAM_ID =~ ^-?[0-9]+$ ]]; do + echo -e "\033[1;32m★★★ 您的TG账号ID输入不正确,请重新输入或按“Ctrl+C”结束安装! ★★★\033[0m" + read -p """请输入使用机器人的telegram账号ID(获取ID机器人@userinfobot)并回车 + Your Telegram ID =>:""" YOUR_TELEGRAM_ID +done + +read -p """请输入转存默认目的地团队盘ID(不指定转存目的地默认改地址,脚本强制要求输入团队盘ID)并回车 + Your Google Team Drive ID =>:""" YOUR_GOOGLE_TEAM_DRIVE_ID +#判断google team drive ID是否正确(团队盘ID长度19位) +while [[ "${#YOUR_GOOGLE_TEAM_DRIVE_ID}" != 19 ]]; do + echo -e "\033[1;32m★★★ 您的Google team drive ID输入不正确,请重新输入或按“Ctrl+C”结束安装! ★★★\033[0m" + read -p """请输入转存默认目的地ID(不指定转存目的地默认该地址,脚本强制要求输入团队盘ID)并回车 + Your Google Team Drive ID =>:""" YOUR_GOOGLE_TEAM_DRIVE_ID +done + +cd ~ && + sed -i "s/bot_token/$YOUR_BOT_TOKEN/g" ./gd-utils/config.js && + sed -i "s/your_tg_username/$YOUR_TELEGRAM_ID/g" ./gd-utils/config.js && + sed -i "s/DEFAULT_TARGET = ''/DEFAULT_TARGET = '$YOUR_GOOGLE_TEAM_DRIVE_ID'/g" ./gd-utils/config.js +echo -e "\033[1;32m----------------------------------------------------------\033[0m" + +echo -e "\033[1;32m“进程守护程序pm2”开始安装......\033[0m" +cd /root/gd-utils && + npm i pm2 -g && pm2 l +echo -e "\033[1;32m启动守护进程......\033[0m" +pm2 start server.js --node-args="--max-old-space-size=4096" +echo -e "\033[1;32m----------------------------------------------------------\033[0m" + +echo -e "\033[1;32m“nginx”开始安装......\033[0m" +cd ~ && + $cmd_install install nginx -y +echo +echo -e "\033[1;32m===== <<配置nginx服务>> ===== \033[0m" +echo +echo -e "\033[1;32m“nginx”起一个web服务......\033[0m" + +cd $nginx_conf +echo "server { + listen 80; + server_name ecbot.nctu.me; + return 301 https://$host$request_uri; +} + +server { + listen 443 ssl; + ssl on; + ssl_certificate /etc/ssl/certificate.crt; + ssl_certificate_key /etc/ssl/private.key; + server_name $YOUR_DOMAIN_NAME; + location / { + proxy_pass http://127.0.0.1:23333/; + } +}" >${nginx_conf}gdutilsbot.conf && + $rm_nginx_default + +ls && + nginx -t && + nginx -c /etc/nginx/nginx.conf && + nginx -s reload && + netstat -tulpen +echo -e "\033[1;32m----------------------------------------------------------\033[0m" + +echo -e "\033[1;32m“检查网站是否部署成功”......\033[0m" +curl $YOUR_DOMAIN_NAME/api/gdurl/count\?fid=124pjM5LggSuwI1n40bcD5tQ13wS0M6wg +echo +echo -e "\033[1;32m设置Webhook服务......\033[0m" +print_webhook=$(curl -F "url=https://$YOUR_DOMAIN_NAME/api/gdurl/tgbot" "https://api.telegram.org/bot$YOUR_BOT_TOKEN/setWebhook") +echo + +# 判断反向代理是否部署成功 +if [[ $print_webhook =~ "true" ]]; then + echo -e "\033[1;32m★★★ 恭喜你!GoogleDrive查询转存机器人部署成功,请回到TG界面给bot发送个“/help”获取使用帮助 ★★★\033[0m" +else + echo -e "\033[32m★★★ 很遗憾!机器人设置失败,请返回检查网站是否部署成功,并重复本安装过程 ★★★\033[0m", exit! +fi +nginx -t && nginx -s reload +echo +echo + +cd ~ +rm -f gdutilsinstall.sh + +###########################gdutils功能建议################################## +# 本部分是对gdutils项目的建议,因为我主要用的是查询功能所以以下建议只涉及查询功能 +# 1-把以下参数放入配置文件设置:sa存放路径 +# 2-改sa“随机”使用为“顺序”分组使用; +# 3-增加输出模式,可以用命令行后带参数选择,具体模式建议: +# ①按一级或者二级文件夹显示数量大小 +# ②可以一次性统计多个磁盘并且输出单个磁盘文件数和大小以及几个磁盘总和 +# ③获取id对应的文件夹名或者磁盘明保存数据库,给个命令能够查询历史记录汇总或者指定汇总 +# 4-查询过程中输出方式不要每次都输出一次,可以固定+数字变化 +# 5-命令参数可加在ID前或后,如果非要固定一种的话就加在ID之前 +# 6-命令行也改为默认sa模式 +############################################################################ diff --git a/src/gd.js b/src/gd.js index 1cac3b7..5300083 100644 --- a/src/gd.js +++ b/src/gd.js @@ -8,7 +8,7 @@ const HttpsProxyAgent = require('https-proxy-agent') const { GoogleToken } = require('gtoken') const handle_exit = require('signal-exit') -const { AUTH, RETRY_LIMIT, PARALLEL_LIMIT, TIMEOUT_BASE, TIMEOUT_MAX, LOG_DELAY, PAGE_SIZE, DEFAULT_TARGET } = require('../config') +const { AUTH, RETRY_LIMIT, PARALLEL_LIMIT, TIMEOUT_BASE, TIMEOUT_MAX, LOG_DELAY, PAGE_SIZE, DEFAULT_TARGET, SA_PATH } = require('../config') const { db } = require('../db') const { make_table, make_tg_table, make_html, summary } = require('./summary') @@ -18,7 +18,7 @@ const { https_proxy } = process.env const axins = axios.create(https_proxy ? { httpsAgent: new HttpsProxyAgent(https_proxy) } : {}) const SA_BATCH_SIZE = 1000 -const SA_FILES = fs.readdirSync(path.join(__dirname, '../sa')).filter(v => v.endsWith('.json')) +const SA_FILES = fs.readdirSync(path.join(__dirname, SA_PATH)).filter(v => v.endsWith('.json')) SA_FILES.flag = 0 let SA_TOKENS = get_sa_batch()