import os import textwrap from optparse import make_option from os.path import expanduser from django.conf import settings from django.core.management.base import BaseCommand, CommandError from orchestra.utils import paths from orchestra.utils.sys import run, check_root, get_default_celeryd_username class Command(BaseCommand): def __init__(self, *args, **kwargs): super(Command, self).__init__(*args, **kwargs) self.option_list = BaseCommand.option_list + ( make_option('--cert', dest='cert', default='', help='Nginx SSL certificate, one will be created by default.'), make_option('--cert-key', dest='cert_key', default='', help='Nginx SSL certificate key.'), make_option('--cert-path', dest='cert_path', default='/etc/nginx/ssl/orchestra.crt', help='Nginx SSL certificate, one will be created by default.'), make_option('--cert-key-path', dest='cert_key_path', default='/etc/nginx/ssl/orchestra.key', help='Nginx SSL certificate key.'), # Cert options make_option('--cert-override', dest='cert_override', action='store_true', default=False, help='Force override cert and keys if exists.'), make_option('--cert-country', dest='cert_country', default='ES', help='Certificate Distinguished Name Country.'), make_option('--cert-state', dest='cert_state', default='Spain', help='Certificate Distinguished Name STATE.'), make_option('--cert-locality', dest='cert_locality', default='Barcelona', help='Certificate Distinguished Name Country.'), make_option('--cert-org_name', dest='cert_org_name', default='Orchestra', help='Certificate Distinguished Name Organization Name.'), make_option('--cert-org_unit', dest='cert_org_unit', default='DevOps', help='Certificate Distinguished Name Organization Unity.'), make_option('--cert-email', dest='cert_email', default='orchestra@orchestra.lan', help='Certificate Distinguished Name Email Address.'), make_option('--cert-common_name', dest='cert_common_name', default=None, help='Certificate Distinguished Name Common Name.'), make_option('--server-name', dest='server_name', default='', help='Nginx SSL certificate key.'), make_option('--user', dest='user', default='', help='uWSGI daemon user.'), make_option('--group', dest='group', default='', help='uWSGI daemon group.'), make_option('--processes', dest='processes', default=4, help='uWSGI number of processes.'), make_option('--noinput', action='store_false', dest='interactive', default=True, help='Tells Django to NOT prompt the user for input of any kind. ' 'You must use --username with --noinput, and must contain the ' 'cleeryd process owner, which is the user how will perform tincd updates'), ) option_list = BaseCommand.option_list help = 'Configures nginx + uwsgi to run with your Orchestra instance.' def generate_certificate(self, **options): override = options.get('cert_override') interactive = options.get('interactive') cert = options.get('cert') key = options.get('cert_key') if bool(cert) != bool(key): raise CommandError("--cert and --cert-key go in tandem") cert_path = options.get('cert_path') key_path = options.get('cert_key_path') run('mkdir -p %s' % os.path.basename(cert_path)) exists = os.path.isfile(cert_path) if not override and exists: self.stdout.write('Your cert and keys are already in place.') self.stdout.write('Use --override in order to regenerate them.') return cert_path, key_path common_name = options.get('cert_common_name') or options.get('server_name') or 'orchestra.lan' country = options.get('cert_country') state = options.get('cert_state') locality = options.get('cert_locality') org_name = options.get('cert_org_name') org_unit = options.get('cert_org_unit') email = options.get('cert_email') if interactive: msg = ('-----\n' 'You are about to be asked to enter information that\n' 'will be incorporated\n' 'into your certificate request.\n' 'What you are about to enter is what is called a\n' 'Distinguished Name or a DN.\n' 'There are quite a few fields but you can leave some blank\n' '-----\n') self.stdout.write(msg) msg = 'Country Name (2 letter code) [%s]: ' % country country = input(msg) or country msg = 'State or Province Name (full name) [%s]: ' % state state = input(msg) or state msg = 'Locality Name (eg, city) [%s]: ' % locality locality = input(msg) or locality msg = 'Organization Name (eg, company) [%s]: ' % org_name org_name = input(msg) or org_name msg = 'Organizational Unit Name (eg, section) [%s]: ' % org_unit org_unit = input(msg) or org_unit msg = 'Email Address [%s]: ' % email email = input(msg) or email self.stdout.write('Common Name: %s' % common_name) subject = { 'C': country, 'S': state, 'L': locality, 'O': org_name, 'OU': org_unit, 'Email': email, 'CN': common_name, } context = { 'subject': ''.join(('/%s=%s' % (k,v) for k,v in subject.items())), 'key_path': key_path, 'cert_path': cert_path, } self.stdout.write('writing new cert to \'%s\'' % cert_path) self.stdout.write('writing new cert key to \'%s\'' % key_path) run('openssl req -x509 -nodes -days 365 -newkey rsa:4096 -keyout %(key_path)s -out %(cert_path)s -subj "%(subject)s"' % context, display=True) return cert_path, key_path @check_root def handle(self, *args, **options): user = options.get('user') if not user: raise CommandError("System user for running uwsgi must be provided.") cert_path, key_path = self.generate_certificate(**options) server_name = options.get('server_name') context = { 'cert_path': cert_path, 'key_path': key_path, 'project_name': paths.get_project_name(), 'project_dir': paths.get_project_dir(), 'site_dir': paths.get_site_dir(), 'static_root': settings.STATIC_ROOT, 'user': user, 'group': options.get('group') or user, 'home': expanduser("~%s" % options.get('user')), 'processes': int(options.get('processes')), 'server_name': 'server_name %s' % server_name if server_name else '' } nginx_conf = textwrap.dedent("""\ server { listen 80; listen [::]:80 ipv6only=on; return 301 https://$host$request_uri; } server { listen 443 ssl; # listen [::]:443 ssl; # add SSL support to IPv6 address %(server_name)s ssl_certificate %(cert_path)s; ssl_certificate_key %(key_path)s; rewrite ^/$ /admin/; client_max_body_size 16m; location / { uwsgi_pass unix:///var/run/uwsgi/app/%(project_name)s/socket; include uwsgi_params; } location /static { alias %(static_root)s; expires 30d; } } """ ) % context uwsgi_conf = textwrap.dedent("""\ [uwsgi] plugins = python chdir = %(site_dir)s module = %(project_name)s.wsgi master = true processes = %(processes)d chmod-socket = 664 stats = /run/uwsgi/%%(deb-confnamespace)/%%(deb-confname)/statsocket vacuum = true uid = %(user)s gid = %(group)s env = HOME=%(home)s touch-reload = %(project_dir)s/wsgi.py enable-threads = true max-requests = 500 """ ) % context nginx_file = '/etc/nginx/conf.d/%(project_name)s.conf' % context if server_name: context['server_name'] = server_name nginx_file = '/etc/nginx/sites-available/%(server_name)s.conf' % context nginx = { 'file': nginx_file, 'conf': nginx_conf } uwsgi = { 'file': '/etc/uwsgi/apps-available/%(project_name)s.ini' % context, 'conf': uwsgi_conf } for extra_context in (nginx, uwsgi): context.update(extra_context) diff = run("echo '%(conf)s' | diff - %(file)s" % context, error_codes=[0,1,2]) if diff.return_code == 2: # File does not exist run("echo '%(conf)s' > %(file)s" % context, display=True) elif diff.return_code == 1: # File is different, save the old one if interactive: msg = ("\n\nFile %(file)s be updated, do you like to overide " "it? (yes/no): " % context) confirm = input(msg) while 1: if confirm not in ('yes', 'no'): confirm = input('Please enter either "yes" or "no": ') continue if confirm == 'no': return break run("cp %(file)s %(file)s.save" % context, display=True) run("echo '%(conf)s' > %(file)s" % context, display=True) self.stdout.write("\033[1;31mA new version of %(file)s has been installed.\n " "The old version has been placed at %(file)s.save\033[m" % context) if server_name: run('ln -s /etc/nginx/sites-available/%(server_name)s.conf /etc/nginx/sites-enabled/' % context, error_codes=[0,1], display=True) run('ln -s /etc/uwsgi/apps-available/%(project_name)s.ini /etc/uwsgi/apps-enabled/' % context, error_codes=[0,1], display=True) rotate = textwrap.dedent("""\ /var/log/nginx/*.log { daily missingok rotate 30 compress delaycompress notifempty create 640 root adm sharedscripts postrotate [ ! -f /var/run/nginx.pid ] || kill -USR1 `cat /var/run/nginx.pid` endscript }""" ) run("echo '%s' > /etc/logrotate.d/nginx" % rotate, display=True) # Allow nginx to write to uwsgi socket run('adduser www-data %(group)s' % context, display=True)