from base64 import urlsafe_b64encode

import ecdsa
from django.core.management.base import BaseCommand
from djangoldp_webpushnotification.models import VAPIDKeyset
from ecdsa import SigningKey


class Command(BaseCommand):
    help = "Generate VAPID key pair"

    def handle(self, *args, **options):
        priv_key = SigningKey.generate(curve=ecdsa.NIST256p)

        VAPIDKeyset.objects.create(
                private_key=urlsafe_b64encode(priv_key.to_string()).strip(b"=")
        )

        self.stdout.write("VAPID Keyset succesfully generated")

        exit(0)