#!/bin/sh
# Copyright (C) Juewuy

[ -n "$__IS_MODULE_OVERRIDE" ] && return
__IS_MODULE_OVERRIDE=1

YAMLSDIR="$CRASHDIR"/yamls
JSONSDIR="$CRASHDIR"/jsons

# 配置文件覆写
override() {
    while true; do
        [ -z "$rule_link" ] && rule_link=1
        [ -z "$server_link" ] && server_link=1
        comp_box "\033[30;47m 欢迎使用配置文件覆写功能！\033[0m"
        content_line "2) 管理\033[36m自定义规则\033[0m"
        echo "$crashcore" | grep -q 'singbox' || {
            content_line "3) 管理\033[33m自定义节点\033[0m"
            content_line "4) 管理\033[36m自定义策略组\033[0m"
        }
        content_line "5) \033[32m自定义\033[0m高级功能"
        [ "$disoverride" != 1 ] && content_line "9) \033[33m禁用\033[0m配置文件覆写"
        content_line ""
        content_line "0) 返回上级菜单"
        separator_line "="
        read -r -p "请输入对应数字> " num
        case "$num" in
        "" | 0)
            break
            ;;
        2)
            setrules
            ;;
        3)
            setproxies
            ;;
        4)
            setgroups
            ;;
        5)
            if echo "$crashcore" | grep -q 'singbox'; then
                set_singbox_adv
            else
                set_clash_adv
            fi
            sleep 3
            ;;
        9)
            comp_box "\033[33m此功能可能会导致严重问题！启用后脚本中大部分功能都将禁用！！！\033[0m" \
                "如果你不是非常了解$crashcore的运行机制，切勿开启！\033[0m" \
                "\033[33m继续后如出现任何问题，请务必自行解决，一切提问恕不受理！\033[0m"
            sleep 2
            btm_box "1) 我确认遇到问题可以自行解决" \
                "0) 返回上级菜单"
            read -r -p "$COMMON_INPUT> " res
            [ "$res" = '1' ] && {
                disoverride=1
                if setconfig disoverride $disoverride; then
                    common_success
                else
                    common_failed
                fi
            }
            ;;
        *)
            errornum
            ;;
        esac
    done
}

