installer.py

# In the name of Allah

import os
import sys
import atexit
import time
import shutil
import json
import random
from distutils.dir_util import copy_tree

colors = {
    'RED': '\033[1;31m',
    'GREEN': '\033[1;32m',
    'YELLOW': '\033[1;33m',
    'MAGENTA': '\033[1;35m',
    'BLUE': '\033[1;34m',
    'CYAN': '\033[1;36m',
    'WHITE': '\033[1;37m'
}

info = {'python': None, 'license_accepted': False,
        'installtion_complete': False, 'username': None, 'home': None, 'php': None,'host':True,'secret':None,'jupyter_installed':False,'show_hidden_files':False,'theme':'monokai','password':None,'ips':[]}
sys.stdout.write(colors['MAGENTA'])
print('In the name of Allah\n')
time.sleep(2)

sys.stdout.write(colors['YELLOW'])
print('MWX\nmwxgaf.ir\nmwxgaf@yahoo.com\nTwitter: @mwxgaf\n')
time.sleep(2)

print('PyLocalhost\n\nmwxgaf.github.io/pylocalhost\n\n')
time.sleep(2)

# Making atexit function


def end():
    if not info['installtion_complete']:
        sys.stdout.write(colors['YELLOW'])
        print('\nInstaller exited from installation process and result was not successfull!\nIf you have an issue please submit it at Github repository of project:\nhttps://github.com/mwxgaf/pylocalhost\n')
    else:
        sys.stdout.write(colors['GREEN'])
        print('\nSaving your system info ...')
        os.chdir('/etc/pylocalhost/')
        info_json_file = open('info.json', 'w')
        json.dump(info, info_json_file)
        info_json_file.close()
        shutil.chown('/etc/pylocalhost/info.json',info['username'])
        print('\nInstallation completed successfully\nThank you for using Pylocalhost :)')


atexit.register(end)

# Defining some required functions

def run(command):
	p = os.popen(command)
	os.waitpid(p._proc.pid,0)

def unknown_error(e):
    sys.stdout.write(colors['RED'])
    print('Unexpected error: {}\n'.format(e))
    sys.exit(500)

def permission_error():
    sys.stdout.write(colors['RED'])
    print('Not enough permissions! run file with sudo\n')
    sys.exit(403)

def remove_or_go(path):
    try:
        os.remove(path)
    except FileNotFoundError:
        pass
    except Exception as e:
        unknown_error(e)

# Getting user info

try:
    preinfofile = open('pylhinfo.json')
except FileNotFoundError:
    sys.stdout.write(colors['RED'])
    print('Could not find pylhinfo.json file!\nPlease read the installation guid from this link:\nhttps://mwxgaf.github.io/pylocalhost')
    sys.exit(404)
preinfo = json.load(preinfofile)
info['home'] = preinfo['home']
info['username'] = preinfo['user']
preinfofile.close()

remove_or_go('pylhinfo.json')
remove_or_go('preinstaller.py')

# Asking user for accepting the license
sys.stdout.write(colors['CYAN'])
print('This software is licensed under MPL-2.0\n(Mozilla public license - version 2.0)\n')
while not info['license_accepted']:
    sys.stdout.write(colors['YELLOW'])
    accept_question = input('Are you agree with the license (y/n)? ')
    print('\n')
    if accept_question.lower() == 'y':
        info['license_accepted'] = True
    elif accept_question.lower() == 'n':
        sys.stdout.write(colors['RED'])
        print('You must be agree with license to install and use this software!\n')
        sys.exit(406)
    else:
        continue


# Check if OS is supported or not
if os.name == 'nt':
    sys.stdout.write(colors['RED'])
    print('Microsoft Windows OS is not supported yet!')
    sys.exit(400)
sys.stdout.write(colors['GREEN'])
print('Operating system supported! Starting dependencies installation check ...\n')

# Check if dependencies are installed or not


def check_installtion(package):
    if not os.popen('which {}'.format(package)).read().startswith('/'):
        sys.stdout.write(colors['RED'])
        print('Clound not detect {} installation!'.format(package))
        sys.exit(404)
    sys.stdout.write(colors['CYAN'])
    print('{} installation found! ...'.format(package))


for p in ['nginx', 'git']:
    check_installtion(p)
sys.stdout.write(colors['GREEN'])
print('Checking completed successfully!\n')

# Checking and finding python 3
for p in ['python --version','python3 --version']:
    try:
        if os.popen(p).read()[7] == '3':
                info['python'] = p.split()[0]
    except:
        pass
if not info['python']:
    sys.stdout.write(colors['RED'])
    print('Could not find python version 3 in your system!')
    sys.exit(404)
sys.stdout.write(colors['GREEN'])
print('Python version 3 found! Starting installtion ...\n')

