!7

!include once #1030462 // BouncyCastle

import nl.martijndwars.webpush.Subscription;
import nl.martijndwars.webpush.PushService;
import nl.martijndwars.webpush.Notification;

concept Entry {
  S publicKey;
  SecretValue<S> privateKey;
  S comment;
  S globalID = aGlobalID();
}

cmodule2 WebPushKeyManager > DynCRUD<Entry> {
  start {
    registerBouncyCastle();
  }
  
  KeyPair generateKeyPair() ctex {
    ECNamedCurveParameterSpec parameterSpec = ECNamedCurveTable.getParameterSpec(nl.martijndwars.webpush.Utils.CURVE);

    KeyPairGenerator keyPairGenerator = KeyPairGenerator.getInstance(nl.martijndwars.webpush.Utils.ALGORITHM,
      org.bouncycastle.jce.provider.BouncyCastleProvider.PROVIDER_NAME);
    keyPairGenerator.initialize(parameterSpec);

    ret keyPairGenerator.generateKeyPair();
  }
  
  enhanceFrame {
    internalFrameMenuItem(f, "Make key pair", rThreadEnter getKeyPair);
  }
  
  // API
  
  PKIKeyPair getKeyPair() {
    lock dbLock();
    
    Entry entry = conceptWhere Entry();
    if (entry == null) {
      KeyPair keyPair = generateKeyPair();
      ECPublicKey publicKey = cast keyPair.getPublic();
      ECPrivateKey privateKey = cast keyPair.getPrivate();

      byte[] encodedPublicKey = nl.martijndwars.webpush.Utils.encode(publicKey);
      byte[] encodedPrivateKey = nl.martijndwars.webpush.Utils.encode(privateKey);
    
      S publicKeyString = base64encode(encodedPublicKey);
      S privateKeyString = base64encode(encodedPrivateKey);   
      
      entry = cnew Entry(publicKey := publicKeyString,
        privateKey := SecretValue(privateKeyString));
    }
    
    ret PKIKeyPair(entry.privateKey!, entry.publicKey);
  }
  
  // subData = subscription data as received from e.g. mozilla.com
  void sendNotification(PKIKeyPair keyPair default getKeyPair(), MapSO subData, S msg) ctex {
    // not using PushAsyncService because lazy
    PushService pushService = new(keyPair.publicKey(),
      keyPair.privateKey());
    SS keys = cast subData.get("keys");
    Subscription subscription = new((S) subData.get("endpoint"),
      new Subscription.Keys(keys.get("p256dh"), keys.get("auth")));
    Notification notification = new(subscription, msg);
    pushService.send(notification);
    print("Sent push msg: " + msg);
  }
}