# 自定义规则
setrules() {
    set_rule_type() {
        comp_box "\033[33m请选择规则类型：\033[0m"
        printf '%s\n' "$rule_type" |
            awk '{for (i = 1; i <= NF; i++) print i") " $i}' |
            while IFS= read -r line; do
                content_line "$line"
            done
        content_line ""
        content_line "0) 返回上级菜单"
        separator_line "="
        read -r -p "请输入对应数字> " num
        case "$num" in
        "" | 0) ;;
        [0-9]*)
            if [ "$num" -gt $(echo $rule_type | awk -F " " '{print NF}') ]; then
                errornum
            else
                rule_type_set=$(echo "$rule_type" | cut -d' ' -f"$num")
                comp_box "\033[33m请输入规则语句，\n可以是域名、泛域名、IP网段或者其他匹配规则类型的内容\033[0m"
                read -r -p "请输入对应规则> " rule_state_set
                if [ -n "$rule_state_set" ]; then
                    set_group_type
                else
                    errornum
                fi
            fi
            ;;
        *)
            errornum
            ;;
        esac
    }

    set_group_type() {
        comp_box "\033[36m请选择具体规则\033[0m" \
            "\033[33m此处规则读取自现有配置文件，如果你后续更换配置文件时运行出错，请尝试重新添加\033[0m"
        printf '%s\n' "$rule_group" |
            awk -F '#' '{for (i = 1; i <= NF; i++) print i") " $i}' |
            while IFS= read -r line; do
                content_line "$line"
            done
        content_line ""
        content_line "0) 返回上级菜单"
        separator_line "="
        read -r -p "请输入对应数字> " num
        case "$num" in
        "" | 0) ;;
        [0-9]*)
            if [ "$num" -gt "$(echo "$rule_group" | awk -F "#" '{print NF}')" ]; then
                errornum
            else
                rule_group_set=$(echo "$rule_group" | cut -d'#' -f"$num")
                rule_all="- ${rule_type_set},${rule_state_set},${rule_group_set}"
                echo "IP-CIDR SRC-IP-CIDR IP-CIDR6" | grep -q -- "$rule_type_set" && rule_all="${rule_all},no-resolve"
                echo "$rule_all" >>"$YAMLSDIR"/rules.yaml
                msg_alert "\033[32m添加成功！\033[0m"
            fi
            ;;
        *)
            errornum
            ;;
        esac
    }

    del_rule_type() {
        while true; do
            comp_box "输入对应数字即可移除相应规则："
            sed -i '/^ *$/d; /^#/d' "$YAMLSDIR"/rules.yaml
            awk -F '#' '!/^#/ {print NR") "$1 $2 $3}' "$YAMLSDIR/rules.yaml" |
                while IFS= read -r line; do
                    content_line "$line"
                done
            content_line ""
            content_line "0) 返回上级菜单"
            separator_line "="
            read -r -p "请输入对应数字> " num
            case "$num" in
            "" | 0)
                break
                ;;
            *)
                if [ "$num" -le "$(wc -l <"$YAMLSDIR"/rules.yaml)" ]; then
                    if sed -i "${num}d" "$YAMLSDIR"/rules.yaml; then
                        common_success
                    else
                        common_failed
                    fi
                    sleep 1
                else
                    errornum
                fi
                ;;
            esac
        done
    }

    get_rule_group() {
        . "$CRASHDIR"/libs/web_save.sh
        get_save http://127.0.0.1:${db_port}/proxies | sed 's/:{/!/g' | awk -F '!' '{for(i=1;i<=NF;i++) print $i}' | grep -aE '"Selector|URLTest|LoadBalance"' | grep -aoE '"name":.*"now":".*",' | awk -F '"' '{print "#"$4}' | tr -d '\n'
    }

    while true; do
        comp_box "\033[33m你可以在这里快捷管理自定义规则\033[0m" \
            "如需批量操作，请手动编辑：\033[36m $YAMLSDIR/rules.yaml\033[0m" \
            "\033[33msingbox和clash共用此处规则，可无缝切换！\033[0m" \
            "大量规则请尽量使用rule-set功能添加，\n\033[31m此处过量添加可能导致启动卡顿！\033[0m"
        content_line "1) 新增自定义规则"
        content_line "2) 移除自定义规则"
        content_line "3) 清空规则列表"
        echo "$crashcore" | grep -q 'singbox' || content_line "4) 配置节点绕过：	\033[36m$proxies_bypass\033[0m"
        content_line ""
        content_line "0) 返回上级菜单"
        separator_line "="
        read -r -p "请输入对应数字> " num
        case "$num" in
        "" | 0)
            break
            ;;
        1)
            rule_type="DOMAIN-SUFFIX DOMAIN-KEYWORD IP-CIDR SRC-IP-CIDR DST-PORT SRC-PORT GEOIP GEOSITE IP-CIDR6 DOMAIN PROCESS-NAME"
            rule_group="DIRECT#REJECT$(get_rule_group)"
            set_rule_type
            ;;
        2)
            if [ -s "$YAMLSDIR"/rules.yaml ]; then
                del_rule_type
            else
                msg_alert "请先添加自定义规则！"
            fi
            ;;
        3)
            comp_box "是否确认清空全部自定义规则？"
            btm_box "1) 是" \
                "0) 否，返回上级菜单"
            read -r -p "$COMMON_INPUT> " res
            if [ "$res" = "1" ]; then
                if sed -i '/^\s*[^#]/d' "$YAMLSDIR"/rules.yaml; then
                    common_success
                else
                    common_failed
                fi
            fi
            ;;
        4)
            if [ "$proxies_bypass" = "OFF" ]; then
                comp_box "\033[33m本功能会自动将当前配置文件中的节点域名或IP设置为直连规则以防止出现双重流量！\033[0m" \
                    "\033[33m请确保下游设备使用的节点与ShellCrash中使用的节点相同，否则无法生效！\033[0m" \
                    "" \
                    "是否启用节点绕过？"
                btm_box
                "1) 是" \
                    "0) 否，返回上级菜单"
                read -r -p "$COMMON_INPUT> " res
                if [ "$res" = "1" ]; then
                    proxies_bypass=ON
                else
                    continue
                fi
            else
                proxies_bypass=OFF
            fi

            if setconfig proxies_bypass "$proxies_bypass"; then
                common_success
            else
                common_failed
            fi
            ;;
        *)
            errornum
            ;;
        esac
    done
}