# Checking PHP_FPM
sys.stdout.write(colors['CYAN'])
print('Checking PHP FPM ...')
for p in ['7.0', '7.1', '7.2', '7.3', '7.4','8.0']:
    if os.popen('which php-fpm'+p).read().startswith('/'):
        info['php'] = p
if not info['php']:
    sys.stdout.write(colors['RED'])
    print('\nCould not find PHP-FPM in your system so PyLocalhost will not be able to serve PHP files \n')
    info['php'] = '7.4'
else:
    sys.stdout.write(colors['GREEN'])
    print('\nPHP-FPM found!')

# Installing virtualenv
sys.stdout.write(colors['CYAN'])
print('Installing virtualenv ...')
try:
	run('sudo -H {} -m pip install virtualenv'.format(info['python']))
except BrokenPipeError:
    pass
check_installtion('virtualenv')

# Initializing venv

sys.stdout.write(colors['CYAN'])
print('Creating main directory ...\n')

try:
    os.chdir('/etc/')
except FileNotFoundError:
    sys.stdout.write(colors['RED'])
    print('Seems that your operating system is not supported!\n')
    sys.exit(404)
except Exception as e:
    unknown_error(e)

tries = 5


def make_main_dir(tries):
    try:
        os.mkdir('./pylocalhost')
    except FileExistsError:
        if tries != 0:
            try:
                shutil.rmtree('./pylocalhost')
            except PermissionError:
                permission_error()
            except Exception as e:
                unknown_error(e)
            tries -= 1
        else:
            unknown_error('Could not make main directory!\n')
        make_main_dir(tries)
    except PermissionError:
        permission_error()
    except Exception as e:
        unknown_error(e)


make_main_dir(tries)

os.chdir('/etc/pylocalhost')
sys.stdout.write(colors['CYAN'])
print('Initializing virtual environment ...\n')

run('sudo {} -m virtualenv .venv'.format(info['python']))

if not os.path.isfile('/etc/pylocalhost/.venv/bin/python'):
    unknown_error('Could not initialize virtual environment\n')

print('Cloning software from Github ...')

run('sudo git clone https://github.com/mwxgaf/pylocalhost.git temp')
copy_tree('/etc/pylocalhost/temp', '/etc/pylocalhost')
shutil.rmtree('/etc/pylocalhost/temp')

if not os.path.isfile('/etc/pylocalhost/app.py'):
    unknown_error('\nCloud not clone git repository successfully!')
        
print('Installing dependencies ... Please wait it may take a while ...')

try:
    run('sudo -H /etc/pylocalhost/.venv/bin/pip install -r requirements.txt')
except BrokenPipeError:
    pass

while not info['jupyter_installed']:
    sys.stdout.write(colors['YELLOW'])
    jupyter_question = input('Do you want to install Jupyter for PyLocalHost (y/n)?')
    if jupyter_question.lower() == 'y':
        sys.stdout.write(colors['CYAN'])
        print('Installing Jupyter please wait it may take a while ...')
        try:
            run('sudo -H /etc/pylocalhost/.venv/bin/pip install jupyter')
        except BrokenPipeError:
            pass
        info['jupyter_installed'] = True
        sys.stdout.write(colors['GREEN'])
        print('Successfully installed Jupyter')
    else:
        sys.stdout.write(colors['GREEN'])
        print('No problem! Jupyter wont be installed.')
        break       

if not os.path.isfile('/etc/pylocalhost/.venv/bin/gunicorn'):
    unknown_error('\nCould not install dependencies!')

# Create and write Nginx and Gunicorn config file
sys.stdout.write(colors['CYAN'])
print('Configuring your system ...')

os.chdir(info['home'])
os.makedirs('Pylocalhost', exist_ok=True)
shutil.chown(os.path.join(info['home'], 'Pylocalhost'), user=info['username'])
os.popen('sudo chmod ug=rwx,o=rx Pylocalhost/')
os.chdir('/etc/pylocalhost')

pylocalhost_file = open('/etc/nginx/sites-available/pylocalhost', 'w')
pylocalhost_file.write('server {\n    listen 80 default_server;\n    listen [::]:80;\n    root %s/Pylocalhost;\n    server_name localhost;\n    location /static {\n        alias /etc/pylocalhost/static;\n        expires 365d;\n    }\n    location /s {\n        index index.html index.php index.htm;\n        alias %s/Pylocalhost;\n\n        location ~ \.php$ {\n            fastcgi_pass unix:/run/php/php%s-fpm.sock;\n            include fastcgi_params;\n            fastcgi_param SCRIPT_FILENAME $request_filename;\n        }\n    }\n    location /t {\n        alias %s/Pylocalhost;\n        types{} default_type "text/plain";\n    }\n    location /d {\n        alias %s/Pylocalhost;\n        types {} default_type "application/octet-stream";\n    }\n    location / {\n        try_files $uri @wsgi;\n    }\n    location @wsgi {\n        proxy_pass http://unix:/tmp/gunicorn.sock;\n        include proxy_params;\n    }\n}' % (info['home'], info['home'], info['php'] , info['home'] , info['home']))
pylocalhost_file.close()

