#!/bin/bash
#
#***************************************************************************
# Author: liwanggui
# Email: liwanggui@163.com
# Date: 2023-04-17
# FileName: install-aria2.sh
# Description: This is a test script
# Copyright (C): 2023 All rights reserved
#***************************************************************************
#

CONFIG_DIR=/etc/aria2
CONFIG_NAME=${CONFIG_DIR}/aria2.conf
LOG_DIR=/var/log/aria2
DL_DIR=/opt/download

config_aria2 () {
    local flag=n
    [ -d $CONFIG_DIR ] || mkdir $CONFIG_DIR
    [ -d $LOG_DIR ] || mkdir $LOG_DIR
    [ -f ${CONFIG_DIR}/aria2.session ] || touch ${CONFIG_DIR}/aria2.session

    if [ -f $CONFIG_NAME ]; then
        echo "Warring: The aria2 configuration already exists. ($CONFIG_NAME)"
        read -p "overwrite the original configuration? [Y/N]" flag
        if [[ $flag == "n" ]] || [[ $flag == "N" ]]; then
            echo "Abort!"
            exit 1
        fi
        mv $CONFIG_NAME ${CONFIG_NAME}-bak-$(date '+%Y%m%d%H%M%S')
    fi

    cat >$CONFIG_NAME<<EOF
dir=$DL_DIR
log=${LOG_DIR}/aria2.log
log-level=info

disable-ipv6=true

#打开rpc的目的是为了给web管理端用
enable-rpc=true
rpc-secret=thisisasecret
rpc-allow-origin-all=true
rpc-listen-all=true
rpc-listen-port=6800

#断点续传
continue=true
input-file=${CONFIG_DIR}/aria2.session
save-session=${CONFIG_DIR}/aria2.session

#最大同时下载任务数
max-concurrent-downloads=20
save-session-interval=120

# Http/FTP 相关
connect-timeout=120
#lowest-speed-limit=10K

#同服务器连接数
max-connection-per-server=10
#max-file-not-found=2
#最小文件分片大小, 下载线程数上限取决于能分出多少片, 对于小文件重要
min-split-size=10M

#单文件最大线程数, 路由建议值: 5
split=10
check-certificate=false
#http-no-cache=true
EOF

    cat >/lib/systemd/system/aria2.service<<EOF
[Unit]
Description=Aria2
After=network.target

[Service]
ExecStart=/usr/bin/aria2c --conf-path=/etc/aria2/aria2.conf
Restart=on-failure

[Install]
WantedBy=multi-user.target
EOF

}

main () {
    local install_bin
    if command -v apt &>/dev/null; then
        install_bin=apt
    elif command -v yum &>/dev/null; then
        install_bin=yum
        rpm -q epel-release >/dev/null || yum install -yq epel-release
    fi
    if [[ -z $install_bin ]]; then
        echo "Error: the current system is not supported."
        exit 1
    fi
    command -v aria2c &>/dev/null || $install_bin install -y aria2
    config_aria2
}

main