# 自定义clash策略组
setgroups() {
    set_group_type() {
        comp_box "\033[33m注意策略组名称必须和【自定义规则】或【自定义节点】功能中指定的策略组一致！\033[0m" \
            "\033[33m建议先创建策略组，之后可在【自定义规则】或【自定义节点】功能中智能指定\033[0m" \
            "\033[33m如需在当前策略组下添加节点，请手动编辑$YAMLSDIR/proxy-groups.yaml\033[0m"
        btm_box "\033[32m请直接输入自定义策略组名称\033[0m\n（不支持纯数字且不要包含特殊字符！）" \
            "或输入 0 返回上级菜单"
        read -r -p "请输入> " new_group_name

        comp_box "\033[32m请选择策略组【$new_group_name】的类型：\033[0m"
        printf '%s\n' "$group_type_cn" |
            awk '{for (i = 1; i <= NF; i++) print i") " $i}' |
            while IFS= read -r line; do
                content_line "$line"
            done
        separator_line "="
        read -r -p "请输入对应数字> " num
        new_group_type=$(echo "$group_type" | awk '{print $'"$num"'}')
        if [ "$num" = "1" ]; then
            unset new_group_url interval
        else
            comp_box "请输入测速地址" \
                "或直接回车使用默认地址：https://www.gstatic.com/generate_204"
            read -r -p "请输入> " new_group_url
            [ -z "$new_group_url" ] && new_group_url=https://www.gstatic.com/generate_204
            new_group_url="url: '$new_group_url'"
            interval="interval: 300"
        fi
        set_group_add
        # 添加自定义策略组
        cat >>"$YAMLSDIR"/proxy-groups.yaml <<EOF
  - name: $new_group_name
    type: $new_group_type
    $new_group_url
    $interval
    proxies:
     - DIRECT
EOF
        sed -i "/^ *$/d" "$YAMLSDIR"/proxy-groups.yaml
        msg_alert "\033[32m添加成功！\033[0m"

    }

    set_group_add() {
        comp_box "\033[36m请选择想要将本策略添加到的策略组\033[0m" \
            "\033[32m如需添加到多个策略组，请一次性输入多个数字并用空格隔开\033[0m"
        printf '%s\n' "$proxy_group" |
            awk -F '#' '{for (i = 1; i <= NF; i++) print i") " $i}' |
            while IFS= read -r line; do
                content_line "$line"
            done
        content_line ""
        content_line "0) 跳过添加"
        separator_line "="
        read -r -p "请输入对应数字（多个用空格分隔）> " char
        case "$char" in
        "" | 0) ;;
        *)
            for num in $char; do
                rule_group_set=$(echo "$proxy_group" | cut -d'#' -f"$num")
                rule_group_add="${rule_group_add}#${rule_group_set}"
            done
            if [ -n "$rule_group_add" ]; then
                new_group_name="$new_group_name$rule_group_add"
                unset rule_group_add
            else
                errornum
            fi
            ;;
        esac
    }

    while true; do
        comp_box "\033[33m你可以在这里快捷管理自定义策略组\033[0m" \
            "\033[36m如需修改或批量操作，请手动编辑：$YAMLSDIR/proxy-groups.yaml\033[0m"
        content_line "1) 添加自定义策略组"
        content_line "2) 查看自定义策略组"
        content_line "3) 清空自定义策略组"
        content_line ""
        content_line "0) 返回上级菜单"
        separator_line "="
        read -r -p "请输入对应数字> " num
        case "$num" in
        "" | 0)
            break
            ;;
        1)
            group_type="select url-test fallback load-balance"
            group_type_cn="手动选择 自动选择 故障转移 负载均衡"
            proxy_group="$(cat "$YAMLSDIR"/proxy-groups.yaml "$YAMLSDIR"/config.yaml 2>/dev/null | sed "/#自定义策略组开始/,/#自定义策略组结束/d" | grep -Ev '^#' | grep -o '\- name:.*' | sed 's/#.*//' | sed 's/- name: /#/g' | tr -d '\n' | sed 's/#//')"
            set_group_type
            ;;
        2)
            line_break
            echo "==========================================================="
            cat "$YAMLSDIR"/proxy-groups.yaml
            echo ""
            echo "==========================================================="
            ;;
        3)
            comp_box "是否确认清空全部自定义策略组？"
            btm_box "1) 是" \
                "0) 否，返回上级菜单"
            read -r -p "$COMMON_INPUT> " res
            if [ "$res" = "1" ]; then
                if echo '#用于添加自定义策略组' >"$YAMLSDIR"/proxy-groups.yaml; then
                    common_success
                else
                    common_failed
                fi
            fi
            ;;
        *)
            errornum
            ;;
        esac
    done
}

