mirror of
https://github.com/whyour/qinglong.git
synced 2025-05-22 22:36:06 +08:00
300 lines
9.5 KiB
Bash
Executable File
300 lines
9.5 KiB
Bash
Executable File
#!/usr/bin/env bash
|
||
|
||
ShellDir=${QL_DIR:-$(
|
||
cd $(dirname $0)
|
||
pwd
|
||
)}
|
||
[[ $QL_DIR ]] && ShellJs=js
|
||
LogDir=$ShellDir/log
|
||
[ ! -d $LogDir ] && mkdir -p $LogDir
|
||
DbDir=$ShellDir/db
|
||
[ ! -d $DbDir ] && mkdir -p $DbDir
|
||
ManualLogDir=$ShellDir/manual_log
|
||
[ ! -d $ManualLogDir ] && mkdir -p $ManualLogDir
|
||
ScriptsDir=$ShellDir/scripts
|
||
ConfigDir=$ShellDir/config
|
||
FileConf=$ConfigDir/config.sh
|
||
CookieConf=$ConfigDir/cookie.sh
|
||
AuthConf=$ConfigDir/auth.json
|
||
ExtraShell=$ConfigDir/extra.sh
|
||
FileConfSample=$ShellDir/sample/config.sh.sample
|
||
ListCronSample=$ShellDir/sample/crontab.list.sample
|
||
ListCronCurrent=$ConfigDir/crontab.list
|
||
ListCronRemote=$ScriptsDir/docker/crontab_list.sh
|
||
ListCurrentTask=$LogDir/task.list
|
||
ListRemoteTask=$LogDir/js.list
|
||
ListJsAdd=$LogDir/js-add.list
|
||
ListJsDrop=$LogDir/js-drop.list
|
||
ContentVersion=$ShellDir/version
|
||
ContentNewTask=$ShellDir/new_task
|
||
ContentDropTask=$ShellDir/drop_task
|
||
SendVersion=$ShellDir/send_version
|
||
isTermux=$ANDROID_RUNTIME_ROOT$ANDROID_ROOT
|
||
ShellURL=https://github.com.cnpmjs.org/whyour/qinglong
|
||
ScriptsURL=https://github.com.cnpmjs.org/gossh520/jd_scripts
|
||
|
||
Import_Conf() {
|
||
if [ ! -s $FileConf ]; then
|
||
echo -e "复制一份 $FileConfSample 示例配置文件\n\n"
|
||
cp -fv $FileConfSample $FileConf
|
||
fi
|
||
if [ ! -s $ListCronCurrent ]; then
|
||
echo -e "复制一份 $ListCronSample 基础定时任务\n\n"
|
||
cp -fv $ListCronSample $ListCronCurrent
|
||
fi
|
||
[ -f $CookieConf ] && . $CookieConf
|
||
[ -f $FileConf ] && . $FileConf
|
||
}
|
||
|
||
# 更新shell
|
||
Git_Pull_Shell() {
|
||
echo -e "更新shell...\n"
|
||
cd $ShellDir
|
||
git remote set-url origin $ShellURL
|
||
git fetch --all
|
||
ExitStatusShell=$?
|
||
git reset --hard origin/master
|
||
git pull
|
||
}
|
||
|
||
Git_Pull_Shell_Next() {
|
||
if [[ $ExitStatusShell -eq 0 ]]; then
|
||
echo -e "更新shell成功...\n"
|
||
[ ! -d $ShellDir/node_modules ] && Npm_Install panel
|
||
[ -f $ShellDir/package.json ] && PanelDependNew=$(cat $ShellDir/package.json)
|
||
[[ "$PanelDependOld" != "$PanelDependNew" ]] && cd $ShellDir && Npm_Install panel
|
||
cp -f $FileConfSample $ConfigDir/config.sh.sample
|
||
Notify_Version
|
||
else
|
||
echo -e "更新shell失败，请检查原因...\n"
|
||
fi
|
||
}
|
||
|
||
## npm install
|
||
Npm_Install() {
|
||
echo -e "检测到 $1 的依赖包有变化，运行 npm install...\n"
|
||
Npm_InstallSub
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "\nnpm install 运行不成功，自动删除 $1/node_modules 后再次尝试一遍..."
|
||
rm -rf node_modules
|
||
fi
|
||
echo
|
||
|
||
if [ ! -d node_modules ]; then
|
||
echo -e "运行 npm install...\n"
|
||
Npm_InstallSub
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "\nnpm install 运行不成功，自动删除 $1/node_modules...\n"
|
||
echo -e "请进入 $1 目录后手动运行 npm install...\n"
|
||
echo -e "3...\n"
|
||
sleep 1
|
||
echo -e "2...\n"
|
||
sleep 1
|
||
echo -e "1...\n"
|
||
sleep 1
|
||
rm -rf node_modules
|
||
fi
|
||
fi
|
||
}
|
||
|
||
## npm install 子程序，判断是否为安卓，判断是否安装有yarn
|
||
function Npm_InstallSub() {
|
||
if [ -n "$isTermux" ]; then
|
||
npm install --no-save --no-bin-links --registry=https://registry.npm.taobao.org || npm install --no-bin-links --no-save
|
||
elif ! type yarn >/dev/null 2>&1; then
|
||
npm install --no-save --registry=https://registry.npm.taobao.org || npm install --no-save
|
||
else
|
||
echo -e "检测到本机安装了 yarn，使用 yarn 替代 npm...\n"
|
||
yarn install --registry=https://registry.npm.taobao.org || yarn install
|
||
fi
|
||
}
|
||
|
||
## 检测配置文件版本
|
||
Notify_Version() {
|
||
## 识别出两个文件的版本号
|
||
VerConfSample=$(grep " Version: " $FileConfSample | perl -pe "s|.+v((\d+\.?){3})|\1|")
|
||
[ -f $FileConf ] && VerConf=$(grep " Version: " $FileConf | perl -pe "s|.+v((\d+\.?){3})|\1|")
|
||
|
||
## 删除旧的发送记录文件
|
||
[ -f "$SendVersion" ] && [[ $(cat $SendVersion) != $VerConfSample ]] && rm -f $SendVersion
|
||
|
||
## 识别出更新日期和更新内容
|
||
UpdateDate=$(grep " Date: " $FileConfSample | awk -F ": " '{print $2}')
|
||
UpdateContent=$(grep " Update Content: " $FileConfSample | awk -F ": " '{print $2}')
|
||
|
||
## 如果是今天，并且版本号不一致，则发送通知
|
||
if [ -f $FileConf ] && [[ "$VerConf" != "$VerConfSample" ]] && [[ $UpdateDate == $(date "+%Y-%m-%d") ]]; then
|
||
if [ ! -f $SendVersion ]; then
|
||
notify "检测到配置文件config.sh.sample有更新" "更新日期: $UpdateDate\n当前版本: $VerConf\n新的版本: $VerConfSample\n更新内容: $UpdateContent\n更新说明: 如需使用新功能请对照config.sh.sample，将相关新参数手动增加到你自己的config.sh中，否则请无视本消息。本消息只在该新版本配置文件更新当天发送一次。"
|
||
fi
|
||
else
|
||
[ -f $ContentVersion ] && rm -f $ContentVersion
|
||
[ -f $SendVersion ] && rm -f $SendVersion
|
||
fi
|
||
}
|
||
|
||
## 每天次数随机，更新时间随机，更新秒数随机，至少6次，至多12次，大部分为8-10次，符合正态分布。
|
||
Random_Pull_Cron() {
|
||
if [[ $(date "+%-H") -le 2 ]]; then
|
||
RanMin=$(($RANDOM % 60))
|
||
RanSleep=$(($RANDOM % 56))
|
||
RanHourArray[0]=$(($RANDOM % 3))
|
||
for ((i = 1; i < 14; i++)); do
|
||
j=$(($i - 1))
|
||
tmp=$(($RANDOM % 3 + ${RanHourArray[j]} + 2))
|
||
[[ $tmp -lt 24 ]] && RanHourArray[i]=$tmp || break
|
||
done
|
||
|
||
RanHour=${RanHourArray[0]}
|
||
for ((i = 1; i < ${#RanHourArray[*]}; i++)); do
|
||
RanHour="$RanHour,${RanHourArray[i]}"
|
||
done
|
||
|
||
perl -i -pe "s|.+(git_pull? .+git_pull\.log.*)|$RanMin $RanHour \* \* \* sleep $RanSleep && \1|" $ListCronCurrent
|
||
crontab $ListCronCurrent
|
||
fi
|
||
}
|
||
|
||
## 克隆scripts
|
||
Git_Clone_Scripts() {
|
||
git clone -b master $ScriptsURL $ScriptsDir
|
||
ExitStatusScripts=$?
|
||
}
|
||
|
||
## 更新scripts
|
||
Git_Pull_Scripts() {
|
||
if [ -d $ScriptsDir/.git ]; then
|
||
echo -e "更新scripts...\n"
|
||
cd $ScriptsDir
|
||
git remote set-url origin $ScriptsURL
|
||
git fetch --all
|
||
ExitStatusScripts=$?
|
||
git reset --hard origin/master
|
||
git pull
|
||
else
|
||
Git_Clone_Scripts
|
||
fi
|
||
}
|
||
|
||
Git_Pull_Scripts_Next() {
|
||
if [[ $ExitStatusShell -eq 0 ]]; then
|
||
echo -e "更新scripts成功...\n"
|
||
[ ! -d $ScriptsDir/node_modules ] && Npm_Install scripts
|
||
[ -f $ScriptsDir/package.json ] && ScriptsDependNew=$(cat $ScriptsDir/package.json)
|
||
[[ "$ScriptsDependOld" != "$ScriptsDependNew" ]] && cd $ScriptsDir && Npm_Install scripts
|
||
Diff_Cron
|
||
if [ -s $ListJsDrop ]; then
|
||
Output_ListJs $ListJsDrop "失效"
|
||
Del_Cron
|
||
fi
|
||
if [ -s $ListJsAdd ]; then
|
||
Output_ListJs $ListJsAdd "新"
|
||
Add_Cron
|
||
fi
|
||
else
|
||
echo -e "更新scripts失败，请检查原因...\n"
|
||
fi
|
||
}
|
||
|
||
Diff_Cron() {
|
||
cat $ListCronRemote | grep -E "node.+j[drx]_\w+\.js" | perl -pe "s|.+(j[drx]_\w+)\.js.+|\1|" | sort -u >$ListRemoteTask
|
||
cat $ListCronCurrent | grep -E "$ShellJs j[drx]_\w+" | perl -pe "s|.*ID=(.*) $ShellJs (j[drx]_\w+)\.*|\2|" | sort -u >$ListCurrentTask
|
||
if [ -s $ListCurrentTask ]; then
|
||
grep -vwf $ListCurrentTask $ListRemoteTask >$ListJsAdd
|
||
else
|
||
cp -f $ListRemoteTask $ListJsAdd
|
||
fi
|
||
if [ -s $ListRemoteTask ]; then
|
||
grep -vwf $ListRemoteTask $ListCurrentTask >$ListJsDrop
|
||
else
|
||
cp -f $ListCurrentTask $ListJsDrop
|
||
fi
|
||
}
|
||
|
||
Del_Cron() {
|
||
if [ $AutoDelCron == true ] && [ -s $ListJsDrop ]; then
|
||
echo -e "开始尝试自动删除定时任务如下：\n"
|
||
cat $ListJsDrop
|
||
echo
|
||
JsDrop=$(cat $ListJsDrop)
|
||
for Cron in $JsDrop; do
|
||
local id=$(cat $ListCronCurrent | grep -E "js $Cron" | perl -pe "s|.*ID=(.*) js $Cron|\1|")
|
||
del_cron_api "$id"
|
||
done
|
||
crontab $ListCronCurrent
|
||
echo -e "成功删除失效的脚本与定时任务\n"
|
||
notify "删除 lxk0301 失效脚本" "$JsDrop"
|
||
fi
|
||
}
|
||
|
||
Add_Cron() {
|
||
if [ $AutoAddCron == true ] && [ -s $ListJsAdd ]; then
|
||
echo -e "开始尝试自动添加定时任务\n"
|
||
JsAdd=$(cat $ListJsAdd)
|
||
for Cron in $JsAdd; do
|
||
if [[ $Cron == jd_bean_sign ]]; then
|
||
echo "4 0,9 * * * $ShellJs $Cron" >>$ListCronCurrent
|
||
else
|
||
local name=$(cat "$ScriptsDir/$Cron.js" | grep -E "new Env\(" | perl -pe "s|(^.+)new Env\(\'*\"*(.+?)'*\"*\).+|\2|")
|
||
local param=$(cat $ListCronRemote | grep -E "\/$Cron\." | perl -pe "s|(^.+) node */scripts/(j[drx]_\w+)\.js.+|\1\:$ShellJs \2|")
|
||
add_cron_api "$param:$name"
|
||
fi
|
||
done
|
||
|
||
if [ $? -eq 0 ]; then
|
||
crontab $ListCronCurrent
|
||
echo -e "成功添加新的定时任务...\n"
|
||
notify "新增lxk0301脚本" "$JsAdd"
|
||
else
|
||
echo -e "添加新的定时任务出错，请手动添加...\n"
|
||
notify "尝试自动添加lxk0301以下新的定时任务出错，请手动添加：" "$JsAdd"
|
||
fi
|
||
fi
|
||
}
|
||
|
||
## 输出定时任务变化
|
||
Output_ListJs() {
|
||
local list=$1
|
||
local type=$2
|
||
if [ -s $list ]; then
|
||
echo -e "检测到有$type的定时任务：\n"
|
||
cat $list
|
||
echo
|
||
fi
|
||
}
|
||
|
||
#################################################################################################################################
|
||
echo -e "\n--------------------------------------------------------------\n"
|
||
echo -n "系统时间："
|
||
echo $(date "+%Y-%m-%d %H:%M:%S")
|
||
if [ "${TZ}" = "UTC" ]; then
|
||
echo -n "北京时间："
|
||
echo $(date -d "8 hour" "+%Y-%m-%d %H:%M:%S")
|
||
fi
|
||
echo -e "\nJS脚本目录：$ScriptsDir\n"
|
||
echo -e "--------------------------------------------------------------\n"
|
||
|
||
. $ShellDir/shell/api.sh
|
||
get_token
|
||
|
||
Import_Conf
|
||
|
||
# 更新shell
|
||
[ -f $ShellDir/package.json ] && PanelDependOld=$(cat $ShellDir/package.json)
|
||
Git_Pull_Shell
|
||
Git_Pull_Shell_Next
|
||
|
||
## 更新scripts
|
||
[ -f $ScriptsDir/package.json ] && ScriptsDependOld=$(cat $ScriptsDir/package.json)
|
||
Git_Pull_Scripts
|
||
Git_Pull_Scripts_Next
|
||
|
||
## 调用用户自定义的extra.sh
|
||
if [[ $EnableExtraShell == true ]]; then
|
||
if [ -f $ExtraShell ]; then
|
||
. $ExtraShell
|
||
else
|
||
echo -e "$ExtraShell 文件不存在，跳过执行DIY脚本...\n"
|
||
fi
|
||
fi
|