configure_server.py

import time
from server_creator import server_creator
import paramiko
import re
from SECRETS import SHADOWSOCKSR_PASSWORD, SSH_PRIVATE_KEY_PASS, EMAIL_SENDER
from sendemail import get_email_info


# 3.Configure servers
def configure_server(list, server_name):
    hetzner_or_not = int(input("If you want to create a new Hetzner server press 1, if you want to configure your own "
                           "server press 0 and then press Enter. (0/1) "))
    if hetzner_or_not == 1:
        # 3.2.Login to server
        server_access_info = server_creator(server_name)

        # Connection details
        host = server_access_info['server_ip']
        host_ir = input('Tunnel host IP:(if you have if not press Enter) ')
        if host_ir == '':
            host_ir = host
        username = 'root'

        # The path to the private key file
        private_key_file = "./file.key"

        # The password to unlock the private key
        private_key_password = SSH_PRIVATE_KEY_PASS

        # Load the private key
        private_key = paramiko.RSAKey.from_private_key_file(private_key_file, password=private_key_password)
    elif hetzner_or_not == 0:
        host = input("Please enter your host's IP: ")
        host_ir = input('Tunnel host IP:(if you have if not press Enter) ')
        if host_ir == '':
            host_ir = host
        username = 'root'
        password = input("Enter your server's password: ")
        ssh_port = input("Enter your server's SSH port: ")

    # Create an instance of the SSH client
    ssh = paramiko.SSHClient()
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

    try:
        time.sleep(20)
        if hetzner_or_not == 1:
            ssh.connect(host, username=username, pkey=private_key)
        elif hetzner_or_not == 0:
            ssh.connect(host, username=username, password=password)
        # Successful connection message
        print("Connected to {} as {}".format(host, username))

    except Exception as e:
        print("Error: {}".format(e), "Sleep time was 20 seconds. Now going to try 30 seconds.")
        time.sleep(30)
        if hetzner_or_not == 1:
            ssh.connect(host, username=username, pkey=private_key)
        elif hetzner_or_not == 0:
            ssh.connect(host, username=username, password=password)
        # Successful connection message
        print("Connected to {} as {}".format(host, username))

    except Exception as e:
        print("Error: {}".format(e), "Sleep time was 30 seconds. Now going to try 40 seconds.")
        time.sleep(40)
        if hetzner_or_not == 1:
            ssh.connect(host, username=username, pkey=private_key)
        elif hetzner_or_not == 0:
            ssh.connect(host, username=username, password=password)
        # Successful connection message
        print("Connected to {} as {}".format(host, username))


    # Run a command on the VPS
    ssh.exec_command("yes" + "\n")
    ssh.exec_command("\n")
    ssh.exec_command("\n")
    ssh.exec_command("\n")

    # wget ShadowsocksR
    try:
        stdin, stdout, stderr = ssh.exec_command('wget -N --no-check-certificate https://raw.githubusercontent.com/ToyoDAdoubi/doubi/master/ssrmu.sh && chmod +x ssrmu.sh')
        exit_status = stdout.channel.recv_exit_status()  # Blocking call
        if exit_status == 0:
            print("ssrmu.sh downloaded")
        else:
            print("Error", exit_status)
        print("ssrmu.sh download is ok.")
    except Exception as e:
        print("Error: {}".format(e))
        exit()

    # Install ShadowsocksR
    port = 2333
    try:
        stdin, stdout, stderr = ssh.exec_command(f"./ssrmu.sh <<< $'1\n{host_ir}\nAdmin\n{port}\n{SHADOWSOCKSR_PASSWORD}\n10\n5\n2\nn\n\n\n\n\n\n'")
        exit_status = stdout.channel.recv_exit_status()  # Blocking call
        if exit_status == 0:
            print("ShadowsocksR installed.")
        else:
            print("Error", exit_status)
        output = stdout.read().decode()
        print(output)
        print("ShadowsocksR installation is ok.")
    except Exception as e:
        print("Error: {}".format(e))
        exit()

    # Get the user link that starts from ssr://
    pattern = r"ssr:\/\/.*[a-zA-Z]\w"
    match = re.search(pattern, output)
    if match:
        user_link = match.group()
        print('This is Admin\'s user link: ')
        print(user_link)
        print('Going to send it.')
        get_email_info("admin", EMAIL_SENDER, user_link)


    # Create users
    for user in list:
        name = user["name"]
        email = user["email"]
        port += 1
        stdin, stdout, stderr = ssh.exec_command(f"./ssrmu.sh <<< $'7\n1\n{name}\n{port}\n{SHADOWSOCKSR_PASSWORD}\n10\n5\n2\nn\n1\n\n\n\n\nn'")
        exit_status = stdout.channel.recv_exit_status()  # Blocking call
        if exit_status == 0:
            print(f"user {name} created successfully.")
        else:
            print("Error", exit_status)
        # Print the output of the command
        output = stdout.read().decode()

        # Get the user link that starts from ssr://
        pattern = r"ssr:\/\/.*[a-zA-Z]\w"
        match = re.search(pattern, output)
        if match:
            user_link = match.group()
            print(f'This is {name}\'s user link: ')
            print(user_link)
            print('Going to send it.')
            get_email_info(name, email, user_link)

    # Close the connection
    ssh.close()