Fixed smore bugs, tried to speed up bot.

This commit is contained in:
Daniil Gentili 2016-04-17 14:00:45 +02:00
parent c33082e585
commit d13b6e25eb
2 changed files with 46 additions and 42 deletions

1
.gitignore vendored
View File

@ -1 +1,2 @@
*~ *~
count

View File

@ -13,14 +13,13 @@
# 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.
# To enable this option in your bot, send the /setinline command to @BotFather. # To enable this option in your bot, send the /setinline command to @BotFather.
INLINE=0 INLINE=0
SCRIPT="$0"
MSG_URL=$URL'/sendMessage' MSG_URL=$URL'/sendMessage'
PHO_URL=$URL'/sendPhoto' PHO_URL=$URL'/sendPhoto'
AUDIO_URL=$URL'/sendAudio' AUDIO_URL=$URL'/sendAudio'
@ -43,6 +42,7 @@ OFFSET=0
declare -A USER MESSAGE URLS CONTACT LOCATION declare -A USER MESSAGE URLS CONTACT LOCATION
send_message() { send_message() {
[ "$2" = "" ] && return 1
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')"
local arg="$3" local arg="$3"
@ -75,13 +75,25 @@ send_message() {
} }
send_text() { send_text() {
echo "$2" | grep -q '^html_parse_mode' && local add="-F \"parse_mode=html\"" case "$2" in
echo "$2" | grep -q '^markdown_parse_mode' && local add="-F \"parse_mode=markdown\"" html_parse_mode*)
res=$(curl -s "$MSG_URL" -F "chat_id=$1" -F "text=$2" $add) send_html_message "$1" "${2//html_parse_mode}"
;;
markdown_parse_mode*)
send_markdown_message "$1" "${2//markdown_parse_mode}"
;;
*)
res=$(curl -s "$MSG_URL" -d "chat_id=$1" -d "text=$2")
;;
esac
} }
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" -d "chat_id=$1" -d "text=$2" -d "parse_mode=markdown")
}
send_html_message() {
res=$(curl -s "$MSG_URL" -F "chat_id=$1" -F "text=$2" -F "parse_mode=html")
} }
answer_inline_query() { answer_inline_query() {
@ -168,7 +180,6 @@ send_keyboard() {
get_file() { 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) [ "$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() { send_file() {
@ -235,11 +246,11 @@ startproc() {
killproc killproc
mkfifo /tmp/$copname mkfifo /tmp/$copname
TMUX= tmux new-session -d -s $copname "$* &>/tmp/$copname; echo imprettydarnsuredatdisisdaendofdacmd>/tmp/$copname" TMUX= tmux new-session -d -s $copname "$* &>/tmp/$copname; echo imprettydarnsuredatdisisdaendofdacmd>/tmp/$copname"
TMUX= tmux new-session -d -s sendprocess_$copname "bash bashbot.sh outproc ${USER[ID]} $copname" TMUX= tmux new-session -d -s sendprocess_$copname "bash $SCRIPT outproc ${USER[ID]} $copname"
} }
killproc() { killproc() {
(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; echo imprettydarnsuredatdisisdaendofdacmd>/tmp/$copname; tmux kill-session -t sendprocess_$copname; rm -r /tmp/$copname)2>/dev/null
} }
inproc() { inproc() {
@ -285,7 +296,7 @@ process_client() {
NAME="$(basename ${URLS[*]} &>/dev/null)" NAME="$(basename ${URLS[*]} &>/dev/null)"
# Tmux # Tmux
copname="$ME"_"${USER[USERNAME]}" copname="$ME"_"${USER[ID]}"
if ! tmux ls | grep -v send | grep -q $copname; then if ! tmux ls | grep -v send | grep -q $copname; then
[ ! -z ${URLS[*]} ] && { [ ! -z ${URLS[*]} ] && {
@ -319,9 +330,8 @@ process_client() {
if [[ $iQUERY_MSG == web ]]; then if [[ $iQUERY_MSG == web ]]; then
answer_inline_query "$iQUERY_ID" "article" "Telegram" "https://telegram.org/" answer_inline_query "$iQUERY_ID" "article" "Telegram" "https://telegram.org/"
fi fi
fi fi
fi
case $MESSAGE in case $MESSAGE in
'/question') '/question')
startproc "./question" startproc "./question"
@ -343,24 +353,16 @@ Written by @topkecleon, Juan Potato (@awkward_potato), Lorenzo Santina (BigNerd9
Contribute to the project: https://github.com/topkecleon/telegram-bot-bash Contribute to the project: https://github.com/topkecleon/telegram-bot-bash
" "
;; ;;
'') '/cancel')
if tmux ls | grep -q $copname; then killproc && send_message "${USER[ID]}" "Command canceled.";else send_message "${USER[ID]}" "No command is currently running.";fi
;; ;;
*) *)
send_message "${USER[ID]}" "$MESSAGE" "safe" if tmux ls | grep -v send | grep -q $copname;then inproc; else send_message "${USER[ID]}" "$MESSAGE" "safe";fi
esac
else
case $MESSAGE in
'/cancel')
killproc
send_message "${USER[ID]}" "Command canceled."
;; ;;
*) inproc;;
esac esac
fi
if [ "$MESSAGE" = "/start" ]; then
tmpcount="COUNT${USER[ID]}" tmpcount="COUNT${USER[ID]}"
cat count | grep -q "$tmpcount" || echo "$tmpcount">>count cat count | grep -q "$tmpcount" || echo "$tmpcount">>count
fi
# To get user count execute bash bashbot.sh count # To get user count execute bash bashbot.sh count
} }
@ -390,10 +392,11 @@ case "$1" in
rm -r /tmp/$3 rm -r /tmp/$3
;; ;;
"count") "count")
wc -l count echo "A total of $(wc -l count | sed 's/count//g')users used me."
;; ;;
"broadcast") "broadcast")
[ $(wc -l count) -gt 300 ] && sleep="sleep 0.5" echo "Sending the broadcast $* to $(wc -l count | sed 's/count//g')users."
[ $(wc -l count | sed 's/ count//g') -gt 300 ] && sleep="sleep 0.5"
shift shift
for f in $(cat count);do send_message ${f//COUNT} "$*"; $sleep;done for f in $(cat count);do send_message ${f//COUNT} "$*"; $sleep;done
;; ;;