PushService.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2022 Lluis Turró Cutiller <http://www.turro.org/>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.turro.push.service;
 
import java.io.IOException;
import java.security.GeneralSecurityException;
import java.security.KeyPair;
import java.util.concurrent.ExecutionException;
import java.util.function.Consumer;
import org.apache.hc.client5.http.async.methods.SimpleHttpRequest;
import org.apache.hc.client5.http.async.methods.SimpleHttpResponse;
import org.apache.hc.client5.http.async.methods.SimpleRequestBuilder;
import org.apache.hc.client5.http.classic.methods.HttpPost;
import org.apache.hc.client5.http.impl.async.CloseableHttpAsyncClient;
import org.apache.hc.client5.http.impl.async.HttpAsyncClientBuilder;
import org.apache.hc.client5.http.impl.classic.CloseableHttpClient;
import org.apache.hc.client5.http.impl.classic.HttpClientBuilder;
import org.apache.hc.core5.concurrent.FutureCallback;
import org.apache.hc.core5.http.ContentType;
import org.apache.hc.core5.http.HttpResponse;
import org.apache.hc.core5.http.Method;
import org.apache.hc.core5.http.io.entity.ByteArrayEntity;
import org.apache.hc.core5.io.CloseMode;
import org.jose4j.lang.JoseException;
 
public class PushService extends AbstractPushService<PushService> {
 
  public PushService() {
  }
 
  public PushService(String gcmApiKey) {
    super(gcmApiKey);
  }
 
  public PushService(KeyPair keyPair) {
    super(keyPair);
  }
 
  public PushService(KeyPair keyPair, String subject) {
    super(keyPair, subject);
  }
 
  public PushService(String publicKey, String privateKey) throws GeneralSecurityException {
    super(publicKey, privateKey);
  }
 
  public PushService(String publicKey, String privateKey, String subject) throws GeneralSecurityException {
    super(publicKey, privateKey, subject);
  }
 
  public HttpResponse send(Notification notification) throws GeneralSecurityException, IOException, JoseException, ExecutionException, InterruptedException {
    return send(notification, Encoding.AESGCM);
  }
 
  public HttpResponse send(Notification notification, Encoding encoding) throws GeneralSecurityException, IOException, JoseException {
    HttpPost httpPost = preparePost(notification, encoding);
    try(CloseableHttpClient client = HttpClientBuilder.create().build()) {
      return client.execute(httpPost);
    }
  }
 
  public HttpPost preparePost(Notification notification, Encoding encoding) throws GeneralSecurityException, IOException, JoseException {
    HttpRequest request = prepareRequest(notification, encoding);
    HttpPost httpPost = new HttpPost(request.getUrl());
    request.getHeaders().forEach(httpPost::addHeader);
    if (request.getBody() != null) {
      httpPost.setEntity(new ByteArrayEntity(request.getBody(), ContentType.APPLICATION_OCTET_STREAM));
    }
    return httpPost;
  }
  
  public void sendAsync(Notification notification, Consumer<SimpleHttpResponse> completed, 
          Consumer<Exception> failed, Runnable cancelled) throws GeneralSecurityException, IOException, JoseException, ExecutionException, InterruptedException {
    sendAsync(notification, Encoding.AESGCM, completed, failed, cancelled);
  }
 
  public void sendAsync(Notification notification, Encoding encoding, Consumer<SimpleHttpResponse> completed, 
          Consumer<Exception> failed, Runnable cancelled) throws GeneralSecurityException, IOException, JoseException {
    SimpleHttpRequest httpPost = prepareAsyncPost(notification, encoding);
    final CloseableHttpAsyncClient httpClient = HttpAsyncClientBuilder.create().build();
    httpClient.start();
    httpClient.execute(httpPost, new FutureCallback<SimpleHttpResponse>() {
      @Override
      public void completed(SimpleHttpResponse result) {
        if(completed != null) completed.accept(result);
        httpClient.close(CloseMode.GRACEFUL);
      }
 
      @Override
      public void failed(Exception ex) {
        if(failed != null) failed.accept(ex);
        httpClient.close(CloseMode.GRACEFUL);
      }
 
      @Override
      public void cancelled() {
        if(cancelled != null) cancelled.run();
        httpClient.close(CloseMode.GRACEFUL);
      }
    });
  }
  
  public SimpleHttpRequest prepareAsyncPost(Notification notification, Encoding encoding) throws GeneralSecurityException, IOException, JoseException {
    HttpRequest request = prepareRequest(notification, encoding);
    SimpleHttpRequest httpPost = SimpleRequestBuilder.create(Method.POST).setUri(request.getUrl()).build();
    request.getHeaders().forEach(httpPost::addHeader);
    httpPost.setBody(request.getBody(), ContentType.APPLICATION_OCTET_STREAM);
    return httpPost;
  }
  
}