#!/usr/bin/python3 -u

# A script for install pzoo user

import subprocess
import argparse
import re
import os
import shutil
import pwd
from subprocess import call
from pathlib import Path
from time import sleep
import distro

parser = argparse.ArgumentParser(description='A script for install leviathan user')

#parser.add_argument('--url', help='The url where notify updates', required=True)
parser.add_argument('--url_stats', help='The url where pastafaristats notify the stats', required=True)
parser.add_argument('--user', help='The user for pastafari', required=True)
parser.add_argument('--pub_key', help='The pub key used in pastafari user', required=True)
parser.add_argument('--group', help='Server group', required=False)
parser.add_argument('--path', help='A path used for save internal things for pastafari', required=True)

args = parser.parse_args()

#url=args.url

check_url = re.compile(
        r'^(?:http|ftp)s?://' # http:// or https://
        r'(?:(?:[A-Z0-9](?:[A-Z0-9-]{0,61}[A-Z0-9])?\.)+(?:[A-Z]{2,6}\.?|[A-Z0-9-]{2,}\.?)|' #domain...
        r'localhost|' #localhost...
        r'\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})' # ...or ip
        r'(?::\d+)?' # optional port
        r'(?:/?|[/?]\S+)$', re.IGNORECASE)
        
print('{"error": 0, "status": 0, "progress": 0, "no_progress":0, "message": "Installing tools for monitoring the server..."}')

#check_url.match(args.url) and 

if check_url.match(args.url_stats):
    
    try:
        u=pwd.getpwnam(args.user)
        
        if call("sudo userdel -r %s" % args.user, shell=True) > 0:
            print('Error, user with same username exists and cannot delete of the server')
            exit(1)
        else:
            print('Cleaning user with the same name')
        
    except:
        pass

    # Create users

    if call("sudo useradd -m -s /bin/bash %s" % args.user, shell=True) > 0:
        
        # Delete user with this name, you should not install it in a old server. 
        
        if call("sudo userdel -r %s" % args.user, shell=True) > 0:
            print('Error, remove the old user')
            exit(1)            
        else:

            if call("sudo useradd -m -s /bin/sh %s" % args.user, shell=True) > 0:
                print('Error, cannot add a new user')
                exit(1)            
        
    else:
        print('Added user')
        
    if call("sudo mkdir -p /home/"+args.user+"/.ssh && sudo chown "+args.user+":"+args.user+" /home/"+args.user+"/.ssh && sudo chmod 700 /home/"+args.user+"/.ssh", shell=True) > 0:
        print('Error, cannot add ssh directory')
        exit(1)
    else:
        print('Added ssh directory')
        
    if call("sudo cp "+args.pub_key+" /home/"+args.user+"/.ssh/authorized_keys && sudo chown "+args.user+":"+args.user+" /home/"+args.user+"/.ssh/authorized_keys && sudo chmod 600 /home/"+args.user+"/.ssh/authorized_keys", shell=True) > 0:
        print('Error, cannot pub key to user')
        exit(1)
    else:
        print('Added pub key to user')
        
    # Edit sudo file

    with open('modules/pastafari2/scripts/system/sudoers.d/spanel') as f:
        sudoers=f.read()

    with open('/etc/sudoers.d/spanel', 'w') as f:
        sudoers=sudoers.replace("spanel", args.user)
        f.write(sudoers)

    # Installing composer things for php alerts...
    
    
    #mkdir $HOME/pzoo/scripts

    #chown $USER:$USER $HOME/pzoo/scripts

    #su - $USER -s /bin/bash -c "composer --working-dir=$HOME/scripts require guzzlehttp/guzzle:~6.0"
    
    args.path=os.path.basename(args.path)
    
    pt=Path('/home/'+args.user+'/'+args.path+'/tasks')
    
    pt.mkdir(mode=0o755, parents=True, exist_ok=True)
    
    shutil.chown('/home/'+args.user+'/'+args.path, args.user, args.user)
    
    shutil.chown('/home/'+args.user+'/'+args.path+'/tasks', args.user, args.user)
    
    # Create scripts pzoo
    """
    ps=Path('/home/'+args.user+'/pzoo/scripts')
    
    ps.mkdir(mode=0o755, parents=True, exist_ok=True)
    
    shutil.chown('/home/'+args.user+'/pzoo', args.user, args.user)
    
    shutil.chown('/home/'+args.user+'/pzoo/scripts', args.user, args.user)

    
    if call("su - "+args.user+" -s /bin/bash -c \"composer --working-dir=/home/"+args.user+"/pzoo/scripts require guzzlehttp/guzzle:~6.0\"", shell=True)>0:
        print('Error, cannot install scripts for use in check scripts')
        exit(1)
    else:
        print('Composer dependencies for check scripts added successfully')    

    # Edit get_updates.py

    with open('scripts/system/get_updates.py') as f:
        get_updates=f.read()

    with open('/etc/cron.daily/get_updates.py', 'w') as f:
        url_updates=args.url.replace('/getinfo/', '/getupdates/')
        get_updates=get_updates.replace("http://url/to/server/token/ip", url_updates)
        f.write(get_updates)

    os.chmod('/etc/cron.daily/get_updates.py', 0o700)
    """
    if call("sudo /opt/pythonenv/bin/pip3 install --upgrade git+https://bitbucket.org/paramecio/pastafaristats", shell=True)>0:
        print('Error, cannot install pastafari stats')
        exit(1)
    else:
        print('Added pastafari stats')
        
    # Add configuration to pastafari stats
    
    if not os.path.isdir('/etc/pastafari'):
        # Create pastafari dir
        p=Path('/etc/pastafari')
        p.mkdir(mode=0o755, parents=False, exist_ok=True)
        
    with open('/etc/pastafari/stats.cfg', 'w') as f:
        
        f.write("[DEFAULT]\n\nurl_server="+args.url_stats+"\ngroup="+args.group)
        
    
    with open('/etc/systemd/system/pastafaristats.service', 'w') as f:
        
        systemd_unit="""
        # Save it in /etc/systemd/system/pastafaristats.service

        [Unit]
        Description=Pastafari Stats
        After=syslog.target
        After=network.target

        [Service]
        Type=simple
        User=pzoo
        Group=pzoo
        ExecStart=pastafaristats
        Restart=always
        Environment=PYTHONUNBUFFERED=1

        [Install]
        WantedBy=multi-user.target
        """
        
        systemd_unit=systemd_unit.replace('pzoo', args.user)
        
        f.write(systemd_unit)
        
    if call("sudo systemctl enable pastafaristats.service && sudo systemctl start pastafaristats.service", shell=True)>0:
        print('Error, cannot start pastafari stats')
        exit(1)
    else:
        print('Pastafari stats ready')    

    print('{"error": 0, "status": 0, "progress": 100, "no_progress":0, "message": "Tools installed..."}')

    sleep(1)
    
    linux_distro=distro.id()
    
    # sudo su - %s -s /bin/bash -c 'mkdir -p %s'
    
    if linux_distro=='arch':
        print('Add utilities for archlinux distro...')
        if call("sudo su - %s -s /bin/bash -c 'sudo pacman -S --noconfirm --needed git base-devel && git clone https://aur.archlinux.org/yay.git && cd yay && makepkg -si  --noconfirm --needed'" % args.user, shell=True) > 0:
            print('Error, cannot add utilities for archlinux')
            exit(1)

else:
    
    print('Error installing the module, not valid url')
    
    exit(1)