Fixed bugs (now you can run multiple background session in multiple bots), added option to count users, added option to parse markdown and html using send_message, updated documentation and decided to put the sending loop for interactive chats in the background using tmux so that if I have to update the bot interactive sessions will continue to send out messages. Also changed the file name and the location (/tmp/ of the fifos for interactive chats) and added option to broadcast messages to all users.

This commit is contained in:
Daniil Gentili 2016-04-16 20:50:05 +02:00
parent d95470e8a9
commit c33082e585
2 changed files with 80 additions and 29 deletions

View File

@ -70,7 +70,7 @@ Clone the repository:
git clone https://github.com/topkecleon/telegram-bot-bash git clone https://github.com/topkecleon/telegram-bot-bash
``` ```
Paste the token on line 12 (instead of tokenhere). Paste the token on line 15 (instead of tokenhere).
Then start editing the commands. Then start editing the commands.
### Recieve data ### Recieve data
@ -104,9 +104,16 @@ To send messages use the ```send_message``` function:
``` ```
send_message "${USER[ID]}" "lol" send_message "${USER[ID]}" "lol"
``` ```
To send html or markdown put the following strings before the text, depending on the parsing mode you want to enable:
```
send_message "${USER[ID]}" "markdown_parse_mode lol <b>bold</b>"
```
```
send_message "${USER[ID]}" "html_parse_mode lol <b>bold</b>"
```
This function also allows a third parameter that disables additional function parsing (for safety use this when reprinting user input): This function also allows a third parameter that disables additional function parsing (for safety use this when reprinting user input):
``` ```
send_message "${USER[ID]}" "lol" "text" send_message "${USER[ID]}" "lol" "safe"
``` ```
To send images, videos, voice files, photos ecc use the ```send_photo``` function (remember to change the safety Regex @ line 94 to allow sending files only from certain directories): To send images, videos, voice files, photos ecc use the ```send_photo``` function (remember to change the safety Regex @ line 94 to allow sending files only from certain directories):
``` ```
@ -157,6 +164,22 @@ If some thing doesn't work as it should, debug with ```bash -x bashbot.sh```.
To use the functions provided in this script in other scripts source bashbot.sh: ```source bashbot.sh source``` To use the functions provided in this script in other scripts source bashbot.sh: ```source bashbot.sh source```
## User count
To count the total number of users that ever used the bot run the following command:
```
bash bashbot.sh count
```
## Sending broadcasts to all users
To send a broadcast to all of users that ever used the bot run the following command:
```
bash bashbot.sh broadcast "Hey! I just wanted to let you know that the bot's been updated!"
```
That's it! That's it!
If you feel that there's something missing or if you found a bug, feel free to submit a pull request! If you feel that there's something missing or if you found a bug, feel free to submit a pull request!

View File

