Files
ShellCrash/scripts/menus/5_task.sh

422 lines
14 KiB
Bash

#!/bin/sh
# Copyright (C) Juewuy
[ -n "$__IS_MODULE_5_TASK_LOADED" ] && return
__IS_MODULE_5_TASK_LOADED=1
# 通用工具
. "$CRASHDIR"/libs/set_config.sh
. "$CRASHDIR"/libs/set_cron.sh
load_lang 5_task
# 任务工具
set_cron() {
[ -z "$week" ] && week=*
[ -z "$hour" ] && hour=*
[ -z "$min" ] && min=0
comp_box "\033[33m$cron_time\033[0m$TASK5_RUN_TASK\033[36m$task_name\033[0m" \
"" \
"$TASK5_CONFIRM_ADD_CRON"
btm_box "1) $TASK5_YES" \
"0) $TASK5_NO"
read -r -p "$COMMON_INPUT> " res
if [ "$res" = '1' ]; then
task_txt="$min $hour * * $week $CRASHDIR/task/task.sh $task_id $cron_time$task_name"
cronset "$cron_time$task_name" "$task_txt"
msg_alert -t 0 "$TASK5_TASK_PREFIX$cron_time$task_name$TASK5_TASK_ADDED"
fi
unset week hour min
sleep 1
}
set_service() {
# 参数1代表要任务类型,参数2代表任务ID,参数3代表任务描述,参数4代表running任务cron时间
task_file="$CRASHDIR"/task/$1
[ -s "$task_file" ] && sed -i "/$3/d" "$task_file"
# 运行时每分钟执行的任务特殊处理
if [ "$1" = "running" ]; then
task_txt="$4 $CRASHDIR/task/task.sh $2 $3"
echo "$task_txt" >>"$task_file"
[ -n "$(pidof CrashCore)" ] && cronset "$3" "$task_txt"
else
echo "$CRASHDIR/task/task.sh $2 $3" >>"$task_file"
fi
content_line "$3】\033[32m$COMMON_SUCCESS\033[0m"
sleep 1
}
# 任务界面
#
# 自定义命令添加
task_user_add() {
while true; do
comp_box "\033[33m$TASK5_USER_ADD_HINT1\033[0m" \
"\033[36m$TASK5_USER_ADD_HINT2\033[0m" \
"$TASK5_USER_ADD_HINT3\033[32m${CRASHDIR}/task/task.user\033[0m$TASK5_USER_ADD_HINT4"
btm_box "\033[36m$TASK5_INPUT_CMD\033[0m" \
"$TASK5_OR_BACK"
read -r -p "$TASK5_INPUT> " script
if [ "$script" = 0 ]; then
break
elif [ -n "$script" ]; then
task_command=$script
comp_box "$TASK5_CHECK_INPUT\033[32m$task_command\033[0m"
# 获取本任务ID
task_max_id=$(awk -F '#' '{print $1}' "$CRASHDIR"/task/task.user 2>/dev/null | sort -n | tail -n 1)
[ -z "$task_max_id" ] && task_max_id=200
task_id=$((task_max_id + 1))
read -r -p "$TASK5_INPUT_REMARK> " txt
[ -n "$txt" ] && task_name=$txt || task_name="$TASK5_CUSTOM_TASK$task_id"
echo "$task_id#$task_command#$task_name" >>"$CRASHDIR"/task/task.user
msg_alert "\033[32m$TASK5_CUSTOM_ADDED\033[0m"
break
else
msg_alert "\033[31m$TASK5_INPUT_ERROR\033[0m"
fi
done
}
# 自定义命令删除
task_user_del() {
while true; do
if grep -Evq '^#' "$CRASHDIR/task/task.user" 2>/dev/null; then
comp_box "$TASK5_USER_DEL_HINT1" \
"$TASK5_USER_DEL_HINT2\033[32m${CRASHDIR}/task/task.user\033[0m"
grep -Ev '^#' "$CRASHDIR/task/task.user" 2>/dev/null |
awk -F '#' '{print $1") "$3}' |
while IFS= read -r line; do
content_line "$line"
done
btm_box "" \
"0) $COMMON_BACK"
read -r -p "$COMMON_INPUT> " num
if [ "$num" = 0 ]; then
break
elif [ -n "$num" ]; then
sed -i "/^$num#/d" "$CRASHDIR"/task/task.user 2>/dev/null
common_success
else
msg_alert "\033[31m$TASK5_INPUT_ERROR\033[0m"
fi
else
msg_alert "\033[33m$TASK5_NO_CUSTOM_TASK\033[0m"
break
fi
done
}
# 任务添加
task_add() {
while true; do
comp_box "\033[36m$TASK5_SELECT_ADD\033[0m"
# 输出任务列表
list=$(cat "$CRASHDIR"/task/task.list "$CRASHDIR"/task/task.user 2>/dev/null | grep -Ev '^(#|$)' | awk -F '#' '{print $3}')
list_box "$list"
btm_box "" \
"0) $COMMON_BACK"
read -r -p "$COMMON_INPUT> " num
case "$num" in
0)
break
;;
[1-9] | [1-9][0-9])
if [ "$num" -le "$(echo "$list" | wc -l)" ]; then
task_id=$(cat "$CRASHDIR"/task/task.list "$CRASHDIR"/task/task.user 2>/dev/null | grep -Ev '^(#|$)' | sed -n "$num p" | awk -F '#' '{print $1}')
task_name=$(cat "$CRASHDIR"/task/task.list "$CRASHDIR"/task/task.user 2>/dev/null | grep -Ev '^(#|$)' | sed -n "$num p" | awk -F '#' '{print $3}')
task_type
break
else
errornum
fi
;;
*)
errornum
;;
esac
done
}
# 任务删除
task_del() {
# 删除定时任务
cronset "$1"
# 删除条件任务
sed -i "/$1/d" "$CRASHDIR"/task/cron 2>/dev/null
sed -i "/$1/d" "$CRASHDIR"/task/bfstart 2>/dev/null
sed -i "/$1/d" "$CRASHDIR"/task/afstart 2>/dev/null
sed -i "/$1/d" "$CRASHDIR"/task/running 2>/dev/null
sed -i "/$1/d" "$CRASHDIR"/task/affirewall 2>/dev/null
}
# 任务条件选择菜单
task_type() {
comp_box "$TASK5_SELECT_COND\033[36m【$task_name】\033[0m$TASK5_SELECT_COND2"
btm_box "1) $TASK5_COND_1" \
"2) $TASK5_COND_2" \
"3) $TASK5_COND_3" \
"4) $TASK5_COND_4" \
"$TASK5_WARN_LINE1" \
"$TASK5_WARN_LINE2" \
"$TASK5_WARN_LINE3" \
"5) $TASK5_COND_5" \
"6) $TASK5_COND_6" \
"7) $TASK5_COND_7" \
"8) $TASK5_COND_8" \
"" \
"0) $COMMON_BACK"
read -r -p "$COMMON_INPUT> " num
case "$num" in
0)
return 1
;;
1)
comp_box "$TASK5_WEEK_HINT1" \
"$TASK5_WEEK_HINT2" \
"$TASK5_WEEK_HINT3"
read -r -p "$TASK5_WEEK_INPUT> " week
# week=`echo ${week/7/0}` # 把7换成0
read -r -p "$TASK5_HOUR_INPUT1> " hour
cron_time="$TASK5_CRON_WEEK$week$TASK5_CRON_WEEK2$hour$TASK5_OCLOCK"
# cron_time=`echo ${cron_time/周0/周日}` # 把0换成日
[ -n "$week" ] && [ -n "$hour" ] && set_cron
;;
2)
comp_box "$TASK5_DAY_HINT1" \
"$TASK5_DAY_HINT2"
read -r -p "$TASK5_HOUR_INPUT2> " hour
read -r -p "$TASK5_MIN_INPUT> " min
cron_time="$TASK5_CRON_DAY$hour$TASK5_POINT$min$TASK5_MINUTE"
[ -n "$min" ] && [ -n "$hour" ] && set_cron
;;
3)
line_break
read -r -p "$TASK5_EVERY_HOUR_INPUT> " num
hour="*/$num"
cron_time="$TASK5_EVERY$num$TASK5_HOUR"
[ -n "$hour" ] && set_cron
;;
4)
line_break
read -r -p "$TASK5_EVERY_MIN_INPUT> " num
min="*/$num"
cron_time="$TASK5_EVERY$num$TASK5_MIN"
[ -n "$min" ] && set_cron
;;
5)
set_service bfstart "$task_id" "$TASK5_BFSTART$task_name"
;;
6)
set_service afstart "$task_id" "$TASK5_AFSTART$task_name"
;;
7)
comp_box "$TASK5_RUNNING_HINT1" \
"$TASK5_RUNNING_HINT2"
read -r -p "$TASK5_RUNNING_INPUT> " num
if [ "$num" -lt 60 ]; then
min="$num"
cron_time="*/$min * * * *"
time_des="$min$TASK5_MIN"
else
hour="$((num / 60))"
cron_time="0 */$hour * * *"
time_des="$hour$TASK5_HOUR"
fi
[ -n "$cron_time" ] && set_service running "$task_id" "$TASK5_RUNNING_PREFIX$time_des$task_name" "$cron_time"
;;
8)
comp_box "$TASK5_AFFW_HINT"
"$TASK5_CONFIRM_CONTINUE"
btm_box "1) $TASK5_YES" \
"0) $TASK5_NO"
read -r -p "$COMMON_INPUT> " res
[ "$res" = 1 ] && set_service affirewall "$task_id" "$TASK5_AFFW_PREFIX$task_name"
;;
*)
errornum
;;
esac
}
# 任务管理列表
task_manager() {
while true; do
# 抽取并生成临时列表
cronload >"$TMPDIR"/task_cronlist
cat "$TMPDIR"/task_cronlist "$CRASHDIR"/task/running 2>/dev/null | sort -u | grep -oE "task/task.sh .*" | cut -d ' ' -f 2- >"$TMPDIR"/task_list
cat "$CRASHDIR"/task/bfstart "$CRASHDIR"/task/afstart "$CRASHDIR"/task/affirewall 2>/dev/null | cut -d ' ' -f 2- >>"$TMPDIR"/task_list
cat "$TMPDIR"/task_cronlist 2>/dev/null | sort -u | grep -oE " #.*" | grep -v "$TASK5_GUARD_WORD" | awk -F '#' '{print "0 '$TASK5_OLD_PREFIX'"$2}' >>"$TMPDIR"/task_list
sed -i '/^ *$/d' "$TMPDIR"/task_list
rm -rf "$TMPDIR"/task_cronlist
# 判断为空则返回
if [ ! -s "$TMPDIR"/task_list ]; then
msg_alert "\033[31m$TASK5_NONE_TO_MANAGE\033[36m"
break
else
comp_box "\033[33m$TASK5_ADDED_TASKS\033[0m"
list_box "$(cat "$TMPDIR"/task_list)"
separator_line "-"
btm_box "a) $TASK5_CLEAR_OLD" \
"d) $TASK5_CLEAR_ALL" \
"" \
"0) $COMMON_BACK"
read -r -p "$COMMON_INPUT> " num
case "$num" in
"" | 0)
break
;;
a)
task_del "#"
msg_alert "\033[31m$TASK5_OLD_CLEARED\033[36m"
;;
d)
task_del "task.sh"
msg_alert "\033[31m$TASK5_ALL_CLEARED\033[36m"
;;
[1-9] | [1-9][0-9])
task_txt=$(sed -n "$num p" "$TMPDIR"/task_list)
task_id=$(echo "$task_txt" | awk '{print $1}')
if [ "$task_id" = 0 ]; then
comp_box "$TASK5_OLD_NOT_SUPPORT"
btm_box "1) $TASK5_YES" \
"0) $TASK5_NO_BACK"
read -r -p "$COMMON_INPUT> " res
if [ "$res" = 1 ]; then
cronname=$(echo "$task_txt" | awk -F '-' '{print $2}')
cronset "$cronname"
sed -i "/$cronname/d" "$CRASHDIR"/task/cron 2>/dev/null
break
fi
else
task_des=$(echo "$task_txt" | awk '{print $2}')
task_name=$(cat "$CRASHDIR"/task/task.list "$CRASHDIR"/task/task.user 2>/dev/null | grep "$task_id" | awk -F '#' '{print $3}')
comp_box "$TASK5_CURRENT_TASK\033[36m$task_des\033[0m"
btm_box "1) $TASK5_EDIT_TASK" \
"2) $TASK5_DEL_TASK" \
"3) $TASK5_RUN_ONCE" \
"4) $TASK5_VIEW_RECORD" \
"" \
"0) $COMMON_BACK"
read -r -p "$COMMON_INPUT> " num
case "$num" in
"" | 0)
continue
;;
1)
task_type && task_del "$task_des"
;;
2)
task_del "$task_des"
common_success
;;
3)
task_command=$(cat "$CRASHDIR"/task/task.list "$CRASHDIR"/task/task.user 2>/dev/null | grep "$task_id" | awk -F '#' '{print $2}')
eval "$task_command" && task_res="$TASK5_RUN_OK" || task_res="$TASK5_RUN_FAIL"
msg_alert "\033[33m$TASK5_TASK_PREFIX$task_des$task_res\033[0m"
;;
4)
if cat "$TMPDIR"/ShellCrash.log | grep -q "$task_name"; then
line_break
echo "==========================================================="
cat "$TMPDIR"/ShellCrash.log | grep "$task_name"
echo "==========================================================="
else
msg_alert "\033[31m$TASK5_RECORD_NOT_FOUND\033[0m"
fi
;;
*)
errornum
;;
esac
fi
;;
*)
errornum
;;
esac
fi
done
}
# 任务推荐
task_recom() {
comp_box "\033[36m$TASK_RECOM_TITLE\033[0m" \
"" \
"$TASK_RECOM_ITEM_1" \
"$TASK_RECOM_ITEM_2" \
"$TASK_RECOM_ITEM_3"
btm_box "1) $TASK5_YES" \
"0) $TASK5_NO"
read -r -p "$COMMON_INPUT>" res
[ "$res" = 1 ] && {
line_break
separator_line "="
set_service running "106" "$TASK_RECOM_ITEM_1" "*/10 * * * *"
set_service afstart "107" "$TASK_RECOM_ITEM_2"
cronset "$TASK_RECOM_ITEM_3" "0 3 * * * ${CRASHDIR}/task/task.sh 103 $TASK_RECOM_ITEM_3" &&
content_line "$TASK_RECOM_ITEM_3】\033[32m$COMMON_SUCCESS\033[0m"
separator_line "="
}
}
# 任务菜单
task_menu() {
while true; do
# 检测并创建自定义任务文件
[ -f "$CRASHDIR"/task/task.user ] || echo "$TASK5_USER_FILE_HEADER" >"$CRASHDIR"/task/task.user
comp_box "\033[30;47m$TASK5_MENU_TITLE\033[0m"
btm_box "1) $TASK5_MENU_1" \
"2) $TASK5_MENU_2" \
"3) $TASK5_MENU_3" \
"4) $TASK5_MENU_4" \
"5) $TASK5_MENU_5" \
"6) $TASK5_MENU_6" \
"7) $TASK5_MENU_7" \
"" \
"0) $COMMON_BACK"
read -r -p "$COMMON_INPUT> " num
case "$num" in
"" | 0)
break
;;
1)
task_add
;;
2)
task_manager
rm -rf "$TMPDIR"/task_list
;;
3)
if cat "$TMPDIR"/ShellCrash.log | grep -q "$TASK5_TASK_GREP"; then
line_break
echo "==========================================================="
cat "$TMPDIR"/ShellCrash.log | grep "$TASK5_TASK_GREP"
echo "==========================================================="
else
msg_alert "\033[31m$TASK5_LOG_NOT_FOUND\033[0m"
fi
;;
4)
msg_alert "\033[36m$TASK5_PUSH_HINT\033[0m"
. "$CRASHDIR"/menus/8_tools.sh && log_pusher
;;
5)
task_user_add
;;
6)
task_user_del
;;
7)
task_recom
;;
*)
errornum
;;
esac
done
}