""" Creates or updates a Site including Partner and SiteConfiguration data. """

from __future__ import unicode_literals
import logging

from django.contrib.sites.models import Site
from django.core.management import BaseCommand
from oscar.core.loading import get_model

from ecommerce.core.models import SiteConfiguration

logger = logging.getLogger(__name__)
Partner = get_model('partner', 'Partner')


class Command(BaseCommand):
    help = 'Create or update Site, Partner, and SiteConfiguration'

    def add_arguments(self, parser):
        parser.add_argument('--site-id',
                            action='store',
                            dest='site_id',
                            type=int,
                            help='ID of the Site to update.')
        parser.add_argument('--site-domain',
                            action='store',
                            dest='site_domain',
                            type=str,
                            required=True,
                            help='Domain of the Site to create or update.')
        parser.add_argument('--site-name',
                            action='store',
                            dest='site_name',
                            type=str,
                            default='',
                            help='Name of the Site to create or update.')
        parser.add_argument('--partner-code',
                            action='store',
                            dest='partner_code',
                            type=str,
                            required=True,
                            help='Partner code to select/create and associate with site.')
        parser.add_argument('--partner-name',
                            action='store',
                            dest='partner_name',
                            type=str,
                            default='',
                            help='Partner name to select/create and associate with site.')
        parser.add_argument('--lms-url-root',
                            action='store',
                            dest='lms_url_root',
                            type=str,
                            required=True,
                            help='Root URL of LMS (e.g. https://localhost:8000)')
        parser.add_argument('--theme-scss-path',
                            action='store',
                            dest='theme_scss_path',
                            type=str,
                            default='',
                            help='Path to theme SCSS relative to STATIC_DIR (e.g. sass/themes/edx.scss)')
        parser.add_argument('--payment-processors',
                            action='store',
                            dest='payment_processors',
                            type=str,
                            default='',
                            help='Comma-delimited list of payment processors (e.g. cybersource,paypal)')
        parser.add_argument('--client-id',
                            action='store',
                            dest='client_id',
                            type=str,
                            required=True,
                            help='client ID')
        parser.add_argument('--client-secret',
                            action='store',
                            dest='client_secret',
                            type=str,
                            required=True,
                            help='client secret')
        parser.add_argument('--segment-key',
                            action='store',
                            dest='segment_key',
                            type=str,
                            required=False,
                            help='segment key')
        parser.add_argument('--from-email',
                            action='store',
                            dest='from_email',
                            type=str,
                            required=True,
                            help='from email')
        parser.add_argument('--enable-enrollment-codes',
                            action='store',
                            dest='enable_enrollment_codes',
                            type=bool,
                            required=False,
                            help='Enable the creation of enrollment codes.')
        parser.add_argument('--payment-support-email',
                            action='store',
                            dest='payment_support_email',
                            type=str,
                            required=False,
                            help='Email address displayed to user for payment support')
        parser.add_argument('--payment-support-url',
                            action='store',
                            dest='payment_support_url',
                            type=str,
                            required=False,
                            help='URL displayed to user for payment support')

    def handle(self, *args, **options):
        site_id = options.get('site_id')
        site_domain = options.get('site_domain')
        site_name = options.get('site_name')
        partner_code = options.get('partner_code')
        partner_name = options.get('partner_name')
        lms_url_root = options.get('lms_url_root')
        client_id = options.get('client_id')
        client_secret = options.get('client_secret')
        segment_key = options.get('segment_key')
        from_email = options.get('from_email')
        enable_enrollment_codes = True if options.get('enable_enrollment_codes') else False
        payment_support_email = options.get('payment_support_email', '')
        payment_support_url = options.get('payment_support_url', '')

        try:
            site = Site.objects.get(id=site_id)
        except Site.DoesNotExist:
            site, site_created = Site.objects.get_or_create(domain=site_domain)
            if site_created:
                logger.info('Site created with domain %s', site_domain)

        site.domain = site_domain
        if site_name:
            site.name = site_name
        site.save()

        partner, partner_created = Partner.objects.get_or_create(code=partner_code)
        if partner_created:
            partner.name = partner_name
            partner.short_code = partner_code
            partner.save()
            logger.info('Partner created with code %s', partner_code)

        site_configuration_defaults = {
            'partner': partner,
            'lms_url_root': lms_url_root,
            'theme_scss_path': options['theme_scss_path'],
            'payment_processors': options['payment_processors'],
            'segment_key': segment_key,
            'from_email': from_email,
            'enable_enrollment_codes': enable_enrollment_codes,
            'oauth_settings': {
                'SOCIAL_AUTH_EDX_OIDC_URL_ROOT': '{lms_url_root}/oauth2'.format(lms_url_root=lms_url_root),
                'SOCIAL_AUTH_EDX_OIDC_KEY': client_id,
                'SOCIAL_AUTH_EDX_OIDC_SECRET': client_secret,
                'SOCIAL_AUTH_EDX_OIDC_ID_TOKEN_DECRYPTION_KEY': client_secret,
                'SOCIAL_AUTH_EDX_OIDC_ISSUERS': [lms_url_root]
            }
        }
        if payment_support_email:
            site_configuration_defaults['payment_support_email'] = payment_support_email
        if payment_support_url:
            site_configuration_defaults['payment_support_url'] = payment_support_url

        SiteConfiguration.objects.update_or_create(
            site=site,
            defaults=site_configuration_defaults
        )