@ -13,6 +13,7 @@
# If you're in Europe, and public domain does not exist, then haha. # If you're in Europe, and public domain does not exist, then haha.
TOKEN='tokenhere' TOKEN='tokenhere'
URL='https://api.telegram.org/bot'$TOKEN URL='https://api.telegram.org/bot'$TOKEN
# Set INLINE to 1 in order to receive inline queries. # Set INLINE to 1 in order to receive inline queries.
@ -31,6 +32,9 @@ LOCATION_URL=$URL'/sendLocation'
ACTION_URL=$URL'/sendChatAction' ACTION_URL=$URL'/sendChatAction'
FORWARD_URL=$URL'/forwardMessage' FORWARD_URL=$URL'/forwardMessage'
INLINE_QUERY=$URL'/answerInlineQuery' INLINE_QUERY=$URL'/answerInlineQuery'
ME_URL=$URL'/getMe'
ME=$(curl -s $ME_URL | ./JSON.sh -s | egrep '\["result","username"\]' | cut -f 2 | cut -d '"' -f 2)
FILE_URL='https://api.telegram.org/file/bot'$TOKEN'/' FILE_URL='https://api.telegram.org/file/bot'$TOKEN'/'
UPD_URL=$URL'/getUpdates?offset=' UPD_URL=$URL'/getUpdates?offset='
@ -41,7 +45,8 @@ declare -A USER MESSAGE URLS CONTACT LOCATION
send_message() { send_message() {
local chat="$1" local chat="$1"
local text="$(echo "$2" | sed 's/ mykeyboardstartshere.*//g;s/ myfilelocationstartshere.*//g;s/ mylatstartshere.*//g;s/ mylongstartshere.*//g')" local text="$(echo "$2" | sed 's/ mykeyboardstartshere.*//g;s/ myfilelocationstartshere.*//g;s/ mylatstartshere.*//g;s/ mylongstartshere.*//g')"
[ "$3" != "text" ] && { local arg="$3"
[ "$3" != "safe" ] && {
local keyboard="$(echo "$2" | sed '/mykeyboardstartshere /!d;s/.*mykeyboardstartshere //g;s/ myfilelocationstartshere.*//g;s/ mylatstartshere.*//g;s/ mylongstartshere.*//g')" local keyboard="$(echo "$2" | sed '/mykeyboardstartshere /!d;s/.*mykeyboardstartshere //g;s/ myfilelocationstartshere.*//g;s/ mylatstartshere.*//g;s/ mylongstartshere.*//g')"
local file="$(echo "$2" | sed '/myfilelocationstartshere /!d;s/.*myfilelocationstartshere //g;s/ mykeyboardstartshere.*//g;s/ mylatstartshere.*//g;s/ mylongstartshere.*//g')" local file="$(echo "$2" | sed '/myfilelocationstartshere /!d;s/.*myfilelocationstartshere //g;s/ mykeyboardstartshere.*//g;s/ mylatstartshere.*//g;s/ mylongstartshere.*//g')"
@ -64,19 +69,23 @@ send_message() {
fi fi
if [ "$sent" != "y" ];then if [ "$sent" != "y" ];then
res=$(curl -s "$MSG_URL" -F "chat_id=$chat" -F "text=$text") send_text "$chat" "$text"
fi fi
} }
send_text() {
echo "$2" | grep -q '^html_parse_mode' && local add="-F \"parse_mode=html\""
echo "$2" | grep -q '^markdown_parse_mode' && local add="-F \"parse_mode=markdown\""
res=$(curl -s "$MSG_URL" -F "chat_id=$1" -F "text=$2" $add)
}
send_markdown_message() { send_markdown_message() {
res=$(curl -s "$MSG_URL" -F "chat_id=$1" -F "text=$2" -F "parse_mode=markdown") res=$(curl -s "$MSG_URL" -F "chat_id=$1" -F "text=$2" -F "parse_mode=markdown")
} }
answer_inline_query() { answer_inline_query() {
case $2 in case $2 in
"article") "article")
InlineQueryResult='[{"type":"'$2'","id":"$RANDOM","title":"'$3'","message_text":"'$4'"}]' InlineQueryResult='[{"type":"'$2'","id":"$RANDOM","title":"'$3'","message_text":"'$4'"}]'
;; ;;
@ -106,10 +115,10 @@ answer_inline_query() {
;; ;;
"venue") "venue")
InlineQueryResult='[{"type":"'$2'","id":"$RANDOM","latitude":"'$3'","longitude":"'$4'","title":"'$5'","address":"'$6'"}]' InlineQueryResult='[{"type":"'$2'","id":"$RANDOM","latitude":"'$3'","longitude":"'$4'","title":"'$5'","address":"'$6'"}]'
;; ;;
"contact") "contact")
InlineQueryResult='[{"type":"'$2'","id":"$RANDOM","phone_number":"'$3'","first_name":"'$4'"}]' InlineQueryResult='[{"type":"'$2'","id":"$RANDOM","phone_number":"'$3'","first_name":"'$4'"}]'
;; ;;
# Cached media stored in Telegram server # Cached media stored in Telegram server
@ -224,18 +233,13 @@ forward() {
startproc() { startproc() {
killproc killproc
mkfifo $copname mkfifo /tmp/$copname
TMUX= tmux new-session -d -s $copname "$* &>$copname; echo >$copname; sleep 5; rm -r $copname" TMUX= tmux new-session -d -s $copname "$* &>/tmp/$copname; echo imprettydarnsuredatdisisdaendofdacmd>/tmp/$copname"
while [ -p "$copname" ];do TMUX= tmux new-session -d -s sendprocess_$copname "bash bashbot.sh outproc ${USER[ID]} $copname"
read -t 10 line
[ "$line" != "" ] && send_message "${USER[ID]}" "$line"
line=
done <$copname
} }
killproc() { killproc() {
(rm -r $copname (tmux kill-session -t $copname; echo imprettydarnsuredatdisisdaendofdacmd>/tmp/$copname; sleep 1; tmux kill-session -t send$copname; rm -r /tmp/$copname)2>/dev/null
tmux kill-session -t $copname)2>/dev/null
} }
inproc() { inproc() {
@ -244,10 +248,14 @@ inproc() {
} }
process_client() { process_client() {
# Message
MESSAGE=$(echo "$res" | egrep '\["result",0,"message","text"\]' | cut -f 2 | cut -d '"' -f 2)
# User # User
USER[ID]=$(echo "$res" | egrep '\["result",0,"message","chat","id"\]' | cut -f 2)
USER[FIRST_NAME]=$(echo "$res" | egrep '\["result",0,"message","chat","first_name"\]' | cut -f 2 | cut -d '"' -f 2) 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[LAST_NAME]=$(echo "$res" | egrep '\["result",0,"message","chat","last_name"\]' | cut -f 2 | cut -d '"' -f 2)
USER[USERNAME]=$(echo "$res" | sed 's/^.*\(username.*\)/\1/g' | cut -d '"' -f3) USER[USERNAME]=$(echo "$res" | egrep '\["result",0,"message","chat","username"\]' | cut -f 2 | cut -d '"' -f 2)
# Audio # Audio
URLS[AUDIO]=$(get_file $(echo "$res" | egrep '\["result",0,"message","audio","file_id"\]' | cut -f 2 | cut -d '"' -f 2)) URLS[AUDIO]=$(get_file $(echo "$res" | egrep '\["result",0,"message","audio","file_id"\]' | cut -f 2 | cut -d '"' -f 2))
@ -277,9 +285,9 @@ process_client() {
NAME="$(basename ${URLS[*]} &>/dev/null)" NAME="$(basename ${URLS[*]} &>/dev/null)"
# Tmux # Tmux
copname="CO${USER[ID]}" copname="$ME"_"${USER[USERNAME]}"
if ! tmux ls | grep -q $copname; then if ! tmux ls | grep -v send | grep -q $copname; then
[ ! -z ${URLS[*]} ] && { [ ! -z ${URLS[*]} ] && {
curl -s ${URLS[*]} -o $NAME curl -s ${URLS[*]} -o $NAME
send_file "${USER[ID]}" "$NAME" "$CAPTION" send_file "${USER[ID]}" "$NAME" "$CAPTION"
@ -316,7 +324,7 @@ process_client() {
case $MESSAGE in case $MESSAGE in
'/question') '/question')
startproc "./question"& startproc "./question"
;; ;;
'/info') '/info')
send_message "${USER[ID]}" "This is bashbot, the Telegram bot written entirely in bash." send_message "${USER[ID]}" "This is bashbot, the Telegram bot written entirely in bash."
@ -338,7 +346,7 @@ Contribute to the project: https://github.com/topkecleon/telegram-bot-bash
'') '')
;; ;;
*) *)
send_message "${USER[ID]}" "$MESSAGE" "text" send_message "${USER[ID]}" "$MESSAGE" "safe"
esac esac
else else
case $MESSAGE in case $MESSAGE in
@ -349,25 +357,45 @@ Contribute to the project: https://github.com/topkecleon/telegram-bot-bash
*) inproc;; *) inproc;;
esac esac
fi fi
if [ "$MESSAGE" = "/start" ]; then
tmpcount="COUNT${USER[ID]}"
cat count | grep -q "$tmpcount" || echo "$tmpcount">>count
fi
# To get user count execute bash bashbot.sh count
} }
# source the script with source as param to use functions in other scripts # source the script with source as param to use functions in other scripts
while [ "$1" != "source" ]; do { while [ "$1" == "" ]; do {
res=$(curl -s $UPD_URL$OFFSET | ./JSON.sh -s) res=$(curl -s $UPD_URL$OFFSET | ./JSON.sh -s)
# Target
USER[ID]=$(echo "$res" | egrep '\["result",0,"message","chat","id"\]' | cut -f 2)
# Offset # Offset
OFFSET=$(echo "$res" | egrep '\["result",0,"update_id"\]' | cut -f 2) 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)
OFFSET=$((OFFSET+1)) OFFSET=$((OFFSET+1))
if [ $OFFSET != 1 ]; then if [ $OFFSET != 1 ]; then
process_client& process_client&
fi fi
}; done }; done
case "$1" in
"outproc")
until [ "$line" = "imprettydarnsuredatdisisdaendofdacmd" ];do
line=
read -t 10 line
[ "$line" != "" -a "$line" != "imprettydarnsuredatdisisdaendofdacmd" ] && send_message "$2" "$line"
done </tmp/$3
rm -r /tmp/$3
;;
"count")
wc -l count
;;
"broadcast")
[ $(wc -l count) -gt 300 ] && sleep="sleep 0.5"
shift
for f in $(cat count);do send_message ${f//COUNT} "$*"; $sleep;done
;;
esac