# 自定义clash节点
setproxies() {

    set_proxy_type() {
        while true; do
            comp_box "\033[33m注意\n节点格式必须是单行、不包括括号、“name:”为开头，例如：\033[0m" \
                "\033[36m【name: \"test\", server: 192.168.1.1, port: 12345, type: socks5, udp: true】\033[0m" \
                "更多写法请参考：\033[32mhttps://juewuy.github.io/\033[0m"
            btm_box "\033[36m请直接输入自定义节点\033[0m" \
                "或输入 0 返回上级菜单"
            read -r -p "请输入> " proxy_state_set
            if [ "$proxy_state_set" = 0 ]; then
                break
            elif echo "$proxy_state_set" | grep -q "#"; then
                msg_alert "\033[33m绝对禁止包含【#】号！\033[0m"
            elif echo "$proxy_state_set" | grep -Eq "^name:"; then
                set_group_add
            else
                errornum
            fi
        done
    }

    set_group_add() {
        comp_box "\033[36m请选择想要将节点添加到的策略组\033[0m" \
            "\033[32m如需添加到多个策略组，请一次性输入多个数字并用空格隔开\033[0m" \
            "\033[33m如需自定义策略组，请先使用【管理自定义策略组功能】添加\033[0m"
        printf '%s\n' "$proxy_group" |
            awk -F '#' '{for (i = 1; i <= NF; i++) print i") " $i}' |
            while IFS= read -r line; do
                content_line "$line"
            done
        content_line ""
        content_line "0) 返回上级菜单"
        separator_line "="
        read -r -p "请输入对应数字（多个用空格分隔）> " char
        case "$char" in
        "" | 0) ;;
        *)
            for num in $char; do
                rule_group_set=$(echo "$proxy_group" | cut -d'#' -f"$num")
                rule_group_add="${rule_group_add}#${rule_group_set}"
            done
            if [ -n "$rule_group_add" ]; then
                echo "- {$proxy_state_set}$rule_group_add" >>"$YAMLSDIR"/proxies.yaml
                msg_alert "\033[32m添加成功！\033[0m"
                unset rule_group_add
            else
                errornum
            fi
            ;;
        esac
    }

    while true; do
        comp_box "\033[33m你可以在这里快捷管理自定义节点\033[0m" \
            "\033[36m如需批量操作，请手动编辑：$YAMLSDIR/proxies.yaml\033[0m"
        content_line "1) 添加自定义节点"
        content_line "2) 管理自定义节点"
        content_line "3) 清空自定义节点"
        content_line "4) 配置节点绕过：	\033[36m$proxies_bypass\033[0m"
        content_line ""
        content_line "0) 返回上级菜单"
        separator_line "="
        read -r -p "请输入对应数字> " num
        case "$num" in
        "" | 0)
            break
            ;;
        1)
            proxy_type="DOMAIN-SUFFIX DOMAIN-KEYWORD IP-CIDR SRC-IP-CIDR DST-PORT SRC-PORT GEOIP GEOSITE IP-CIDR6 DOMAIN MATCH"
            proxy_group="$(cat "$YAMLSDIR"/proxy-groups.yaml "$YAMLSDIR"/config.yaml 2>/dev/null | sed "/#自定义策略组开始/,/#自定义策略组结束/d" | grep -Ev '^#' | grep -o '\- name:.*' | sed 's/#.*//' | sed 's/- name: /#/g' | tr -d '\n' | sed 's/#//')"
            set_proxy_type
            ;;
        2)
            sed -i '/^ *$/d' "$YAMLSDIR"/proxies.yaml 2>/dev/null
            if [ -s "$YAMLSDIR"/proxies.yaml ]; then
                comp_box "\033[33m输入节点对应数字可以移除对应节点\033[0m" \
                    "当前已添加的自定义节点为："
                grep -Ev '^#' "$YAMLSDIR/proxies.yaml" |
                    awk -F '[,}]' '{print NR") " $1 " " $NF}' |
                    sed 's/- {//g' |
                    while IFS= read -r line; do
                        content_line "$line"
                    done
                content_line ""
                content_line "0) 返回上级菜单"
                separator_line "="
                read -r -p "请输入对应数字> " num
                if [ "$num" = 0 ]; then
                    continue
                elif [ "$num" -le $(cat "$YAMLSDIR"/proxies.yaml | grep -Ev '^#' | wc -l) ]; then
                    if sed -i "$num{/^\s*[^#]/d}" "$YAMLSDIR"/proxies.yaml; then
                        common_success
                    else
                        common_failed
                    fi
                else
                    errornum
                fi
            else
                msg_alert "请先添加自定义节点！"
            fi
            ;;
        3)
            comp_box "是否确认清空全部自定义节点？"
            btm_box "1) 是" \
                "0) 否，返回上级菜单"
            read -r -p "$COMMON_INPUT> " res
            if [ "$res" = "1" ]; then
                if sed -i '/^\s*[^#]/d' "$YAMLSDIR"/proxies.yaml 2>/dev/null; then
                    common_success
                else
                    common_failed
                fi
            else
                continue
            fi
            ;;
        4)
            if [ "$proxies_bypass" = "OFF" ]; then
                comp_box "\033[33m本功能会自动将当前配置文件中的节点域名或IP设置为直连规则以防止出现双重流量！\033[0m" \
                    "\033[33m请确保下游设备使用的节点与ShellCrash中使用的节点相同，否则无法生效！\033[0m" \
                    "" \
                    "是否确定启用节点绕过："
                btm_box "1) 是" \
                    "0) 否，返回上级菜单"
                read -r -p "$COMMON_INPUT> " res
                if [ "$res" = "1" ]; then
                    proxies_bypass=ON
                else
                    continue
                fi
            else
                proxies_bypass=OFF
            fi
            setconfig proxies_bypass "$proxies_bypass"
            sleep 1
            setrules
            break
            ;;
        *)
            errornum
            ;;
        esac
    done
}

