util.sh

#!/bin/bash
source .token.sh
source util.logging.sh

if [ -z "$TOKEN" ]; then
    log -f util "Cannot load token, exiting"
    exit 1
fi

readonly API="https://api.telegram.org/bot$TOKEN"

log -i util "Getting my ID"
readonly BOT_ID=$(curl -s $API/getMe | jq .result.id)
log -i util "ID Acquired"

tg() {
    case $1 in
    --sendmsg)
        shift
        local CHAT_ID=$1
        local MSG=$2
        local RESULT=$(curl -s "$API/sendMessage" -d "chat_id=$CHAT_ID" -d "text=$MSG")
        SENT_MSG_ID=$(echo "$RESULT" | jq '.result | .message_id')
        echo $RESULT | jq .
        ;;
    --sendmsghtml)
        shift
        local CHAT_ID=$1
        local MSG=$2
        local RESULT=$(curl -s "$API/sendMessage" --form-string "chat_id=$CHAT_ID" --form-string "text=$MSG" --form-string "parse_mode=HTML" --form-string "disable_web_page_preview=True" | jq .)
        SENT_MSG_ID=$(echo "$RESULT" | jq '.result | .message_id')
        echo $RESULT | jq .
        ;;
    --editmsg | --editmarkdownv2msg)
        local PARAM=$1
        shift
        local CHAT_ID=$1
        local MSG_ID=$2
        local NEW_TEXT=$3
        if [[ "$PARAM" =~ "--editmarkdownv2msg" ]]; then
            curl -s "$API/editMessageText" -d "chat_id=$CHAT_ID" -d "message_id=$MSG_ID" -d "text=$NEW_TEXT" -d "parse_mode=MarkdownV2" | jq .
        else
            curl -s "$API/editMessageText" -d "chat_id=$CHAT_ID" -d "message_id=$MSG_ID" -d "text=$NEW_TEXT" | jq .
        fi
        ;;
    --editmsgurl)
        local PARAM=$1
        shift
        local CHAT_ID=$1
        local MSG_ID=$2
        local NEW_TEXT=$3
        curl -s "$API/editMessageText" -F "chat_id=$CHAT_ID" -F "message_id=$MSG_ID" -F "text=$NEW_TEXT" | jq .
        ;;
    --editmsghtml)
        shift
        local CHAT_ID=$1
        local MSG_ID=$2
        local MSG=$3
        local RESULT=$(curl -s "$API/editMessageText" --form-string "chat_id=$CHAT_ID" --form-string "message_id=$MSG_ID" --form-string "text=$MSG" --form-string "parse_mode=HTML" --form-string "disable_web_page_preview=True" | jq .)
        SENT_MSG_ID=$(echo "$RESULT" | jq '.result | .message_id')
        echo $RESULT | jq .
        ;;
    --sendmarkdownv2msg)
        shift
        local CHAT_ID=$1
        local MSG=$2
        local RESULT=$(curl -s "$API/sendMessage" -d "chat_id=$CHAT_ID" -d "parse_mode=MarkdownV2" -d "text=$MSG")
        SENT_MSG_ID=$(echo "$RESULT" | jq '.result | .message_id')
        echo $RESULT | jq .
        ;;
    --replymsg)
        shift
        local CHAT_ID=$1
        local MSG_ID=$2
        local MSG=$3
        local RESULT=$(curl -s "$API/sendMessage" -d "chat_id=$CHAT_ID" -d "reply_to_message_id=$MSG_ID" -d "text=$MSG" | jq .)
        SENT_MSG_ID=$(echo "$RESULT" | jq '.result | .message_id')
        echo $RESULT | jq .
        ;;
    --replyfile)
        shift
        local CHAT_ID=$1
        local MSG_ID=$2
        local MSG=$3
        local RESULT=$(curl -s "$API/sendDocument" -F "chat_id=$CHAT_ID" -F "reply_to_message_id=$MSG_ID" -F "document=@\"$MSG\"" | jq .)
        SENT_MSG_ID=$(echo "$RESULT" | jq '.result | .message_id')
        echo $RESULT | jq .
        ;;
    --replymsghtml)
        shift
        local CHAT_ID=$1
        local MSG_ID=$2
        local MSG=$3
        local RESULT=$(curl -s "$API/sendMessage" --form-string "chat_id=$CHAT_ID" --form-string "reply_to_message_id=$MSG_ID" --form-string "text=$MSG" --form-string "parse_mode=HTML" --form-string "disable_web_page_preview=True" | jq .)
        SENT_MSG_ID=$(echo "$RESULT" | jq '.result | .message_id')
        echo $RESULT | jq .
        ;;
    --replymsgmarkdown)
        shift
        local CHAT_ID=$1
        local MSG_ID=$2
        local MSG=$3
        local RESULT=$(curl -s "$API/sendMessage" -d "chat_id=$CHAT_ID" -d "reply_to_message_id=$MSG_ID" --data-urlencode "text=$MSG" | jq .)
        SENT_MSG_ID=$(echo "$RESULT" | jq '.result | .message_id')
        echo $RESULT | jq .
        ;;
    --replymarkdownv2msg)
        shift
        local CHAT_ID=$1
        local MSG_ID=$2
        local MSG=$3
        local RESULT=$(curl -s "$API/sendMessage" -d "chat_id=$CHAT_ID" -d "reply_to_message_id=$MSG_ID" -d "text=$MSG" -d "parse_mode=MarkdownV2" | jq .)
        SENT_MSG_ID=$(echo "$RESULT" | jq '.result | .message_id')
        echo "$RESULT" | jq .
        ;;
    --delmsg)
        shift
        local CHAT_ID=$1
        local MSG_ID=$2
        curl -s "$API/deleteMessage" -d "chat_id=$CHAT_ID" -d "message_id=$MSG_ID" | jq .
        ;;
    --sendsticker | --replysticker)
        local PARAM=$1
        shift
        local CHAT_ID=$1
        local FILE_ID=$2
        if [[ "$PARAM" =~ "--replysticker" ]]; then
            local MSG_ID=$2
            local FILE_ID=$3
            curl -s "$API/sendSticker" -d "chat_id=$CHAT_ID" -d "sticker=$FILE_ID" -d "reply_to_message_id=$MSG_ID" | jq .
        else
            curl -s "$API/sendSticker" -d "chat_id=$CHAT_ID" -d "sticker=$FILE_ID" | jq .
        fi
        ;;
    --fwdmsg | --cpmsg)
        local PARAM=$1 # Save this to check for --cpmsg
        shift
        local FROM=$1
        local TO=$2
        local MSG_ID=$3
        if [ "$PARAM" = "--cpmsg" ]; then
            local MODE=copyMessage
        else
            local MODE=forwardMessage
        fi
        curl -s "$API/$MODE" -d "from_chat_id=$FROM" -d "chat_id=$TO" -d "message_id=$MSG_ID"
        ;;
    --pinmsg)
        shift
        local CHAT_ID=$1
        local MSG_ID=$2
        curl -s "$API/pinChatMessage" -d "chat_id=$CHAT_ID" -d "message_id=$MSG_ID"
        ;;
    --unpinmsg)
        shift
        local CHAT_ID=$1
        local MSG_ID=$2
        curl -s "$API/unpinChatMessage" -d "chat_id=$CHAT_ID" -d "message_id=$MSG_ID"
        ;;
    --getuserpfp) #pfp -> Profile Photo
        shift
        local USER_ID=$1
        local RESULT=$(curl -s "$API/getUserProfilePhotos" -d "user_id=$USER_ID" | jq . )
        echo $RESULT | jq .
        FILE_ID=$(echo "$RESULT" | jq -r '.result.photos[0][-1].file_id')
        FILE_PATH=$(echo "$RESULT" | jq -r '.result.photos[0][-1].file_path')
        ;;
    # --downloadfile -> OUTPUT_FILE should be complete name of file including MIME type / extension.
    --downloadfile)
        shift
        local FILE_ID=$1
        local OUTPUT_FILE=$2
        local RESULT=$(curl -s "$API/getFile" -d "file_id=$FILE_ID" | jq .)
        echo $RESULT | jq .
        FILE_ID=$(echo "$RESULT" | jq -r '.result.file_id')
        FILE_PATH=$(echo "$RESULT" | jq -r '.result.file_path')
        curl -s "https://api.telegram.org/file/bot$TOKEN/$FILE_PATH" -o $OUTPUT_FILE
        ;;
    --ban)
        shift
        local CHAT_ID=$1
        local USER_ID=$2
        curl -s $API/banChatMember -F "chat_id=$CHAT_ID" -F "user_id=$USER_ID" | jq .
        ;;
    --unban)
        shift
        local CHAT_ID=$1
        local USER_ID=$2
        curl -s $API/unbanChatMember -F "chat_id=$CHAT_ID" -F "user_id=$USER_ID" -F only_if_banned=true | jq .
        ;;
    --sendphoto)
        shift
        local CHAT_ID=$1
        local PHOTO=$2
        local CAPTION=$3
        local RESULT=$(curl -s "$API/sendPhoto" -F "chat_id=$CHAT_ID" -F "photo=@\"$PHOTO\"" -F "caption=$CAPTION" | jq . )
        echo $RESULT | jq .
        SENT_MSG_ID=$(echo "$RESULT" | jq '.result | .message_id')
        ;;
    --replyphoto)
        shift
        local CHAT_ID=$1
        local MSG_ID=$2
        local PHOTO=$3
        local CAPTION=$4
        local RESULT=$(curl -s "$API/sendPhoto" -F "chat_id=$CHAT_ID" -F "reply_to_message_id=$MSG_ID" -F "photo=@\"$PHOTO\"" -F "caption=$CAPTION" | jq . )
        echo $RESULT | jq .
        SENT_MSG_ID=$(echo "$RESULT" | jq '.result | .message_id')
        ;;
    #--sendaudiofile -> File must have the correct MIME type (e.g., audio/mp3 )
    --sendaudiofile)
        shift
        local CHAT_ID=$1
        local AUDIO=$2
        local CAPTION=$3
        local RESULT=$(curl -s "$API/sendAudio" -F "chat_id=$CHAT_ID" -F "audio=@\"$AUDIO\"" -F "caption=$CAPTION" | jq . )
        echo $RESULT | jq .
        SENT_MSG_ID=$(echo "$RESULT" | jq '.result | .message_id')
        ;;
    #--sendvoicefile -> The file must have the type audio/ogg and be no more than 1MB in size. 1-20MB voice notes will be sent as files.
    --sendvoicefile)
        shift
        local CHAT_ID=$1
        local VOICE=$2
        local CAPTION=$3
        local RESULT=$(curl -s "$API/sendVoice" -F "chat_id=$CHAT_ID" -F "audio=@\"$VOICE\"" -F "caption=$CAPTION" | jq . )
        echo $RESULT | jq .
        SENT_MSG_ID=$(echo "$RESULT" | jq '.result | .message_id')
        ;;
    --sendpreviewvideo)
        shift
        local CHAT_ID=$1
        local VIDEO=$2
        local resolution=$(ffprobe -v error -select_streams v:0 -show_entries stream=width,height -of csv=s=x:p=0 "$VIDEO" )
        local width=$(echo $resolution | cut -d 'x' -f 1)
        local height=$(echo $resolution | cut -d 'x' -f 2)
        local RESULT=$(curl -s $API/sendVideo -F "chat_id=$CHAT_ID" -F "video=@\"$VIDEO\"" -F "width=$width" -F "height=$height" | jq . )
        echo $RESULT | jq .
        SENT_MSG_ID=$(echo "$RESULT" | jq '.result | .message_id')
        ;;
    --replypreviewvideo)
        shift
        local CHAT_ID=$1
        local MSG_ID=$2
        local VIDEO=$3
        local resolution=$(ffprobe -v error -select_streams v:0 -show_entries stream=width,height -of csv=s=x:p=0 "$VIDEO" )
        local width=$(echo $resolution | cut -d 'x' -f 1)
        local height=$(echo $resolution | cut -d 'x' -f 2)
        local RESULT=$(curl -s $API/sendVideo -F "chat_id=$CHAT_ID" -F "reply_to_message_id=$MSG_ID" -F "video=@\"$VIDEO\"" -F "width=$width" -F "height=$height" | jq . )
        echo $RESULT | jq .
        SENT_MSG_ID=$(echo "$RESULT" | jq '.result | .message_id')
        ;;
    --sendwithinlinebutton)
        shift
        local CHAT_ID=$1
        local MSG=$2
        local BUTTON_TEXT=$3
        CALLBACK_DATA=$(echo $RANDOM | md5sum | head -c 20)
        local RESULT=$(curl -s -X POST "$API/sendMessage" -H "Content-Type: application/json" -d "{\"chat_id\":\"$CHAT_ID\", \"text\":\"$MSG\", \"reply_markup\": {\"inline_keyboard\": [[{\"text\":\"$BUTTON_TEXT\", \"callback_data\": \"$CALLBACK_DATA\"}]]} }")
        SENT_MSG_ID=$(echo "$RESULT" | jq '.result.message_id')
        QUERY_SENT_CHAT_ID=$CHAT_ID
        QUERY_ID=$(echo "$RESULT" | jq '.result.message_id')
        echo $RESULT | jq .
        ;;
    esac
}

