main.py

from bin import sentinel
from lib import config
from lib import desire_config
import time
import datetime
import sys
import os
import shutil
import string
import random
import math
import logging
import traceback

from builtins import input

try:
    from colorama import init
    from termcolor import colored
    from pynput import keyboard

    DISPLAY_AVAILABLE = True
except:
    DISPLAY_AVAILABLE = False


def random_string(n):
    rnd = random.SystemRandom()
    pool = string.ascii_lowercase + string.ascii_uppercase + string.digits
    m = len(pool) - 1

    return ''.join(pool[rnd.randint(0, m)] for _ in range(n))

def run_sentinel():
    logger = logging.getLogger('sentinel')
    print(colored('Sentinel runs every 1 minute', 'green'))

    while True:
        now = datetime.datetime.now().strftime("%Y-%m-%d %H:%M")

        print(colored('{} Running sentinel'.format(now), 'green'))
        try:
            sentinel.entrypoint()
        except Exception as e:
            print(colored('Error: {}'.format(e), 'red'))

            logger.error(str(e))
            logger.error(traceback.format_exc())
            logger.error('--------------------')
        
        time.sleep(60) # Wait for a minute

def fix_masternode(data_folder):
    try:
        wallet_file = os.path.join(data_folder, 'wallet.dat')

        if not os.path.isfile(wallet_file):
            print('It seems like the data folder (the one containing wallet.dat) is not the same as the folder where desire.conf is')
            data_folder = input('Please, write the data folder path: ')
            return fix_masternode(data_folder)

        print(colored('Make a copy of "wallet.dat" and "desire.conf" before continuing.\nThis program will try it best not to touch them, but just in case!', attrs=['bold']))
        confirm = input('Once done, press [ENTER], or write cancel + [ENTER] to exit\n')
        if confirm.lower() == "cancel":
            return

        print(colored('Removing conflicting files', 'green'))
        for filename in os.listdir(data_folder):
            if filename in ('wallet.dat', 'desire.conf'):
                continue
            
            realpath = os.path.join(data_folder, filename)
            if os.path.isfile(realpath):
                os.remove(realpath)
            elif os.path.isdir(realpath):
                shutil.rmtree(realpath)
            
            print('\t{}'.format(filename))

        print(colored('Done removing', 'green'))
        print('')

        print(colored('Checking desire.conf contents', 'green'))
        tokens = desire_config.DesireConfig.tokenize(config.desire_conf)

        # Check if needed tokens are there, set them if not
        tokens['rpcuser'] = tokens.get('rpcuser', random_string(18))
        tokens['rpcpassword'] = tokens.get('rpcpassword', random_string(18))
        tokens['rpcconnect'] = tokens.get('rpcconnect', '127.0.0.1')
        tokens['rpcport'] = tokens.get('rpcport', 9918)
        tokens['server'] = 1

        # It must have a masternode=1
        try:
            if int(tokens['masternode']) != 1:
                raise Exception
        except:
            r = None
            while r not in ('y', 'n'):
                print(colored('It seems like your desire.conf is not setting the wallet as masternode!', 'red'))
                print(colored('If your are running a COLD wallet, this must be executed on the MN wallet, not the collateral', 'red'))
                print(colored('If you are using MNs hosts like HostMNs DO NOT continue, this should be done by them, not you!', 'red'))
                r = input(colored('Continuing will set masternode=1, don\'t do it if this is a collateral wallet, [y/n]: ', 'red', attrs=['bold']))

                if r == 'n':
                    rewrite = False
                elif r != 'y':
                    print('Please write y or n')

            if r == 'y':
                tokens['masternode'] = 1
            else:
                sys.exit(1)

        # It must have a masternodeprivkey
        if 'masternodeprivkey' not in tokens:
            print(colored('Your desire.conf does not contain a \'masternodeprivkey\' entry, please set it up before opening your wallet', 'red'))

        # Rewrite config
        with open(config.desire_conf, 'w') as fp:
            for k, v in tokens.items():
                fp.write('{}={}\n'.format(k, v))

        print(colored('Done checking\n', 'green'))
        print(colored('Please open your wallet (desired on linux) and wait for it completely load/open before continuing', attrs=['bold']))
        input('Press [ENTER] to continue')
        print('')

        print(colored('Sentinel will automatically start in 15s', 'green', attrs=['bold']))
        print(colored('Until the wallet is fully synced, sentinel will show errors related to sync status, that\'s fine!', 'green'))
        print(colored('If it says it can not connect, check your desire.conf settings', 'yellow'))

        r = 15
        while r > 0:
            sys.stdout.write('Starting in {}s \r'.format(r))
            sys.stdout.flush()
            r -= 1
            time.sleep(1)

        print('\n')
        run_sentinel()
    except Exception as e:
        print(colored('Error: {}'.format(e), 'red'))

        logger.error(str(e))
        logger.error(traceback.format_exc())
        logger.error('--------------------')
    
# @https://stackoverflow.com/questions/24582233/python-flush-input-before-raw-input
def flush_input():
    try:
        import sys, termios
        termios.tcflush(sys.stdin, termios.TCIOFLUSH)
    except ImportError:
        import msvcrt
        while msvcrt.kbhit():
            msvcrt.getch()

def menu():
    args = config.get_args()

    if args.repair and args.sentinel:
        print(colored('Do not specify --repair and --sentinel at the same time', 'red'))
        sys.exit(1)
    elif args.sentinel:
        return 1
    elif args.repair:
        return 2

    print('Select an option:')
    print('\t1. Start sentinel')
    print('\t2. Fix wallet and masternode')

    global menu_option 
    menu_option = 0

    # No PyInput means no auto-start if no command line option was given, sry!
    if not DISPLAY_AVAILABLE:
        while menu_option not in (1, 2):
            try:
                menu_option = int(input('Please write 1 or 2 and press [ENTER]: '))
            except:
                pass
                
        return menu_option

    print('')
    

    def on_release(key):
        global menu_option 

        try:
            if key.char in ('1', '2'):
                menu_option = int(key.char)
                return False
            else:
                print(colored('Option not valid', 'red'))
        except:
            pass

    # Collect events until released
    with keyboard.Listener(on_release=on_release) as listener:
        r = 30
        while menu_option == 0 and r > 0:
            sys.stdout.write('\rPress 1 or 2. Sentinel will automatically start in {}s  '.format(int(math.ceil(r))))
            sys.stdout.flush()

            r -= 0.1
            time.sleep(0.1)

        listener.stop()

    if menu_option == 0:
        menu_option = 1

    flush_input()

    print('\n')
    return menu_option

if __name__ == '__main__':
    init()

    # Setup logging
    logger = logging.getLogger('sentinel')
    hdlr = logging.FileHandler('sentinel.log')
    formatter = logging.Formatter('%(asctime)s %(levelname)s %(message)s')
    hdlr.setFormatter(formatter)
    logger.addHandler(hdlr) 
    logger.setLevel(logging.INFO)

    if os.path.isfile(config.sentinel_config_file):
        print(colored('Using sentinel.conf: {}'.format(config.sentinel_config_file), 'green'))

    print(colored('Using desire.conf: {}'.format(config.desire_conf), 'green'))

    option = menu()
    if option == 1: 
        run_sentinel()
    elif option == 2: 
        # Use default data folder
        data_folder = os.path.dirname(config.desire_conf)
        fix_masternode(data_folder)