from discord.ext.commands import Bot
import paramiko
from contextlib import contextmanager

TOKEN = 'NTMwNzgzNjIxNDY0ODUwNDcz.DxEbgA.0x0tIateVzkA8nJuYP9YmA-iiWA'
BOT_PREFIX = ('!')

client = Bot(command_prefix=BOT_PREFIX)

@client.command()
async def on_message(message):

    if message.author == client.user:
        return

    if message.content.startswith('$start server'):
        await client.send_message(message.channel, 'IP address? Type $IP xx.xx.xx.xx')

        def check(msg):
            return msg.content.startswith('$IP')

        message = await client.wait_for_message(author=message.author, check=check)
        IP = message.content[len('$IP'):].strip()
        await client.send_message(message.channel, 'port number? Type $port xxxxx')

        def check2(msg):
            return msg.content.startswith('$port')

        message = await client.wait_for_message(author=message.author, check=check2)
        port = message.content[len('$port'):].strip()
        await client.send_message(message.channel, 'Username? Type $username xxxxxx')

        def check3(msg):
            return msg.content.startswith('$username')

        message = await client.wait_for_message(author=message.author, check=check3)
        username = message.content[len('$username'):].strip()
        await client.send_message(message.channel, 'Password? Type $password xxxxxxx')

        def check4(msg):
            return msg.content.startswith('$password')

        message = await client.wait_for_message(author=message.author, check=check4)
        password = message.content[len('$password'):].strip()
        await client.send_message(message.channel, 'Connecting to server')

        ssh = paramiko.SSHClient()
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        ssh.connect(IP, port=port, username=username, password=password)
        await client.send_message(message.channel, 'Connected')

@client.event
async def on_ready():
    print('Logged in as')
    print(client.user.name)
    print(client.user.id)
    print('------')

client.run(TOKEN)