gunicorn_file = open('/etc/systemd/system/gunicorn.service', 'w')
gunicorn_file.write('[Unit]\nDescription=gunicorn daemon for Pylocalhost\nAfter=network.target\n\n[Service]\nUser={}\nGroup=www-data\nRuntimeDirectory=gunicorn\nWorkingDirectory=/etc/pylocalhost/\nExecStart=/etc/pylocalhost/.venv/bin/gunicorn --bind=unix:/tmp/gunicorn.sock --workers=2 app:app\nExecReload=/bin/kill -s HUP $MAINPID\nExecStop=/bin/kill -s TERM $MAINPID\n\n[Install]\nWantedBy=multi-user.target'.format(info['username']))
gunicorn_file.close()

shutil.move('/etc/nginx/nginx.conf', '/etc/nginx/nginx.conf.default')
shutil.copy('/etc/pylocalhost/nginx.conf', '/etc/nginx/nginx.conf')

remove_or_go('/etc/nginx/sites-enabled/default')
remove_or_go('/etc/nginx/sites-enabled/pylocalhost')

os.symlink('/etc/nginx/sites-available/pylocalhost',
           '/etc/nginx/sites-enabled/pylocalhost')

hosts_file = open('/etc/hosts','a')
hosts_file.write('\n127.0.0.1  pylh\n127.0.0.1  pylocalhost')
hosts_file.close()

print('Reloading Systemctl daemon ...')

run('sudo systemctl daemon-reload')

print('Intsalling CLI ...')

remove_or_go('/usr/bin/pylocalhost')
remove_or_go('/usr/bin/pylh')

shutil.copy('/etc/pylocalhost/pylocalhost.sh', '/usr/bin/pylocalhost')
shutil.chown('/usr/bin/pylocalhost', user='root')
os.popen('sudo chmod u=rwx,og=rx /usr/bin/pylocalhost')
os.symlink('/usr/bin/pylocalhost', '/usr/bin/pylh')

if not os.path.isfile('/usr/bin/pylocalhost') or not os.path.islink('/usr/bin/pylh'):
    unknown_error('Could not install CLI')

print('Downloading CodeMirror files ...')
os.mkdir('/etc/pylocalhost/static/codemirror')
os.chdir('/etc/pylocalhost/static/codemirror')

for url in ['http://bayanbox.ir/view/4819108218995809028/cobalt.css','http://bayanbox.ir/view/5845364735760611073/codemirror.css','http://bayanbox.ir/view/8723841575246188377/codemirror.js','http://bayanbox.ir/view/2163356795121549378/css.js','http://bayanbox.ir/view/5667818355064209123/eclipse.css','http://bayanbox.ir/view/1474701487801988577/xml.js','http://bayanbox.ir/view/1421977402385602506/isotope.css','http://bayanbox.ir/view/8235731700109004525/javascript.js','http://bayanbox.ir/view/9118313191000262204/markdown.js','http://bayanbox.ir/view/1267880735759285854/monokai.css','http://bayanbox.ir/view/9178237286963248335/neat.css','http://bayanbox.ir/view/3399704187251334873/php.js','http://bayanbox.ir/view/3788822247937739817/python.js','http://bayanbox.ir/view/1577133095062831655/sublime.js']:
	run(f'wget {url}')

if not os.path.isfile('/etc/pylocalhost/static/codemirror/python.js'):
	unknown_error('Could not download CodeMirror files')

print('Creating test files ...')
try:
    copy_tree('/etc/pylocalhost/templates/test',os.path.join(info['home'],'Pylocalhost'))
    shutil.chown(os.path.join(info['home'],'Pylocalhost','test.html'),user=info['username'])
    shutil.chown(os.path.join(info['home'],'Pylocalhost','test.php'),user=info['username'])
except:
    pass

print('Making secret key ...')
info['secret'] = str(random.randint(10000000000000,99999999999999))
info['password'] = str(random.randint(10000,99999))

print('Finalizing installtion ...')

time.sleep(2)
info['installtion_complete'] = True

sys.stdout.write(colors['GREEN'])
print('\nDone! :)\n')
sys.stdout.write(colors['MAGENTA'])
print('>>> Your Password is ' + info['password'] + ' <<< Use it to login')

sys.exit(0)