|
|
|
@ -18,13 +18,10 @@
|
|
|
|
|
package io.entgra.device.mgt.core.device.mgt.extensions.push.notification.provider.fcm;
|
|
|
|
|
|
|
|
|
|
import com.google.auth.oauth2.GoogleCredentials;
|
|
|
|
|
import com.google.gson.JsonArray;
|
|
|
|
|
import com.google.gson.JsonObject;
|
|
|
|
|
import com.google.gson.JsonPrimitive;
|
|
|
|
|
import com.hazelcast.aws.utility.Environment;
|
|
|
|
|
import io.entgra.device.mgt.core.device.mgt.core.operation.mgt.OperationManagerImpl;
|
|
|
|
|
import io.entgra.device.mgt.core.device.mgt.extensions.logger.spi.EntgraLogger;
|
|
|
|
|
import io.entgra.device.mgt.core.notification.logger.impl.EntgraDeviceConnectivityLoggerImpl;
|
|
|
|
|
import io.entgra.device.mgt.core.device.mgt.core.config.DeviceConfigurationManager;
|
|
|
|
|
import io.entgra.device.mgt.core.device.mgt.core.config.push.notification.ContextMetadata;
|
|
|
|
|
import io.entgra.device.mgt.core.device.mgt.core.config.push.notification.PushNotificationConfiguration;
|
|
|
|
|
import org.apache.commons.logging.Log;
|
|
|
|
|
import org.apache.commons.logging.LogFactory;
|
|
|
|
|
import io.entgra.device.mgt.core.device.mgt.common.Device;
|
|
|
|
@ -34,28 +31,34 @@ import io.entgra.device.mgt.core.device.mgt.common.push.notification.Notificatio
|
|
|
|
|
import io.entgra.device.mgt.core.device.mgt.common.push.notification.PushNotificationConfig;
|
|
|
|
|
import io.entgra.device.mgt.core.device.mgt.common.push.notification.PushNotificationExecutionFailedException;
|
|
|
|
|
import io.entgra.device.mgt.core.device.mgt.extensions.push.notification.provider.fcm.internal.FCMDataHolder;
|
|
|
|
|
import org.wso2.carbon.utils.CarbonUtils;
|
|
|
|
|
|
|
|
|
|
import java.io.*;
|
|
|
|
|
import java.io.File;
|
|
|
|
|
import java.io.IOException;
|
|
|
|
|
import java.io.OutputStream;
|
|
|
|
|
import java.net.HttpURLConnection;
|
|
|
|
|
import java.net.URL;
|
|
|
|
|
import java.nio.file.Files;
|
|
|
|
|
import java.nio.file.Path;
|
|
|
|
|
import java.nio.file.Paths;
|
|
|
|
|
import java.nio.file.StandardOpenOption;
|
|
|
|
|
import java.util.Arrays;
|
|
|
|
|
import java.util.List;
|
|
|
|
|
import java.util.Properties;
|
|
|
|
|
|
|
|
|
|
public class FCMNotificationStrategy implements NotificationStrategy {
|
|
|
|
|
|
|
|
|
|
private static final Log log = LogFactory.getLog(FCMNotificationStrategy.class);
|
|
|
|
|
|
|
|
|
|
private static final String NOTIFIER_TYPE_FCM = "FCM";
|
|
|
|
|
private static final String FCM_TOKEN = "FCM_TOKEN";
|
|
|
|
|
private static final String FCM_ENDPOINT = "https://fcm.googleapis.com/v1/projects/myproject-b5ae1/messages:send";
|
|
|
|
|
private static final String FCM_API_KEY = "fcmAPIKey";
|
|
|
|
|
private static final int TIME_TO_LIVE = 2419199; // 1 second less that 28 days
|
|
|
|
|
private static final int TIME_TO_LIVE = 2419199; // 1 second less than 28 days
|
|
|
|
|
private static final int HTTP_STATUS_CODE_OK = 200;
|
|
|
|
|
private final PushNotificationConfig config;
|
|
|
|
|
private static final String FCM_SERVICE_ACCOUNT_PATH = CarbonUtils.getCarbonHome() + File.separator +
|
|
|
|
|
"repository" + File.separator + "resources" + File.separator + "service-account.json";
|
|
|
|
|
private static final String[] FCM_SCOPES = { "https://www.googleapis.com/auth/firebase.messaging" };
|
|
|
|
|
private static final String FCM_ENDPOINT_KEY = "FCM_SERVER_ENDPOINT";
|
|
|
|
|
private Properties contextMetadataProperties;
|
|
|
|
|
private volatile GoogleCredentials defaultApplication;
|
|
|
|
|
|
|
|
|
|
public FCMNotificationStrategy(PushNotificationConfig config) {
|
|
|
|
|
this.config = config;
|
|
|
|
@ -63,23 +66,25 @@ public class FCMNotificationStrategy implements NotificationStrategy {
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public void init() {
|
|
|
|
|
|
|
|
|
|
initContextConfigs();
|
|
|
|
|
initDefaultOAuthApplication();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public void execute(NotificationContext ctx) throws PushNotificationExecutionFailedException {
|
|
|
|
|
String token = getFcmOauthToken();
|
|
|
|
|
try {
|
|
|
|
|
if (NOTIFIER_TYPE_FCM.equals(config.getType())) {
|
|
|
|
|
Device device = FCMDataHolder.getInstance().getDeviceManagementProviderService()
|
|
|
|
|
.getDeviceWithTypeProperties(ctx.getDeviceId());
|
|
|
|
|
if(device.getProperties() != null && getFCMToken(device.getProperties()) != null) {
|
|
|
|
|
this.sendWakeUpCall(ctx.getOperation().getCode(), device, token);
|
|
|
|
|
defaultApplication.refresh();
|
|
|
|
|
sendWakeUpCall(defaultApplication.getAccessToken().getTokenValue(),
|
|
|
|
|
getFCMToken(device.getProperties()));
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
if (log.isDebugEnabled()) {
|
|
|
|
|
log.debug("Not using FCM notifier as notifier type is set to " + config.getType() +
|
|
|
|
|
" in Platform Configurations.");
|
|
|
|
|
" in Platform Configurations.");
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
} catch (DeviceManagementException e) {
|
|
|
|
@ -89,100 +94,93 @@ public class FCMNotificationStrategy implements NotificationStrategy {
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private String getFcmOauthToken() {
|
|
|
|
|
GoogleCredentials googleCredentials = null;
|
|
|
|
|
try {
|
|
|
|
|
googleCredentials = GoogleCredentials
|
|
|
|
|
.fromStream(new FileInputStream("/etc/service-account.json"))
|
|
|
|
|
.createScoped(Arrays.asList("https://www.googleapis.com/auth/firebase.messaging"));
|
|
|
|
|
googleCredentials.refresh();
|
|
|
|
|
if (null != googleCredentials) {
|
|
|
|
|
writeLog("========= Google Credentials created " + googleCredentials.getAccessToken());
|
|
|
|
|
} else {
|
|
|
|
|
writeLog("========= Google Credentials is null");
|
|
|
|
|
private void initDefaultOAuthApplication() {
|
|
|
|
|
if (defaultApplication == null) {
|
|
|
|
|
synchronized (FCMNotificationStrategy.class) {
|
|
|
|
|
if (defaultApplication == null) {
|
|
|
|
|
Path serviceAccountPath = Paths.get(FCM_SERVICE_ACCOUNT_PATH);
|
|
|
|
|
try {
|
|
|
|
|
this.defaultApplication = GoogleCredentials.
|
|
|
|
|
fromStream(Files.newInputStream(serviceAccountPath)).
|
|
|
|
|
createScoped(FCM_SCOPES);
|
|
|
|
|
} catch (IOException e) {
|
|
|
|
|
log.error("Fail to initialize default OAuth application for FCM communication");
|
|
|
|
|
throw new IllegalStateException(e);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return googleCredentials.getAccessToken().getTokenValue();
|
|
|
|
|
} catch (IOException e) {
|
|
|
|
|
log.error("Error occurred while getting the FCM OAuth token.", e);
|
|
|
|
|
throw new RuntimeException(e);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public NotificationContext buildContext() {
|
|
|
|
|
return null;
|
|
|
|
|
private void initContextConfigs() {
|
|
|
|
|
PushNotificationConfiguration pushNotificationConfiguration = DeviceConfigurationManager.getInstance().
|
|
|
|
|
getDeviceManagementConfig().getPushNotificationConfiguration();
|
|
|
|
|
List<ContextMetadata> contextMetadata = pushNotificationConfiguration.getContextMetadata();
|
|
|
|
|
Properties properties = new Properties();
|
|
|
|
|
if (contextMetadata != null) {
|
|
|
|
|
for (ContextMetadata metadata : contextMetadata) {
|
|
|
|
|
properties.setProperty(metadata.getKey(), metadata.getValue());
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
contextMetadataProperties = properties;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public void undeploy() {
|
|
|
|
|
private void sendWakeUpCall(String accessToken, String registrationId) throws IOException,
|
|
|
|
|
PushNotificationExecutionFailedException {
|
|
|
|
|
OutputStream os = null;
|
|
|
|
|
HttpURLConnection conn = null;
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
String fcmServerEndpoint = contextMetadataProperties.getProperty(FCM_ENDPOINT_KEY);
|
|
|
|
|
if(fcmServerEndpoint == null) {
|
|
|
|
|
String msg = "Encountered configuration issue. " + FCM_ENDPOINT_KEY + " is not defined";
|
|
|
|
|
log.error(msg);
|
|
|
|
|
throw new PushNotificationExecutionFailedException(msg);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
try {
|
|
|
|
|
byte[] bytes = getFCMRequest(registrationId).getBytes();
|
|
|
|
|
URL url = new URL(fcmServerEndpoint);
|
|
|
|
|
conn = (HttpURLConnection) url.openConnection();
|
|
|
|
|
conn.setRequestProperty("Content-Type", "application/json");
|
|
|
|
|
conn.setRequestProperty("Authorization", "Bearer " + accessToken);
|
|
|
|
|
conn.setRequestMethod("POST");
|
|
|
|
|
conn.setDoOutput(true);
|
|
|
|
|
|
|
|
|
|
os = conn.getOutputStream();
|
|
|
|
|
os.write(bytes);
|
|
|
|
|
|
|
|
|
|
private void sendWakeUpCall(String message, Device device, String token) throws IOException,
|
|
|
|
|
PushNotificationExecutionFailedException {
|
|
|
|
|
if (device.getProperties() != null) {
|
|
|
|
|
writeLog("===== Calling senWakeupCall " + device);
|
|
|
|
|
OutputStream os = null;
|
|
|
|
|
byte[] bytes = getFCMRequest(message, getFCMToken(device.getProperties())).getBytes();
|
|
|
|
|
|
|
|
|
|
HttpURLConnection conn = null;
|
|
|
|
|
try {
|
|
|
|
|
conn = (HttpURLConnection) new URL(FCM_ENDPOINT).openConnection();
|
|
|
|
|
conn.setRequestProperty("Content-Type", "application/json");
|
|
|
|
|
conn.setRequestProperty("Authorization", "Bearer " + token);
|
|
|
|
|
conn.setRequestMethod("POST");
|
|
|
|
|
conn.setDoOutput(true);
|
|
|
|
|
os = conn.getOutputStream();
|
|
|
|
|
os.write(bytes);
|
|
|
|
|
} finally {
|
|
|
|
|
if (os != null) {
|
|
|
|
|
os.close();
|
|
|
|
|
}
|
|
|
|
|
if (conn != null) {
|
|
|
|
|
conn.disconnect();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
int status = conn.getResponseCode();
|
|
|
|
|
if (log.isDebugEnabled()) {
|
|
|
|
|
log.debug("Result code: " + status + ", Message: " + conn.getResponseMessage());
|
|
|
|
|
if (status != 200) {
|
|
|
|
|
log.error("Response Status: " + status + ", Response Message: " + conn.getResponseMessage());
|
|
|
|
|
}
|
|
|
|
|
} finally {
|
|
|
|
|
if (os != null) {
|
|
|
|
|
os.close();
|
|
|
|
|
}
|
|
|
|
|
if (status != HTTP_STATUS_CODE_OK) {
|
|
|
|
|
throw new PushNotificationExecutionFailedException("Push notification sending failed with the HTTP " +
|
|
|
|
|
"error code '" + status + "'");
|
|
|
|
|
if (conn != null) {
|
|
|
|
|
conn.disconnect();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private static String getFCMRequest(String message, String registrationId) {
|
|
|
|
|
JsonObject fcmRequest = new JsonObject();
|
|
|
|
|
private static String getFCMRequest(String registrationId) {
|
|
|
|
|
JsonObject messageObject = new JsonObject();
|
|
|
|
|
messageObject.addProperty("token", registrationId);
|
|
|
|
|
JsonObject notification = new JsonObject();
|
|
|
|
|
notification.addProperty("title", "FCM Message");
|
|
|
|
|
notification.addProperty("body", message);
|
|
|
|
|
messageObject.add("notification", notification);
|
|
|
|
|
fcmRequest.add("message", messageObject);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/*fcmRequest.addProperty("delay_while_idle", false);
|
|
|
|
|
fcmRequest.addProperty("time_to_live", TIME_TO_LIVE);
|
|
|
|
|
fcmRequest.addProperty("priority", "high");
|
|
|
|
|
JsonObject fcmRequest = new JsonObject();
|
|
|
|
|
fcmRequest.add("message", messageObject);
|
|
|
|
|
|
|
|
|
|
//Add message to FCM request
|
|
|
|
|
JsonObject data = new JsonObject();
|
|
|
|
|
if (message != null && !message.isEmpty()) {
|
|
|
|
|
data.addProperty("data", message);
|
|
|
|
|
fcmRequest.add("data", data);
|
|
|
|
|
}
|
|
|
|
|
return fcmRequest.toString();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
//Set device reg-id
|
|
|
|
|
JsonArray regIds = new JsonArray();
|
|
|
|
|
regIds.add(new JsonPrimitive(registrationId));
|
|
|
|
|
@Override
|
|
|
|
|
public NotificationContext buildContext() {
|
|
|
|
|
return null;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fcmRequest.add("registration_ids", regIds);*/
|
|
|
|
|
@Override
|
|
|
|
|
public void undeploy() {
|
|
|
|
|
|
|
|
|
|
writeLog("========= FCM Request " + fcmRequest);
|
|
|
|
|
return fcmRequest.toString();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private static String getFCMToken(List<Device.Property> properties) {
|
|
|
|
@ -196,21 +194,8 @@ public class FCMNotificationStrategy implements NotificationStrategy {
|
|
|
|
|
return fcmToken;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private static void writeLog(String message) {
|
|
|
|
|
try (FileWriter fw = new FileWriter("/opt/entgra/migration/entgra-uem-ultimate-6.0.3.0/log.txt", true);
|
|
|
|
|
BufferedWriter bw = new BufferedWriter(fw)) {
|
|
|
|
|
bw.write(message);
|
|
|
|
|
bw.newLine();
|
|
|
|
|
bw.flush();
|
|
|
|
|
} catch (IOException e) {
|
|
|
|
|
e.printStackTrace();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public PushNotificationConfig getConfig() {
|
|
|
|
|
return config;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|