update() {
    FETCH=$(curl -s "$API/getUpdates" -d "offset=$UPDATE_ID" -d "timeout=60" | jq '.result[]' 2>/dev/null)
    if [ -n "$FETCH" ]; then
        UPDATE_ID=$((UPDATE_ID + 1))

        # IDs
        RET_MSG_ID=$(echo "$FETCH" | jq '.message.message_id')
        RET_CHAT_ID=$(echo "$FETCH" | jq '.message.chat.id')
        RET_CHAT_TYPE=$(echo "$FETCH" | jq -r '.message.chat.type')
        RET_CHAT_TITLE=$(echo "$FETCH" | jq -r '.message.chat.title')
        MSGGER=$(echo "$FETCH" | jq '.message.from.id')
        RET_FILE_ID=$(echo "$FETCH" | jq -r '.message.document.file_id')

        # Strings
        RET_MSG_TEXT=$(echo "$FETCH" | jq -r '.message.text')
        FIRST_NAME=$(echo "$FETCH" | jq -r '.message.from.first_name')
        LAST_NAME=$(echo "$FETCH" | jq -r '.message.from.last_name')
        USERNAME=$(echo "$FETCH" | jq  -r '.message.from.username')

        # Replies
        RET_REPLIED_MSG_ID=$(echo "$FETCH" | jq '.message.reply_to_message.message_id')
        RET_REPLIED_MSGGER_ID=$(echo "$FETCH" | jq '.message.reply_to_message.from.id')
        RET_REPLIED_MSGGER_FIRST_NAME=$(echo "$FETCH" | jq -r '.message.reply_to_message.from.first_name')
        RET_REPLIED_MSGGER_LAST_NAME=$(echo "$FETCH" | jq -r '.message.reply_to_message.from.last_name')
        RET_REPLIED_MSGGER_USERNAME=$(echo "$FETCH" | jq -r '.message.reply_to_message.from.username')
        RET_REPLIED_MSG_TEXT=$(echo "$FETCH" | jq -r '.message.reply_to_message.text')
        RET_REPLIED_FILE_ID=$(echo "$FETCH" | jq -r '.message.reply_to_message.document.file_id')

        # Stickers
        STICKER_EMOJI=$(echo "$FETCH" | jq -r '.message.sticker.emoji')
        STICKER_FILE_ID=$(echo "$FETCH" | jq -r '.message.sticker.file_id')
        STICKER_PACK_NAME=$(echo "$FETCH" | jq -r '.message.sticker.set_name')

        # Callback Query
        CALL_BACK_QUERY_ID=$(echo "$FETCH" | jq -r '.callback_query.id')
        CALL_BACK_QUERY_FROM=$(echo "$FETCH" | jq -r '.callback_query.from')
        CALL_BACK_QUERY_DATA=$(echo "$FETCH" | jq -r '.callback_query.data')
        CALL_BACK_QUERY_MESSAGE=$(echo "$FETCH" | jq -r '.callback_query.message')

    fi
}

