import os
import base64
from django.contrib.auth.models import User
from django.core.management.base import BaseCommand
from oauth2_provider.models import Application


class Command(BaseCommand):
    """Management command for create Oauth2 application"""

    def handle(self, *args, **options):
        client_id = base64.b64decode(os.environ.get(
            'SOCIAL_AUTH_DRYCC_CONTROLLER_KEY')) if os.environ.get(
            'SOCIAL_AUTH_DRYCC_CONTROLLER_KEY') else None
        client_secret = base64.b64decode(os.environ.get(
            'SOCIAL_AUTH_DRYCC_CONTROLLER_SECRET')) if os.environ.get(
            'SOCIAL_AUTH_DRYCC_CONTROLLER_SECRET') else None
        controller_domain = os.environ.get('DRYCC_CONTROLLER_DOMAIN')
        if not all([client_id, client_secret, controller_domain]):
            self.stdout.write('client_id or client_secret non-existent')
            return
        user = User.objects.get(username='admin')
        application, created = Application.objects.get_or_create(
            client_id=client_id,
            client_secret=client_secret,
            defaults={
                'name': 'Drycc Controller',
                'user': user,
                'redirect_uris': f'{controller_domain}/complete/drycc/',
                'authorization_grant_type': 'authorization-code',
                'client_type': 'Public',
                'algorithm': 'RS256'
            }
        )
        if created:
            self.stdout.write('Drycc controller app created')
        else:
            self.stdout.write("Drycc controller app already exists")