#!/bin/bash
# 在ssh中执行：bash <(curl -s https://static.imily.net/sh/frpc/install.sh) "$(pwd)"
if [ -n "$1" ]; then
    cd $1
else
    cd $(dirname $0)
fi

# 判断架构，x86_64或arm
arch=$(uname -m)
if [ "$arch" == "x86_64" ]; then
    arch="x64"
elif [[ "$arch" == arm* ]]; then
    arch="arm"
elif [[ "$arch" == "aarch64" ]]; then
    arch="arm"
fi


# 下载frpc_arm_v6.zip 或 frpc_x64_v6.zip
echo "download frpc_${arch}_v6.zip"
curl -o ./frpc_${arch}.zip https://static.imily.net/sh/frpc/frpc_${arch}_v6.zip

# 解压frpc_arm.zip 或 frpc_x86_64.zip 到当前文件夹 如果存在则覆盖
unzip -o ./frpc_${arch}.zip -d ./frpc
rm -rf ./frpc_${arch}.zip

# 转到frpc文件夹
cd ./frpc
chmod +x frpc

# 提示用户输入映射名称
echo "请输入映射名称(英文)："
read -e -p "(默认名称: hd_xm2):" server_name

# 如果用户输入为空，则使用默认名称
if [ -z "$server_name" ]; then
    server_name="hd_xm2"
fi

echo "请输入本地端口号："
read -e -p "(默认端口号: 22):" local_port

# 如果用户输入为空，则使用默认端口号
if [ -z "$local_port" ]; then
    local_port="22"
fi

echo "请输入远程端口号："
read -e -p "(默认端口号: 9122):" remote_port

# 如果用户输入为空，则使用默认端口号
if [ -z "$remote_port" ]; then
    remote_port="9122"
fi


echo "请输入auth.token："
read -e -p "(默认token: 123):" token

# 如果用户输入为空，则使用默认token
if [ -z "$token" ]; then
    token="123"
fi


# 创建frpc.toml文件
cat > frpc.toml <<EOF
serverAddr = "117.158.104.55"
serverPort = 9100
auth.additionalScopes = ["NewWorkConns"]
auth.method = "token"
auth.token = "${token}"
log.to = "./frpc.log"

[[proxies]]
name = "${server_name}"
type = "tcp"
localIP = "127.0.0.1"
localPort = ${local_port}
remotePort = ${remote_port}

EOF

# 如果存在frpc进程，杀死
pid=$(ps -ef | grep frpc | grep -v grep | awk '{print $2}')
if [ -n "$pid" ]; then
    kill -9 $pid
fi


# 下载frpc.service文件到当前文件夹
curl -o ./frpc.service https://static.imily.net/sh/frpc/frpc.service

# 修改frpc.service文件，替换文件中的所有{{work_dir}}为当前目录，注意不是替换work_dir, 而是{{work_dir}}
sed -i "s/{{work_dir}}/$(pwd | sed 's/\//\\\//g')/g" ./frpc.service

# 复制到/etc/systemd/system
cp ./frpc.service /etc/systemd/system/

# 重新加载systemd配置
systemctl daemon-reload

# 启动frpc服务
systemctl enable frpc.service
systemctl start frpc.service

echo "frpc安装完成"