update_init() {
    until [ -n "$UPDATE_ID" ]; do
        UPDATE_ID=$(($(curl -s "$API/getUpdates" -d "offset=-1" | jq '.result[].update_id') + 1))
    done
}

is_botowner() {
    [ "$MSGGER" = "$BOT_OWNER_ID" ] && return 0
    return 1
}

err_not_botowner() {
    tg --replymsg "$RET_CHAT_ID" "$RET_MSG_ID" "You are not allowed to use this command."
}

# contains: returns true if an element is present in an array
# (kang from fish)
contains() {
    local element=$1
    shift
    local array=("$@")

    for elem in "${array[@]}"; do
        if [ "$elem" = "$element" ]; then
            return 0
        fi
    done

    return 1
}

# is_admin: return 0 if all passed users are admin, otherwise return 1
is_admin() {
    local chat_id=$1
    shift
    local user_id=("$@")
    local everyone_is_admin=true
    local chat_admins=$(curl -s "$API/getChatAdministrators" -d "chat_id=$chat_id" | jq .result[].user.id)
    chat_admins=($chat_admins) # Convert to an array
    for user in "${user_id[@]}"; do
        if ! contains "$user" "${chat_admins[@]}"; then
            return 1 # This user is not admin, thus not everyone is admin
        fi
    done

    # The loop finished successfully, which means everyone is admin
    return 0
}