CloudBot.py

import json
import requests
import subprocess
import argparse
import logging
import sys


class botCli(object):

    def __init__(self,path,url,token):
        self.path = path
        self.url= url.format(token)

    def get_url(self,url):
        try:
            response = requests.get(url)
            content = response.content.decode("utf8")
            return content
        except requests.exceptions.ConnectionError:
            logging.error("Problem with Telegram Connection")

    def send_message(self,text, chat_id):
        url = self.url + "sendMessage?text={}&chat_id={}".format(text, chat_id)
        self.get_url(url)

    def get_json_from_url(self,url):
        js = {}
        try:
            content = self.get_url(url)
            js = json.loads(content)
        except TypeError:
            logging.error("Problem parsing JSON")
        return js

    def get_updates(self):
        url = self.url + "getUpdates?timeout=5"
        js = self.get_json_from_url(url)
        return js

    def get_last_chat_id_and_text(self,updates):
            num_updates = len(updates["result"])
            last_update = num_updates - 1
            text = updates["result"][last_update]["message"]["text"]
            chat_id = updates["result"][last_update]["message"]["chat"]["id"]
            m_id = updates["result"][last_update]["update_id"]
            return (chat_id,text,m_id)


    def execute_analysis_aws(self,chat_id,args):
        try:
            cmd =[self.path,"-M","mono"]
            args.pop(0)
            for i in args:
                cmd.append(i)
            p = subprocess.Popen(cmd,
                                 stdout=subprocess.PIPE,
                                 stderr=subprocess.STDOUT)

            for line in iter(p.stdout.readline, b''):
                logging.info(line.decode('utf-8'))
                self.send_message(line.decode('utf-8'),chat_id)
        except FileNotFoundError:
            logging.error("Prowler not found")
    def execute_nmap(self,chat_id,args):
        cmd = ["nmap"]
        args.pop(0)
        try:
            for i in args:
                cmd.append(i)
            p = subprocess.Popen(cmd,
                                 stdout=subprocess.PIPE,
                                 stderr=subprocess.STDOUT)

            for line in iter(p.stdout.readline, b''):
                logging.info(line.decode('utf-8'))
                self.send_message(line.decode('utf-8'),chat_id)
        except FileNotFoundError:
            logging.error("Nmap not found")

    def get_initID(self):
        init_update = 0
        if(len(self.get_json_from_url(self.url + "getUpdates?")["result"])==0):
           init_update = 0
        else:
            _, _, init_update = self.get_last_chat_id_and_text(self.get_updates())

        return init_update

    def run(self,chat_id):
        last_id = self.get_initID()
        logging.info("[*][*][*][*][*] Running Telegram Server Bot ... [*][*][*][*][*]")
        while True:
            try:
                user_id, msg, current_id = self.get_last_chat_id_and_text(self.get_updates())

                if user_id in chat_id and current_id > last_id:
                    last_id = current_id
                    if msg.split(' ', 1)[0] == "/ScanAWS":
                        if len(msg.split())==1:
                            self.send_message(
                                "Unable to locate credentials. You can configure credentials by running \"aws configure\"",
                                user_id)
                        else:
                            self.execute_analysis_aws(user_id, msg.split())
                            logging.info("Request of %i User to execute ScanAWS",user_id)
                    if msg.split(' ', 1)[0] == "/Nmap":
                        self.execute_nmap(user_id,msg.split())
                        logging.info("Request of %i User to execute Nmap", user_id)

                    else:
                        self.send_message("To scan your AWS Account use /ScanAWS -p \"profileAWS\", to scan another network could use /Nmap with {nmap args}",user_id)
                        logging.info("Request Failed of %i User without corrects parameters", user_id)

            except IndexError:
                logging.error("Problem parsing JSON")
            except TypeError:
                logging.error("Problem parsing JSON")
            except KeyError:
                logging.error("Problem parsing JSON")

if __name__ == '__main__':

    if sys.version_info < (3, 5,):
        print("To run Telegram Bot Server you must use Python 3.5+")
        sys.exit(0)

    FORMAT = '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
    logging.basicConfig(format=FORMAT,level=logging.INFO)


    #Arguments
    parser = argparse.ArgumentParser(description='[+][+] Telegram Bot Server to audit AWS Security Checks')
    parser.add_argument('--token','-t', type=str, required=True ,help='Token API Telegram Bot')
    parser.add_argument('--path',"-p", type=str,  required=True, help='Prowler Path')
    parser.add_argument('--users', '-u',type=int ,required=True , nargs='+', help='Users allowed')

    args = parser.parse_args()

    url = "https://api.telegram.org/bot{}/"

    #Create de cli
    botcli = botCli(args.path,url,args.token)
    #Your ID's user
    botcli.run(args.users)