199 lines
6.4 KiB
Python
199 lines
6.4 KiB
Python
#!/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
|
|
|
|
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 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)
|
|
|
|
else:
|
|
|
|
print('Error installing the module, not valid url')
|
|
|
|
exit(1)
|