telegram-bot-bash/bashbot.sh

207 lines
6.7 KiB
Bash
Raw Normal View History

#!/bin/bash
2015-07-10 07:43:08 +02:00
# bashbot, the Telegram bot written in bash.
2016-01-05 17:01:28 +01:00
# Written by @topkecleon, Juan Potato (@awkward_potato), Lorenzo Santina (BigNerd95) and Daniil Gentili (danog)
# https://github.com/topkecleon/telegram-bot-bash
# Depends on ./JSON.sh (http://github.com/dominictarr/./JSON.sh),
# which is MIT/Apache-licensed
# And on tmux (https://github.com/tmux/tmux),
# which is BSD-licensed
2015-07-10 07:43:08 +02:00
# This file is public domain in the USA and all free countries.
# If you're in Europe, and public domain does not exist, then haha.
TOKEN='tokenhere'
2015-07-10 07:43:08 +02:00
URL='https://api.telegram.org/bot'$TOKEN
2016-01-06 17:11:56 +01:00
FORWARD_URL=$URL'/forwardMessage'
2015-11-25 04:07:39 +01:00
MSG_URL=$URL'/sendMessage'
PHO_URL=$URL'/sendPhoto'
2016-01-06 17:11:56 +01:00
AUDIO_URL=$URL'/sendAudio'
DOCUMENT_URL=$URL'/sendDocument'
STICKER_URL=$URL'/sendSticker'
VIDEO_URL=$URL'/sendVideo'
VOICE_URL=$URL'/sendVoice'
LOCATION_URL=$URL'/sendLocation'
ACTION_URL=$URL'/sendChatAction'
2016-01-05 18:26:27 +01:00
FILE_URL='https://api.telegram.org/file/bot'$TOKEN'/'
2015-07-10 07:43:08 +02:00
UPD_URL=$URL'/getUpdates?offset='
2016-01-06 17:11:56 +01:00
GET_URL=$URL'/getFile'
2015-07-10 07:43:08 +02:00
OFFSET=0
2016-01-06 17:11:56 +01:00
declare -A USER URLS CONTACT LOCATION
2015-07-10 07:43:08 +02:00
send_message() {
2016-01-05 17:01:28 +01:00
local chat="$1"
local text="$(echo "$2" | sed 's/ mykeyboardstartshere.*//g;s/ myimagelocationstartshere.*//g')"
local keyboard="$(echo "$2" | sed '/mykeyboardstartshere /!d;s/.*mykeyboardstartshere //g;s/ myimagelocationstartshere.*//g')"
local image="$(echo "$2" | sed '/myimagelocationstartshere /!d;s/.*myimagelocationstartshere //g;s/ mykeyboardstartshere.*//g;')"
if [ "$keyboard" != "" ]; then
2016-01-05 17:01:28 +01:00
send_keyboard "$chat" "$text" "$keyboard"
local sent=y
fi
if [ "$image" != "" ]; then
send_photo "$chat" "$image"
local sent=y
fi
if [ "$sent" != "y" ];then
res=$(curl -s "$MSG_URL" -F "chat_id=$chat" -F "text=$text")
2016-01-05 17:01:28 +01:00
fi
2016-01-03 02:14:07 +01:00
}
2016-01-05 17:01:28 +01:00
2016-01-03 02:14:07 +01:00
send_keyboard() {
2016-01-05 17:01:28 +01:00
local chat="$1"
local text="$2"
shift 2
local keyboard=init
for f in $*;do local keyboard="$keyboard, [\"$f\"]";done
local keyboard=${keyboard/init, /}
2016-01-05 17:01:28 +01:00
res=$(curl -s "$MSG_URL" --header "content-type: multipart/form-data" -F "chat_id=$chat" -F "text=$text" -F "reply_markup={\"keyboard\": [$keyboard],\"one_time_keyboard\": true}")
2015-11-25 04:07:39 +01:00
}
send_photo() {
2016-01-05 17:01:28 +01:00
res=$(curl -s "$PHO_URL" -F "chat_id=$1" -F "photo=@$2")
2015-07-10 09:32:34 +02:00
}
2016-01-06 17:11:56 +01:00
send_audio() {
}
get_file() {
[ "$1" != "" ] && echo $FILE_URL$(curl -s "$GET_URL" -F "file_id=$1" | ./JSON.sh -s | egrep '\["result","file_path"\]' | cut -f 2 | cut -d '"' -f 2)
}
send_file() {
}
2016-01-05 17:01:28 +01:00
startproc() {
local copname="$1"
2016-01-06 17:11:56 +01:00
local USER="$2"
2016-01-05 17:01:28 +01:00
mkdir -p "$copname"
mkfifo $copname/out
2016-01-06 17:11:56 +01:00
tmux new-session -d -s $copname "./question 2>&1>$copname/out"
2016-01-05 17:01:28 +01:00
local pid=$(ps aux | sed '/tmux/!d;/'$copname'/!d;/sed/d;s/'$USER'\s*//g;s/\s.*//g')
echo $pid>$copname/pid
while ps aux | grep -v grep | grep -q $pid;do
read -t 10 line
2016-01-06 17:11:56 +01:00
[ "$line" != "" ] && send_message "${USER[ID]}" "$line"
2016-01-05 17:01:28 +01:00
line=
done <$copname/out
2016-01-02 21:51:45 +01:00
}
2016-01-06 17:11:56 +01:00
2016-01-03 02:14:07 +01:00
inproc() {
2016-01-05 17:01:28 +01:00
local copname="$1"
local copid="$2"
2016-01-05 18:26:27 +01:00
local MESSAGE="$3"
2016-01-05 17:01:28 +01:00
shift 2
2016-01-05 18:26:27 +01:00
tmux send-keys -t $copname "$MESSAGE
2016-01-05 17:01:28 +01:00
"
ps aux | grep -v grep | grep -q "$copid" || { rm -r $copname; };
2016-01-03 02:14:07 +01:00
}
process_client() {
2016-01-06 17:11:56 +01:00
local copname="CO${USER[ID]}"
2016-01-05 17:01:28 +01:00
local copidname="$copname/pid"
local copid="$(cat $copidname 2>/dev/null)"
if [ "$copid" = "" ]; then
2016-01-02 21:51:45 +01:00
case $MESSAGE in
2016-01-05 17:01:28 +01:00
'/question')
2016-01-06 17:11:56 +01:00
startproc "$copname" "${USER[ID]}"&
2016-01-05 17:01:28 +01:00
;;
'/info')
2016-01-06 17:11:56 +01:00
send_message "${USER[ID]}" "This is bashbot, the Telegram bot written entirely in bash."
2016-01-05 17:01:28 +01:00
;;
2016-01-05 17:05:26 +01:00
'/start')
2016-01-06 17:11:56 +01:00
send_message "${USER[ID]}" "This is bashbot, the Telegram bot written entirely in bash.
2016-01-05 17:05:26 +01:00
Features background tasks and interactive chats.
Can serve as an interface for cli programs.
Currently can send messages, custom keyboards and photos.
2016-01-05 17:14:47 +01:00
Available commands:
/start: Start bot and get this message.
/info: Get shorter info message about this bot.
/question: Start interactive chat.
/cancel: Cancel any currently running interactive chats.
2016-01-05 17:05:26 +01:00
Written by @topkecleon, Juan Potato (@awkward_potato), Lorenzo Santina (BigNerd95) and Daniil Gentili (danog)
https://github.com/topkecleon/telegram-bot-bash
2016-01-05 17:05:26 +01:00
"
;;
2016-01-05 17:01:28 +01:00
*)
2016-01-06 17:11:56 +01:00
send_message "${USER[ID]}" "$MESSAGE"
2016-01-02 21:51:45 +01:00
esac
2016-01-05 17:01:28 +01:00
else
2016-01-02 21:51:45 +01:00
case $MESSAGE in
2016-01-05 17:01:28 +01:00
'/cancel')
kill $copid
rm -r $copname
2016-01-06 17:11:56 +01:00
send_message "${USER[ID]}" "Command canceled."
2016-01-05 17:01:28 +01:00
;;
2016-01-06 17:11:56 +01:00
*) inproc "$copname" "$copid" "$MESSAGE";;
2016-01-02 21:51:45 +01:00
esac
2016-01-05 17:01:28 +01:00
fi
}
2015-07-10 07:43:08 +02:00
while true; do {
2016-01-06 17:11:56 +01:00
res=$(curl -s $UPD_URL$OFFSET | ./JSON.sh -s)
2015-07-10 07:43:08 +02:00
2016-01-06 17:11:56 +01:00
# Target
USER[ID]=$(echo "$res" | egrep '\["result",0,"message","chat","id"\]' | cut -f 2)
# Offset
OFFSET=$(echo "$res" | egrep '\["result",0,"update_id"\]' | cut -f 2)
# Message
MESSAGE=$(echo "$res" | egrep '\["result",0,"message","text"\]' | cut -f 2 | cut -d '"' -f 2)
2015-07-10 07:43:08 +02:00
OFFSET=$((OFFSET+1))
2015-07-10 09:32:34 +02:00
if [ $OFFSET != 1 ]; then
2016-01-06 17:11:56 +01:00
# User
USER[FIRST_NAME]=$(echo "$res" | egrep '\["result",0,"message","chat","first_name"\]' | cut -f 2 | cut -d '"' -f 2)
USER[LAST_NAME]=$(echo "$res" | egrep '\["result",0,"message","chat","last_name"\]' | cut -f 2 | cut -d '"' -f 2)
USER[USERNAME]=$(echo "$res" | egrep '\["result",0,"message","chat","username"\]' | cut -f 2 | cut -d '"' -f 2)
# Audio
AUDIO_ID=$(echo "$res" | egrep '\["result",0,"message","audio","file_id"\]' | cut -f 2 | cut -d '"' -f 2)
URLS[AUDIO]=$(get_file "$AUDIO_ID")
# Document
DOCUMENT_ID=$(echo "$res" | egrep '\["result",0,"message","document","file_id"\]' | cut -f 2 | cut -d '"' -f 2)
URLS[DOCUMENT]=$(get_file "$DOCUMENT_ID")
# Photo
PHOTO_ID=$(echo "$res" | egrep '\["result",0,"message","photo",.*,"file_id"\]' | cut -f 2 | cut -d '"' -f 2 | sed -n '$p')
URLS[PHOTO]=$(get_file "$PHOTO_ID")
# Sticker
STICKER_ID=$(echo "$res" | egrep '\["result",0,"message","sticker","file_id"\]' | cut -f 2 | cut -d '"' -f 2)
URLS[STICKER]=$(get_file "$STICKER_ID")
# Video
VIDEO_ID=$(echo "$res" | egrep '\["result",0,"message","video","file_id"\]' | cut -f 2 | cut -d '"' -f 2)
URLS[VIDEO]=$(get_file "$VIDEO_ID")
# Voice
VOICE_ID=$(echo "$res" | egrep '\["result",0,"message","voice","file_id"\]' | cut -f 2 | cut -d '"' -f 2)
URLS[VOICE]=$(get_file "$VOICE_ID")
# Contact
CONTACT[NUMBER]=$(echo "$res" | egrep '\["result",0,"message","contact","phone_number"\]' | cut -f 2 | cut -d '"' -f 2)
CONTACT[FIRST_NAME]=$(echo "$res" | egrep '\["result",0,"message","contact","first_name"\]' | cut -f 2 | cut -d '"' -f 2)
CONTACT[LAST_NAME]=$(echo "$res" | egrep '\["result",0,"message","contact","last_name"\]' | cut -f 2 | cut -d '"' -f 2)
CONTACT[USER_ID]=$(echo "$res" | egrep '\["result",0,"message","contact","user_id"\]' | cut -f 2 | cut -d '"' -f 2)
# Caption
CAPTION=$(echo "$res" | egrep '\["result",0,"message","caption"\]' | cut -f 2 | cut -d '"' -f 2)
# Location
LOCATION[LONGITUDE]=$(echo "$res" | egrep '\["result",0,"message","location","longitude"\]' | cut -f 2 | cut -d '"' -f 2)
LOCATION[LATITUDE]=$(echo "$res" | egrep '\["result",0,"message","location","latitude"\]' | cut -f 2 | cut -d '"' -f 2)
process_client&
2015-07-10 09:32:34 +02:00
fi
2015-07-10 07:43:08 +02:00
2016-01-05 17:01:28 +01:00
}; done
2016-01-06 17:11:56 +01:00