# 自定义clash高级规则
set_clash_adv() {
    [ ! -f "$YAMLSDIR"/user.yaml ] && cat >"$YAMLSDIR"/user.yaml <<EOF
#用于编写自定义设定(可参考https://lancellc.gitbook.io/clash/clash-config-file/general 或 https://docs.metacubex.one/function/general)
#端口之类请在脚本中修改，否则不会加载
#port: 7890
EOF
    [ ! -f "$YAMLSDIR"/others.yaml ] && cat >"$YAMLSDIR"/others.yaml <<EOF
#用于编写自定义的锚点、入站、proxy-providers、sub-rules、rule-set、script等功能
#可参考 https://github.com/MetaCubeX/Clash.Meta/blob/Meta/docs/config.yaml 或 https://lancellc.gitbook.io/clash/clash-config-file/an-example-configuration-file
#此处内容会被添加在配置文件的“proxy-group：”模块的末尾与“rules：”模块之前的位置
#例如：
#proxy-providers:
#rule-providers:
#sub-rules:
#tunnels:
#script:
#listeners:
EOF

    comp_box "\033[32m已经创建自定义设定文件：$YAMLSDIR/user.yaml ！\033[0m" \
        "\033[33m可用于编写自定义的DNS等功能\033[0m" \
        "" \
        "\033[32m已经创建自定义功能文件：$YAMLSDIR/others.yaml ！\033[0m" \
        "\033[33m可用于编写自定义的锚点、入站、proxy-providers、rule-set、sub-rules、script等功能\033[0m"

    btm_box "Windows下请使用\033[33mWinSCP软件\033[0m进行编辑！\033[0m" \
        "MacOS下请使用\033[33mSecureFX软件\033[0m进行编辑！\033[0m" \
        "Linux可使用\033[33mvim\033[0m进行编辑（路由设备若不显示中文请勿使用）！\033[0m"
}

# s自定义singbox配置文件
set_singbox_adv() {
    comp_box "支持覆盖脚本设置的模块有：\033[0m" \
        "\033[36mlog dns ntp certificate experimental\033[0m" \
        "支持与内置功能合并（但不可冲突）的模块有：\033[0m" \
        "\033[36mendpoints inbounds outbounds providers route services\033[0m" \
        "将相应json文件放入\033[33m$JSONSDIR\033[0m目录后即可在启动时自动加载" \
        "" \
        "使用前请务必参考配置教程：\033[32;4m https://juewuy.github.io/nWTjEpkSK \033[0m"
}