Fixed the bugs and switched to tmux

This commit is contained in:
Daniil Gentili 2016-01-05 17:01:28 +01:00
parent 7ef846b8c0
commit fb0c473c90
2 changed files with 79 additions and 50 deletions

View File

@ -1,6 +1,6 @@
#!/bin/bash #!/bin/bash
# bashbot, the Telegram bot written in bash. # bashbot, the Telegram bot written in bash.
# Written by @topkecleon, Juan Potato (@awkward_potato) and Lorenzo Santina (BigNerd95) # Written by @topkecleon, Juan Potato (@awkward_potato), Lorenzo Santina (BigNerd95) and Daniil Gentili (danog)
# http://github.com/topkecleon/bashbot # http://github.com/topkecleon/bashbot
# Depends on JSON.sh (http://github.com/dominictarr/JSON.sh), # Depends on JSON.sh (http://github.com/dominictarr/JSON.sh),
@ -9,7 +9,6 @@
# This file is public domain in the USA and all free countries. # 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. # If you're in Europe, and public domain does not exist, then haha.
TOKEN='' TOKEN=''
URL='https://api.telegram.org/bot'$TOKEN URL='https://api.telegram.org/bot'$TOKEN
MSG_URL=$URL'/sendMessage' MSG_URL=$URL'/sendMessage'
@ -18,77 +17,90 @@ UPD_URL=$URL'/getUpdates?offset='
OFFSET=0 OFFSET=0
send_message() { send_message() {
[ "$2" != "" ] && res=$(curl "$MSG_URL" -F "chat_id=$1" -F "text=$2") local chat="$1"
local text="$(echo "$2" | sed 's/ mykeyboardstartshere.*//g')"
local keyboard="$(echo "$2" | sed '/mykeyboardstartshere /!d;s/.*mykeyboardstartshere //g')"
if [ "$keyboard" = "" ]; then
res=$(curl -s "$MSG_URL" -F "chat_id=$chat" -F "text=$text")
else
send_keyboard "$chat" "$text" "$keyboard"
fi
} }
send_keyboard() { send_keyboard() {
for f in ${@/1\|2/};do keyboard="$keyboard, [\"$f\"]";done local chat="$1"
keyboard=${keyboard/^, /} local text="$2"
res=$(curl "$MSG_URL" -F "chat_id=$1" -F "text=$2" -F "reply_markup={\"keyboard\": $keyboard, \"one_time_keyboard\": true}") shift 2
keyboard=init
for f in $*;do keyboard="$keyboard, [\"$f\"]";done
keyboard=${keyboard/init, /}
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}")
} }
send_photo() { send_photo() {
res=$(curl "$PHO_URL" -F "chat_id=$1" -F "photo=@$2") res=$(curl -s "$PHO_URL" -F "chat_id=$1" -F "photo=@$2")
} }
question() { startproc() {
TARGET="$1' local copname="$1"
echo "Why hello there. local TARGET="$2"
Would you like some tea (y/n)?" mkdir -p "$copname"
read answer mkfifo $copname/out
[[ $answer =~ ^([yY][eE][sS]|[yY])$ ]] && echo "OK then, here you go: http://www.rivertea.com/blog/wp-content/uploads/2013/12/Green-Tea.jpg" || echo "OK then." tmux new-session -d -n $copname "./question $TARGET 2>&1>$copname/out"
until [ "$SUCCESS" = "y" ] ;do local pid=$(ps aux | sed '/tmux/!d;/'$copname'/!d;/sed/d;s/'$USER'\s*//g;s/\s.*//g')
send_keyboard "$TARGET" "Do you like Music?" "Yass!" "No" echo $pid>$copname/pid
read answer while ps aux | grep -v grep | grep -q $pid;do
case $answer in read -t 10 line
'Yass!') echo "Goody!";SUCCESS=y;; [ "$line" != "" ] && send_message "$TARGET" "$line"
'No') echo "Well that's weird";SUCCESS=y;; line=
*) SUCCESS=n;; done <$copname/out
esac
done
} }
inproc() { inproc() {
copname="$1" local copname="$1"
msg="${@/1/}" local copid="$2"
echo "$msg" >&${$1["0"]} shift 2
tmux send-keys -t $copname "$@
"
ps aux | grep -v grep | grep -q "$copid" || { rm -r $copname; };
} }
outproc() {
copname="$1"
TARGET="$2"
while true; do read msg <&${$copname["0"]}; [ "$?" != "0" ] && return || send_message "$TARGET" "$msg";done
}
process_client() { process_client() {
local MESSAGE=$1 local MESSAGE=$1
local TARGET=$2 local TARGET=$2
local msg="" local msg=""
local copname="coproc$TARGET" local copname="CO$TARGET"
local copidname="$copname"_PID local copidname="$copname/pid"
local copid="${$copid}" local copid="$(cat $copidname 2>/dev/null)"
[ "$copid" = "" ] { if [ "$copid" = "" ]; then
case $MESSAGE in case $MESSAGE in
'/info') send_message "$TARGET" "This is bashbot, the Telegram bot written entirely in bash.";; '/question')
'/question') coproc "$copname" { question "$TARGET"; } &>&1; outproc "$copname" "$TARGET"; return;; startproc "$copname" "$TARGET"&
*) send_message "$TARGET" "$MESSAGE";; ;;
'/info')
send_message "$TARGET" "This is bashbot, the Telegram bot written entirely in bash."
;;
*)
send_message "$TARGET" "$MESSAGE"
esac esac
} || { else
case $MESSAGE in case $MESSAGE in
'/cancel') kill $copid;; '/cancel')
*) inproc "$copname" "$MESSAGE";; kill $copid
rm -r $copname
send_message "$TARGET" "Command canceled."
;;
*) inproc "$copname" "$copid" "$MESSAGE";;
esac esac
} fi
} }
while true; do { while true; do {
res=$(curl $UPD_URL$OFFSET) res=$(curl -s $UPD_URL$OFFSET)
TARGET=$(echo $res | ./JSON.sh | egrep '\["result",0,"message","chat","id"\]' | cut -f 2) TARGET=$(echo $res | JSON.sh | egrep '\["result",0,"message","chat","id"\]' | cut -f 2)
OFFSET=$(echo $res | ./JSON.sh | egrep '\["result",0,"update_id"\]' | cut -f 2) OFFSET=$(echo $res | JSON.sh | egrep '\["result",0,"update_id"\]' | cut -f 2)
MESSAGE=$(echo $res | ./JSON.sh -s | egrep '\["result",0,"message","text"\]' | cut -f 2 | cut -d '"' -f 2) MESSAGE=$(echo $res | JSON.sh -s | egrep '\["result",0,"message","text"\]' | cut -f 2 | cut -d '"' -f 2)
OFFSET=$((OFFSET+1)) OFFSET=$((OFFSET+1))
@ -96,4 +108,5 @@ while true; do {
process_client "$MESSAGE" "$TARGET"& process_client "$MESSAGE" "$TARGET"&
fi fi
} &>/dev/null; done }; done

16
question Executable file
View File

@ -0,0 +1,16 @@
#!/bin/bash
TARGET="$1"
echo "Why hello there.
Would you like some tea (y/n)?"
read answer
[[ $answer =~ ^([yY][eE][sS]|[yY])$ ]] && echo "OK then, here you go: http://www.rivertea.com/blog/wp-content/uploads/2013/12/Green-Tea.jpg" || echo "OK then."
until [ "$SUCCESS" = "y" ] ;do
echo "Do you like Music? mykeyboardstartshere Yass! No"
read answer
case $answer in
'Yass!') echo "Goody!";SUCCESS=y;;
'No') echo "Well that's weird";SUCCESS=y;;
*) SUCCESS=n;;
esac
done
exit