ssh2tg.sh

#!/bin/bash
# 检查是否安装jq模块，若未安装则自动安装
if ! command -v jq &> /dev/null; then
    echo "jq is not installed. Installing jq..."
    if command -v apt-get &> /dev/null; then
        sudo apt-get update
        sudo apt-get install -y jq
    elif command -v yum &> /dev/null; then
        sudo yum install -y epel-release
        sudo yum install -y jq
    else
        echo "Error: Unable to install jq. Please install jq manually."
        exit 1
    fi
fi
# 登录计数文件路径
count_file="/var/log/ssh_login_count.log"

# 检查计数文件是否存在，如果不存在则创建
if [ ! -f $count_file ]; then
    touch $count_file
    count=0   # 设置默认初始值为0
else
    # 从计数文件中读取当前登录次数
    count=$(cat $count_file)
fi

# 显示当前登录次数，并将次数加一
echo "当前登录次数: $count"
count=$((count+1))

# 将新的登录次数写入计数文件
echo $count > $count_file




# Telegram Bot相关信息
TELEGRAM_BOT_TOKEN=replacetoken
CHAT_ID=replaceid

# 获取登录信息
IP=$(echo $SSH_CONNECTION | awk '{print $1}')
TIME=$(date +"%Y年%m月%d日 %H:%M:%S")
# 查询IP地址对应的地区信息
#LOCATION=$(curl -s https://ipapi.co/$IP/json/ | jq -r '.city')
 LOCATION=$(curl -s "http://opendata.baidu.com/api.php?query=$IP&co=&resource_id=6006&oe=utf8&format=json" | jq -r '.data[0].location')
# 获取当前用户名
 USERNAME=$(whoami)
 HOSTNAME=$(hostname)
# 发送Telegram消息
MESSAGE="登录信息：
登录机器：$HOSTNAME
登录次数：$count
登录名：$USERNAME
登录IP：$IP
登录时间：$TIME
登录地区：$LOCATION"

curl -s -X POST "https://api.telegram.org/bot$TELEGRAM_BOT_TOKEN/sendMessage" -d "chat_id=$CHAT_ID&text=$MESSAGE" >> /root/nohupssh2tg.out 2>&1 & disown
echo "您此台机器最后5次的登录记录如下！"
cat nohupssh2tg.out |jq '.result.text'|tail -5|sort -r|tr -d 'n''"'|sed 's/\\//g'|nl -w 2 -s '、'